Add support for configuring automatic migration (and default to false) (

warrant-dev#85)
difftim · Apr 18, 2023 · 342c090 · 342c090
1 parent 9327a94
commit 342c090
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 18 deletions.
diff --git a/cmd/warrant/main.go b/cmd/warrant/main.go
@@ -57,9 +57,11 @@ func (env *ServiceEnv) InitDB(config config.Config) error {
 			return err
 		}
 
-		err = db.Migrate(ctx, MySQLDatastoreMigrationVersion)
-		if err != nil {
-			return err
+		if config.AutoMigrate {
+			err = db.Migrate(ctx, MySQLDatastoreMigrationVersion)
+			if err != nil {
+				return err
+			}
 		}
 
 		env.Datastore = db
@@ -73,9 +75,11 @@ func (env *ServiceEnv) InitDB(config config.Config) error {
 			return err
 		}
 
-		err = db.Migrate(ctx, PostgresDatastoreMigrationVersion)
-		if err != nil {
-			return err
+		if config.AutoMigrate {
+			err = db.Migrate(ctx, PostgresDatastoreMigrationVersion)
+			if err != nil {
+				return err
+			}
 		}
 
 		env.Datastore = db
@@ -89,9 +93,11 @@ func (env *ServiceEnv) InitDB(config config.Config) error {
 			return err
 		}
 
-		err = db.Migrate(ctx, SQLiteDatastoreMigrationVersion)
-		if err != nil {
-			return err
+		if config.AutoMigrate {
+			err = db.Migrate(ctx, SQLiteDatastoreMigrationVersion)
+			if err != nil {
+				return err
+			}
 		}
 
 		env.Datastore = db
@@ -112,9 +118,11 @@ func (env *ServiceEnv) InitEventDB(config config.Config) error {
 			return err
 		}
 
-		err = db.Migrate(ctx, MySQLEventstoreMigrationVersion)
-		if err != nil {
-			return err
+		if config.AutoMigrate {
+			err = db.Migrate(ctx, MySQLEventstoreMigrationVersion)
+			if err != nil {
+				return err
+			}
 		}
 
 		env.Eventstore = db
@@ -128,9 +136,11 @@ func (env *ServiceEnv) InitEventDB(config config.Config) error {
 			return err
 		}
 
-		err = db.Migrate(ctx, PostgresEventstoreMigrationVersion)
-		if err != nil {
-			return err
+		if config.AutoMigrate {
+			err = db.Migrate(ctx, PostgresEventstoreMigrationVersion)
+			if err != nil {
+				return err
+			}
 		}
 
 		env.Eventstore = db
@@ -144,9 +154,11 @@ func (env *ServiceEnv) InitEventDB(config config.Config) error {
 			return err
 		}
 
-		err = db.Migrate(ctx, SQLiteEventstoreMigrationVersion)
-		if err != nil {
-			return err
+		if config.AutoMigrate {
+			err = db.Migrate(ctx, SQLiteEventstoreMigrationVersion)
+			if err != nil {
+				return err
+			}
 		}
 
 		env.Eventstore = db

diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -29,6 +29,7 @@ type Config struct {
 	Port            int               `mapstructure:"port"`
 	LogLevel        int8              `mapstructure:"logLevel"`
 	EnableAccessLog bool              `mapstructure:"enableAccessLog"`
+	AutoMigrate     bool              `mapstructure:"autoMigrate"`
 	Datastore       *DatastoreConfig  `mapstructure:"datastore"`
 	Eventstore      *EventstoreConfig `mapstructure:"eventstore"`
 	ApiKey          string            `mapstructure:"apiKey"`
@@ -89,6 +90,7 @@ func NewConfig() Config {
 	viper.SetDefault("port", 8000)
 	viper.SetDefault("levelLevel", zerolog.DebugLevel)
 	viper.SetDefault("enableAccessLog", true)
+	viper.SetDefault("autoMigrate", false)
 	viper.SetDefault("datastore.mysql.migrationSource", DefaultMySQLDatastoreMigrationSource)
 	viper.SetDefault("datastore.postgres.migrationSource", DefaultPostgresDatastoreMigrationSource)
 	viper.SetDefault("datastore.sqlite.migrationSource", DefaultSQLiteDatastoreMigrationSource)