analyze/internal/model/do/init.go

package do

import (
	"analyze/internal/model"
	elastic "app.yhyue.com/moapp/jybase/es"
	m "app.yhyue.com/moapp/jybase/mongodb"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gctx"
)

var (
	Mgo m.MongodbSim
	Es  *elastic.EsV7
)

func init() {
	g.Log().Info(model.Ctx, "--数据库信息初始化 start")
	if g.Cfg().MustGet(gctx.New(), "es").IsMap() {
		Es = &elastic.EsV7{
			Address:  g.Cfg().MustGet(gctx.New(), "es.address").String(),
			Size:     g.Cfg().MustGet(gctx.New(), "es.size").Int(),
			UserName: g.Cfg().MustGet(gctx.New(), "es.userName").String(),
			Password: g.Cfg().MustGet(gctx.New(), "es.password").String(),
		}
		Es.Init()
		g.Log().Info(model.Ctx, "初始化 elasticsearch")
	}
	//
	if g.Cfg().MustGet(gctx.New(), "mgo").IsMap() {
		g.Log().Info(model.Ctx, "初始化 mongodb main")
		Mgo = m.MongodbSim{
			MongodbAddr: g.Cfg().MustGet(gctx.New(), "mgo.address").String(),
			Size:        g.Cfg().MustGet(gctx.New(), "mgo.size").Int(),
			DbName:      g.Cfg().MustGet(gctx.New(), "mgo.dbName").String(),
		}
		Mgo.InitPool()
	}
	g.Log().Info(model.Ctx, "--数据库信息初始化 end")
}