251 lines
8.0 KiB
PHP
251 lines
8.0 KiB
PHP
<?php
|
||
// +----------------------------------------------------------------------
|
||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||
// +----------------------------------------------------------------------
|
||
// | Copyright (c) 2006-2016 http://thinkphp.cn All rights reserved.
|
||
// +----------------------------------------------------------------------
|
||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||
// +----------------------------------------------------------------------
|
||
// | Author: luofei614 <weibo.com/luofei614>
|
||
// +----------------------------------------------------------------------
|
||
|
||
namespace think\log\driver;
|
||
|
||
use think\App;
|
||
|
||
/**
|
||
* github: https://github.com/luofei614/SocketLog
|
||
* @author luofei614<weibo.com/luofei614>
|
||
*/
|
||
class Socket
|
||
{
|
||
public $port = 1116; //SocketLog 服务的http的端口号
|
||
|
||
protected $config = [
|
||
// socket服务器地址
|
||
'host' => 'localhost',
|
||
// 是否显示加载的文件列表
|
||
'show_included_files' => false,
|
||
// 日志强制记录到配置的client_id
|
||
'force_client_ids' => [],
|
||
// 限制允许读取日志的client_id
|
||
'allow_client_ids' => [],
|
||
];
|
||
|
||
protected $css = [
|
||
'sql' => 'color:#009bb4;',
|
||
'sql_warn' => 'color:#009bb4;font-size:14px;',
|
||
'error' => 'color:#f4006b;font-size:14px;',
|
||
'page' => 'color:#40e2ff;background:#171717;',
|
||
'big' => 'font-size:20px;color:red;',
|
||
];
|
||
|
||
protected $allowForceClientIds = []; //配置强制推送且被授权的client_id
|
||
|
||
/**
|
||
* 构造函数
|
||
* @param array $config 缓存参数
|
||
* @access public
|
||
*/
|
||
public function __construct(array $config = [])
|
||
{
|
||
if (!empty($config)) {
|
||
$this->config = array_merge($this->config, $config);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 调试输出接口
|
||
* @access public
|
||
* @param array $log 日志信息
|
||
* @return bool
|
||
*/
|
||
public function save(array $log = [], $append = false)
|
||
{
|
||
if (!$this->check()) {
|
||
return false;
|
||
}
|
||
$trace = [];
|
||
if (App::$debug) {
|
||
$runtime = round(microtime(true) - THINK_START_TIME, 10);
|
||
$reqs = $runtime > 0 ? number_format(1 / $runtime, 2) : '∞';
|
||
$time_str = ' [运行时间：' . number_format($runtime, 6) . 's][吞吐率：' . $reqs . 'req/s]';
|
||
$memory_use = number_format((memory_get_usage() - THINK_START_MEM) / 1024, 2);
|
||
$memory_str = ' [内存消耗：' . $memory_use . 'kb]';
|
||
$file_load = ' [文件加载：' . count(get_included_files()) . ']';
|
||
|
||
if (isset($_SERVER['HTTP_HOST'])) {
|
||
$current_uri = $_SERVER['HTTP_HOST'] . $_SERVER['REQUEST_URI'];
|
||
} else {
|
||
$current_uri = 'cmd:' . implode(' ', $_SERVER['argv']);
|
||
}
|
||
// 基本信息
|
||
$trace[] = [
|
||
'type' => 'group',
|
||
'msg' => $current_uri . $time_str . $memory_str . $file_load,
|
||
'css' => $this->css['page'],
|
||
];
|
||
}
|
||
|
||
foreach ($log as $type => $val) {
|
||
$trace[] = [
|
||
'type' => 'groupCollapsed',
|
||
'msg' => '[ ' . $type . ' ]',
|
||
'css' => isset($this->css[$type]) ? $this->css[$type] : '',
|
||
];
|
||
foreach ($val as $msg) {
|
||
if (!is_string($msg)) {
|
||
$msg = var_export($msg, true);
|
||
}
|
||
$trace[] = [
|
||
'type' => 'log',
|
||
'msg' => $msg,
|
||
'css' => '',
|
||
];
|
||
}
|
||
$trace[] = [
|
||
'type' => 'groupEnd',
|
||
'msg' => '',
|
||
'css' => '',
|
||
];
|
||
}
|
||
|
||
if ($this->config['show_included_files']) {
|
||
$trace[] = [
|
||
'type' => 'groupCollapsed',
|
||
'msg' => '[ file ]',
|
||
'css' => '',
|
||
];
|
||
$trace[] = [
|
||
'type' => 'log',
|
||
'msg' => implode("\n", get_included_files()),
|
||
'css' => '',
|
||
];
|
||
$trace[] = [
|
||
'type' => 'groupEnd',
|
||
'msg' => '',
|
||
'css' => '',
|
||
];
|
||
}
|
||
|
||
$trace[] = [
|
||
'type' => 'groupEnd',
|
||
'msg' => '',
|
||
'css' => '',
|
||
];
|
||
|
||
$tabid = $this->getClientArg('tabid');
|
||
if (!$client_id = $this->getClientArg('client_id')) {
|
||
$client_id = '';
|
||
}
|
||
|
||
if (!empty($this->allowForceClientIds)) {
|
||
//强制推送到多个client_id
|
||
foreach ($this->allowForceClientIds as $force_client_id) {
|
||
$client_id = $force_client_id;
|
||
$this->sendToClient($tabid, $client_id, $trace, $force_client_id);
|
||
}
|
||
} else {
|
||
$this->sendToClient($tabid, $client_id, $trace, '');
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 发送给指定客户端
|
||
* @author Zjmainstay
|
||
* @param $tabid
|
||
* @param $client_id
|
||
* @param $logs
|
||
* @param $force_client_id
|
||
*/
|
||
protected function sendToClient($tabid, $client_id, $logs, $force_client_id)
|
||
{
|
||
$logs = [
|
||
'tabid' => $tabid,
|
||
'client_id' => $client_id,
|
||
'logs' => $logs,
|
||
'force_client_id' => $force_client_id,
|
||
];
|
||
$msg = @json_encode($logs);
|
||
$address = '/' . $client_id; //将client_id作为地址， server端通过地址判断将日志发布给谁
|
||
$this->send($this->config['host'], $msg, $address);
|
||
}
|
||
|
||
protected function check()
|
||
{
|
||
$tabid = $this->getClientArg('tabid');
|
||
//是否记录日志的检查
|
||
if (!$tabid && !$this->config['force_client_ids']) {
|
||
return false;
|
||
}
|
||
//用户认证
|
||
$allow_client_ids = $this->config['allow_client_ids'];
|
||
if (!empty($allow_client_ids)) {
|
||
//通过数组交集得出授权强制推送的client_id
|
||
$this->allowForceClientIds = array_intersect($allow_client_ids, $this->config['force_client_ids']);
|
||
if (!$tabid && count($this->allowForceClientIds)) {
|
||
return true;
|
||
}
|
||
|
||
$client_id = $this->getClientArg('client_id');
|
||
if (!in_array($client_id, $allow_client_ids)) {
|
||
return false;
|
||
}
|
||
} else {
|
||
$this->allowForceClientIds = $this->config['force_client_ids'];
|
||
}
|
||
return true;
|
||
}
|
||
|
||
protected function getClientArg($name)
|
||
{
|
||
static $args = [];
|
||
|
||
$key = 'HTTP_USER_AGENT';
|
||
|
||
if (isset($_SERVER['HTTP_SOCKETLOG'])) {
|
||
$key = 'HTTP_SOCKETLOG';
|
||
}
|
||
|
||
if (!isset($_SERVER[$key])) {
|
||
return;
|
||
}
|
||
if (empty($args)) {
|
||
if (!preg_match('/SocketLog\((.*?)\)/', $_SERVER[$key], $match)) {
|
||
$args = ['tabid' => null];
|
||
return;
|
||
}
|
||
parse_str($match[1], $args);
|
||
}
|
||
if (isset($args[$name])) {
|
||
return $args[$name];
|
||
}
|
||
return;
|
||
}
|
||
|
||
/**
|
||
* @param string $host - $host of socket server
|
||
* @param string $message - 发送的消息
|
||
* @param string $address - 地址
|
||
* @return bool
|
||
*/
|
||
protected function send($host, $message = '', $address = '/')
|
||
{
|
||
$url = 'http://' . $host . ':' . $this->port . $address;
|
||
$ch = curl_init();
|
||
curl_setopt($ch, CURLOPT_URL, $url);
|
||
curl_setopt($ch, CURLOPT_POST, true);
|
||
curl_setopt($ch, CURLOPT_POSTFIELDS, $message);
|
||
curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
|
||
curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, 1);
|
||
curl_setopt($ch, CURLOPT_TIMEOUT, 10);
|
||
$headers = [
|
||
"Content-Type: application/json;charset=UTF-8",
|
||
];
|
||
curl_setopt($ch, CURLOPT_HTTPHEADER, $headers); //设置header
|
||
return curl_exec($ch);
|
||
}
|
||
|
||
}
|