自动设置数据库

2019-11-10 16:51:27 +08:00
parent fa641b9dcd
commit 7d479f181a
2 changed files with 14 additions and 3 deletions
@@ -9,8 +9,10 @@ import (

 func SetDB(appIns *hotime.Application) {
 	if appIns.Config.GetString("dbType") == "sqlite" || strings.Index(appIns.Config.GetString("dbName"), ".db") == len(appIns.Config.GetString("dbName"))-3 {
+		appIns.Config["dbType"] = "sqlite"
 		sqlite.SetDB(appIns)
 	} else {
+		appIns.Config["dbType"] = "mysql"
 		mysql.SetDB(appIns)
 	}
 }