From f49fc848e464d21bc9960044c193a05cd5161e89 Mon Sep 17 00:00:00 2001 From: Mo Tarbin Date: Sat, 23 Nov 2024 20:51:25 -0500 Subject: [PATCH] use embed File System for Migration script --- internal/database/migration.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/internal/database/migration.go b/internal/database/migration.go index 1546310..5034640 100644 --- a/internal/database/migration.go +++ b/internal/database/migration.go @@ -1,6 +1,7 @@ package database import ( + "embed" "fmt" "os" "path/filepath" @@ -17,6 +18,9 @@ import ( "gorm.io/gorm" ) +//go:embed migrations/*.sql +var embeddedMigrations embed.FS + func Migration(db *gorm.DB) error { if err := db.AutoMigrate(uModel.User{}, chModel.Chore{}, chModel.ChoreHistory{}, @@ -41,8 +45,9 @@ func Migration(db *gorm.DB) error { } func MigrationScripts(gormDB *gorm.DB, cfg *config.Config) error { - migrations := &migrate.FileMigrationSource{ - Dir: migrationDir(), + migrations := &migrate.EmbedFileSystemMigrationSource{ + FileSystem: embeddedMigrations, + Root: "migrations", } path := os.Getenv("DT_SQLITE_PATH")