chore: add version checker

This commit is contained in:
boojack 2022-07-02 01:06:28 +08:00
parent 0d317839d2
commit 3c58953e56
2 changed files with 64 additions and 2 deletions

View File

@ -153,12 +153,20 @@ func (db *DB) compareMigrationHistory() error {
}
currentVersion := common.Version
migrationHistory, err := upsertMigrationHistory(db.Db, currentVersion)
migrationHistoryFind := MigrationHistoryFind{
Version: currentVersion,
}
migrationHistory, err := findMigrationHistory(db.Db, &migrationHistoryFind)
if err != nil {
return err
}
if migrationHistory == nil {
return fmt.Errorf("failed to upsert migration history")
// ...do schema migration,
// then upsert the newest version to migration_history
_, err := upsertMigrationHistory(db.Db, currentVersion)
if err != nil {
return err
}
}
return nil

View File

@ -2,6 +2,7 @@ package db
import (
"database/sql"
"strings"
)
type MigrationHistory struct {
@ -9,6 +10,59 @@ type MigrationHistory struct {
Version string
}
type MigrationHistoryFind struct {
Version string
}
func findMigrationHistoryList(db *sql.DB, find *MigrationHistoryFind) ([]*MigrationHistory, error) {
where, args := []string{"1 = 1"}, []interface{}{}
where, args = append(where, "version = ?"), append(args, find.Version)
rows, err := db.Query(`
SELECT
version,
created_ts
FROM
migration_history
WHERE `+strings.Join(where, " AND ")+`
ORDER BY created_ts DESC`,
args...,
)
if err != nil {
return nil, err
}
defer rows.Close()
migrationHistoryList := make([]*MigrationHistory, 0)
for rows.Next() {
var migrationHistory MigrationHistory
if err := rows.Scan(
&migrationHistory.Version,
&migrationHistory.CreatedTs,
); err != nil {
return nil, err
}
migrationHistoryList = append(migrationHistoryList, &migrationHistory)
}
return migrationHistoryList, nil
}
func findMigrationHistory(db *sql.DB, find *MigrationHistoryFind) (*MigrationHistory, error) {
list, err := findMigrationHistoryList(db, find)
if err != nil {
return nil, err
}
if len(list) == 0 {
return nil, nil
} else {
return list[0], nil
}
}
func upsertMigrationHistory(db *sql.DB, version string) (*MigrationHistory, error) {
row, err := db.Query(`
INSERT INTO migration_history (