Open-IM-Server/pkg/common/db/model.go

package db

import (
	"Open_IM/pkg/common/config"
	"Open_IM/pkg/common/log"
//	"context"
//	"fmt"
	"github.com/garyburd/redigo/redis"
	"gopkg.in/mgo.v2"
	"time"

	//"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
//	"go.mongodb.org/mongo-driver/mongo/options"

)

var DB DataBases

type DataBases struct {
	MysqlDB    mysqlDB
	mgoSession *mgo.Session
	redisPool  *redis.Pool
	mongoClient *mongo.Client
}

func key(dbAddress, dbName string) string {
	return dbAddress + "_" + dbName
}

func init() {
	var mgoSession *mgo.Session
	var mongoClient *mongo.Client
	var err1 error
	//mysql init
	initMysqlDB()
	// mongo init
	// "mongodb://sysop:moon@localhost/records"
	// uri := "mongodb://user:pass@sample.host:27017/?maxPoolSize=20&w=majority"
	//uri := fmt.Sprintf("mongodb://%s:%s@%s/%s/?maxPoolSize=%d",
	//	config.Config.Mongo.DBUserName, config.Config.Mongo.DBPassword,
	//	config.Config.Mongo.DBAddress[0],config.Config.Mongo.DBDatabase,
	//	config.Config.Mongo.DBMaxPoolSize)
	//
	//mongoClient, err := mongo.Connect(context.TODO(), options.Client().ApplyURI(uri))
	//if err != nil{
	//	log.NewError(" mongo.Connect  failed, try ", err.Error(), uri)
	//	time.Sleep(time.Duration(30) * time.Second)
	//	mongoClient, err1 = mongo.Connect(context.TODO(), options.Client().ApplyURI(uri))
	//	if err1 != nil {
	//		log.NewError(" mongo.Connect  failed, panic", err.Error(), uri)
	//		panic(err1.Error())
	//	}
	//}



	mgoDailInfo := &mgo.DialInfo{
		Addrs:     config.Config.Mongo.DBAddress,
		Direct:    config.Config.Mongo.DBDirect,
		Timeout:   time.Second * time.Duration(config.Config.Mongo.DBTimeout),
		Database:  config.Config.Mongo.DBDatabase,
		Source:    config.Config.Mongo.DBSource,
		Username:  config.Config.Mongo.DBUserName,
		Password:  config.Config.Mongo.DBPassword,
		PoolLimit: config.Config.Mongo.DBMaxPoolSize,
	}
	mgoSession, err := mgo.DialWithInfo(mgoDailInfo)

	if err != nil {

		mgoSession, err1 = mgo.DialWithInfo(mgoDailInfo)
		if err1 != nil {
//			log.NewError(" mongo.Connect  failed, panic", err.Error(), uri)
			panic(err1.Error())
		}
	}
	DB.mongoClient = mongoClient
	DB.mgoSession = mgoSession
	DB.mgoSession.SetMode(mgo.Monotonic, true)
	c := DB.mgoSession.DB(config.Config.Mongo.DBDatabase).C(cChat)
	err = c.EnsureIndexKey("uid")
	if err != nil {
		panic(err.Error())
	}


	// redis pool init
	DB.redisPool = &redis.Pool{
		MaxIdle:     config.Config.Redis.DBMaxIdle,
		MaxActive:   config.Config.Redis.DBMaxActive,
		IdleTimeout: time.Duration(config.Config.Redis.DBIdleTimeout) * time.Second,
		Dial: func() (redis.Conn, error) {
			return redis.Dial(
				"tcp",
				config.Config.Redis.DBAddress,
				redis.DialReadTimeout(time.Duration(1000)*time.Millisecond),
				redis.DialWriteTimeout(time.Duration(1000)*time.Millisecond),
				redis.DialConnectTimeout(time.Duration(1000)*time.Millisecond),
				redis.DialDatabase(0),
				redis.DialPassword(config.Config.Redis.DBPassWord),
			)
		},
	}
}