|
- <?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);
- }
-
- }
|