[优化] go 引用调整

2026-10-03 07:06:36 +08:00 · 2026-06-06 10:26:20 +08:00
parent ee1110b752
commit 3cfefb4367
552 changed files with 1642 additions and 2185 deletions
@@ -0,0 +1,123 @@
+package main
+
+import (
+	"context"
+	"embed"
+	"fmt"
+	"log/slog"
+	"os"
+	"strconv"
+
+	"github.com/rain-kl/openflare/openflare-server/common"
+	_ "github.com/rain-kl/openflare/openflare-server/docs"
+	"github.com/rain-kl/openflare/openflare-server/job"
+	"github.com/rain-kl/openflare/openflare-server/middleware"
+	"github.com/rain-kl/openflare/openflare-server/model"
+	"github.com/rain-kl/openflare/openflare-server/router"
+	"github.com/rain-kl/openflare/openflare-server/service"
+	"github.com/rain-kl/openflare/openflare-server/utils/geoip"
+
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-contrib/sessions/cookie"
+	"github.com/gin-contrib/sessions/redis"
+	"github.com/gin-gonic/gin"
+)
+
+//go:embed all:web/build
+var buildFS embed.FS
+
+//go:embed web/build/index.html
+var indexPage []byte
+
+// @title OpenFlare Server API
+// @version 3.0
+// @description OpenFlare Server 管理端与 Agent API 文档。
+// @BasePath /
+// @schemes http https
+// @securityDefinitions.apikey OpenFlareTokenAuth
+// @in header
+// @name OpenFlare-Token
+// @description 管理端 API 使用登录后返回的用户 Token
+// @securityDefinitions.apikey AccessTokenAuth
+// @in header
+// @name X-Agent-Token
+// @description Agent API 使用节点专属 Agent Token 或全局 Discovery Token
+func main() {
+	common.ParseFlags()
+	common.SetupGinLog()
+	slog.Info("OpenFlare started", "version", common.Version)
+	if os.Getenv("GIN_MODE") != "debug" {
+		gin.SetMode(gin.ReleaseMode)
+	}
+	// Initialize SQL Database
+	defer service.ShutdownWSHubs()
+
+	err := model.InitDB()
+	if err != nil {
+		slog.Error("initialize database failed", "error", err)
+		os.Exit(1)
+	}
+	defer func() {
+		err := model.CloseDB()
+		if err != nil {
+			slog.Error("close database failed", "error", err)
+			os.Exit(1)
+		}
+	}()
+
+	// Initialize Redis
+	err = common.InitRedisClient()
+	if err != nil {
+		slog.Error("initialize redis failed", "error", err)
+		os.Exit(1)
+	}
+
+	// Initialize options
+	model.InitOptionMap()
+	middleware.InitJWTMiddleware()
+	geoip.InitGeoIP(common.GeoIPProvider)
+	backgroundCtx, cancelBackgroundTasks := context.WithCancel(context.Background())
+	defer cancelBackgroundTasks()
+	service.StartDatabaseAutoCleanupScheduler(backgroundCtx)
+
+	job.InitCronJobs()
+	defer job.StopCronJobs()
+
+	// Initialize HTTP server
+	server := gin.Default()
+	//server.Use(gzip.Gzip(gzip.DefaultCompression))
+	server.Use(middleware.CORS())
+
+	// Initialize session store
+	if common.RedisEnabled {
+		opt := common.ParseRedisOption()
+		store, _ := redis.NewStore(opt.MinIdleConns, opt.Network, opt.Addr, opt.Password, []byte(common.SessionSecret))
+		server.Use(sessions.Sessions("session", store))
+	} else {
+		store := cookie.NewStore([]byte(common.SessionSecret))
+		server.Use(sessions.Sessions("session", store))
+	}
+
+	router.SetRouter(server, buildFS, indexPage)
+	var port = os.Getenv("PORT")
+	if port == "" {
+		port = strconv.Itoa(*common.Port)
+	}
+	dbBackend := "sqlite"
+	if common.SQLDSN != "" {
+		dbBackend = "postgres"
+	}
+	slog.Info("server config", "port", port, "gin_mode", gin.Mode(), "log_level", common.GetLogLevel(), "db_backend", dbBackend, "sqlite_path", common.SQLitePath, "redis_enabled", common.RedisEnabled, "log_dir", valueOrDefault(*common.LogDir, "stdout"), "access_token_configured", common.AccessToken != "", "node_offline_threshold", common.NodeOfflineThreshold)
+	slog.Info("server listening", "address", fmt.Sprintf(":%s", port))
+	err = server.Run(":" + port)
+	if err != nil {
+		slog.Error("server run failed", "error", err)
+	}
+}
+
+func valueOrDefault(value string, fallback string) string {
+	if value == "" {
+		return fallback
+	}
+	return value
+}