jybxseo/utility/db.go

package utility

import (
	m "app.yhyue.com/moapp/jybase/mongodb"
	"context"
	"fmt"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gctx"
)

type (
	mgoConf struct {
		Address         string
		Size            int
		DbName          string
		ReplSet         string
		UserName        string
		Password        string
		Collection      string
		Collection_back string
	}
	mongo struct {
		Default mgoConf
		Bidding mgoConf
	}
)

var (
	mongdbConf mongo
	Mgo        m.MongodbSim
	BiddingMgo m.MongodbSim
)

func init() {
	initCtx := gctx.New()
	if err := g.Cfg().MustGet(initCtx, "mongodb").Struct(&mongdbConf); err != nil {
		panic(err)
	}
	Mgo = m.MongodbSim{
		MongodbAddr: mongdbConf.Default.Address,
		Size:        mongdbConf.Default.Size,
		DbName:      mongdbConf.Default.DbName,
		UserName:    mongdbConf.Default.UserName,
		Password:    mongdbConf.Default.Password,
		ReplSet:     mongdbConf.Default.ReplSet,
	}
	Mgo.InitPool()

	fmt.Println("mongdbConf.Bidding.Address", mongdbConf.Bidding.UserName, mongdbConf.Bidding.Password)
	BiddingMgo = m.MongodbSim{
		MongodbAddr: mongdbConf.Bidding.Address,
		Size:        mongdbConf.Bidding.Size,
		DbName:      mongdbConf.Bidding.DbName,
		UserName:    mongdbConf.Bidding.UserName,
		Password:    mongdbConf.Bidding.Password,
		ReplSet:     mongdbConf.Bidding.ReplSet,
	}
	BiddingMgo.InitPool()

	if err := g.Cfg().MustGet(context.Background(), "entMongodb").Struct(&mongdbConf); err != nil {
		panic(err)
	}
}