Merge branch 'main' into feat/aws-regions-migration

pkg/signoz/provider.go

@@ -194,6 +194,7 @@ func NewSQLMigrationProviderFactories(
 		sqlmigration.NewDeprecateAPIKeyFactory(sqlstore, sqlschema),
 		sqlmigration.NewServiceAccountAuthzactory(sqlstore),
 		sqlmigration.NewDropUserDeletedAtFactory(sqlstore, sqlschema),
+		sqlmigration.NewMigrateAWSAllRegionsFactory(sqlstore),
 	)
 }
 

pkg/sqlmigration/077_migrate_aws_all_regions.go

@@ -0,0 +1,116 @@
+package sqlmigration
+
+import (
+	"context"
+	"encoding/json"
+	"slices"
+	"sort"
+
+	"github.com/SigNoz/signoz/pkg/factory"
+	"github.com/SigNoz/signoz/pkg/sqlstore"
+	"github.com/SigNoz/signoz/pkg/types/cloudintegrationtypes"
+	"github.com/uptrace/bun"
+	"github.com/uptrace/bun/migrate"
+)
+
+type migrateAWSAllRegions struct {
+	sqlstore sqlstore.SQLStore
+}
+
+type cloudIntegrationAWSMigrationRow struct {
+	bun.BaseModel `bun:"table:cloud_integration"`
+
+	ID     string `bun:"id"`
+	Config string `bun:"config"`
+}
+
+type awsConfig struct {
+	Regions []string `json:"regions"`
+}
+
+func NewMigrateAWSAllRegionsFactory(sqlstore sqlstore.SQLStore) factory.ProviderFactory[SQLMigration, Config] {
+	return factory.NewProviderFactory(
+		factory.MustNewName("migrate_aws_all_regions"),
+		func(ctx context.Context, ps factory.ProviderSettings, c Config) (SQLMigration, error) {
+			return &migrateAWSAllRegions{sqlstore: sqlstore}, nil
+		},
+	)
+}
+
+func (migration *migrateAWSAllRegions) Register(migrations *migrate.Migrations) error {
+	if err := migrations.Register(migration.Up, migration.Down); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (migration *migrateAWSAllRegions) Up(ctx context.Context, db *bun.DB) error {
+	tx, err := db.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+
+	defer func() {
+		_ = tx.Rollback()
+	}()
+
+	accounts := make([]*cloudIntegrationAWSMigrationRow, 0)
+	err = tx.NewSelect().
+		Model(&accounts).
+		Where("provider = ?", cloudintegrationtypes.CloudProviderTypeAWS).
+		Where("removed_at IS NULL").
+		Scan(ctx)
+	if err != nil {
+		return err
+	}
+
+	idsToUpdate := make([]string, 0)
+	for _, account := range accounts {
+		var cfg awsConfig
+		if err := json.Unmarshal([]byte(account.Config), &cfg); err != nil {
+			continue
+		}
+		if containsAllRegion(cfg.Regions) {
+			idsToUpdate = append(idsToUpdate, account.ID)
+		}
+	}
+
+	if len(idsToUpdate) == 0 {
+		return tx.Commit()
+	}
+
+	cfg := awsConfig{Regions: allValidAWSRegions()}
+	newBytes, err := json.Marshal(cfg)
+	if err != nil {
+		return err
+	}
+
+	_, err = tx.NewUpdate().
+		TableExpr("cloud_integration").
+		Set("config = ?", string(newBytes)).
+		Where("id IN (?)", bun.In(idsToUpdate)).
+		Exec(ctx)
+	if err != nil {
+		return err
+	}
+
+	return tx.Commit()
+}
+
+func (migration *migrateAWSAllRegions) Down(context.Context, *bun.DB) error {
+	return nil
+}
+
+func containsAllRegion(regions []string) bool {
+	return slices.Contains(regions, "all")
+}
+
+func allValidAWSRegions() []string {
+	awsRegions := cloudintegrationtypes.SupportedRegions[cloudintegrationtypes.CloudProviderTypeAWS]
+	out := make([]string, 0, len(awsRegions))
+	for _, r := range awsRegions {
+		out = append(out, r.StringValue())
+	}
+	sort.Strings(out)
+	return out
+}