clz/bill-server-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add migrate sql data

Browse Source
This commit is contained in:
Liangzhao Che 2023-06-01 20:07:02 +08:00
parent 8540470c62
commit 49a21a0449
2 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/infrastructure/sqlite.go
View File

@ -5,6 +5,7 @@ import (
"bill-go-fiber/internal/label" "bill-go-fiber/internal/label"
"gorm.io/driver/sqlite" "gorm.io/driver/sqlite"
"gorm.io/gorm" "gorm.io/gorm"
"os"
) )
func ConnectToSqlite() (*gorm.DB, error) { func ConnectToSqlite() (*gorm.DB, error) {
@ -24,7 +25,14 @@ func ConnectToSqlite() (*gorm.DB, error) {
panic("failed to migrate database.\n") panic("failed to migrate database.\n")
} }
db.Create(&label.Label{ID: 1, Name: "bill"}) // migrate data
go func() {
sql, err := os.ReadFile("scripts/sqliteMigrations.sql")
if err != nil {
panic("failed to read migration sql file: " + err.Error())
}
db.Exec(string(sql))
}()
return db, nil return db, nil
} }

8
scripts/sqliteMigrations.sql Normal file
View File

@ -0,0 +1,8 @@
INSERT INTO labels (id, type, name, count)
VALUES (1, '', 'bill', 0),
(2, '', 'bill', 0),
(3, '', 'bill', 0),
(4, '', 'bill', 0),
(5, '', 'bill', 0),
(6, '', 'bill', 0),
(7, '', 'bill', 0);