自动设置数据库

2019-11-10 16:49:05 +08:00
parent 58b908f8ba
commit fa641b9dcd
2 changed files with 13 additions and 12 deletions
@@ -8,7 +8,7 @@ import (
 )

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