优化结构

internal/data/integration_defaults.go

@@ -0,0 +1,39 @@
+package data
+
+import (
+	"encoding/json"
+	"errors"
+
+	"kra/internal/biz"
+
+	"gorm.io/gorm"
+)
+
+func ensureCommunicationIntegrationConfigs(db *gorm.DB) error {
+	defaults := []struct {
+		kind     string
+		provider string
+	}{
+		{kind: biz.IntegrationKindMQ, provider: "emqx"},
+		{kind: biz.IntegrationKindMQ, provider: "rabbitmq"},
+		{kind: biz.IntegrationKindWebSocket, provider: "melody"},
+	}
+	for _, item := range defaults {
+		var row integrationConfigPO
+		err := db.Where("kind = ? AND provider = ?", item.kind, item.provider).First(&row).Error
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			values, marshalErr := json.Marshal(biz.DefaultIntegrationConfig(item.kind, item.provider))
+			if marshalErr != nil {
+				return marshalErr
+			}
+			if err = db.Create(&integrationConfigPO{Kind: item.kind, Provider: item.provider, Enabled: false, Config: string(values)}).Error; err != nil {
+				return err
+			}
+			continue
+		}
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}