fix: invalid path for cleanup-test-schemas

go.mod

@@ -29,6 +29,7 @@ require (
 	github.com/trinodb/trino-go-client v0.315.0
 	github.com/youmark/pkcs8 v0.0.0-20240424034433-3c2c7870ae76
 	golang.org/x/crypto v0.25.0
+	golang.org/x/sync v0.7.0
 	google.golang.org/api v0.189.0
 )
 
@@ -139,7 +140,6 @@ require (
 	golang.org/x/mod v0.18.0 // indirect
 	golang.org/x/net v0.27.0 // indirect
 	golang.org/x/oauth2 v0.21.0 // indirect
-	golang.org/x/sync v0.7.0 // indirect
 	golang.org/x/sys v0.22.0 // indirect
 	golang.org/x/term v0.22.0 // indirect
 	golang.org/x/text v0.16.0 // indirect

sqlconnect/cmd/cleanup.go → sqlconnect/cmd/cleanup/cleanup.go

@@ -7,6 +7,7 @@
 	"strings"
 
 	"github.com/tidwall/sjson"
+	"golang.org/x/sync/errgroup"
 
 	"github.com/rudderlabs/sqlconnect-go/sqlconnect"
 	"github.com/rudderlabs/sqlconnect-go/sqlconnect/internal/bigquery"
@@ -33,32 +34,42 @@
 		{Env: "TRINO_TEST_ENVIRONMENT_CREDENTIALS", Type: trino.DatabaseType},
 	}
 
+	g, ctx := errgroup.WithContext(context.Background())
+	g.SetLimit(4)
 	for _, c := range cleanupConfigs {
-		configJSON, ok := os.LookupEnv(c.Env)
-		if !ok {
-			log.Fatalf("%s environment variable not set", c.Env)
-		}
-		if c.Fn != nil {
-			configJSON = c.Fn(configJSON)
-		}
-		db, err := sqlconnect.NewDB(c.Type, []byte(configJSON))
-		if err != nil {
-			log.Fatalf("[%s] failed to create db: %v", c.Type, err)
-		}
-		schemas, err := db.ListSchemas(context.Background())
-		if err != nil {
-			log.Fatalf("[%s] failed to list schemas: %v", c.Type, err)
-		}
-		for _, schema := range schemas {
-			if strings.Contains(strings.ToLower(schema.Name), "tsqlcon_") {
-				err := db.DropSchema(context.Background(), schema)
-				if err != nil {
-					log.Printf("[%s] failed to drop schema: %v", c.Type, err)
-				} else {
-					log.Printf("[%s] dropped schema %s", c.Type, schema)
+		c := c
+		g.Go(func() error {
+			configJSON, ok := os.LookupEnv(c.Env)
+			if !ok {
+				log.Fatalf("%s environment variable not set", c.Env)
+			}
+			if c.Fn != nil {
+				configJSON = c.Fn(configJSON)
+			}
+			db, err := sqlconnect.NewDB(c.Type, []byte(configJSON))
+			if err != nil {
+				log.Fatalf("[%s] failed to create db: %v", c.Type, err)
+			}
+			schemas, err := db.ListSchemas(ctx)
+			if err != nil {
+				log.Fatalf("[%s] failed to list schemas: %v", c.Type, err)
+			}
+			for _, schema := range schemas {
+				if strings.Contains(strings.ToLower(schema.Name), "tsqlcon_") {
+					err := db.DropSchema(ctx, schema)
+					if err != nil {
+						log.Printf("[%s] failed to drop schema: %v", c.Type, err)
+					} else {
+						log.Printf("[%s] dropped schema %s", c.Type, schema)
+					}
 				}
 			}
-		}
+			return nil
+		})
+	}
+
+	if err := g.Wait(); err != nil {
+		log.Fatalf("cleanup failed: %v", err)
 	}
 }