初始化仓库
This commit is contained in:
109
vendor/topthink/think-trace/src/TraceDebug.php
vendored
Normal file
109
vendor/topthink/think-trace/src/TraceDebug.php
vendored
Normal file
@ -0,0 +1,109 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006~2019 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
declare (strict_types = 1);
|
||||
|
||||
namespace think\trace;
|
||||
|
||||
use Closure;
|
||||
use think\App;
|
||||
use think\Config;
|
||||
use think\event\LogWrite;
|
||||
use think\Request;
|
||||
use think\Response;
|
||||
use think\response\Redirect;
|
||||
|
||||
/**
|
||||
* 页面Trace中间件
|
||||
*/
|
||||
class TraceDebug
|
||||
{
|
||||
|
||||
/**
|
||||
* Trace日志
|
||||
* @var array
|
||||
*/
|
||||
protected $log = [];
|
||||
|
||||
/**
|
||||
* 配置参数
|
||||
* @var array
|
||||
*/
|
||||
protected $config = [];
|
||||
|
||||
/** @var App */
|
||||
protected $app;
|
||||
|
||||
public function __construct(App $app, Config $config)
|
||||
{
|
||||
$this->app = $app;
|
||||
$this->config = $config->get('trace');
|
||||
}
|
||||
|
||||
/**
|
||||
* 页面Trace调试
|
||||
* @access public
|
||||
* @param Request $request
|
||||
* @param Closure $next
|
||||
* @return void
|
||||
*/
|
||||
public function handle($request, Closure $next)
|
||||
{
|
||||
$debug = $this->app->isDebug();
|
||||
|
||||
// 注册日志监听
|
||||
if ($debug) {
|
||||
$this->log = [];
|
||||
$this->app->event->listen(LogWrite::class, function ($event) {
|
||||
if (empty($this->config['channel']) || $this->config['channel'] == $event->channel) {
|
||||
$this->log = array_merge_recursive($this->log, $event->log);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
$response = $next($request);
|
||||
|
||||
// Trace调试注入
|
||||
if ($debug) {
|
||||
$data = $response->getContent();
|
||||
$this->traceDebug($response, $data);
|
||||
$response->content($data);
|
||||
}
|
||||
|
||||
return $response;
|
||||
}
|
||||
|
||||
public function traceDebug(Response $response, &$content)
|
||||
{
|
||||
$config = $this->config;
|
||||
$type = $config['type'] ?? 'Html';
|
||||
|
||||
unset($config['type']);
|
||||
|
||||
$trace = App::factory($type, '\\think\\trace\\', $config);
|
||||
|
||||
if ($response instanceof Redirect) {
|
||||
//TODO 记录
|
||||
} else {
|
||||
$log = $this->app->log->getLog($config['channel'] ?? '');
|
||||
$log = array_merge_recursive($this->log, $log);
|
||||
$output = $trace->output($this->app, $response, $log);
|
||||
if (is_string($output)) {
|
||||
// trace调试信息注入
|
||||
$pos = strripos($content, '</body>');
|
||||
if (false !== $pos) {
|
||||
$content = substr($content, 0, $pos) . $output . substr($content, $pos);
|
||||
} else {
|
||||
$content = $content . $output;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user