自动设置数据库

2019-11-10 16:42:49 +08:00
parent 177eb86eae
commit 58b908f8ba
6 changed files with 114 additions and 21 deletions
@@ -0,0 +1,16 @@
+package db
+
+import (
+	"code.hoteas.com/hoteas/hotime"
+	"code.hoteas.com/hoteas/hotime/dri/mysql"
+	"code.hoteas.com/hoteas/hotime/dri/sqlite"
+	"strings"
+)
+
+func SetDB(appIns *hotime.Application) {
+	if appIns.Config.GetString("dbType") == "sqlite" || strings.Contains(appIns.Config.GetString("dbName"), ".db") {
+		sqlite.SetDB(appIns)
+	} else {
+		mysql.SetDB(appIns)
+	}
+}