package initialimport ("context" // 引入 context 包，以支持帶 context 的初始化函數"errors""fmt""your_project/internal/model/dao""your_project/internal/pkg/config""your_project/internal/pkg/logger""your_project/internal/pkg/trace""your_project/internal/service/adapter""your_project/internal/service/core"
)// InitializerType 定義了初始化器的類型標識符
type InitializerType string// 定義各種初始化器常量
const (CONFIG  InitializerType = "config"TRACE   InitializerType = "trace"LOGGER  InitializerType = "logger"STORE   InitializerType = "store"MANAGER InitializerType = "manager"ADAPTER InitializerType = "adapter"
)// 定義不同類型的初始化函數或接口，以便于在 initializers 映射中存儲
type Initializer interface {Init() error
}
type InitializerFunc func() error
type InitializerFuncWithCtx func(ctx context.Context) errorvar (// init_graph 定義了初始化器的依賴關系：key 依賴 value(s)// 例如: TRACE 依賴 CONFIG，意味著 CONFIG 必須在 TRACE 之前初始化。init_graph = map[InitializerType][]InitializerType{TRACE:   {CONFIG},  // TRACE 模塊依賴 CONFIG 模塊CONFIG:  {LOGGER},  // CONFIG 模塊依賴 LOGGER 模塊LOGGER:  {},        // LOGGER 模塊沒有外部依賴STORE:   {TRACE, CONFIG, LOGGER}, // STORE 模塊依賴 TRACE, CONFIG, LOGGERMANAGER: {STORE},   // MANAGER 模塊依賴 STORE 模塊ADAPTER: {MANAGER}, // ADAPTER 模塊依賴 MANAGER 模塊}// initializers 映射了初始化器類型到具體的初始化函數或接口實現initializers = map[InitializerType]interface{}{LOGGER:  InitializerFunc(logger.Init),      // 假設 logger.Init 是 func() errorCONFIG:  InitializerFunc(config.Init),      // 假設 config.Init 是 func() errorTRACE:   InitializerFunc(trace.Init),       // 假設 trace.Init 是 func() errorSTORE:   InitializerFunc(dao.Init),         // 假設 dao.Init 是 func() errorMANAGER: InitializerFunc(core.InitManager), // 假設 core.InitManager 是 func() errorADAPTER: InitializerFunc(adapter.Init),     // 假設 adapter.Init 是 func() error}
)// Run 執行所有組件的初始化，并按照依賴關系進行排序
func Run() error {// 1. 生成初始化順序init_order, err := topoSort(init_graph)if err != nil {return fmt.Errorf("failed to generate initialization order: %w", err)}// 2. 按照拓撲排序的順序逐一初始化組件for _, it := range init_order {if i, ok := initializers[it]; ok {switch initializer := i.(type) {case Initializer:if err := initializer.Init(); err != nil {return fmt.Errorf("failed to initialize %s: %w", it, err)}case InitializerFunc:if err := initializer(); err != nil {return fmt.Errorf("failed to initialize %s: %w", it, err)}case InitializerFuncWithCtx:// 對于需要 context 的初始化函數，這里傳入 nil context，實際應用中可能需要更具體的 contextif err := initializer(nil); err != nil {return fmt.Errorf("failed to initialize %s: %w", it, err)}default:return fmt.Errorf("unknown initializer type for %s", it)}}}return nil
}// topoSort 執行初始化依賴圖的拓撲排序 (Kahn's Algorithm)
// graph: key 依賴 value(s)，表示如果 key 存在，它依賴 value(s) 中的所有節點。
// 換句話說，在依賴圖中，存在從 value -> key 的邊。
func topoSort(graph map[InitializerType][]InitializerType) ([]InitializerType, error) {// 1. 構建鄰接表 (adjList) 和計算入度 (inDegree)// adjList[node]: 存儲 node 指向的所有節點 (即 node 是誰的依賴)// inDegree[node]: 存儲有多少條邊指向 node (即 node 被多少個其他節點依賴)adjList := make(map[InitializerType][]InitializerType)inDegree := make(map[InitializerType]int)// 初始化所有在依賴圖中出現的節點for node := range graph {// 確保所有作為依賴出現但未作為主鍵的節點也被初始化if _, exists := adjList[node]; !exists {adjList[node] = []InitializerType{}}if _, exists := inDegree[node]; !exists {inDegree[node] = 0}}for _, dependencies := range graph {for _, dep := range dependencies {if _, exists := adjList[dep]; !exists {adjList[dep] = []InitializerType{}}if _, exists := inDegree[dep]; !exists {inDegree[dep] = 0}}}// 填充 adjList 和計算入度for dependent, dependencies := range graph {for _, dep := range dependencies {// 如果 dependent 依賴 dep (即 dep -> dependent 有一條邊)// adjList[dep] 存儲 dep 依賴的節點// inDegree[dependent]++adjList[dep] = append(adjList[dep], dependent)inDegree[dependent]++}}// 2. 找到所有入度為 0 的節點，放入隊列var queue []InitializerTypefor node, degree := range inDegree {if degree == 0 {queue = append(queue, node)}}// 3. 循環處理隊列中的節點var topoOrder []InitializerTypeprocessedNodesCount := 0 // 記錄已處理的節點數量for len(queue) > 0 {node := queue[0] // 取出隊列頭部節點queue = queue[1:]topoOrder = append(topoOrder, node) // 將節點加入拓撲排序結果processedNodesCount++// 遍歷當前節點所指向的所有鄰居（即依賴它的節點），減少它們的入度// 根據 `adjList` 的構建方式，adjList[node] 存儲的是 node 依賴的節點// 它們在圖中是以 node 為起點的邊所指向的節點for _, neighbor := range adjList[node] {inDegree[neighbor]--if inDegree[neighbor] == 0 {queue = append(queue, neighbor)}}}// 4. 檢查是否存在循環依賴// 如果拓撲排序結果的節點數量不等于圖中所有節點的數量，則存在循環if processedNodesCount != len(inDegree) {return nil, errors.New("cycle detected in graph")}return topoOrder, nil
}