Allow DB override via hidden parameter (#482)

xmtp · Feb 13, 2025 · d911295 · d911295
1 parent e8b01a5
commit d911295
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/cmd/replication/main.go b/cmd/replication/main.go
@@ -65,10 +65,14 @@ func main() {
 	tracing.GoPanicWrap(ctx, &wg, "main", func(ctx context.Context) {
 		var dbInstance *sql.DB
 		if options.Replication.Enable || options.Sync.Enable || options.Indexer.Enable {
+			namespace := options.DB.NameOverride
+			if namespace == "" {
+				namespace = utils.BuildNamespace(options)
+			}
 			dbInstance, err = db.NewNamespacedDB(
 				ctx,
 				options.DB.WriterConnectionString,
-				utils.BuildNamespace(options),
+				namespace,
 				options.DB.WaitForDB,
 				options.DB.ReadTimeout,
 			)

diff --git a/pkg/config/options.go b/pkg/config/options.go
@@ -25,6 +25,7 @@ type DbOptions struct {
 	WriteTimeout           time.Duration `long:"write-timeout"            env:"XMTPD_DB_WRITE_TIMEOUT"            description:"Timeout for writing to the database"            default:"10s"`
 	MaxOpenConns           int           `long:"max-open-conns"           env:"XMTPD_DB_MAX_OPEN_CONNS"           description:"Maximum number of open connections"             default:"80"`
 	WaitForDB              time.Duration `long:"wait-for"                 env:"XMTPD_DB_WAIT_FOR"                 description:"wait for DB on start, up to specified duration"`
+	NameOverride           string        `long:"name-override"            env:"XMTPD_DB_NAME_OVERRIDE"            description:"Override the automatically generated DB name"                 hidden:"true"`
 }
 
 type IndexerOptions struct {