优化

internal/biz/system_init.go

@@ -1,6 +1,16 @@
 package biz
 
-import "context"
+import (
+	"context"
+	"errors"
+)
+
+var ErrTaskRuntimeReload = errors.New("task runtime reload failed")
+
+type TaskRuntimeReloadError struct{ Err error }
+
+func (e *TaskRuntimeReloadError) Error() string        { return e.Err.Error() }
+func (e *TaskRuntimeReloadError) Is(target error) bool { return target == ErrTaskRuntimeReload }
 
 type DatabaseConfig struct {
 	Driver, Host, Port, User, Password, Name, Path, Config, Template, AdminPassword string
@@ -16,10 +26,13 @@ type InitializationRepo interface {
 	ReloadConfig(context.Context) error
 }
 
-type SystemConfigUsecase struct{ repo InitializationRepo }
+type SystemConfigUsecase struct {
+	repo  InitializationRepo
+	tasks TaskRuntime
+}
 
-func NewSystemConfigUsecase(repo InitializationRepo) *SystemConfigUsecase {
-	return &SystemConfigUsecase{repo: repo}
+func NewSystemConfigUsecase(repo InitializationRepo, tasks TaskRuntime) *SystemConfigUsecase {
+	return &SystemConfigUsecase{repo: repo, tasks: tasks}
 }
 
 func (uc *SystemConfigUsecase) IsInitialized(ctx context.Context) (bool, error) {
@@ -29,7 +42,13 @@ func (uc *SystemConfigUsecase) Initialize(ctx context.Context, config *DatabaseC
 	if config == nil || len(config.AdminPassword) < 6 {
 		return ErrInvalidCredentials
 	}
-	return uc.repo.Initialize(ctx, config)
+	if err := uc.repo.Initialize(ctx, config); err != nil {
+		return err
+	}
+	if err := uc.tasks.Reload(ctx); err != nil {
+		return &TaskRuntimeReloadError{Err: err}
+	}
+	return nil
 }
 func (uc *SystemConfigUsecase) PersistConfig(ctx context.Context) error {
 	return uc.repo.PersistConfig(ctx)
@@ -41,5 +60,11 @@ func (uc *SystemConfigUsecase) PersistRuntimeConfig(ctx context.Context, data, a
 	return uc.repo.PersistRuntimeConfig(ctx, data, admin)
 }
 func (uc *SystemConfigUsecase) ReloadConfig(ctx context.Context) error {
-	return uc.repo.ReloadConfig(ctx)
+	if err := uc.repo.ReloadConfig(ctx); err != nil {
+		return err
+	}
+	if err := uc.tasks.Reload(ctx); err != nil {
+		return &TaskRuntimeReloadError{Err: err}
+	}
+	return nil
 }