anthrove/otter-space-sdk
Archived
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 10 Wiki Activity

feat(db): embed migration files

Browse Source 
Signed-off-by: SoXX <soxx@fenpa.ws>
This commit is contained in:
SoXX 2024-06-04 13:28:14 +02:00
parent 549c1ca573
commit b65e942f24
12 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
db/migrations/001_create_enum_Rating_2024_06_04.sql → pkg/database/migrations/001_create_enum_Rating_2024_06_04.sql
View File

0
db/migrations/002_create_enum_TagType_2024_06_04.sql → pkg/database/migrations/002_create_enum_TagType_2024_06_04.sql
View File

0
db/migrations/003_create_table_Post_2024_06_04.sql → pkg/database/migrations/003_create_table_Post_2024_06_04.sql
View File

0
db/migrations/004_create_table_Source_2024_06_04.sql → pkg/database/migrations/004_create_table_Source_2024_06_04.sql
View File

0
db/migrations/005_create_table_Tag_2024_06_04.sql → pkg/database/migrations/005_create_table_Tag_2024_06_04.sql
View File

0
db/migrations/006_create_table_User_2024_06_04.sql → pkg/database/migrations/006_create_table_User_2024_06_04.sql
View File

0
db/migrations/007_create_table_PostReference_2024_06_04.sql → pkg/database/migrations/007_create_table_PostReference_2024_06_04.sql
View File

0
db/migrations/008_create_table_TagAlias_2024_06_04.sql → pkg/database/migrations/008_create_table_TagAlias_2024_06_04.sql
View File

0
db/migrations/009_create_table_TagGroup_2024_06_04.sql → pkg/database/migrations/009_create_table_TagGroup_2024_06_04.sql
View File

0
db/migrations/010_create_table_UserFavorite_2024_06_04.sql → pkg/database/migrations/010_create_table_UserFavorite_2024_06_04.sql
View File

0
db/migrations/011_create_table_UserSource_2024_06_04.sql → pkg/database/migrations/011_create_table_UserSource_2024_06_04.sql
View File

21
pkg/database/postgres.go
View File

@ -3,6 +3,7 @@ package database
import ( import (
"context" "context"
"database/sql" "database/sql"
"embed"
"fmt" "fmt"
"git.dragse.it/anthrove/otter-space-sdk/internal/postgres" "git.dragse.it/anthrove/otter-space-sdk/internal/postgres"
@ -10,10 +11,14 @@ import (
"git.dragse.it/anthrove/otter-space-sdk/pkg/models/graphModels" "git.dragse.it/anthrove/otter-space-sdk/pkg/models/graphModels"
_ "github.com/lib/pq" _ "github.com/lib/pq"
migrate "github.com/rubenv/sql-migrate" migrate "github.com/rubenv/sql-migrate"
log "github.com/sirupsen/logrus"
gormPostgres "gorm.io/driver/postgres" gormPostgres "gorm.io/driver/postgres"
"gorm.io/gorm" "gorm.io/gorm"
) )
//go:embed migrations/*.sql
var embedMigrations embed.FS
type postgresqlConnection struct { type postgresqlConnection struct {
db *gorm.DB db *gorm.DB
} }
@ -142,20 +147,24 @@ func (p postgresqlConnection) GetSourceByURL(ctx context.Context, sourceUrl stri
func (p postgresqlConnection) migrateDatabase(connectionString string) error { func (p postgresqlConnection) migrateDatabase(connectionString string) error {
dialect := "postgres" dialect := "postgres"
migrations := &migrate.FileMigrationSource{ migrations := &migrate.EmbedFileSystemMigrationSource{FileSystem: embedMigrations, Root: "migrations"}
Dir: "db/migrations",
}
db, err := sql.Open(dialect, connectionString) db, err := sql.Open(dialect, connectionString)
if err != nil { if err != nil {
return err return fmt.Errorf("postgres migration: %v", err)
} }
n, err := migrate.Exec(db, dialect, migrations, migrate.Up) n, err := migrate.Exec(db, dialect, migrations, migrate.Up)
if err != nil { if err != nil {
return err return fmt.Errorf("postgres migration: %v", err)
}
if n != 0 {
log.Infof("postgres migration: applied %d migrations!", n)
} else {
log.Info("postgres migration: nothing to migrate")
} }
fmt.Printf("Applied %d migrations!\n", n)
return nil return nil
} }