chore(server): enable to set visualizer db name by .env (#1345)

reearth · Jan 10, 2025 · 52f66e4 · 52f66e4
1 parent 887173f
commit 52f66e4
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/server/.env.example b/server/.env.example
@@ -1,6 +1,8 @@
 # General
 PORT=8080
-REEARTH_DB=mongodb://localhost
+REEARTH_DB=mongodb://localhost # for reearth database url
+REEARTH_DB_ACCOUNT=reearth_account # for reearth account database name
+REEARTH_DB_VIS=reearth # for reearth visualizer database name
 REEARTH_HOST=https://localhost:8080
 REEARTH_HOST_WEB=https://localhost:3000
 REEARTH_ASSETBASEURL=https://localhost:8080/assets

diff --git a/server/internal/app/config/config.go b/server/internal/app/config/config.go
@@ -31,6 +31,7 @@ type Config struct {
 	DB               string            `default:"mongodb://localhost"`
 	DB_Account       string            `pp:",omitempty"`
 	DB_Users         []appx.NamedURI   `pp:",omitempty"`
+	DB_Vis           string            `pp:",omitempty"`
 	GraphQL          GraphQLConfig     `pp:",omitempty"`
 	Published        PublishedConfig   `pp:",omitempty"`
 	GCPProject       string            `envconfig:"GOOGLE_CLOUD_PROJECT" pp:",omitempty"`

diff --git a/server/internal/app/repo.go b/server/internal/app/repo.go
@@ -25,7 +25,8 @@ import (
 	"go.opentelemetry.io/contrib/instrumentation/go.mongodb.org/mongo-driver/mongo/otelmongo"
 )
 
-const databaseName = "reearth"
+const defaultVisDatabase = "reearth"
+const defaultAccountDatabase = "reearth_account"
 
 func initReposAndGateways(ctx context.Context, conf *config.Config, debug bool) (*repo.Container, *gateway.Container, *accountrepo.Container, *accountgateway.Container) {
 	gateways := &gateway.Container{}
@@ -42,11 +43,11 @@ func initReposAndGateways(ctx context.Context, conf *config.Config, debug bool)
 		log.Fatalf("mongo error: %+v\n", err)
 	}
 
-	// repos
+	// for account database
 	accountDatabase := conf.DB_Account
 	accountRepoCompat := false
 	if accountDatabase == "" {
-		accountDatabase = databaseName
+		accountDatabase = defaultAccountDatabase
 		accountRepoCompat = true
 	}
 
@@ -66,7 +67,13 @@ func initReposAndGateways(ctx context.Context, conf *config.Config, debug bool)
 		log.Fatalf("Failed to init mongo: %+v\n", err)
 	}
 
-	repos, err := mongorepo.NewWithExtensions(ctx, client.Database(databaseName), accountRepos, txAvailable, conf.Ext_Plugin)
+	// for reearth visualizer database
+	visDatabase := conf.DB_Vis
+	if visDatabase == "" {
+		visDatabase = defaultVisDatabase
+	}
+
+	repos, err := mongorepo.NewWithExtensions(ctx, client.Database(visDatabase), accountRepos, txAvailable, conf.Ext_Plugin)
 	if err != nil {
 		log.Fatalf("Failed to init mongo: %+v\n", err)
 	}