biService/entity/entity.go

package entity

import (
	"log"

	"encoding/json"

	util "app.yhyue.com/moapp/jybase/common"
	elastic "app.yhyue.com/moapp/jybase/esv1"
	"app.yhyue.com/moapp/jybase/mongodb"
	"app.yhyue.com/moapp/jybase/mysql"
	"github.com/nsqio/go-nsq"
	"github.com/zeromicro/go-zero/core/logx"
)

var (
	JyMysql    *mysql.Mysql
	JyDocMysql *mysql.Mysql
	JyBiMysql  *mysql.Mysql
	JyTidb     *mysql.Mysql
	JyBiTidb   *mysql.Mysql
	Mgo        *mongodb.MongodbSim
	Es         *elastic.Elastic
	AreaCode   = map[string]string{}
)

type Handler struct {
}

func InitMysql(n, x, y, z, s *mysql.Mysql) {
	JyMysql = &mysql.Mysql{
		Address:      n.Address,
		UserName:     n.UserName,
		PassWord:     n.PassWord,
		DBName:       n.DBName,
		MaxOpenConns: n.MaxOpenConns,
		MaxIdleConns: n.MaxIdleConns,
	}
	JyMysql.Init()
	JyDocMysql = &mysql.Mysql{
		Address:      x.Address,
		UserName:     x.UserName,
		PassWord:     x.PassWord,
		DBName:       x.DBName,
		MaxOpenConns: x.MaxOpenConns,
		MaxIdleConns: x.MaxIdleConns,
	}
	JyDocMysql.Init()
	JyBiMysql = &mysql.Mysql{
		Address:      y.Address,
		UserName:     y.UserName,
		PassWord:     y.PassWord,
		DBName:       y.DBName,
		MaxOpenConns: y.MaxOpenConns,
		MaxIdleConns: y.MaxIdleConns,
	}
	JyBiMysql.Init()
	JyTidb = &mysql.Mysql{
		Address:      z.Address,
		UserName:     z.UserName,
		PassWord:     z.PassWord,
		DBName:       z.DBName,
		MaxOpenConns: z.MaxOpenConns,
		MaxIdleConns: z.MaxIdleConns,
	}
	JyTidb.Init()
	JyBiTidb = &mysql.Mysql{
		Address:      s.Address,
		UserName:     s.UserName,
		PassWord:     s.PassWord,
		DBName:       s.DBName,
		MaxOpenConns: s.MaxOpenConns,
		MaxIdleConns: s.MaxIdleConns,
	}
	JyBiTidb.Init()
	logx.Info("初始化mysql")
}

func InitMongo(addr, db string, size int) {
	Mgo = mongodb.NewMgo(addr, db, size)
}

func InitEs(address string, size int) {
	Es = &elastic.Elastic{
		S_esurl: address,
		I_size:  size,
	}
	Es.InitElasticSize()
}

func InitArea() {
	info := JyBiMysql.Find("d_area_code_back", nil, "", "", -1, -1)
	if info != nil && len(*info) > 0 {
		for _, m := range *info {
			var key string
			for i, v := range []string{"area", "city", "district"} {
				if i == 0 && util.ObjToString(m[v]) != "" {
					key = util.ObjToString(m[v])
				} else if util.ObjToString(m[v]) != "" {
					key += "," + util.ObjToString(m[v])
				}
			}
			AreaCode[key] = util.ObjToString(m["code"])
		}
	}
	log.Println("AreaCodeLen ", len(AreaCode))
}

func (h *Handler) HandleMessage(m *nsq.Message) error {
	defer util.Catch()
	if len(m.Body) == 0 {
		log.Println("Nsq接收到空")
		return nil
	}
	var msg *map[string]interface{}
	err := json.Unmarshal(m.Body, &msg)
	if err != nil {
		log.Println(err)
		return err
	}
	entPostionId := util.Int64All((*msg)["entPostionId"])
	userPositionId := util.Int64All((*msg)["userPositionId"])
	entId := util.Int64All((*msg)["entId"])
	entUserName := util.ObjToString((*msg)["entUserName"])
	entName := ""
	log.Println(userPositionId, entPostionId, entId, entUserName, entName)
	if userPositionId == 0 || entPostionId == 0 {
		log.Println("职位id为空")
		return nil
	}
	data := JyMysql.FindOne("entniche_info", map[string]interface{}{"id": entId}, "", "")
	if data != nil && len(*data) > 0 {
		entName = util.ObjToString((*data)["name"])
	}
	ok := JyBiMysql.Update("dwd_f_report_data_baseinfo", map[string]interface{}{"position_id": userPositionId}, map[string]interface{}{"ent_id": entId, "position_type": 1, "user_name": entUserName, "company_name": entName})
	if ok {
		log.Println("个人销售数据同步企业数据成功！！！")
	} else {
		log.Println("个人销售数据同步企业数据失败-----")
	}
	return nil
}