From 97bf025984460ed990b94b0f2dc44611bd85f358 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Sun, 31 Dec 2023 21:43:16 +0800
Subject: [PATCH] Fix lint

---
 migrate/migrate.go | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/migrate/migrate.go b/migrate/migrate.go
index 1fb94232..99def671 100644
--- a/migrate/migrate.go
+++ b/migrate/migrate.go
@@ -87,7 +87,11 @@ func (m *Migrate) Migrate() error {
 		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()
 	}
 
@@ -200,11 +204,10 @@ func (m *Migrate) migrationDidRun(mig *Migration) (bool, error) {
 	return count > 0, err
 }
 
-func (m *Migrate) isFirstRun() bool {
-	row := m.db.DB().QueryRow(fmt.Sprintf("SELECT COUNT(*) FROM %s", m.options.TableName))
+func (m *Migrate) isFirstRun() (bool, error) {
 	var count int
-	_ = row.Scan(&count)
-	return count == 0
+	_, err := m.db.SQL(fmt.Sprintf("SELECT COUNT(*) FROM %s", m.options.TableName)).Get(&count)
+	return count == 0, err
 }
 
 func (m *Migrate) insertMigration(id string) error {