HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix lint

This commit is contained in:
Lunny Xiao 2023-12-31 21:43:16 +08:00
parent 7d4e191660
commit 97bf025984
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
migrate/migrate.go
View File

@ -87,7 +87,11 @@ func (m *Migrate) Migrate() error {
return err return err
} }
if m.initSchema != nil && m.isFirstRun() { isFirstRun, err := m.isFirstRun()
if err != nil {
return err
}
if m.initSchema != nil && isFirstRun {
return m.runInitSchema() return m.runInitSchema()
} }
@ -200,11 +204,10 @@ func (m *Migrate) migrationDidRun(mig *Migration) (bool, error) {
return count > 0, err return count > 0, err
} }
func (m *Migrate) isFirstRun() bool { func (m *Migrate) isFirstRun() (bool, error) {
row := m.db.DB().QueryRow(fmt.Sprintf("SELECT COUNT(*) FROM %s", m.options.TableName))
var count int var count int
_ = row.Scan(&count) _, err := m.db.SQL(fmt.Sprintf("SELECT COUNT(*) FROM %s", m.options.TableName)).Get(&count)
return count == 0 return count == 0, err
} }
func (m *Migrate) insertMigration(id string) error { func (m *Migrate) insertMigration(id string) error {