Update.

Author: xOS

cmd/dashboard/main.go

@@ -3,11 +3,16 @@ package main
 import (
 	"fmt"
 	"log"
+	"net/http"
 	"os"
+	"os/signal"
 	"path/filepath"
+	"syscall"
 	"time"
 
 	"github.com/spf13/pflag"
+	"github.com/xos/serverstatus/cmd/dashboard/controller"
+	"github.com/xos/serverstatus/cmd/dashboard/rpc"
 	"github.com/xos/serverstatus/model"
 	"github.com/xos/serverstatus/proto"
 	"github.com/xos/serverstatus/service/singleton"
@@ -43,21 +48,6 @@ func init() {
 	pflag.Parse()
 }
 
-func initSystem() {
-	// 启动 singleton 包下的所有服务
-	singleton.LoadSingleton()
-
-	// 每天的3:30 对 监控记录 和 流量记录 进行清理
-	if _, err := singleton.Cron.AddFunc("0 30 3 * * *", singleton.CleanMonitorHistory); err != nil {
-		panic(err)
-	}
-
-	// 每小时对流量记录进行打点
-	if _, err := singleton.Cron.AddFunc("0 0 * * * *", singleton.RecordTransferHourlyUsage); err != nil {
-		panic(err)
-	}
-}
-
 func main() {
 	var err error
 	hostName, err = os.Hostname()
@@ -113,9 +103,77 @@ func main() {
 		return
 	}
 
-	// 这里是原来的main函数代码
+	// 初始化系统
 	log.Println("NG>> 服务启动中...")
-	initSystem()
+	singleton.LoadSingleton()
+
+	// 添加定时任务
+	log.Println("NG>> 初始化定时任务...")
+	// 每天的3:30 对 监控记录 和 流量记录 进行清理
+	if _, err := singleton.Cron.AddFunc("0 30 3 * * *", singleton.CleanMonitorHistory); err != nil {
+		panic(err)
+	}
+	// 每小时对流量记录进行打点
+	if _, err := singleton.Cron.AddFunc("0 0 * * * *", singleton.RecordTransferHourlyUsage); err != nil {
+		panic(err)
+	}
+
+	// 启动HTTP和gRPC服务
+	startServices()
+
+	// 添加这段代码，等待中断信号以优雅地关闭服务
+	log.Println("NG>> 所有服务已启动，正在运行...")
+	waitForInterrupt()
+}
+
+// startServices 启动HTTP和gRPC服务
+func startServices() {
+	// 启动RPC服务
+	go startRPCServer()
+
+	// 启动HTTP服务
+	go startHTTPServer()
+
+	// 启动其他必要的服务
+	go dispatchReportInfoTask()
+}
+
+// startRPCServer 启动gRPC服务
+func startRPCServer() {
+	log.Printf("NG>> gRPC服务启动在端口 %d", singleton.Conf.GRPCPort)
+
+	// 使用rpc包中的服务器
+	rpc.ServeRPC(singleton.Conf.GRPCPort)
+}
+
+// startHTTPServer 启动HTTP服务
+func startHTTPServer() {
+	log.Printf("NG>> HTTP服务启动在端口 %d", singleton.Conf.HTTPPort)
+
+	// 使用controller包中的ServeWeb函数
+	srv := controller.ServeWeb(singleton.Conf.HTTPPort)
+
+	if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+		log.Fatalf("NG>> HTTP服务启动失败: %v", err)
+	}
+}
+
+// waitForInterrupt 等待中断信号
+func waitForInterrupt() {
+	// 创建一个通道接收系统信号
+	c := make(chan os.Signal, 1)
+	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
+
+	// 阻塞等待信号
+	<-c
+
+	// 收到信号后，执行清理工作
+	log.Println("NG>> 收到中断信号，服务正在关闭...")
+
+	// 保存最后的流量统计
+	singleton.RecordTransferHourlyUsage()
+
+	log.Println("NG>> 服务已安全关闭")
 }
 
 func dispatchReportInfoTask() {