2024-09-30 00:58:49 +02:00
|
|
|
package database
|
|
|
|
|
|
|
|
import (
|
2024-11-18 00:10:58 +01:00
|
|
|
"database/sql"
|
2024-09-30 00:58:49 +02:00
|
|
|
"embed"
|
|
|
|
"io/fs"
|
|
|
|
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
|
|
)
|
|
|
|
|
|
|
|
//go:embed sql/*.sql
|
|
|
|
var schemaFiles embed.FS
|
|
|
|
|
|
|
|
func (db *DB) migrate() error {
|
|
|
|
statements := make([]string, 0)
|
|
|
|
err := fs.WalkDir(schemaFiles, ".", func(path string, d fs.DirEntry, err error) error {
|
|
|
|
if d.IsDir() || err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
var stmts []byte
|
|
|
|
if stmts, err = schemaFiles.ReadFile(path); err != nil {
|
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
statements = append(statements, string(stmts))
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2024-11-18 00:10:58 +01:00
|
|
|
return db.WithTransaction(func(tx *sql.Tx) error {
|
|
|
|
var version int
|
|
|
|
if err = tx.QueryRowContext(db.ctx, `SELECT version FROM schema_version;`).Scan(&version); err != nil {
|
|
|
|
if err.Error() == "no such table: schema_version" {
|
|
|
|
version = 0
|
|
|
|
} else {
|
|
|
|
return err
|
|
|
|
}
|
2024-09-30 00:58:49 +02:00
|
|
|
}
|
|
|
|
|
2024-11-18 00:10:58 +01:00
|
|
|
for version < len(statements) {
|
|
|
|
if _, err = tx.ExecContext(db.ctx, statements[version]); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
version++
|
2024-09-30 00:58:49 +02:00
|
|
|
}
|
2024-11-18 00:10:58 +01:00
|
|
|
_, err = tx.ExecContext(db.ctx, `DELETE FROM schema_version; INSERT INTO schema_version (version) VALUES (?);`, version)
|
2024-09-30 00:58:49 +02:00
|
|
|
return err
|
2024-11-18 00:10:58 +01:00
|
|
|
})
|
2024-09-30 00:58:49 +02:00
|
|
|
}
|