From c6b1cadc0214c6791a5c5bcc62ab57601e38f457 Mon Sep 17 00:00:00 2001
From: liuxiaolong <liuxiaolong@aiotlink.com>
Date: 星期四, 13 八月 2020 11:31:08 +0800
Subject: [PATCH] add db restriction
---
models/db.go | 10 +++++++---
1 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/models/db.go b/models/db.go
index f4be3a1..72258a9 100644
--- a/models/db.go
+++ b/models/db.go
@@ -1,14 +1,18 @@
package models
import (
+ "car-service/extend/util"
"github.com/astaxie/beego"
"github.com/astaxie/beego/orm"
- _ "github.com/go-sql-driver/mysql" //瀵煎叆鏁版嵁搴撻┍鍔�
+ _ "github.com/mattn/go-sqlite3"
)
func InitDb() {
dbUrl := beego.AppConfig.String("dbUrl")
- _ = orm.RegisterDataBase("default", "mysql", dbUrl)
- orm.RegisterModel(new(User))
+ rootPath := util.GetAppRootPath()
+ dbPath := rootPath +"/"+ dbUrl
+ orm.RegisterDriver("sqlite", orm.DRSqlite)
+ _ = orm.RegisterDataBase("default", "sqlite3", dbPath)
+ orm.RegisterModel(new(User), new(UserCar), new(SysUpgrade), new(UserClient))
_ = orm.RunSyncdb("default", false, true)
}
--
Gitblit v1.8.0