优化结构

internal/data/integration_runtime.go

@@ -1,29 +1,13 @@
 package data
 
 import (
-	"errors"
-
-	"kra/internal/integration/runtimeconfig"
-
 	"gorm.io/gorm"
+	dataintegration "kra/internal/data/integration"
+	"kra/internal/integration/runtimeconfig"
 )
 
 func readIntegrationRuntime(db *gorm.DB) ([]runtimeconfig.Config, error) {
-	if db == nil || !db.Migrator().HasTable(&integrationConfigPO{}) {
-		return nil, nil
-	}
-	var rows []integrationConfigPO
-	if err := db.Session(&gorm.Session{NewDB: true}).
-		Where("kind IN ?", []string{"mq", "websocket"}).
-		Order("kind ASC, provider ASC").
-		Find(&rows).Error; err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
-		return nil, err
-	}
-	configs := make([]runtimeconfig.Config, 0, len(rows))
-	for _, row := range rows {
-		configs = append(configs, runtimeconfig.Config{Kind: row.Kind, Provider: row.Provider, Enabled: row.Enabled, Values: []byte(row.Config)})
-	}
-	return configs, nil
+	return dataintegration.ReadRuntime(db)
 }
 
 func (d *Data) loadIntegrationRuntime(db *gorm.DB) error {