message/db/db.go

package db

import (
	"github.com/gogf/gf/v2/os/gcfg"
	"github.com/gogf/gf/v2/os/gctx"
	"log"

	"app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/redis"
)

var (
	Mgo_Log *mongodb.MongodbSim
	Mgo     *mongodb.MongodbSim
)

type MgoConf struct {
	Address         string
	Size            int
	DbName          string
	ReplSet         string
	UserName        string
	Password        string
	Collection      string
	Collection_back string
}

func init() {
	redis.InitRedis(gcfg.Instance().MustGet(gctx.New(), "databases.redis", "").String())
	log.Println("初始化 redis")

	var mgoConf MgoConf
	if err := gcfg.Instance().MustGet(gctx.New(), "databases.mogLog").Scan(&mgoConf); err == nil {
		Mgo_Log = &mongodb.MongodbSim{
			MongodbAddr: mgoConf.Address,
			Size:        mgoConf.Size,
			DbName:      mgoConf.DbName,
			ReplSet:     mgoConf.ReplSet,
			UserName:    mgoConf.UserName,
			Password:    mgoConf.Password,
		}
		Mgo_Log.InitPool()
		log.Printf("初始化 mongodb log 完成 %+v\n", mgoConf)
	}
	if err := gcfg.Instance().MustGet(gctx.New(), "databases.mongodb").Scan(&mgoConf); err == nil {
		Mgo = &mongodb.MongodbSim{
			MongodbAddr: mgoConf.Address,
			Size:        mgoConf.Size,
			DbName:      mgoConf.DbName,
			ReplSet:     mgoConf.ReplSet,
			UserName:    mgoConf.UserName,
			Password:    mgoConf.Password,
		}
		Mgo.InitPool()
		log.Printf("初始化 mongodb 完成 %+v\n", mgoConf)
	}

}