添加重启endpoint

dns/server.go

@@ -65,16 +65,18 @@ type Server struct {
 	resolvedDomainsMutex sync.RWMutex
 	resolvedDomains      map[string]*BlockedDomain // 用于记录解析的域名
 	clientStatsMutex     sync.RWMutex
-	clientStats          map[string]*ClientStats // 用于记录客户端统计
+	clientStats          map[string]*ClientStats   // 用于记录客户端统计
 	hourlyStatsMutex     sync.RWMutex
-	hourlyStats          map[string]int64 // 按小时统计屏蔽数量
+	hourlyStats          map[string]int64          // 按小时统计屏蔽数量
 	dailyStatsMutex      sync.RWMutex
-	dailyStats           map[string]int64 // 按天统计屏蔽数量
+	dailyStats           map[string]int64          // 按天统计屏蔽数量
 	monthlyStatsMutex    sync.RWMutex
-	monthlyStats         map[string]int64 // 按月统计屏蔽数量
+	monthlyStats         map[string]int64          // 按月统计屏蔽数量
 	saveTicker           *time.Ticker     // 用于定时保存数据
 	startTime            time.Time        // 服务器启动时间
 	saveDone             chan struct{}    // 用于通知保存协程停止
+	stopped              bool             // 服务器是否已经停止
+	stoppedMutex         sync.Mutex       // 保护stopped标志的互斥锁
 }
 
 // Stats DNS服务器统计信息
@@ -102,9 +104,9 @@ func NewServer(config *config.DNSConfig, shieldConfig *config.ShieldConfig, shie
 			Net:     "udp",
 			Timeout: time.Duration(config.Timeout) * time.Millisecond,
 		},
-		ctx:       ctx,
-		cancel:    cancel,
-		startTime: time.Now(), // 记录服务器启动时间
+		ctx:            ctx,
+		cancel:         cancel,
+		startTime:      time.Now(), // 记录服务器启动时间
 		stats: &Stats{
 			Queries:           0,
 			Blocked:           0,
@@ -123,6 +125,7 @@ func NewServer(config *config.DNSConfig, shieldConfig *config.ShieldConfig, shie
 		dailyStats:      make(map[string]int64),
 		monthlyStats:    make(map[string]int64),
 		saveDone:        make(chan struct{}),
+		stopped:         false, // 初始化为未停止状态
 	}
 
 	// 加载已保存的统计数据