data_validation/src/util/config.go

package util

import (
	"mongodb"
	qu "qfw/util"
	"qfw/util/elastic"
	"sync"
)

//const JYDATAFROMDB = "jyqyfw"
//const JYDATAFROMCOLL = "usermail_history"
//const JYDATATODB = "jyqyfw"

var (
	Sysconfig           map[string]interface{} //配置文件
	Quaconfig           map[string]interface{} //质量配置文件
	Mgo                 *mongodb.MongodbSim
	AllToColl           string              //所有标注数据汇总表
	Password            string              //默认登陆密码
	MgoB                *mongodb.MongodbSim //bidding
	BidColl1            string
	BidColl2            string
	MgoE                *mongodb.MongodbSim //extract
	ExtColl1            string
	ExtColl2            string
	ProjectColl         string
	MgoJy               *mongodb.MongodbSim //jy
	JyHistory           string
	JyUser              string
	Es                  *elastic.Elastic //es
	Index               string
	Itype               string
	CustomerFieldMap_EH map[string]string
	CustomerFieldMap_HE map[string]string
)


var (
	Qy_Es                  					*elastic.Elastic 		//企业es
	Mgo_QY									*mongodb.MongodbSim		//企业mgo
	Qy_Lock 								sync.Mutex         	   	//锁-多线程备用
	Ext_Type,Ext_From						map[string]interface{}	//抽取来源，方式分
	Buyer_Score,S_Winner_Score				map[string]interface{}	//字段规则分
	Budget_Score,Bidamount_Score			map[string]interface{}
	Projectname_Score,Projectcode_Score		map[string]interface{}
)



var SE = qu.SimpleEncrypt{Key: "topJYBX2019"}

const BIDDINGSTARTID = "5a862f0640d2d9bbe88e3cec"
const PROJECTCOLLNAME = "f_project"
const TASKCOLLNAME = "f_task"

func InitConfig() {
	Mgo = &mongodb.MongodbSim{
		MongodbAddr: qu.ObjToString(Sysconfig["mgodb"]),
		Size:        qu.IntAll(Sysconfig["dbsize"]),
		DbName:      qu.ObjToString(Sysconfig["dbname"]),
	}
	Mgo.InitPool()
	AllToColl = qu.ObjToString(Sysconfig["alltocoll"])
	Password = "123"

	//bid
	bid := Sysconfig["bidding"].(map[string]interface{})
	BidColl1 = qu.ObjToString(bid["coll1"])
	BidColl2 = qu.ObjToString(bid["coll2"])
	MgoB = &mongodb.MongodbSim{
		MongodbAddr: qu.ObjToString(bid["addr"]),
		DbName:      qu.ObjToString(bid["db"]),
		Size:        qu.IntAll(bid["size"]),
		//UserName:    qu.ObjToString(bid["username"]),
		//Password:    qu.ObjToString(bid["password"]),
	}
	MgoB.InitPool()

	//ext
	ext := Sysconfig["extract"].(map[string]interface{})
	ExtColl1 = qu.ObjToString(ext["coll1"])
	ExtColl2 = qu.ObjToString(ext["coll2"])
	ProjectColl = qu.ObjToString(ext["projectcoll"])
	MgoE = &mongodb.MongodbSim{
		MongodbAddr: qu.ObjToString(ext["addr"]),
		DbName:      qu.ObjToString(ext["db"]),
		Size:        qu.IntAll(ext["size"]),
	}
	MgoE.InitPool()

	//jy
	jy := Sysconfig["jy"].(map[string]interface{})
	JyHistory = qu.ObjToString(jy["history"])
	JyUser = qu.ObjToString(jy["user"])
	MgoJy = &mongodb.MongodbSim{
		MongodbAddr: qu.ObjToString(jy["addr"]),
		DbName:      qu.ObjToString(jy["db"]),
		Size:        qu.IntAll(jy["size"]),
	}
	MgoJy.InitPool()

	//es
	//es := Sysconfig["es"].(map[string]interface{})
	//Index = qu.ObjToString(es["index"])
	//Itype = qu.ObjToString(es["itype"])
	//Es = &elastic.Elastic{
	//	S_esurl: qu.ObjToString(es["addr"]),
	//	I_size:  qu.IntAll(es["pool"]),
	//}
	//Es.InitElasticSize()

	//other
	CustomerFieldMap_EH = map[string]string{}
	CustomerFieldMap_HE = map[string]string{}
	for k, v := range Sysconfig["customerfield"].(map[string]interface{}) {
		val := v.(string)
		CustomerFieldMap_HE[k] = val
		CustomerFieldMap_EH[val] = k
	}



	initQuaConfig()
}




//质量相关初始化配置
func initQuaConfig()  {
	//线上-需替换-172.17.145.163:27083,172.17.4.187:27082，
	qyxyconf := Quaconfig["qy_mongodb"].(map[string]interface{})
	Mgo_QY = &mongodb.MongodbSim{
		MongodbAddr: qyxyconf["qy_addrName"].(string),
		DbName:      qyxyconf["qy_dbName"].(string),
		Size:        qu.IntAllDef(qyxyconf["qy_pool"], 5),
		UserName:    qyxyconf["qy_username"].(string),
		Password:    qyxyconf["qy_password"].(string),
	}
	Mgo_QY.InitPool()

	//查询企业库-可使用es - 也可以mongo  http://172.17.145.170:9800
	//Qy_Es = &elastic.Elastic{
	//	S_esurl: qu.ObjToString(Quaconfig["qyxy_es"]),
	//	I_size:  10,
	//}
	//Es.InitElasticSize()

	Ext_Type = Quaconfig["ext_type"].(map[string]interface{})
	Ext_From = Quaconfig["ext_from"].(map[string]interface{})

	Buyer_Score = Quaconfig["buyer_score"].(map[string]interface{})
	S_Winner_Score = Quaconfig["s_winner_score"].(map[string]interface{})

	Budget_Score = Quaconfig["budget_score"].(map[string]interface{})
	Bidamount_Score = Quaconfig["bidamount_score"].(map[string]interface{})

	Projectname_Score = Quaconfig["projectname_score"].(map[string]interface{})
	Projectcode_Score = Quaconfig["projectcode_score"].(map[string]interface{})
}