mirror of
https://github.com/woodchen-ink/proxy-go.git
synced 2025-07-18 08:31:55 +08:00
- Introduced safe type conversion functions for memory usage, status code stats, top paths, and recent requests to prevent panics when accessing metrics data. - Updated MetricsHandler to utilize these safe functions, improving the robustness of metrics handling. - Refactored GetStats method in Collector to ensure all fields are initialized and to streamline the retrieval of metrics data. - Enhanced the overall structure of metrics data for better clarity and reliability in reporting.
841 lines
26 KiB
Go
841 lines
26 KiB
Go
package handler
|
||
|
||
import (
|
||
"encoding/json"
|
||
"log"
|
||
"net/http"
|
||
"proxy-go/internal/metrics"
|
||
"proxy-go/internal/models"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
type Metrics struct {
|
||
// 基础指标
|
||
Uptime string `json:"uptime"`
|
||
ActiveRequests int64 `json:"active_requests"`
|
||
TotalRequests int64 `json:"total_requests"`
|
||
TotalErrors int64 `json:"total_errors"`
|
||
ErrorRate float64 `json:"error_rate"`
|
||
|
||
// 系统指标
|
||
NumGoroutine int `json:"num_goroutine"`
|
||
MemoryUsage string `json:"memory_usage"`
|
||
|
||
// 性能指标
|
||
AverageResponseTime string `json:"avg_response_time"`
|
||
RequestsPerSecond float64 `json:"requests_per_second"`
|
||
|
||
// 新增字段
|
||
TotalBytes int64 `json:"total_bytes"`
|
||
BytesPerSecond float64 `json:"bytes_per_second"`
|
||
StatusCodeStats map[string]int64 `json:"status_code_stats"`
|
||
LatencyPercentiles map[string]float64 `json:"latency_percentiles"`
|
||
TopPaths []models.PathMetrics `json:"top_paths"`
|
||
RecentRequests []models.RequestLog `json:"recent_requests"`
|
||
TopReferers []models.PathMetrics `json:"top_referers"`
|
||
}
|
||
|
||
func (h *ProxyHandler) MetricsHandler(w http.ResponseWriter, r *http.Request) {
|
||
uptime := time.Since(h.startTime)
|
||
collector := metrics.GetCollector()
|
||
stats := collector.GetStats()
|
||
|
||
if stats == nil {
|
||
http.Error(w, "Failed to get metrics", http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
// 添加安全的类型转换函数
|
||
safeInt64 := func(v interface{}) int64 {
|
||
if v == nil {
|
||
return 0
|
||
}
|
||
if i, ok := v.(int64); ok {
|
||
return i
|
||
}
|
||
return 0
|
||
}
|
||
|
||
safeInt := func(v interface{}) int {
|
||
if v == nil {
|
||
return 0
|
||
}
|
||
if i, ok := v.(int); ok {
|
||
return i
|
||
}
|
||
return 0
|
||
}
|
||
|
||
// 添加安全的字符串转换函数
|
||
safeString := func(v interface{}) string {
|
||
if v == nil {
|
||
return "0 B" // 返回默认值
|
||
}
|
||
if s, ok := v.(string); ok {
|
||
return s
|
||
}
|
||
return "0 B" // 返回默认值
|
||
}
|
||
|
||
totalRequests := safeInt64(stats["total_requests"])
|
||
metrics := Metrics{
|
||
Uptime: uptime.String(),
|
||
ActiveRequests: safeInt64(stats["active_requests"]),
|
||
TotalRequests: totalRequests,
|
||
TotalErrors: safeInt64(stats["total_errors"]),
|
||
ErrorRate: float64(safeInt64(stats["total_errors"])) / float64(max(totalRequests, 1)),
|
||
NumGoroutine: safeInt(stats["num_goroutine"]),
|
||
MemoryUsage: safeString(stats["memory_usage"]), // 使用安全转换
|
||
AverageResponseTime: metrics.FormatDuration(time.Duration(safeInt64(stats["avg_latency"]))),
|
||
TotalBytes: safeInt64(stats["total_bytes"]),
|
||
BytesPerSecond: float64(safeInt64(stats["total_bytes"])) / metrics.Max(uptime.Seconds(), 1),
|
||
RequestsPerSecond: float64(totalRequests) / metrics.Max(uptime.Seconds(), 1),
|
||
StatusCodeStats: safeStatusCodeStats(stats["status_code_stats"]), // 添加安全转换
|
||
TopPaths: safePathMetrics(stats["top_paths"]), // 添加安全转换
|
||
RecentRequests: safeRequestLogs(stats["recent_requests"]), // 添加安全转换
|
||
TopReferers: safePathMetrics(stats["top_referers"]), // 添加安全转换
|
||
}
|
||
|
||
w.Header().Set("Content-Type", "application/json")
|
||
if err := json.NewEncoder(w).Encode(metrics); err != nil {
|
||
log.Printf("Error encoding metrics: %v", err)
|
||
}
|
||
}
|
||
|
||
// 辅助函数
|
||
func max(a, b int64) int64 {
|
||
if a > b {
|
||
return a
|
||
}
|
||
return b
|
||
}
|
||
|
||
// 修改模板,添加登录页面
|
||
var loginTemplate = `
|
||
<!DOCTYPE html>
|
||
<html>
|
||
<head>
|
||
<title>Proxy-Go Metrics Login</title>
|
||
<meta charset="UTF-8">
|
||
<style>
|
||
body {
|
||
font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif;
|
||
display: flex;
|
||
justify-content: center;
|
||
align-items: center;
|
||
height: 100vh;
|
||
margin: 0;
|
||
background: #f5f5f5;
|
||
}
|
||
.login-card {
|
||
background: white;
|
||
padding: 30px;
|
||
border-radius: 8px;
|
||
box-shadow: 0 2px 4px rgba(0,0,0,0.1);
|
||
width: 300px;
|
||
}
|
||
.login-title {
|
||
text-align: center;
|
||
margin-bottom: 20px;
|
||
color: #333;
|
||
}
|
||
.input-group {
|
||
margin-bottom: 15px;
|
||
}
|
||
input {
|
||
width: 100%;
|
||
padding: 8px;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
box-sizing: border-box;
|
||
}
|
||
button {
|
||
width: 100%;
|
||
padding: 10px;
|
||
background: #4CAF50;
|
||
color: white;
|
||
border: none;
|
||
border-radius: 4px;
|
||
cursor: pointer;
|
||
}
|
||
button:hover {
|
||
background: #45a049;
|
||
}
|
||
.error {
|
||
color: red;
|
||
text-align: center;
|
||
margin-bottom: 15px;
|
||
display: none;
|
||
}
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<div class="login-card">
|
||
<h2 class="login-title">Metrics Login</h2>
|
||
<div id="error" class="error">密码错误</div>
|
||
<div class="input-group">
|
||
<input type="password" id="password" placeholder="请输入密码">
|
||
</div>
|
||
<button onclick="login()">登录</button>
|
||
</div>
|
||
|
||
<script>
|
||
function login() {
|
||
const password = document.getElementById('password').value;
|
||
fetch('/metrics/auth', {
|
||
method: 'POST',
|
||
headers: {
|
||
'Content-Type': 'application/json',
|
||
},
|
||
body: JSON.stringify({password: password})
|
||
})
|
||
.then(response => {
|
||
if (response.ok) {
|
||
// 登录成功,保存token并跳转
|
||
response.json().then(data => {
|
||
localStorage.setItem('metricsToken', data.token);
|
||
window.location.href = '/metrics/dashboard';
|
||
});
|
||
} else {
|
||
// 显示错误信息
|
||
document.getElementById('error').style.display = 'block';
|
||
}
|
||
})
|
||
.catch(error => {
|
||
console.error('Error:', error);
|
||
document.getElementById('error').style.display = 'block';
|
||
});
|
||
}
|
||
</script>
|
||
</body>
|
||
</html>
|
||
`
|
||
|
||
// 修改原有的 metricsTemplate,添加 token 检查
|
||
var metricsTemplate = `
|
||
<!DOCTYPE html>
|
||
<html>
|
||
<head>
|
||
<title>Proxy-Go Metrics</title>
|
||
<meta charset="UTF-8">
|
||
<style>
|
||
body {
|
||
font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
|
||
max-width: 1200px;
|
||
margin: 0 auto;
|
||
padding: 20px;
|
||
background: #f5f5f5;
|
||
}
|
||
.card {
|
||
background: white;
|
||
border-radius: 8px;
|
||
padding: 20px;
|
||
margin-bottom: 20px;
|
||
box-shadow: 0 2px 4px rgba(0,0,0,0.1);
|
||
}
|
||
.metric {
|
||
display: flex;
|
||
justify-content: space-between;
|
||
padding: 10px 0;
|
||
border-bottom: 1px solid #eee;
|
||
}
|
||
.metric:last-child {
|
||
border-bottom: none;
|
||
}
|
||
.metric-label {
|
||
color: #666;
|
||
}
|
||
.metric-value {
|
||
font-weight: bold;
|
||
color: #333;
|
||
}
|
||
h1 {
|
||
color: #333;
|
||
margin-bottom: 30px;
|
||
}
|
||
h2 {
|
||
color: #666;
|
||
margin: 0 0 15px 0;
|
||
}
|
||
.refresh {
|
||
position: fixed;
|
||
top: 20px;
|
||
right: 20px;
|
||
padding: 10px 20px;
|
||
background: #4CAF50;
|
||
color: white;
|
||
border: none;
|
||
border-radius: 4px;
|
||
cursor: pointer;
|
||
}
|
||
.refresh:hover {
|
||
background: #45a049;
|
||
}
|
||
#lastUpdate {
|
||
position: fixed;
|
||
top: 20px;
|
||
right: 140px;
|
||
color: #666;
|
||
}
|
||
/* 添加表格样式 */
|
||
table {
|
||
width: 100%;
|
||
border-collapse: collapse;
|
||
margin: 10px 0;
|
||
}
|
||
th, td {
|
||
padding: 8px;
|
||
text-align: left;
|
||
border-bottom: 1px solid #eee;
|
||
}
|
||
th {
|
||
background: #f8f9fa;
|
||
color: #666;
|
||
}
|
||
.status-badge {
|
||
padding: 3px 8px;
|
||
border-radius: 12px;
|
||
font-size: 12px;
|
||
color: white;
|
||
}
|
||
.status-2xx { background: #28a745; }
|
||
.status-3xx { background: #17a2b8; }
|
||
.status-4xx { background: #ffc107; }
|
||
.status-5xx { background: #dc3545; }
|
||
.grid-container {
|
||
display: grid;
|
||
grid-template-columns: repeat(2, 1fr);
|
||
gap: 20px;
|
||
margin-bottom: 20px;
|
||
}
|
||
.grid-container .card {
|
||
margin-bottom: 0;
|
||
}
|
||
.chart-container {
|
||
margin-top: 20px;
|
||
}
|
||
.chart {
|
||
height: 200px;
|
||
margin-bottom: 20px;
|
||
}
|
||
#timeRange {
|
||
padding: 8px;
|
||
border-radius: 4px;
|
||
border: 1px solid #ddd;
|
||
margin-bottom: 15px;
|
||
}
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<h1>Proxy-Go Metrics</h1>
|
||
|
||
<div class="grid-container">
|
||
<div class="card">
|
||
<h2>基础指标</h2>
|
||
<div class="metric">
|
||
<span class="metric-label">运行时间</span>
|
||
<span class="metric-value" id="uptime"></span>
|
||
</div>
|
||
<div class="metric">
|
||
<span class="metric-label">当前活跃请求</span>
|
||
<span class="metric-value" id="activeRequests"></span>
|
||
</div>
|
||
<div class="metric">
|
||
<span class="metric-label">总请求数</span>
|
||
<span class="metric-value" id="totalRequests"></span>
|
||
</div>
|
||
<div class="metric">
|
||
<span class="metric-label">错误数</span>
|
||
<span class="metric-value" id="totalErrors"></span>
|
||
</div>
|
||
<div class="metric">
|
||
<span class="metric-label">错误率</span>
|
||
<span class="metric-value" id="errorRate"></span>
|
||
</div>
|
||
</div>
|
||
|
||
<div class="card">
|
||
<h2>系统指标</h2>
|
||
<div class="metric">
|
||
<span class="metric-label">Goroutine数量</span>
|
||
<span class="metric-value" id="numGoroutine"></span>
|
||
</div>
|
||
<div class="metric">
|
||
<span class="metric-label">内存使用</span>
|
||
<span class="metric-value" id="memoryUsage"></span>
|
||
</div>
|
||
</div>
|
||
|
||
<div class="card">
|
||
<h2>性能指标</h2>
|
||
<div class="metric">
|
||
<span class="metric-label">平均响应时间</span>
|
||
<span class="metric-value" id="avgResponseTime"></span>
|
||
</div>
|
||
<div class="metric">
|
||
<span class="metric-label">每秒请求数</span>
|
||
<span class="metric-value" id="requestsPerSecond"></span>
|
||
</div>
|
||
</div>
|
||
|
||
<div class="card">
|
||
<h2>流量统计</h2>
|
||
<div class="metric">
|
||
<span class="metric-label">总传输字节</span>
|
||
<span class="metric-value" id="totalBytes"></span>
|
||
</div>
|
||
<div class="metric">
|
||
<span class="metric-label">每秒传输</span>
|
||
<span class="metric-value" id="bytesPerSecond"></span>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
<div class="card">
|
||
<h2>状态码统计</h2>
|
||
<div id="statusCodes"></div>
|
||
</div>
|
||
|
||
<div class="card">
|
||
<h2>热门路径 (Top 10)</h2>
|
||
<table id="topPaths">
|
||
<thead>
|
||
<tr>
|
||
<th>路径</th>
|
||
<th>请求数</th>
|
||
<th>错误数</th>
|
||
<th>平均延迟</th>
|
||
<th>传输大小</th>
|
||
</tr>
|
||
</thead>
|
||
<tbody></tbody>
|
||
</table>
|
||
</div>
|
||
|
||
<div class="card">
|
||
<h2>最近请求</h2>
|
||
<table id="recentRequests">
|
||
<thead>
|
||
<tr>
|
||
<th>时间</th>
|
||
<th>路径</th>
|
||
<th>状态</th>
|
||
<th>延迟</th>
|
||
<th>大小</th>
|
||
<th>客户端IP</th>
|
||
</tr>
|
||
</thead>
|
||
<tbody></tbody>
|
||
</table>
|
||
</div>
|
||
|
||
<div class="card">
|
||
<h2>热门引用来源 (Top 10)</h2>
|
||
<table id="topReferers">
|
||
<thead>
|
||
<tr>
|
||
<th>来源</th>
|
||
<th>请求数</th>
|
||
</tr>
|
||
</thead>
|
||
<tbody></tbody>
|
||
</table>
|
||
</div>
|
||
|
||
<div class="card">
|
||
<h2>历史数据</h2>
|
||
<select id="timeRange">
|
||
<option value="1">最近1小时</option>
|
||
<option value="6">最近6小时</option>
|
||
<option value="24">最近24小时</option>
|
||
<option value="168">最近7天</option>
|
||
<option value="720">最近30天</option>
|
||
</select>
|
||
<div id="historyChart"></div>
|
||
</div>
|
||
|
||
<span id="lastUpdate"></span>
|
||
<button class="refresh" onclick="refreshMetrics()">刷新</button>
|
||
|
||
<script>
|
||
// 检查登录状态
|
||
const token = localStorage.getItem('metricsToken');
|
||
if (!token) {
|
||
window.location.href = '/metrics/ui';
|
||
}
|
||
|
||
function formatBytes(bytes) {
|
||
if (bytes === 0) return '0 B';
|
||
const k = 1024;
|
||
const sizes = ['B', 'KB', 'MB', 'GB'];
|
||
const i = Math.floor(Math.log(bytes) / Math.log(k));
|
||
return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
|
||
}
|
||
|
||
function formatDate(dateStr) {
|
||
const date = new Date(dateStr);
|
||
return date.toLocaleTimeString();
|
||
}
|
||
|
||
function formatLatency(nanoseconds) {
|
||
if (nanoseconds < 1000) {
|
||
return nanoseconds + ' ns';
|
||
} else if (nanoseconds < 1000000) {
|
||
return (nanoseconds / 1000).toFixed(2) + ' µs';
|
||
} else if (nanoseconds < 1000000000) {
|
||
return (nanoseconds / 1000000).toFixed(2) + ' ms';
|
||
} else {
|
||
return (nanoseconds / 1000000000).toFixed(2) + ' s';
|
||
}
|
||
}
|
||
|
||
function updateMetrics(data) {
|
||
// 更新现有指标
|
||
document.getElementById('uptime').textContent = data.uptime;
|
||
document.getElementById('activeRequests').textContent = data.active_requests;
|
||
document.getElementById('totalRequests').textContent = data.total_requests;
|
||
document.getElementById('totalErrors').textContent = data.total_errors;
|
||
document.getElementById('errorRate').textContent = (data.error_rate * 100).toFixed(2) + '%';
|
||
document.getElementById('numGoroutine').textContent = data.num_goroutine;
|
||
document.getElementById('memoryUsage').textContent = data.memory_usage;
|
||
document.getElementById('avgResponseTime').textContent = data.avg_response_time;
|
||
document.getElementById('requestsPerSecond').textContent = data.requests_per_second.toFixed(2);
|
||
|
||
// 更新流量统计
|
||
document.getElementById('totalBytes').textContent = formatBytes(data.total_bytes);
|
||
document.getElementById('bytesPerSecond').textContent = formatBytes(data.bytes_per_second) + '/s';
|
||
|
||
// 更新状态码统计
|
||
const statusCodesHtml = Object.entries(data.status_code_stats)
|
||
.map(([status, count]) => {
|
||
const statusClass = 'status-' + status.charAt(0) + 'xx';
|
||
return '<div class="metric">' +
|
||
'<span class="metric-label">' +
|
||
'<span class="status-badge ' + statusClass + '">' + status + '</span>' +
|
||
'</span>' +
|
||
'<span class="metric-value">' + count + '</span>' +
|
||
'</div>';
|
||
})
|
||
.join('');
|
||
document.getElementById('statusCodes').innerHTML = statusCodesHtml;
|
||
|
||
// 更新热门路径
|
||
const topPathsHtml = data.top_paths.map(path =>
|
||
'<tr>' +
|
||
'<td>' + path.path + '</td>' +
|
||
'<td>' + path.request_count + '</td>' +
|
||
'<td>' + path.error_count + '</td>' +
|
||
'<td>' + path.avg_latency + '</td>' +
|
||
'<td>' + formatBytes(path.bytes_transferred) + '</td>' +
|
||
'</tr>'
|
||
).join('');
|
||
document.querySelector('#topPaths tbody').innerHTML = topPathsHtml;
|
||
|
||
// 更新最近请求
|
||
const recentRequestsHtml = data.recent_requests.map(req =>
|
||
'<tr>' +
|
||
'<td>' + formatDate(req.Time) + '</td>' +
|
||
'<td>' + req.Path + '</td>' +
|
||
'<td><span class="status-badge status-' + Math.floor(req.Status/100) + 'xx">' + req.Status + '</span></td>' +
|
||
'<td>' + formatLatency(req.Latency) + '</td>' +
|
||
'<td>' + formatBytes(req.BytesSent) + '</td>' +
|
||
'<td>' + req.ClientIP + '</td>' +
|
||
'</tr>'
|
||
).join('');
|
||
document.querySelector('#recentRequests tbody').innerHTML = recentRequestsHtml;
|
||
|
||
// 更新热门引用来源
|
||
const topReferersHtml = data.top_referers.map(referer =>
|
||
'<tr>' +
|
||
'<td>' + referer.path + '</td>' +
|
||
'<td>' + referer.request_count + '</td>' +
|
||
'</tr>'
|
||
).join('');
|
||
document.querySelector('#topReferers tbody').innerHTML = topReferersHtml;
|
||
|
||
document.getElementById('lastUpdate').textContent = '最后更新: ' + new Date().toLocaleTimeString();
|
||
}
|
||
|
||
function refreshMetrics() {
|
||
fetch('/metrics', {
|
||
headers: {
|
||
'Authorization': 'Bearer ' + token
|
||
}
|
||
})
|
||
.then(response => {
|
||
if (response.status === 401) {
|
||
// token 无效,跳转到登录页
|
||
localStorage.removeItem('metricsToken');
|
||
window.location.href = '/metrics/ui';
|
||
return;
|
||
}
|
||
return response.json();
|
||
})
|
||
.then(data => {
|
||
if (data) updateMetrics(data);
|
||
})
|
||
.catch(error => console.error('Error:', error));
|
||
}
|
||
|
||
// 初始加载
|
||
refreshMetrics();
|
||
|
||
// 每5秒自动刷新
|
||
setInterval(refreshMetrics, 5000);
|
||
|
||
// <20><>加图表相关代码
|
||
function loadHistoryData() {
|
||
const hours = document.getElementById('timeRange').value;
|
||
fetch('/metrics/history?hours=' + hours, {
|
||
headers: {
|
||
'Authorization': 'Bearer ' + token
|
||
}
|
||
})
|
||
.then(response => response.json())
|
||
.then(data => {
|
||
const labels = data.map(m => {
|
||
const date = new Date(m.timestamp);
|
||
if (hours <= 24) {
|
||
return date.toLocaleTimeString();
|
||
} else if (hours <= 168) {
|
||
return date.toLocaleDateString() + ' ' + date.toLocaleTimeString();
|
||
} else {
|
||
return date.toLocaleDateString();
|
||
}
|
||
});
|
||
|
||
// 清除旧图表
|
||
document.getElementById('historyChart').innerHTML = '';
|
||
|
||
// 创建新图表
|
||
document.getElementById('historyChart').innerHTML = '<div class="chart-container">' +
|
||
'<h3>请求数</h3>' +
|
||
'<div class="chart">' +
|
||
'<canvas id="requestsChart"></canvas>' +
|
||
'</div>' +
|
||
'<h3>错误率</h3>' +
|
||
'<div class="chart">' +
|
||
'<canvas id="errorRateChart"></canvas>' +
|
||
'</div>' +
|
||
'<h3>流量</h3>' +
|
||
'<div class="chart">' +
|
||
'<canvas id="bytesChart"></canvas>' +
|
||
'</div>' +
|
||
'</div>';
|
||
|
||
// 绘制图表
|
||
new Chart(document.getElementById('requestsChart'), {
|
||
type: 'line',
|
||
data: {
|
||
labels: labels,
|
||
datasets: [{
|
||
label: '总请求数',
|
||
data: data.map(m => m.total_requests),
|
||
borderColor: '#4CAF50',
|
||
fill: false
|
||
}]
|
||
},
|
||
options: {
|
||
scales: {
|
||
x: {
|
||
ticks: {
|
||
maxRotation: 45,
|
||
minRotation: 45
|
||
}
|
||
}
|
||
}
|
||
}
|
||
});
|
||
|
||
new Chart(document.getElementById('errorRateChart'), {
|
||
type: 'line',
|
||
data: {
|
||
labels: labels,
|
||
datasets: [{
|
||
label: '错误率',
|
||
data: data.map(m => (m.error_rate * 100).toFixed(2)),
|
||
borderColor: '#dc3545',
|
||
fill: false
|
||
}]
|
||
},
|
||
options: {
|
||
scales: {
|
||
x: {
|
||
ticks: {
|
||
maxRotation: 45,
|
||
minRotation: 45
|
||
}
|
||
}
|
||
}
|
||
}
|
||
});
|
||
|
||
new Chart(document.getElementById('bytesChart'), {
|
||
type: 'line',
|
||
data: {
|
||
labels: labels,
|
||
datasets: [{
|
||
label: '传输字节',
|
||
data: data.map(m => m.total_bytes / 1024 / 1024), // 转换为MB
|
||
borderColor: '#17a2b8',
|
||
fill: false
|
||
}]
|
||
},
|
||
options: {
|
||
scales: {
|
||
x: {
|
||
ticks: {
|
||
maxRotation: 45,
|
||
minRotation: 45
|
||
}
|
||
}
|
||
}
|
||
}
|
||
});
|
||
});
|
||
}
|
||
|
||
// 监听时间范围变化
|
||
document.getElementById('timeRange').addEventListener('change', loadHistoryData);
|
||
|
||
// 初始加载历史数据
|
||
loadHistoryData();
|
||
</script>
|
||
|
||
<!-- 添加 Chart.js -->
|
||
<script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
|
||
</body>
|
||
</html>
|
||
`
|
||
|
||
// 添加认证中间件
|
||
func (h *ProxyHandler) AuthMiddleware(next http.HandlerFunc) http.HandlerFunc {
|
||
return func(w http.ResponseWriter, r *http.Request) {
|
||
auth := r.Header.Get("Authorization")
|
||
if auth == "" || !strings.HasPrefix(auth, "Bearer ") {
|
||
http.Error(w, "Unauthorized", http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
token := strings.TrimPrefix(auth, "Bearer ")
|
||
if !h.auth.validateToken(token) {
|
||
http.Error(w, "Invalid token", http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
next(w, r)
|
||
}
|
||
}
|
||
|
||
// 修改处理器
|
||
func (h *ProxyHandler) MetricsPageHandler(w http.ResponseWriter, r *http.Request) {
|
||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
w.Write([]byte(loginTemplate))
|
||
}
|
||
|
||
func (h *ProxyHandler) MetricsDashboardHandler(w http.ResponseWriter, r *http.Request) {
|
||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
metricsTemplate = strings.Replace(metricsTemplate,
|
||
`</div>
|
||
|
||
<span id="lastUpdate"></span>`,
|
||
`</div>
|
||
|
||
<div class="card">
|
||
<h2>历史数据</h2>
|
||
<select id="timeRange">
|
||
<option value="1">最近1小时</option>
|
||
<option value="6">最近6小时</option>
|
||
<option value="24">最近24小时</option>
|
||
<option value="168">最近7天</option>
|
||
<option value="720">最近30天</option>
|
||
</select>
|
||
<div id="historyChart"></div>
|
||
</div>
|
||
|
||
<span id="lastUpdate"></span>`, 1)
|
||
|
||
w.Write([]byte(metricsTemplate))
|
||
}
|
||
|
||
func (h *ProxyHandler) MetricsAuthHandler(w http.ResponseWriter, r *http.Request) {
|
||
if r.Method != http.MethodPost {
|
||
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
|
||
return
|
||
}
|
||
|
||
var req struct {
|
||
Password string `json:"password"`
|
||
}
|
||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||
http.Error(w, "Invalid request", http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
if req.Password != h.config.Metrics.Password {
|
||
http.Error(w, "Invalid password", http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
token := h.auth.generateToken()
|
||
h.auth.addToken(token, time.Duration(h.config.Metrics.TokenExpiry)*time.Second)
|
||
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(map[string]string{
|
||
"token": token,
|
||
})
|
||
}
|
||
|
||
// 添加历史数据查询接口
|
||
func (h *ProxyHandler) MetricsHistoryHandler(w http.ResponseWriter, r *http.Request) {
|
||
hours := 24 // 默认24小时
|
||
if h := r.URL.Query().Get("hours"); h != "" {
|
||
if parsed, err := strconv.Atoi(h); err == nil && parsed > 0 {
|
||
hours = parsed
|
||
}
|
||
}
|
||
|
||
collector := metrics.GetCollector()
|
||
metrics, err := collector.GetDB().GetRecentMetrics(hours)
|
||
if err != nil {
|
||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
w.Header().Set("Content-Type", "application/json")
|
||
json.NewEncoder(w).Encode(metrics)
|
||
}
|
||
|
||
// 添加安全的类型转换辅助函数
|
||
func safeStatusCodeStats(v interface{}) map[string]int64 {
|
||
if v == nil {
|
||
return make(map[string]int64)
|
||
}
|
||
if m, ok := v.(map[string]int64); ok {
|
||
return m
|
||
}
|
||
return make(map[string]int64)
|
||
}
|
||
|
||
func safePathMetrics(v interface{}) []models.PathMetrics {
|
||
if v == nil {
|
||
return []models.PathMetrics{}
|
||
}
|
||
if m, ok := v.([]models.PathMetrics); ok {
|
||
return m
|
||
}
|
||
return []models.PathMetrics{}
|
||
}
|
||
|
||
func safeRequestLogs(v interface{}) []models.RequestLog {
|
||
if v == nil {
|
||
return []models.RequestLog{}
|
||
}
|
||
if m, ok := v.([]models.RequestLog); ok {
|
||
return m
|
||
}
|
||
return []models.RequestLog{}
|
||
}
|