优化结构

app/system/internal/data/integration_config.go

@@ -19,6 +19,7 @@ const (
 	integrationKindStorage = "storage"
 	integrationKindEmail   = "email"
 	integrationKindPayment = "payment"
+	integrationKindMQ      = "mq"
 )
 
 // integrationConfigPO stores credentials and provider-specific options for
@@ -34,6 +35,88 @@ type integrationConfigPO struct {
 	Config    string `gorm:"type:text;not null"`
 }
 
+func defaultMQIntegrationConfig() *conf.AdminBackend_MQ {
+	return &conf.AdminBackend_MQ{CleanSession: true, KeepAlive: 30, ConnectTimeout: 10}
+}
+
+func saveMQIntegrationConfig(db *gorm.DB, config *conf.AdminBackend_MQ) error {
+	if config == nil {
+		config = defaultMQIntegrationConfig()
+	}
+	raw, err := protojson.MarshalOptions{UseProtoNames: true, EmitDefaultValues: true}.Marshal(config)
+	if err != nil {
+		return fmt.Errorf("encode emqx integration configuration: %w", err)
+	}
+	enabled := config.Enabled && strings.TrimSpace(config.Broker) != ""
+	clean := db.Session(&gorm.Session{NewDB: true})
+	var row integrationConfigPO
+	err = clean.Where("kind = ? AND provider = ?", integrationKindMQ, "emqx").First(&row).Error
+	switch {
+	case errors.Is(err, gorm.ErrRecordNotFound):
+		return clean.Create(&integrationConfigPO{Kind: integrationKindMQ, Provider: "emqx", Enabled: enabled, Config: string(raw)}).Error
+	case err != nil:
+		return err
+	default:
+		return clean.Model(&row).Updates(map[string]any{"enabled": enabled, "config": string(raw)}).Error
+	}
+}
+
+func loadMQIntegrationConfig(db *gorm.DB) (*conf.AdminBackend_MQ, bool, error) {
+	var row integrationConfigPO
+	err := db.Session(&gorm.Session{NewDB: true}).Where("kind = ? AND provider = ?", integrationKindMQ, "emqx").First(&row).Error
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, false, nil
+	}
+	if err != nil {
+		return nil, false, err
+	}
+	if !json.Valid([]byte(row.Config)) {
+		return nil, false, errors.New("invalid emqx integration configuration")
+	}
+	config := defaultMQIntegrationConfig()
+	if err = (protojson.UnmarshalOptions{DiscardUnknown: true}).Unmarshal([]byte(row.Config), config); err != nil {
+		return nil, false, fmt.Errorf("decode emqx integration configuration: %w", err)
+	}
+	config.Enabled = row.Enabled
+	return config, true, nil
+}
+
+func resolveMQIntegrationConfig(db *gorm.DB, legacy *conf.AdminBackend_MQ) (*conf.AdminBackend_MQ, error) {
+	clean := db.Session(&gorm.Session{NewDB: true})
+	if !clean.Migrator().HasTable(&integrationConfigPO{}) {
+		if legacy == nil {
+			return defaultMQIntegrationConfig(), nil
+		}
+		return proto.Clone(legacy).(*conf.AdminBackend_MQ), nil
+	}
+	loaded, found, err := loadMQIntegrationConfig(clean)
+	if err != nil {
+		return nil, err
+	}
+	if found {
+		return loaded, nil
+	}
+	if legacy == nil {
+		legacy = defaultMQIntegrationConfig()
+	}
+	if err = saveMQIntegrationConfig(clean, legacy); err != nil {
+		return nil, err
+	}
+	loaded, _, err = loadMQIntegrationConfig(clean)
+	return loaded, err
+}
+
+func (d *Data) persistMQIntegrationConfig(ctx context.Context, config *conf.AdminBackend_MQ) error {
+	if !d.databaseReady.Load() {
+		return errors.New("database is not initialized")
+	}
+	db := d.gormDB.WithContext(ctx)
+	if !db.Migrator().HasTable(&integrationConfigPO{}) {
+		return errors.New("integration configuration table does not exist")
+	}
+	return saveMQIntegrationConfig(db, config)
+}
+
 func (integrationConfigPO) TableName() string { return "sys_integration_configs" }
 
 var storageProviderNames = []string{