mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
2a0ebd16fa
- 修复 ClickHouse Batch Writer 与 RiskControl LogWriter 优雅停机，确保退出前队列数据正确刷盘。 - 修复 OpenFlare 系统配置参数全局变量并发读写的 Data Race 冲突，在读取配置时引入读锁保护。 - 修复 upload 模块的 API 错误响应格式，使用 response.Abort* 代替原始的 c.AbortWithStatus 与 c.JSON，保证全局信封格式统一。 - 修复 pkg/utils/network 的 isPrivateIPv4 以使用标准库 net.IP.IsPrivate() 校验，修复 format 的 Bytes2Size 边界，修改大小单位因子变量为只读常量。 - 重构 upload 文件服务与路由器测试，使用 t.TempDir() 代替硬编码的 uploads 相对路径写入和删除，解决测试目录文件污染问题。
105 lines
3.0 KiB
Go
105 lines
3.0 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package bootstrap wires cross-module integrations and process-level subsystem initialization.
|
|
// All registrations use sync.Once so entry points can call them safely without import-order side effects.
|
|
package bootstrap
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
admin_push "github.com/Rain-kl/Wavelet/internal/apps/admin/push"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/admin/push/custom_events"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/openflare/chwriter"
|
|
"github.com/Rain-kl/Wavelet/internal/apps/risk_control"
|
|
taskhandlers "github.com/Rain-kl/Wavelet/internal/task/handlers"
|
|
"github.com/Rain-kl/Wavelet/pkg/logger"
|
|
)
|
|
|
|
// Options selects role-specific runtime bootstrap steps for the current process.
|
|
type Options struct {
|
|
// API enables HTTP-only subsystems such as the ClickHouse access-log writer.
|
|
API bool
|
|
}
|
|
|
|
var (
|
|
registerTasksOnce sync.Once
|
|
registerPushDomainEventsOnce sync.Once
|
|
registerTaskListenersOnce sync.Once
|
|
initRuntimeOnce sync.Once
|
|
)
|
|
|
|
// RegisterTasks registers all built-in task handlers and metadata.
|
|
func RegisterTasks() {
|
|
registerTasksOnce.Do(func() {
|
|
taskhandlers.Register()
|
|
})
|
|
}
|
|
|
|
// RegisterPushDomainEvents wires push notification handlers for domain events.
|
|
func RegisterPushDomainEvents() {
|
|
registerPushDomainEventsOnce.Do(func() {
|
|
custom_events.Register()
|
|
})
|
|
}
|
|
|
|
// RegisterTaskListeners wires operational listeners to task framework hooks.
|
|
func RegisterTaskListeners() {
|
|
registerTaskListenersOnce.Do(func() {
|
|
admin_push.RegisterTaskListeners()
|
|
})
|
|
}
|
|
|
|
// RegisterAPI wires integrations required by the HTTP API process.
|
|
func RegisterAPI() {
|
|
RegisterTasks()
|
|
RegisterPushDomainEvents()
|
|
}
|
|
|
|
// RegisterWorker wires integrations required by the task worker process.
|
|
func RegisterWorker() {
|
|
RegisterTasks()
|
|
RegisterTaskListeners()
|
|
}
|
|
|
|
// RegisterScheduler wires integrations required by the task scheduler process.
|
|
func RegisterScheduler() {
|
|
RegisterTasks()
|
|
}
|
|
|
|
// RegisterAll wires integrations for fused mode (API + Worker + Scheduler).
|
|
func RegisterAll() {
|
|
RegisterTasks()
|
|
RegisterPushDomainEvents()
|
|
RegisterTaskListeners()
|
|
}
|
|
|
|
// Init runs shared runtime bootstrap exactly once per process.
|
|
// Call from cmd entry points after wiring registration and database migration, not from router.
|
|
func Init(ctx context.Context, opts Options) {
|
|
initRuntimeOnce.Do(func() {
|
|
if err := admin_push.SyncEvents(ctx); err != nil {
|
|
logger.ErrorF(ctx, "[Bootstrap] sync push events failed: %v", err)
|
|
}
|
|
if opts.API {
|
|
risk_control.InitLogWriter(ctx)
|
|
chwriter.Init(ctx)
|
|
}
|
|
})
|
|
}
|
|
|
|
// Stop stops all batch writers and background resources.
|
|
func Stop(ctx context.Context) {
|
|
if err := risk_control.StopLogWriter(ctx); err != nil {
|
|
logger.ErrorF(ctx, "[Bootstrap] stop risk_control log writer failed: %v", err)
|
|
}
|
|
if err := chwriter.Stop(ctx); err != nil {
|
|
logger.ErrorF(ctx, "[Bootstrap] stop chwriter failed: %v", err)
|
|
}
|
|
}
|
|
|
|
// ResetInitRuntimeOnceForTest clears initRuntimeOnce so Init can run again in unit tests.
|
|
func ResetInitRuntimeOnceForTest() {
|
|
initRuntimeOnce = sync.Once{}
|
|
} |