package subject_util

import (
	log "github.com/donnie4w/go-logger/logger"
	qu "qfw/util"
)

var (
	SourceMgo, QyxyMgo, SpiMgo *MongodbSim
	TimeLayout                 = "2006-01-02 15:04:05"
	MysqlGlobalTool            *Mysql
	BuyerClassData             = map[string]string{}
	RegionCodeData             = map[string]string{}
	IsLocal                    bool
)

const (
	S_Coll_Name      = "bidding"
	O_Coll_Name      = "zktest_subject_data"
	G_Units_Baseinfo = "dws_f_ent_baseinfo"
	G_Units_Contact  = "dws_f_ent_contact"
	G_Units_Tags     = "dws_f_ent_tags"
	G_Units_Warning  = "dws_f_ent_warning"
)

func InitClass() {
	initMgo()
	initMysql()
	initVCode()
}

//初始化mgo
func initMgo() {
	if IsLocal {
		SourceMgo = &MongodbSim{
			MongodbAddr: "127.0.0.1:27017",
			DbName:      "mixdata",
			Size:        10,
			UserName:    "",
			Password:    "",
		}
		SourceMgo.InitPool()

		QyxyMgo = &MongodbSim{
			MongodbAddr: "127.0.0.1:27017",
			DbName:      "mixdata",
			Size:        10,
			UserName:    "",
			Password:    "",
		}
		QyxyMgo.InitPool()

		SpiMgo = &MongodbSim{
			MongodbAddr: "127.0.0.1:27017",
			DbName:      "mixdata",
			Size:        10,
			UserName:    "",
			Password:    "",
		}
		SpiMgo.InitPool()
	} else {
		SourceMgo = &MongodbSim{
			MongodbAddr: "172.17.145.163:27083,172.17.4.187:27082",
			DbName:      "qfw",
			Size:        10,
			UserName:    "zhengkun",
			Password:    "zk@123123",
		}
		SourceMgo.InitPool()

		QyxyMgo = &MongodbSim{
			MongodbAddr: "172.17.145.163:27083,172.17.4.187:27082",
			DbName:      "mixdata",
			Size:        10,
			UserName:    "zhengkun",
			Password:    "zk@123123",
		}
		QyxyMgo.InitPool()

		SpiMgo = &MongodbSim{
			MongodbAddr: "172.17.4.181:27001",
			DbName:      "mixdata",
			Size:        10,
			UserName:    "",
			Password:    "",
		}
		SpiMgo.InitPool()
	}
}

func initMysql() {
	username, password := "root", "=PDT49#80Z!RVv52_z"
	address := "192.168.3.217:4000"
	if !IsLocal {
		username = "zhengkun"
		password = "Zk#20220824"
		address = "172.17.4.242:4000"
	}
	MysqlGlobalTool = &Mysql{
		Address:  address,
		UserName: username,
		PassWord: password,
		DBName:   "global_common_data",
	}
	MysqlGlobalTool.Init()
}

//加载代码表
func initVCode() {
	data_types := MysqlGlobalTool.Find("code_buyerclass", nil, "", "", -1, -1)
	//先构建所有一级数据
	for _, v := range *data_types {
		name := qu.ObjToString(v["name"])
		code := qu.ObjToString(v["code"])
		BuyerClassData[name] = code
	}
	BuyerClassData["其它"] = "00"
	log.Debug("招标行业分类表~", len(BuyerClassData))

	data_regions := MysqlGlobalTool.Find("code_area", nil, "", "", -1, -1)
	//先构建所有一级数据
	for _, v := range *data_regions {
		area := qu.ObjToString(v["area"])
		city := qu.ObjToString(v["city"])
		district := qu.ObjToString(v["district"])
		code := qu.ObjToString(v["code"])
		key := area + "~" + city + "~" + district + "~"
		RegionCodeData[key] = code
	}
	log.Debug("地域信息数量~", len(RegionCodeData))
}

//插入数据
func InsertMysqlData(name string, data map[string]interface{}, mark string) int64 {
	inb := MysqlGlobalTool.Insert(name, data)
	if inb == -1 {
		log.Debug("插入数据异常...", name, "~", mark)
	}
	return inb
}