sqldb: fix dirty migration in v0.19.0-rc1

This commit is contained in:
Andras Banki-Horvath
2025-03-26 11:22:30 +01:00
parent dae212697d
commit 7e54682493
3 changed files with 56 additions and 0 deletions

View File

@@ -198,3 +198,15 @@ func (s *PostgresStore) GetSchemaVersion() (int, bool, error) {
return version, dirty, nil
}
// SetSchemaVersion sets the schema version of the Postgres database.
//
// NOTE: This alters the internal database schema tracker. USE WITH CAUTION!!!
func (s *PostgresStore) SetSchemaVersion(version int, dirty bool) error {
driver, err := pgx_migrate.WithInstance(s.DB, &pgx_migrate.Config{})
if err != nil {
return errPostgresMigration(err)
}
return driver.SetVersion(version, dirty)
}