chore(redshift): retry schema does not exist errors in list schemas

sqlconnect/internal/redshift/schemaadmin.go

@@ -0,0 +1,22 @@
+package redshift
+
+import (
+	"context"
+	"regexp"
+
+	"github.com/rudderlabs/sqlconnect-go/sqlconnect"
+)
+
+var schemaDoesNotExistRegex = regexp.MustCompile(`schema "(.*)" does not exist`)
+
+func (db *DB) ListSchemas(ctx context.Context) ([]sqlconnect.SchemaRef, error) {
+	// If the list schemas query is executed while a schema is being deleted, the query will fail with a schema does not exist error.
+	retryableError := func(err error) bool {
+		return schemaDoesNotExistRegex.MatchString(err.Error())
+	}
+	schemas, err := db.DB.ListSchemas(ctx)
+	if err != nil && retryableError(err) {
+		return db.ListSchemas(ctx)
+	}
+	return schemas, err
+}

sqlconnect/internal/redshift/schemaadmin_test.go

@@ -0,0 +1,11 @@
+package redshift
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestSchemaDoesNotExistRegex(t *testing.T) {
+	require.True(t, schemaDoesNotExistRegex.MatchString(`ERROR: schema "tsqlcon_zvtquuiqulpz_1728651180" does not exist`))
+}