更新Swagger API文档

main.go

@@ -16,12 +16,26 @@ import (
 )
 
 func main() {
-	// 解析命令行参数
-	configPath := flag.String("config", "config.json", "配置文件路径")
+	// 命令行参数解析
+	var configFile string
+	var daemonMode bool
+	flag.StringVar(&configFile, "config", "config.json", "配置文件路径")
+	flag.BoolVar(&daemonMode, "daemon", false, "以守护进程模式运行")
 	flag.Parse()
 
+	// 如果是守护进程模式，创建守护进程
+	if daemonMode {
+		if err := daemonize(); err != nil {
+			log.Fatalf("创建守护进程失败: %v", err)
+		}
+		// 父进程退出
+		os.Exit(0)
+	}
+
 	// 初始化配置
-	cfg, err := config.LoadConfig(*configPath)
+	var cfg *config.Config
+	var err error
+	cfg, err = config.LoadConfig(configFile)
 	if err != nil {
 		log.Fatalf("加载配置失败: %v", err)
 	}
@@ -61,14 +75,51 @@ func main() {
 	logger.Info(fmt.Sprintf("DNS服务器已启动，监听端口: %d", cfg.DNS.Port))
 	logger.Info(fmt.Sprintf("HTTP控制台已启动，监听端口: %d", cfg.HTTP.Port))
 
-	// 等待退出信号
-	sigChan := make(chan os.Signal, 1)
-	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
-	<-sigChan
+	// 监听信号
+	sigCh := make(chan os.Signal, 1)
+	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
+	<-sigCh
 
-	logger.Info("正在关闭服务...")
+	// 清理资源
+	log.Println("正在关闭服务...")
 	dnsServer.Stop()
 	httpServer.Stop()
 	shieldManager.StopAutoUpdate()
-	logger.Info("所有服务已关闭")
+	// 守护进程模式下不需要删除PID文件
+
+	log.Println("服务已关闭")
+}
+
+// daemonize 创建守护进程
+func daemonize() error {
+	// 使用更简单的方式创建守护进程：直接在当前进程中进行守护化处理
+	// 1. 重定向标准输入、输出、错误
+	nullFile, err := os.OpenFile("/dev/null", os.O_RDWR, 0)
+	if err != nil {
+		return fmt.Errorf("打开/dev/null失败: %w", err)
+	}
+	defer nullFile.Close()
+	
+	// 重定向文件描述符
+	err = syscall.Dup2(int(nullFile.Fd()), int(os.Stdin.Fd()))
+	if err != nil {
+		return fmt.Errorf("重定向stdin失败: %w", err)
+	}
+	err = syscall.Dup2(int(nullFile.Fd()), int(os.Stdout.Fd()))
+	if err != nil {
+		return fmt.Errorf("重定向stdout失败: %w", err)
+	}
+	err = syscall.Dup2(int(nullFile.Fd()), int(os.Stderr.Fd()))
+	if err != nil {
+		return fmt.Errorf("重定向stderr失败: %w", err)
+	}
+	
+	// 2. 创建新的会话和进程组
+	_, err = syscall.Setsid()
+	if err != nil {
+		return fmt.Errorf("创建新会话失败: %w", err)
+	}
+	
+	fmt.Println("守护进程已启动")
+	return nil
 }