chore: improve database migration

0xPolygonID · Jun 26, 2024 · 807f2e0 · 807f2e0
1 parent 7e67052
commit 807f2e0
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/cmd/migrate/main.go b/cmd/migrate/main.go
@@ -4,26 +4,31 @@ import (
 	"context"
 	"os"
 
-	"github.com/polygonid/sh-id-platform/internal/config"
+	"github.com/joho/godotenv"
 	"github.com/polygonid/sh-id-platform/internal/db/schema"
 	"github.com/polygonid/sh-id-platform/internal/log"
 
 	_ "github.com/lib/pq"
 )
 
+const IssuerDatabaseUrl = "ISSUER_DATABASE_URL"
+
 func main() {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
-	cfg, err := config.Load("")
-	if err != nil {
-		log.Error(ctx, "cannot load config", "err", err)
+	if os.Getenv(IssuerDatabaseUrl) == "" {
+		err := godotenv.Load(".env-issuer")
+		if err != nil {
+			log.Error(ctx, "Error loading .env-issuer file")
+		}
 	}
 
-	log.Config(cfg.Log.Level, cfg.Log.Mode, os.Stdout)
-	log.Debug(ctx, "database", "url", cfg.Database.URL)
+	databaseUrl := os.Getenv(IssuerDatabaseUrl)
+	log.Config(log.LevelDebug, log.LevelDebug, os.Stdout)
+	log.Debug(ctx, "database", "url", databaseUrl)
 
-	if err := schema.Migrate(cfg.Database.URL); err != nil {
+	if err := schema.Migrate(databaseUrl); err != nil {
 		log.Error(ctx, "error migrating database", "err", err)
 		return
 	}