修复响应时间异常

dns/server.go

@@ -327,8 +327,8 @@ func (s *Server) handleDNSRequest(w dns.ResponseWriter, r *dns.Msg) {
 		response.SetRcode(r, dns.RcodeRefused)
 		w.WriteMsg(response)
 
-		// 计算响应时间
-		responseTime := time.Since(startTime).Milliseconds()
+		// 缓存命中，响应时间设为0ms
+		responseTime := int64(0)
 		s.updateStats(func(stats *Stats) {
 			stats.TotalResponseTime += responseTime
 			if stats.Queries > 0 {
@@ -344,8 +344,8 @@ func (s *Server) handleDNSRequest(w dns.ResponseWriter, r *dns.Msg) {
 	// 检查hosts文件是否有匹配
 	if ip, exists := s.shieldManager.GetHostsIP(domain); exists {
 		s.handleHostsResponse(w, r, ip)
-		// 计算响应时间
-		responseTime := time.Since(startTime).Milliseconds()
+		// 缓存命中，响应时间设为0ms
+		responseTime := int64(0)
 		s.updateStats(func(stats *Stats) {
 			stats.TotalResponseTime += responseTime
 			if stats.Queries > 0 {
@@ -404,14 +404,19 @@ func (s *Server) handleDNSRequest(w dns.ResponseWriter, r *dns.Msg) {
 	}
 
 	// 缓存未命中，转发到上游DNS服务器
-	response, _ := s.forwardDNSRequestWithCache(r, domain)
+	response, rtt := s.forwardDNSRequestWithCache(r, domain)
 	if response != nil {
 		// 写入响应给客户端
 		w.WriteMsg(response)
 	}
 
-	// 计算响应时间
-	responseTime := time.Since(startTime).Milliseconds()
+	// 使用上游服务器的实际响应时间（转换为毫秒）
+	responseTime := int64(rtt.Milliseconds())
+	// 如果rtt为0（查询失败），则使用本地计算的时间
+	if responseTime == 0 {
+		responseTime = time.Since(startTime).Milliseconds()
+	}
+
 	s.updateStats(func(stats *Stats) {
 		stats.TotalResponseTime += responseTime
 		if stats.Queries > 0 {