package main

import (
	util "app.yhyue.com/data_processing/common_utils"
	"app.yhyue.com/data_processing/common_utils/elastic"
	"app.yhyue.com/data_processing/common_utils/mongodb"
	"app.yhyue.com/data_processing/common_utils/redis"
	"fmt"
	"net"
	"net/http"
	"net/rpc"
	"os"
	"regexp"
)

var (
	Sysconfig  map[string]interface{}
	FileType   []interface{}
	MongoTool  *mongodb.MongodbSim
	Dbname     string
	MgoColl    string
	MgoColl1   string
	Es         *elastic.Elastic
	Index      string
	RpcServer  string
	ClientAddr string
)

var (
	_ent     = regexp.MustCompile(".{2,100}(集团|公司|学校|中心|家具城|门诊|\\\\[大中小\\\\]学|部|院|局|厂|店|所|队|社|室|厅|段|会|场|行)$")
	down_url = "https://jy-datafile.oss-cn-beijing.aliyuncs.com/"
)

func init() {
	util.ReadConfig(&Sysconfig)
	FileType = Sysconfig["file_type"].([]interface{})
	Dbname = Sysconfig["mongodbName"].(string)
	MgoColl = Sysconfig["mongodbColl"].(string)
	MgoColl1 = Sysconfig["mongodbColl1"].(string)
	MongoTool = &mongodb.MongodbSim{
		MongodbAddr: Sysconfig["mongodbServers"].(string),
		Size:        util.IntAllDef(Sysconfig["mongodbPoolSize"], 5),
		DbName:      Dbname,
		UserName:    Sysconfig["uname"].(string),
		Password:    Sysconfig["upwd"].(string),
	}
	MongoTool.InitPool()
	es := Sysconfig["es"].(map[string]interface{})
	Es = &elastic.Elastic{
		S_esurl:  util.ObjToString(es["addr"]),
		I_size:   util.IntAllDef(es["pool"], 10),
		Username: util.ObjToString(es["user"]),
		Password: util.ObjToString(es["password"]),
	}
	Index = util.ObjToString(es["index"])
	Es.InitElasticSize()
	ClientAddr = Sysconfig["serverCenter"].(string)
	RpcServer = Sysconfig["rpcServer"].(string)
	//初始化redis
	redis.InitRedisBySize(util.ObjToString(Sysconfig["redis"]), 50, 30, 240)
}

func initRpc() {
	rpc.Register(new(JyService)) // 注册rpc服务
	rpc.HandleHTTP()             // 采用http协议作为rpc载体
	lis, err := net.Listen("tcp", RpcServer)
	if err != nil {
		util.Debug("fatal error: ", err)
	}
	_, _ = fmt.Fprintf(os.Stdout, "%s\n", "start connection")
	_ = http.Serve(lis, nil)
}

func main() {
	initRpc()
}