优化结构

internal/biz/task_registry.go

@@ -0,0 +1,28 @@
+package biz
+
+import platformtask "kra/pkg/task"
+
+type TaskMethodFunc = platformtask.MethodFunc
+type TaskMethod = platformtask.Method
+
+type TaskMethodRegistry interface {
+	Register(platformtask.Method)
+	Lookup(string) (TaskMethodFunc, bool)
+	List() []TaskMethod
+}
+
+var defaultTaskMethods = platformtask.NewRegistry()
+
+// RegisterTaskMethod remains as a compatibility helper for tests and callers
+// that have not moved to constructor injection yet.
+func RegisterTaskMethod(name, description string, fn TaskMethodFunc) {
+	defaultTaskMethods.Register(platformtask.Method{Name: name, Description: description, Run: fn})
+}
+
+func TaskMethodByName(name string) (TaskMethodFunc, bool) {
+	return defaultTaskMethods.Lookup(name)
+}
+
+func RegisteredTaskMethods() []TaskMethod { return defaultTaskMethods.List() }
+
+func DefaultTaskMethodRegistry() TaskMethodRegistry { return defaultTaskMethods }