50 lines
915 B
Go
50 lines
915 B
Go
package db
|
|
|
|
import (
|
|
"common/config"
|
|
"common/db/mongo"
|
|
"common/db/mysql"
|
|
"common/log"
|
|
"common/utils"
|
|
"fmt"
|
|
"github.com/jinzhu/configor"
|
|
"math/rand"
|
|
"os"
|
|
)
|
|
|
|
// 用于单元测试初始化数据库
|
|
|
|
func InitTestDB(env string) {
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
fmt.Println("无法获取工作目录:", err)
|
|
return
|
|
}
|
|
|
|
var commonCfg config.CommonConfig
|
|
err = configor.Load(&commonCfg, fmt.Sprintf(wd+"/config/common-config-%v.json", env))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Init(&commonCfg.Logger)
|
|
utils.InitSnowflake(int64(rand.Intn(1000)))
|
|
//// Redis
|
|
//if err := redis.Init(&commonCfg.Redis); err != nil {
|
|
// panic(err)
|
|
//}
|
|
// MySQL
|
|
if err := mysql.Init(&commonCfg.MysqlConfig); err != nil {
|
|
panic(err)
|
|
}
|
|
// Mongo
|
|
if err := mongo.Init(&commonCfg.MongoConfig); err != nil {
|
|
panic(err)
|
|
}
|
|
//// Etcd
|
|
//if err := etcd.Init(&commonCfg.Etcd); err != nil {
|
|
// panic(err)
|
|
//}
|
|
|
|
}
|