iedc-go/dri/db/auto.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)
	}
}