From e8e4f7ce931b26a8063ed41892f7a8dba28e86a3 Mon Sep 17 00:00:00 2001
From: SampsonYe <271232507@qq.com>
Date: Mon, 14 Mar 2022 00:10:07 +0800
Subject: [PATCH] =?UTF-8?q?migration=20=E8=BF=81=E7=A7=BB=E5=88=B0?=
 =?UTF-8?q?=E7=8B=AC=E7=AB=8B=E7=9B=AE=E5=BD=95=E4=B8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/migrations/migrations.go | 25 +++++++++++++++++++++++++
 internal/models/models.go         | 26 +++-----------------------
 2 files changed, 28 insertions(+), 23 deletions(-)
 create mode 100644 internal/migrations/migrations.go

diff --git a/internal/migrations/migrations.go b/internal/migrations/migrations.go
new file mode 100644
index 0000000..4c62689
--- /dev/null
+++ b/internal/migrations/migrations.go
@@ -0,0 +1,25 @@
+package migrations
+
+import (
+	"github.com/astaxie/beego/orm"
+	"time"
+)
+
+// Migration db migration base interface
+type Migration interface {
+	GetCreateAt() time.Time
+	Upgrade(ormer orm.Ormer)
+}
+
+// InitMigration db migration register
+func InitMigration() {
+	migrationTypes := []Migration{
+		new(Migration20220101),
+		new(Migration20220309),
+	}
+
+	//数据迁移
+	for _, m := range migrationTypes {
+		m.Upgrade(orm.NewOrm())
+	}
+}
diff --git a/internal/models/models.go b/internal/models/models.go
index 26c4d1d..1b8b386 100644
--- a/internal/models/models.go
+++ b/internal/models/models.go
@@ -35,12 +35,6 @@ type Addons struct {
 	DeleteAt *time.Time `orm:"column(delete_at);type(datetime);index;null" json:"delete_at"`
 }
 
-// db migration base interface
-type Migration interface {
-	GetCreateAt() time.Time
-	Upgrade(ormer orm.Ormer)
-}
-
 // TableNamePrefix ..
 const TableNamePrefix = "atom"
 
@@ -66,9 +60,8 @@ func (a *Addons) MarkDeleted() {
 }
 
 var (
-	dbName         string
-	tableNames     []string
-	migrationTypes []Migration
+	dbName     string
+	tableNames []string
 )
 
 func initOrm() {
@@ -156,25 +149,12 @@ func initOrm() {
 
 }
 
-//db migration register
-func initMigration() {
-	migrationTypes = []Migration{
-		new(migrations.Migration20220101),
-		new(migrations.Migration20220309),
-	}
-
-	//数据迁移
-	for _, m := range migrationTypes {
-		m.Upgrade(orm.NewOrm())
-	}
-}
-
 // Init ...
 func init() {
 	if len(os.Args) > 1 && os.Args[1][:5] == "-test" {
 		return
 	}
 	initOrm()
-	initMigration()
+	migrations.InitMigration()
 	// orm.RunSyncdb("default", false, true)
 }
-- 
GitLab