hotime/dri/db/auto.go
2019-11-10 16:49:05 +08:00

17 lines
411 B
Go

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.Index(appIns.Config.GetString("dbName"), ".db") == len(appIns.Config.GetString("dbName"))-3 {
sqlite.SetDB(appIns)
} else {
mysql.SetDB(appIns)
}
}