add

cmd/qqbotd/main.go

@@ -0,0 +1,90 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log/slog"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/sky22333/qqbot/config"
+	"github.com/sky22333/qqbot/internal/collector"
+	"github.com/sky22333/qqbot/internal/httpserver"
+	"github.com/sky22333/qqbot/internal/notifier"
+	"github.com/sky22333/qqbot/internal/targets"
+)
+
+func main() {
+	configPath := flag.String("config", "configs/config.toml", "配置文件路径")
+	flag.Parse()
+
+	cfg, err := config.Load(*configPath)
+	if err != nil {
+		panic(err)
+	}
+
+	logLevel := slog.LevelInfo
+	if cfg.App.LogLevel == "debug" {
+		logLevel = slog.LevelDebug
+	}
+	logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: logLevel}))
+
+	flushInterval, err := cfg.TargetsFlushInterval()
+	if err != nil {
+		panic(err)
+	}
+	targetStore, err := targets.NewStore(cfg.Targets.FilePath, cfg.Targets.MaxRecords, flushInterval)
+	if err != nil {
+		panic(err)
+	}
+
+	targetCollector, err := collector.New(cfg, logger, targetStore)
+	if err != nil {
+		panic(err)
+	}
+
+	targetCollector.Start()
+	logger.Info("采集器已启动")
+
+	n, err := notifier.New(cfg, logger)
+	if err != nil {
+		panic(err)
+	}
+	n.SetTargetStore(targetStore)
+
+	server, err := httpserver.New(cfg, logger, n, targetStore)
+	if err != nil {
+		panic(err)
+	}
+
+	errCh := make(chan error, 1)
+	go func() {
+		errCh <- server.Start()
+	}()
+
+	stopCh := make(chan os.Signal, 1)
+	signal.Notify(stopCh, syscall.SIGTERM, syscall.SIGINT)
+
+	select {
+	case sig := <-stopCh:
+		logger.Info("收到退出信号，停止程序", "signal", sig.String())
+	case err := <-errCh:
+		if err != nil && err != http.ErrServerClosed {
+			logger.Error("服务异常退出", "error", err)
+		}
+	}
+
+	shutdownTimeout, err := cfg.ShutdownTimeout()
+	if err != nil {
+		shutdownTimeout = 10
+	}
+	ctx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)
+	defer cancel()
+	_ = server.Shutdown(ctx)
+	targetCollector.Stop()
+	n.Close()
+	_ = targetStore.Close()
+	logger.Info("服务已退出")
+}