jy-data-extract/src/main.go

package main

import (
	_ "jy/admin"
	_ "jy/admin/audit"
	_ "jy/admin/distribution"
	_ "jy/admin/task"
	"jy/extract"
	_ "jy/front"
	. "jy/router"
	"jy/util"
	"net/http"
	_ "net/http/pprof"
	qu "qfw/util"

	"qfw/util/elastic"
	"qfw/util/redis"

	log "github.com/donnie4w/go-logger/logger"
)

func init() {
	log.SetConsole(false)
	log.SetLevel(log.DEBUG)
	log.SetRollingDaily("./", "out.log")
	qu.ReadConfig(&util.Config)
	qu.ReadConfig("./res/brandrule.json", &util.BrandRules)
	qu.ReadConfig("./res/goods.json", &util.GoodsConfig)
	qu.ReadConfig("./res/brand.json", &util.BrandConfig)
	//初始化品牌和商品
	util.InitBrand()
	util.InitGoods()
	//初始化util
	util.UtilInit()
	//初始化redis
	redis.InitRedisBySize(qu.ObjToString(util.Config["redis"]), 50, 30, 240)
	//初始化elastic连接
	//"winner=172.17.145.179:2710,buyer=172.17.145.179:2711"
	elastic.InitElasticSize(qu.ObjToString(util.Config["elasticsearch"]), qu.IntAllDef(3, 30))
	//if eClient, err := elastic.NewClient(elastic.SetURL(qu.ObjToString(util.Config["elasticsearch"]))); err != nil {
	//	log.Debug(util.Config["elasticsearch"])
	//	log.Fatal("ElasticClient err:", err)
	//} else {
	//	util.ElasticClient = eClient

	util.QYK_RedisName = qu.ObjToString(util.Config["redis_qyk"])
	util.WinnerDB = qu.IntAll(util.Config["redis_winner_db"])
	util.BuyerDB = qu.IntAll(util.Config["redis_buyer_db"])
	util.AgencyDB = qu.IntAll(util.Config["redis_agency_db"])
	//winner
	util.ElasticClientIndex = qu.ObjToString(util.Config["elasticsearch_index"])
	util.ElasticClientType = qu.ObjToString(util.Config["elasticsearch_type"])
	util.ElasticClientDB = qu.ObjToString(util.Config["elasticsearch_db"])

	//buyer
	util.ElasticClientBuyerIndex = qu.ObjToString(util.Config["elasticsearch_buyer_index"])
	util.ElasticClientBuyerType = qu.ObjToString(util.Config["elasticsearch_buyer_type"])
	util.ElasticClientBuyerDB = qu.ObjToString(util.Config["elasticsearch_buyer_db"])

	//buyer
	util.ElasticClientAgencyIndex = qu.ObjToString(util.Config["elasticsearch_agency_index"])
	util.ElasticClientAgencyType = qu.ObjToString(util.Config["elasticsearch_agency_type"])
	util.ElasticClientAgencyDB = qu.ObjToString(util.Config["elasticsearch_agency_db"])

	//}
}

func main() {
	extract.ExtractUdp() //udp通知抽取
	extract.ClearUdp()   //udp通知清理
	go extract.Export()
	go Router.Run(":" + qu.ObjToString(util.Config["port"]))
	go log.Debug("启动..", qu.ObjToString(util.Config["port"]))
	go func() {
		http.ListenAndServe("localhost:10000", nil)
	}()
	lock := make(chan bool)
	<-lock
}