fieldproject/data_credible/init.go

package main

import (
	"data_credible/config"
	"fmt"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/log"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
	"os"
)

var (
	Mongo, MongoS, MongoE, MongoBz *mongodb.MongodbSim

	SiteMap map[string]interface{}
	CodeMap map[string]interface{}

	Dbname string
	Coll   string

	savePool chan map[string]interface{}
	saveSp   chan bool
	saveSize int
)

func init() {
	config.Init("./common.toml")
	InitLog()
	Mongo = &mongodb.MongodbSim{
		MongodbAddr: config.Conf.DB.Mongo.Addr,
		Size:        config.Conf.DB.Mongo.Size,
		DbName:      config.Conf.DB.Mongo.Dbname,
		UserName:    config.Conf.DB.Mongo.User,
		Password:    config.Conf.DB.Mongo.Password,
	}
	Mongo.InitPool()
	MongoBz = &mongodb.MongodbSim{
		MongodbAddr: config.Conf.DB.MongoBz.Addr,
		Size:        config.Conf.DB.MongoBz.Size,
		DbName:      config.Conf.DB.MongoBz.Dbname,
	}
	MongoBz.InitPool()
	MongoE = &mongodb.MongodbSim{
		MongodbAddr: config.Conf.DB.MongoE.Addr,
		Size:        config.Conf.DB.MongoE.Size,
		DbName:      config.Conf.DB.MongoE.Dbname,
	}
	MongoE.InitPool()
	MongoS = &mongodb.MongodbSim{
		MongodbAddr: config.Conf.DB.MongoS.Addr,
		Size:        config.Conf.DB.MongoS.Size,
		DbName:      config.Conf.DB.MongoS.Dbname,
	}
	MongoS.InitPool()

	SiteMap = make(map[string]interface{}, 50000)
	CodeMap = make(map[string]interface{}, 80000)

	savePool = make(chan map[string]interface{}, 5000)
	saveSp = make(chan bool, 1)
	saveSize = 200
}

// InitLog @Description
// @Author J 2022/7/26 15:30
func InitLog() {
	logcfg := config.Conf.Log

	err := log.InitLog(
		log.Path(logcfg.LogPath),
		log.Level(logcfg.LogLevel),
		log.Compress(logcfg.Compress),
		log.MaxSize(logcfg.MaxSize),
		log.MaxBackups(logcfg.MaxBackups),
		log.MaxAge(logcfg.MaxAge),
		log.Format(logcfg.Format),
	)
	if err != nil {
		fmt.Printf("InitLog failed: %v\n", err)
		os.Exit(1)
	}
}