web日志查询增加修复以及日志数据持久化

dns/server.go

@@ -638,7 +638,7 @@ func (s *Server) GetStats() *Stats {
 }
 
 // GetQueryLogs 获取查询日志
-func (s *Server) GetQueryLogs(limit, offset int, sortField, sortDirection string) []QueryLog {
+func (s *Server) GetQueryLogs(limit, offset int, sortField, sortDirection, resultFilter, searchTerm string) []QueryLog {
 	s.queryLogsMutex.RLock()
 	defer s.queryLogsMutex.RUnlock()
 
@@ -650,9 +650,26 @@ func (s *Server) GetQueryLogs(limit, offset int, sortField, sortDirection string
 		limit = 100 // 默认返回100条日志
 	}
 
-	// 创建日志副本用于排序
-	logsCopy := make([]QueryLog, len(s.queryLogs))
-	copy(logsCopy, s.queryLogs)
+	// 创建日志副本用于过滤和排序
+	var logsCopy []QueryLog
+
+	// 先过滤日志
+	for _, log := range s.queryLogs {
+		// 应用结果过滤
+		if resultFilter != "" && log.Result != resultFilter {
+			continue
+		}
+
+		// 应用搜索过滤
+		if searchTerm != "" {
+			// 搜索域名或客户端IP
+			if !strings.Contains(log.Domain, searchTerm) && !strings.Contains(log.ClientIP, searchTerm) {
+				continue
+			}
+		}
+
+		logsCopy = append(logsCopy, log)
+	}
 
 	// 排序日志
 	if sortField != "" {
@@ -957,6 +974,58 @@ func (s *Server) loadStatsData() {
 	s.clientStatsMutex.Unlock()
 
 	logger.Info("统计数据加载成功")
+
+	// 加载查询日志
+	s.loadQueryLogs()
+}
+
+// loadQueryLogs 从文件加载查询日志
+func (s *Server) loadQueryLogs() {
+	if s.config.StatsFile == "" {
+		return
+	}
+
+	// 获取绝对路径
+	statsFilePath, err := filepath.Abs(s.config.StatsFile)
+	if err != nil {
+		logger.Error("获取统计文件绝对路径失败", "path", s.config.StatsFile, "error", err)
+		return
+	}
+
+	// 构建查询日志文件路径
+	queryLogPath := filepath.Join(filepath.Dir(statsFilePath), "querylog.json")
+
+	// 检查文件是否存在
+	if _, err := os.Stat(queryLogPath); os.IsNotExist(err) {
+		logger.Info("查询日志文件不存在，将使用空列表", "file", queryLogPath)
+		return
+	}
+
+	// 读取文件内容
+	data, err := ioutil.ReadFile(queryLogPath)
+	if err != nil {
+		logger.Error("读取查询日志文件失败", "error", err)
+		return
+	}
+
+	// 解析数据
+	var logs []QueryLog
+	err = json.Unmarshal(data, &logs)
+	if err != nil {
+		logger.Error("解析查询日志失败", "error", err)
+		return
+	}
+
+	// 更新查询日志
+	s.queryLogsMutex.Lock()
+	s.queryLogs = logs
+	// 确保日志数量不超过限制
+	if len(s.queryLogs) > s.maxQueryLogs {
+		s.queryLogs = s.queryLogs[:s.maxQueryLogs]
+	}
+	s.queryLogsMutex.Unlock()
+
+	logger.Info("查询日志加载成功", "count", len(logs))
 }
 
 // saveStatsData 保存统计数据到文件
@@ -1045,6 +1114,37 @@ func (s *Server) saveStatsData() {
 	}
 
 	logger.Info("统计数据保存成功", "file", statsFilePath)
+
+	// 保存查询日志到文件
+	s.saveQueryLogs(statsDir)
+}
+
+// saveQueryLogs 保存查询日志到文件
+func (s *Server) saveQueryLogs(dataDir string) {
+	// 构建查询日志文件路径
+	queryLogPath := filepath.Join(dataDir, "querylog.json")
+
+	// 获取查询日志数据
+	s.queryLogsMutex.RLock()
+	logsCopy := make([]QueryLog, len(s.queryLogs))
+	copy(logsCopy, s.queryLogs)
+	s.queryLogsMutex.RUnlock()
+
+	// 序列化数据
+	jsonData, err := json.MarshalIndent(logsCopy, "", "  ")
+	if err != nil {
+		logger.Error("序列化查询日志失败", "error", err)
+		return
+	}
+
+	// 写入文件
+	err = os.WriteFile(queryLogPath, jsonData, 0644)
+	if err != nil {
+		logger.Error("保存查询日志到文件失败", "file", queryLogPath, "error", err)
+		return
+	}
+
+	logger.Info("查询日志保存成功", "file", queryLogPath)
 }
 
 // startCpuUsageMonitor 启动CPU使用率监控