add init migrattions

This commit is contained in:
sentriz
2020-02-19 23:00:30 +00:00
parent a42edc3441
commit 8251f399f8
4 changed files with 87 additions and 5 deletions

View File

@@ -8,6 +8,7 @@ import (
"github.com/jinzhu/gorm"
"github.com/pkg/errors"
"gopkg.in/gormigrate.v1"
)
var (
@@ -35,12 +36,14 @@ func New(path string) (*DB, error) {
db.SetLogger(log.New(os.Stdout, "gorm ", 0))
db.DB().SetMaxOpenConns(dbMaxOpenConns)
db.Exec("PRAGMA journal_mode=WAL;")
// TODO: don't log if user already exists
db.FirstOrCreate(&User{}, User{
Name: "admin",
Password: "admin",
IsAdmin: true,
migr := gormigrate.New(db, gormigrate.DefaultOptions, []*gormigrate.Migration{
&migrationCreateInitUser,
&migrationInitSchema,
&migrationMergePlaylist,
})
if err = migr.Migrate(); err != nil {
return nil, errors.Wrap(err, "migrating to latest version")
}
return &DB{DB: db}, nil
}

View File

@@ -1 +1,66 @@
//nolint:deadcode,varcheck
package db
import (
"github.com/jinzhu/gorm"
"gopkg.in/gormigrate.v1"
)
var migrationCreateInitUser = gormigrate.Migration{
ID: "202002192019",
Migrate: func(tx *gorm.DB) error {
const (
initUsername = "admin"
initPassword = "admin"
)
err := tx.
Where("name = ?", initUsername).
First(&User{}).
Error
if gorm.IsRecordNotFoundError(err) {
tx.Create(&User{
Name: initUsername,
Password: initPassword,
IsAdmin: true,
})
return nil
}
return err
},
}
var migrationInitSchema = gormigrate.Migration{
ID: "202002192100",
Migrate: func(tx *gorm.DB) error {
return tx.AutoMigrate(
Artist{},
Track{},
User{},
Setting{},
Play{},
Album{},
Playlist{},
PlayQueue{},
).
Error
},
}
var migrationMergePlaylist = gormigrate.Migration{
ID: "202002192222",
Migrate: func(tx *gorm.DB) error {
return tx.Exec(`
UPDATE playlists
SET items = (
SELECT group_concat(track_id) FROM (
SELECT track_id
FROM playlist_items
WHERE playlist_items.playlist_id=playlists.id
ORDER BY created_at
)
);
DROP TABLE playlist_items;`,
).
Error
},
}