优化结构

internal/data/repository/integration_config.go

@@ -8,7 +8,7 @@ import (
 	"time"
 
 	"kra/internal/biz"
-	"kra/internal/integrationruntime"
+	"kra/internal/integration/runtimeconfig"
 
 	"gorm.io/gorm"
 )
@@ -27,6 +27,10 @@ func (integrationConfigPO) TableName() string { return "sys_integration_configs"
 
 type integrationConfigRepo struct{ data Provider }
 
+type integrationRuntimeProvider interface {
+	IntegrationRuntime() *runtimeconfig.Store
+}
+
 func NewIntegrationConfigRepo(data Provider) biz.IntegrationConfigRepo {
 	return &integrationConfigRepo{data: data}
 }
@@ -93,18 +97,25 @@ func (r *integrationConfigRepo) DeleteIntegrationConfig(ctx context.Context, kin
 	if err := r.data.DB().WithContext(ctx).Where("kind = ? AND provider = ?", kind, provider).Delete(&integrationConfigPO{}).Error; err != nil {
 		return err
 	}
-	if runtime := r.data.IntegrationRuntime(); runtime != nil {
+	if runtime := integrationRuntime(r.data); runtime != nil {
 		runtime.Delete(kind, provider)
 	}
 	return nil
 }
 
 func (r *integrationConfigRepo) publish(kind, provider string, enabled bool, values []byte) {
-	if runtime := r.data.IntegrationRuntime(); runtime != nil {
-		runtime.Set(integrationruntime.Config{Kind: kind, Provider: provider, Enabled: enabled, Values: values})
+	if runtime := integrationRuntime(r.data); runtime != nil {
+		runtime.Set(runtimeconfig.Config{Kind: kind, Provider: provider, Enabled: enabled, Values: values})
 	}
 }
 
+func integrationRuntime(provider Provider) *runtimeconfig.Store {
+	if value, ok := provider.(integrationRuntimeProvider); ok {
+		return value.IntegrationRuntime()
+	}
+	return nil
+}
+
 func integrationConfigFromPO(row integrationConfigPO) *biz.IntegrationConfig {
 	values := integrationObject(json.RawMessage(row.Config))
 	maskIntegrationSecrets(row.Kind, row.Provider, values)