mirror of
https://gitee.com/fastadminnet/framework.git
synced 2026-07-01 20:52:48 +08:00
383 lines
12 KiB
PHP
383 lines
12 KiB
PHP
<?php
|
||
/**
|
||
* github: https://github.com/luofei614/SocketLog
|
||
* @author luofei614<weibo.com/luofei614>
|
||
*/
|
||
namespace think;
|
||
|
||
use think\Exception;
|
||
|
||
class Slog
|
||
{
|
||
public static $start_time = 0;
|
||
public static $start_memory = 0;
|
||
public static $port = 1116; //SocketLog 服务的http的端口号
|
||
public static $log_types = ['log', 'info', 'error', 'warn', 'table', 'alert'];
|
||
|
||
protected static $config = [
|
||
'enable' => true, //是否记录日志的开关
|
||
'host' => 'localhost',
|
||
//是否显示利于优化的参数，如果允许时间，消耗内存等
|
||
'optimize' => false,
|
||
'show_included_files' => false,
|
||
'error_handler' => false,
|
||
//日志强制记录到配置的client_id
|
||
'force_client_id' => '',
|
||
//限制允许读取日志的client_id
|
||
'allow_client_ids' => [],
|
||
];
|
||
|
||
protected static $logs = [];
|
||
|
||
protected static $css = [
|
||
'sql' => 'color:#009bb4;',
|
||
'sql_warn' => 'color:#009bb4;font-size:14px;',
|
||
'error_handler' => 'color:#f4006b;font-size:14px;',
|
||
'page' => 'color:#40e2ff;background:#171717;',
|
||
'big' => 'font-size:20px;color:red;',
|
||
];
|
||
|
||
public static function sql($sql, $pdo)
|
||
{
|
||
|
||
if (is_object($pdo)) {
|
||
if (!self::check()) {
|
||
return;
|
||
}
|
||
$css = self::$css['sql'];
|
||
if (preg_match('/^SELECT /i', $sql)) {
|
||
//explain
|
||
try {
|
||
$obj = $pdo->query("EXPLAIN " . $sql);
|
||
if (is_object($obj) && method_exists($obj, 'fetch')) {
|
||
$arr = $obj->fetch(\PDO::FETCH_ASSOC);
|
||
self::sqlexplain($arr, $sql, $css);
|
||
}
|
||
} catch (Exception $e) {
|
||
|
||
}
|
||
}
|
||
self::sqlwhere($sql, $css);
|
||
self::trace($sql, 2, $css);
|
||
} else {
|
||
throw new Exception('SocketLog can not support this database link');
|
||
}
|
||
|
||
}
|
||
|
||
public static function trace($msg, $trace_level = 2, $css = '')
|
||
{
|
||
if (!self::check()) {
|
||
return;
|
||
}
|
||
self::record('groupCollapsed', $msg, $css);
|
||
$traces = debug_backtrace(false);
|
||
$traces = array_reverse($traces);
|
||
$max = count($traces) - $trace_level;
|
||
for ($i = 0; $i < $max; $i++) {
|
||
$trace = $traces[$i];
|
||
$fun = isset($trace['class']) ? $trace['class'] . '::' . $trace['function'] : $trace['function'];
|
||
$file = isset($trace['file']) ? $trace['file'] : 'unknown file';
|
||
$line = isset($trace['line']) ? $trace['line'] : 'unknown line';
|
||
$trace_msg = '#' . $i . ' ' . $fun . ' called at [' . $file . ':' . $line . ']';
|
||
if (!empty($trace['args'])) {
|
||
self::record('groupCollapsed', $trace_msg);
|
||
self::record('log', $trace['args']);
|
||
self::record('groupEnd');
|
||
} else {
|
||
self::record('log', $trace_msg);
|
||
}
|
||
}
|
||
self::record('groupEnd');
|
||
}
|
||
|
||
private static function sqlexplain($arr, &$sql, &$css)
|
||
{
|
||
$arr = array_change_key_case($arr, CASE_LOWER);
|
||
if (false !== strpos($arr['extra'], 'Using filesort')) {
|
||
$sql .= ' <---################[Using filesort]';
|
||
$css = self::$css['sql_warn'];
|
||
}
|
||
if (false !== strpos($arr['extra'], 'Using temporary')) {
|
||
$sql .= ' <---################[Using temporary]';
|
||
$css = self::$css['sql_warn'];
|
||
}
|
||
}
|
||
private static function sqlwhere(&$sql, &$css)
|
||
{
|
||
//判断sql语句是否有where
|
||
if (preg_match('/^UPDATE |DELETE /i', $sql) && !preg_match('/WHERE.*(=|>|<|LIKE|IN)/i', $sql)) {
|
||
$sql .= '<---###########[NO WHERE]';
|
||
$css = self::$css['sql_warn'];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 接管报错
|
||
*/
|
||
public static function registerErrorHandler()
|
||
{
|
||
if (!self::check()) {
|
||
return;
|
||
}
|
||
|
||
set_error_handler([__CLASS__, 'error_handler']);
|
||
register_shutdown_function([__CLASS__, 'fatalError']);
|
||
}
|
||
|
||
public static function error_handler($errno, $errstr, $errfile, $errline)
|
||
{
|
||
switch ($errno) {
|
||
case E_WARNING:
|
||
$severity = 'E_WARNING';
|
||
break;
|
||
case E_NOTICE:
|
||
$severity = 'E_NOTICE';
|
||
break;
|
||
case E_USER_ERROR:
|
||
$severity = 'E_USER_ERROR';
|
||
break;
|
||
case E_USER_WARNING:
|
||
$severity = 'E_USER_WARNING';
|
||
break;
|
||
case E_USER_NOTICE:
|
||
$severity = 'E_USER_NOTICE';
|
||
break;
|
||
case E_STRICT:
|
||
$severity = 'E_STRICT';
|
||
break;
|
||
case E_RECOVERABLE_ERROR:
|
||
$severity = 'E_RECOVERABLE_ERROR';
|
||
break;
|
||
case E_DEPRECATED:
|
||
$severity = 'E_DEPRECATED';
|
||
break;
|
||
case E_USER_DEPRECATED:
|
||
$severity = 'E_USER_DEPRECATED';
|
||
break;
|
||
case E_ERROR:
|
||
$severity = 'E_ERR';
|
||
break;
|
||
case E_PARSE:
|
||
$severity = 'E_PARSE';
|
||
break;
|
||
case E_CORE_ERROR:
|
||
$severity = 'E_CORE_ERROR';
|
||
break;
|
||
case E_COMPILE_ERROR:
|
||
$severity = 'E_COMPILE_ERROR';
|
||
break;
|
||
case E_USER_ERROR:
|
||
$severity = 'E_USER_ERROR';
|
||
break;
|
||
default:
|
||
$severity = 'E_UNKNOWN_ERROR_' . $errno;
|
||
break;
|
||
}
|
||
$msg = "{$severity}: {$errstr} in {$errfile} on line {$errline} -- SocketLog error handler";
|
||
self::trace($msg, 2, self::$css['error_handler']);
|
||
}
|
||
|
||
public static function fatalError()
|
||
{
|
||
// 保存日志记录
|
||
if ($e = error_get_last()) {
|
||
self::error_handler($e['type'], $e['message'], $e['file'], $e['line']);
|
||
self::sendLog();
|
||
}
|
||
}
|
||
|
||
protected static function check()
|
||
{
|
||
if (!SLOG_ON) {
|
||
return false;
|
||
}
|
||
$tabid = self::getClientArg('tabid');
|
||
//是否记录日志的检查
|
||
if (!$tabid && !self::$config['force_client_id']) {
|
||
return false;
|
||
}
|
||
//用户认证
|
||
$allow_client_ids = self::$config['allow_client_ids'];
|
||
if (!empty($allow_client_ids)) {
|
||
if (!$tabid && in_array(self::$config['force_client_id'], $allow_client_ids)) {
|
||
return true;
|
||
}
|
||
|
||
$client_id = self::getClientArg('client_id');
|
||
if (!in_array($client_id, $allow_client_ids)) {
|
||
return false;
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
protected static function getClientArg($name)
|
||
{
|
||
static $args = [];
|
||
|
||
$key = 'HTTP_USER_AGENT';
|
||
|
||
if (isset($_SERVER['HTTP_SOCKETLOG'])) {
|
||
$key = 'HTTP_SOCKETLOG';
|
||
}
|
||
|
||
if (!isset($_SERVER[$key])) {
|
||
return null;
|
||
}
|
||
if (empty($args)) {
|
||
if (!preg_match('/SocketLog\((.*?)\)/', $_SERVER[$key], $match)) {
|
||
$args = ['tabid' => null];
|
||
return null;
|
||
}
|
||
parse_str($match[1], $args);
|
||
}
|
||
if (isset($args[$name])) {
|
||
return $args[$name];
|
||
}
|
||
return null;
|
||
}
|
||
|
||
//设置配置
|
||
public static function config($config)
|
||
{
|
||
$config = array_merge(self::$config, $config);
|
||
self::$config = $config;
|
||
if (self::check()) {
|
||
if ($config['optimize']) {
|
||
self::$start_time = microtime(true);
|
||
self::$start_memory = memory_get_usage();
|
||
}
|
||
|
||
if ($config['error_handler']) {
|
||
self::registerErrorHandler();
|
||
}
|
||
}
|
||
}
|
||
|
||
//获得配置
|
||
public static function getConfig($name)
|
||
{
|
||
return isset(self::$config[$name]) ? self::$config[$name] : null;
|
||
}
|
||
|
||
//记录日志
|
||
public static function record($type, $msg = '', $css = '')
|
||
{
|
||
if (!self::check()) {
|
||
return;
|
||
}
|
||
|
||
self::$logs[] = [
|
||
'type' => $type,
|
||
'msg' => $msg,
|
||
'css' => isset(self::$css[$css]) ? self::$css[$css] : $css,
|
||
];
|
||
}
|
||
|
||
/**
|
||
* @param null $host - $host of socket server
|
||
* @param string $message - 发送的消息
|
||
* @param string $address - 地址
|
||
* @return bool
|
||
*/
|
||
public static function send($host, $message = '', $address = '/')
|
||
{
|
||
$url = 'http://' . $host . ':' . self::$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
|
||
$txt = curl_exec($ch);
|
||
return true;
|
||
}
|
||
|
||
public static function sendLog()
|
||
{
|
||
if (!self::check()) {
|
||
return;
|
||
}
|
||
|
||
$time_str = '';
|
||
$memory_str = '';
|
||
if (self::$start_time) {
|
||
$runtime = microtime(true) - self::$start_time;
|
||
$reqs = number_format(1 / $runtime, 2);
|
||
$time_str = "[运行时间：{$runtime}s][吞吐率：{$reqs}req/s]";
|
||
}
|
||
if (self::$start_memory) {
|
||
$memory_use = number_format(memory_get_usage() - self::$start_memory / 1024, 2);
|
||
$memory_str = "[内存消耗：{$memory_use}kb]";
|
||
}
|
||
|
||
if (isset($_SERVER['HTTP_HOST'])) {
|
||
$current_uri = $_SERVER['HTTP_HOST'] . $_SERVER['REQUEST_URI'];
|
||
} else {
|
||
$current_uri = "cmd:" . implode(' ', $_SERVER['argv']);
|
||
}
|
||
array_unshift(self::$logs, [
|
||
'type' => 'group',
|
||
'msg' => $current_uri . $time_str . $memory_str,
|
||
'css' => self::$css['page'],
|
||
]);
|
||
|
||
if (self::$config['show_included_files']) {
|
||
self::$logs[] = [
|
||
'type' => 'groupCollapsed',
|
||
'msg' => 'included_files',
|
||
'css' => '',
|
||
];
|
||
self::$logs[] = [
|
||
'type' => 'log',
|
||
'msg' => implode("\n", get_included_files()),
|
||
'css' => '',
|
||
];
|
||
self::$logs[] = [
|
||
'type' => 'groupEnd',
|
||
'msg' => '',
|
||
'css' => '',
|
||
];
|
||
}
|
||
|
||
self::$logs[] = [
|
||
'type' => 'groupEnd',
|
||
'msg' => '',
|
||
'css' => '',
|
||
];
|
||
|
||
$tabid = self::getClientArg('tabid');
|
||
if (!$client_id = self::getClientArg('client_id')) {
|
||
$client_id = '';
|
||
}
|
||
if ($force_client_id = self::$config['force_client_id']) {
|
||
$client_id = $force_client_id;
|
||
}
|
||
$logs = [
|
||
'tabid' => $tabid,
|
||
'client_id' => $client_id,
|
||
'logs' => self::$logs,
|
||
'force_client_id' => $force_client_id,
|
||
];
|
||
$msg = @json_encode($logs);
|
||
$address = '/' . $client_id; //将client_id作为地址， server端通过地址判断将日志发布给谁
|
||
self::send(self::$config['host'], $msg, $address);
|
||
|
||
}
|
||
|
||
public static function __callStatic($method, $args)
|
||
{
|
||
if (in_array($method, self::$log_types)) {
|
||
array_unshift($args, $method);
|
||
return call_user_func_array(['\think\Slog', 'record'], $args);
|
||
}
|
||
}
|
||
|
||
}
|