mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-02 14:56:38 +08:00
ryan
416603b616
- Replace db.DB(ctx) with database.DB(ctx) from plugins/infra/database
- Replace db.Redis/db.PrefixedKey/db.GetJSON/db.SetJSON with cachepkg.* from plugins/infra/cache
- Replace pkg/persistence/idgen with pkg/idgen (already exists)
- Replace pkg/persistence/batchwriter with pkg/batchwriter (already exists)
- Replace pkg/persistence/migrator with pkg/migrator (already exists)
- Replace pkg/persistence/logstore with plugins/domain/risk_control/logstore
- Delete defunct pkg/{persistence,cap,message_gateway,push,shared,task}
- Fix vet issues: db alias in domain_test.go, driver_asynq_worker.TaskHandler reference
- Update Makefile architecture guard
- Update docs and skill references
- Update go.mod: gorilla/sessions promotion to direct dependency
101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package driver_asynq_worker
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
// TaskParam 任务参数定义
|
|
//
|
|
//nolint:revive // TaskParam 保留完整名称以避免与通用 Param 混淆
|
|
type TaskParam struct {
|
|
Name string `json:"name"` // 参数键名
|
|
Label string `json:"label"` // 显示名称
|
|
Type string `json:"type"` // 类型：string, text, number, boolean
|
|
Required bool `json:"required"` // 是否必填
|
|
Placeholder string `json:"placeholder"` // 占位符
|
|
Description string `json:"description"` // 描述
|
|
}
|
|
|
|
// TaskMeta 任务元数据
|
|
//
|
|
//nolint:revive // TaskMeta 保留完整名称以避免与通用 Meta 混淆
|
|
type TaskMeta struct {
|
|
Type string `json:"type"`
|
|
AsynqTask string `json:"asynq_task"`
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
SupportsTime bool `json:"supports_time"`
|
|
MaxRetry int `json:"max_retry"`
|
|
Queue string `json:"queue"`
|
|
Retryable bool `json:"retryable"` // 是否支持手动重试
|
|
Params []TaskParam `json:"params,omitempty"`
|
|
}
|
|
|
|
var (
|
|
dispatchableTasksMutex sync.RWMutex
|
|
dispatchableTasks []TaskMeta
|
|
)
|
|
|
|
// RegisterTaskMeta 注册任务元数据到全局列表
|
|
func RegisterTaskMeta(meta TaskMeta) {
|
|
dispatchableTasksMutex.Lock()
|
|
defer dispatchableTasksMutex.Unlock()
|
|
for _, t := range dispatchableTasks {
|
|
if t.Type == meta.Type {
|
|
return
|
|
}
|
|
}
|
|
dispatchableTasks = append(dispatchableTasks, meta)
|
|
}
|
|
|
|
// GetDispatchableTasks 获取所有已注册的元数据列表（返回副本以避免并发读写冲突）
|
|
func GetDispatchableTasks() []TaskMeta {
|
|
dispatchableTasksMutex.RLock()
|
|
defer dispatchableTasksMutex.RUnlock()
|
|
|
|
metas := make([]TaskMeta, len(dispatchableTasks))
|
|
copy(metas, dispatchableTasks)
|
|
return metas
|
|
}
|
|
|
|
// GetTaskMeta 根据任务类型获取元数据
|
|
func GetTaskMeta(taskType string) *TaskMeta {
|
|
dispatchableTasksMutex.RLock()
|
|
defer dispatchableTasksMutex.RUnlock()
|
|
for _, t := range dispatchableTasks {
|
|
if t.Type == taskType {
|
|
copied := t
|
|
return &copied
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetTaskMetaByAsynqTask 根据 Asynq 任务名称获取元数据
|
|
func GetTaskMetaByAsynqTask(asynqTask string) *TaskMeta {
|
|
dispatchableTasksMutex.RLock()
|
|
defer dispatchableTasksMutex.RUnlock()
|
|
for _, t := range dispatchableTasks {
|
|
if t.AsynqTask == asynqTask {
|
|
copied := t
|
|
return &copied
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetRegisteredAsynqTasks 返回所有已注册的 Asynq 任务名称，以便动态注册路由
|
|
func GetRegisteredAsynqTasks() []string {
|
|
handlerRegistryMutex.RLock()
|
|
defer handlerRegistryMutex.RUnlock()
|
|
|
|
keys := make([]string, 0, len(handlerRegistry))
|
|
for k := range handlerRegistry {
|
|
keys = append(keys, k)
|
|
}
|
|
return keys
|
|
}
|