不同服务注册不同组件模块

startup/startup.go

@@ -0,0 +1,122 @@
+// Package startup 提供服务启动时的组件初始化控制
+// 各服务可以按需初始化所需组件，避免不必要的资源占用
+package startup
+
+import (
+	"context"
+	"sync"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/glog"
+)
+
+// Components 组件配置
+type Components struct {
+	Consul   bool // Consul 服务注册发现（所有服务都需要）
+	Jaeger   bool // Jaeger 链路追踪（所有服务都需要）
+	Redis    bool // Redis 缓存
+	RabbitMQ bool // RabbitMQ 消息队列
+	MongoDB  bool // MongoDB 数据库
+	RAGFlow  bool // RAGFlow AI 客户端
+	ES       bool // Elasticsearch
+}
+
+var (
+	initialized bool
+	initOnce    sync.Once
+	components  *Components
+)
+
+// Init 初始化指定的组件
+// 示例:
+//
+//	bootstrap.Init(ctx, &bootstrap.Components{
+//	    Consul:   true,
+//	    Jaeger:   true,
+//	    Redis:    true,
+//	    RabbitMQ: true,
+//	})
+func Init(ctx context.Context, c *Components) {
+	initOnce.Do(func() {
+		components = c
+		initialized = true
+		glog.Infof(ctx, "Bootstrap 初始化完成: %+v", c)
+	})
+}
+
+// IsInitialized 检查是否已初始化
+func IsInitialized() bool {
+	return initialized
+}
+
+// GetComponents 获取组件配置
+func GetComponents() *Components {
+	if components == nil {
+		// 默认配置：从配置文件读取
+		return loadFromConfig()
+	}
+	return components
+}
+
+// NeedRedis 是否需要 Redis
+func NeedRedis() bool {
+	c := GetComponents()
+	return c != nil && c.Redis
+}
+
+// NeedRabbitMQ 是否需要 RabbitMQ
+func NeedRabbitMQ() bool {
+	c := GetComponents()
+	return c != nil && c.RabbitMQ
+}
+
+// NeedMongoDB 是否需要 MongoDB
+func NeedMongoDB() bool {
+	c := GetComponents()
+	return c != nil && c.MongoDB
+}
+
+// NeedRAGFlow 是否需要 RAGFlow
+func NeedRAGFlow() bool {
+	c := GetComponents()
+	return c != nil && c.RAGFlow
+}
+
+// NeedES 是否需要 Elasticsearch
+func NeedES() bool {
+	c := GetComponents()
+	return c != nil && c.ES
+}
+
+// loadFromConfig 从配置文件加载组件配置
+// 如果配置文件中没有 startup 配置，则默认全部启动
+func loadFromConfig() *Components {
+	ctx := context.Background()
+
+	// 检查是否有 startup 配置节
+	startupCfg := g.Cfg().MustGet(ctx, "startup")
+	if startupCfg.IsEmpty() {
+		// 没有配置 startup，默认全部启动
+		glog.Debug(ctx, "未找到 startup 配置，默认启动所有组件")
+		return &Components{
+			Consul:   true,
+			Jaeger:   true,
+			Redis:    true,
+			RabbitMQ: true,
+			MongoDB:  true,
+			RAGFlow:  true,
+			ES:       true,
+		}
+	}
+
+	// 有配置则按配置来，未配置的项默认 true
+	return &Components{
+		Consul:   g.Cfg().MustGet(ctx, "startup.consul", true).Bool(),
+		Jaeger:   g.Cfg().MustGet(ctx, "startup.jaeger", true).Bool(),
+		Redis:    g.Cfg().MustGet(ctx, "startup.redis", true).Bool(),
+		RabbitMQ: g.Cfg().MustGet(ctx, "startup.rabbitmq", true).Bool(),
+		MongoDB:  g.Cfg().MustGet(ctx, "startup.mongodb", true).Bool(),
+		RAGFlow:  g.Cfg().MustGet(ctx, "startup.ragflow", true).Bool(),
+		ES:       g.Cfg().MustGet(ctx, "startup.es", true).Bool(),
+	}
+}