aboutsummaryrefslogtreecommitdiff
path: root/weed/worker/tasks/schema_provider.go
diff options
context:
space:
mode:
Diffstat (limited to 'weed/worker/tasks/schema_provider.go')
-rw-r--r--weed/worker/tasks/schema_provider.go51
1 files changed, 51 insertions, 0 deletions
diff --git a/weed/worker/tasks/schema_provider.go b/weed/worker/tasks/schema_provider.go
new file mode 100644
index 000000000..4d69556b1
--- /dev/null
+++ b/weed/worker/tasks/schema_provider.go
@@ -0,0 +1,51 @@
+package tasks
+
+import (
+ "sync"
+
+ "github.com/seaweedfs/seaweedfs/weed/admin/config"
+)
+
+// TaskConfigSchema defines the schema for task configuration
+type TaskConfigSchema struct {
+ config.Schema // Embed common schema functionality
+ TaskName string `json:"task_name"`
+ DisplayName string `json:"display_name"`
+ Description string `json:"description"`
+ Icon string `json:"icon"`
+}
+
+// TaskConfigSchemaProvider is an interface for providing task configuration schemas
+type TaskConfigSchemaProvider interface {
+ GetConfigSchema() *TaskConfigSchema
+}
+
+// schemaRegistry maintains a registry of schema providers by task type
+type schemaRegistry struct {
+ providers map[string]TaskConfigSchemaProvider
+ mutex sync.RWMutex
+}
+
+var globalSchemaRegistry = &schemaRegistry{
+ providers: make(map[string]TaskConfigSchemaProvider),
+}
+
+// RegisterTaskConfigSchema registers a schema provider for a task type
+func RegisterTaskConfigSchema(taskType string, provider TaskConfigSchemaProvider) {
+ globalSchemaRegistry.mutex.Lock()
+ defer globalSchemaRegistry.mutex.Unlock()
+ globalSchemaRegistry.providers[taskType] = provider
+}
+
+// GetTaskConfigSchema returns the schema for the specified task type
+func GetTaskConfigSchema(taskType string) *TaskConfigSchema {
+ globalSchemaRegistry.mutex.RLock()
+ provider, exists := globalSchemaRegistry.providers[taskType]
+ globalSchemaRegistry.mutex.RUnlock()
+
+ if !exists {
+ return nil
+ }
+
+ return provider.GetConfigSchema()
+}