application/api/internal/service/employService.go

package service

import (
	"app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/date"
	elastic "app.yhyue.com/moapp/jybase/es"
	"app.yhyue.com/moapp/jypkg/ent/util"
	"app.yhyue.com/moapp/jypkg/public"
	MC "bp.jydev.jianyu360.cn/CRM/application/api/common"
	"bp.jydev.jianyu360.cn/CRM/application/api/internal/types"
	"bp.jydev.jianyu360.cn/CRM/application/entity"
	"database/sql"
	"fmt"
	"github.com/RoaringBitmap/roaring"
	"github.com/gogf/gf/v2/util/gconv"
	"go.mongodb.org/mongo-driver/bson"
	"log"
	"strings"
	"time"
)

const (
	BuyerIndex = "buyer" // 采购单位index
	BuyerType  = "buyer"
)

type EmPloyService struct{}
type PersonSmmary struct {
	EntUserId int64
	EntDeptId int64
	Summary   map[int64]*roaring.Bitmap
}

// InfoEmployinfo 收录情况查询
func (e *EmPloyService) InfoEmployinfo(in *types.InfoEmployinfoReq) []map[string]interface{} {
	var data []map[string]interface{}
	depIDArr := ""
	if in.EntNicheDis > 0 {
		//查询所有部门标识
		deptArr := MC.MainMysql.SelectBySql("select GROUP_CONCAT(DISTINCT b.id)  as depIDArr from entniche_department_parent a"+
			"     INNER JOIN entniche_department b "+
			"on (b.ent_id=? and (b.id=? or (a.pid=? and a.id=b.id)))", in.EntId, in.EntDeptId, in.EntDeptId)
		if len(*deptArr) == 0 {
			return data
		}
		depIDArr = common.InterfaceToStr((*deptArr)[0]["depIDArr"])
		if depIDArr == "" {
			return data
		}
	}
	for _, v := range strings.Split(in.IdArr, ",") {
		id := ""
		id = idFormat(v, in.EmployType)
		table, findKey, employKey, source := employKeyFormat(in.EmployType, id)
		if id == "" {
			log.Println(v, in.EmployType, "该信息查询不到数据")
			break
		}
		if in.EntNicheDis > 0 {
			data = append(data, employInfoEnt(v, table, id, employKey, findKey, depIDArr, source, in))
		} else {
			data = append(data, employInfoPerson(v, table, id, employKey, findKey, source, in))
		}
	}
	return data
}

// EmployOperate 收录操作
func (e *EmPloyService) EmployOperate(in *types.EmployOperateReq) (bool, string) {
	log.Println("收录入参", in)
	//收录汇总表
	//summaryMap := SummaryFormat(in.PositionId)
	msg := ""
	return MC.CrmMysql.ExecTx("收录操作", func(tx *sql.Tx) bool {
		fool := true
		for _, v1 := range strings.Split(in.IdArr, ",") {

			//id转中文
			id := ""
			findMap := map[string]interface{}{}
			findMap["position_id"] = in.PositionId
			id = idFormat(v1, in.EmployType)
			table, _, employKey, source := employKeyFormat(in.EmployType, id)
			findMap[employKey] = id
			findMap["source"] = source
			if id == "" {
				log.Println(v1, in.EmployType, "该信息查询不到数据")
				break
			}
			if in.IsEmploy {
				if MC.CrmMysql.Count(table, findMap) > 0 {
					MC.CrmMysql.UpdateByTx(tx, table, findMap, map[string]interface{}{
						"status": 1,
					})
				} else {
					//收录新增
					data := map[string]interface{}{}
					if in.EmployType == 1 || in.EmployType == 4 {
						data = infoFind(id, v1)
					} else if in.EmployType == 2 {
						data = customFind(id, in.EmployType)
						data["jybx_url"] = "/swordfish/page_big_pc/svip/ent_ser_portrait/" + v1
					} else if in.EmployType == 3 {
						data = customFind(id, in.EmployType)
						data["jybx_url"] = "/swordfish/page_big_pc/unit_portrayal/" + v1
					}
					data["source"] = source
					data["position_id"] = in.PositionId
					data["ent_id"] = in.EntId
					data["ent_dept_id"] = in.EntDeptId
					data["ent_user_id"] = in.EntUserId
					ok := MC.CrmMysql.InsertByTx(tx, table, data)
					if ok <= 0 {
						log.Println(v1, id, in.EmployType, "收录失败")
						break
					}
				}
				fool = true
			} else {
				//取消收录
				if MC.CrmMysql.Count(table, findMap) > 0 {
					ok := MC.CrmMysql.UpdateByTx(tx, table, findMap, map[string]interface{}{
						"status": -1,
					})
					if !ok {
						log.Println(v1, id, in.EmployType, "取消收录失败")
					}

				} else {
					fool = false
					if in.EmployType == 1 {
						msg = "该资讯已有员工收录，不可取消收录。"
					} else {
						msg = "该企业已有员工收录，不可取消收录。"
					}
				}
			}
			return fool
		}
		return fool
	}), msg
}

// customFind 企业信息查询
func customFind(id string, employType int64) map[string]interface{} {
	data := map[string]interface{}{}
	//company_id 企业id name户名称 address 地址 industry 行业
	if employType == 2 {
		//企业详情
		entinfo, _ := MC.MgoEnt.FindOneByField("qyxy_std", map[string]interface{}{"_id": id}, map[string]interface{}{
			"company_address": 1, //注册地
			"company_name":    1,
		})
		if entinfo != nil && len(*entinfo) > 0 {
			data["name"] = (*entinfo)["company_name"]
			data["address"] = (*entinfo)["company_address"]
			data["company_id"] = id
		}
	} else {
		//采购单位详情
		rs := elastic.Get(BuyerIndex, BuyerType, fmt.Sprintf(`{"query":{"bool":{"must":[{"terms":{"_id":["%s"]}}]}},"size":1,"_source":["buyer_name","city"]}`, id))
		if rs != nil && len(*rs) > 0 {
			data["name"] = (*rs)[0]["buyer_name"]
			data["address"] = (*rs)[0]["city"]
			data["company_id"] = id
		}
	}
	data["status"] = 1
	data["create_time"] = time.Now().Format(date.Date_Full_Layout)
	return data
}

// infoFind 标讯信息查询
func infoFind(id string, v1 string) map[string]interface{} {
	data := map[string]interface{}{}
	//source_id  信息id、项目id- title 标题-area 省 -city 市 -subtype 信息类型二级分类
	//buyerclass 采购单位行业 -budget 预算 -bidamount 中标金额 buyer采购单位 annex有无附件  publishtime发布时间  projectname 项目时间
	//ownerclass  业主类型(拟在建搜索) expurasing_time 预计采购时间   jybx_url 标讯详情页
	//标讯信息
	obj := map[string]interface{}{}
	brobj, ok := MC.Mgo.Find("bidding_rec", bson.M{"s_id": id}, `{"l_recoverydate":-1}`, public.MgoBiddingFields, false, 0, 1)
	if ok && (*brobj) != nil && len(*brobj) == 1 && (*brobj)[0] != nil {
		obj = (*brobj)[0]
	} else {
		aobj, ok := MC.MgoBidding.FindById("bidding", id, public.MgoBiddingFields)
		if ok && (aobj == nil || *aobj == nil || len(*aobj) == 0) {
			aobj, ok = MC.MgoBidding.FindById("bidding_back", id, public.MgoBiddingFields)
		}
		obj = *aobj
	}
	if ok && obj != nil && len(obj) > 0 {
		toptype := common.InterfaceToStr(obj["toptype"])
		subtype := common.InterfaceToStr(obj["subtype"])
		data["title"] = common.ObjToString(obj["title"])
		data["city"] = obj["city"]
		data["area"] = obj["area"]
		data["subtype"] = obj["subtype"]
		data["buyerclass"] = obj["buyerclass"]
		data["budget"] = obj["budget"]
		data["bidamount"] = obj["bidamount"]
		data["annex"] = 0
		data["buyer"] = obj["buyer"]
		data["winner"] = obj["s_winner"]
		data["jybx_url"] = "/article/content/" + v1 + ".html"
		//类型处理
		pushTime := time.Unix(common.Int64All(obj["publishtime"]), 0).Format(date.Date_Full_Layout)
		if subtype == "拟建" {
			data = niJianHandle(id, data, v1, pushTime)
			data["type"] = 1
		} else {
			if toptype == "采购意向" || toptype == "预告" || toptype == "招标" {
				data["type"] = 2
			} else {
				if subtype == "废标" || subtype == "流标" || subtype == "其它" {
					data["type"] = 2
				} else if subtype == "成交" || subtype == "中标" || subtype == "合同" || subtype == "验收" || subtype == "违规" {
					data["type"] = 3
				} else {
					data["type"] = 1
				}
			}
		}
		if obj["projectinfo"] != nil {
			projectinfo := common.ObjToMap(obj["projectinfo"])
			if (*projectinfo)["attachments"] != nil {
				data["annex"] = 1
			}
		}
		data["publishtime"] = pushTime
		data["show_time"] = pushTime
		data["projectname"] = obj["projectname"]
		procurementlist, _ := obj["procurementlist"].([]interface{})
		for _, vs := range procurementlist { //1.采购意向清单数据集打码处理
			vsMap, _ := vs.(map[string]interface{})
			if vsMap["expurasingtime"] != nil {
				data["expurasing_time"] = vsMap["expurasingtime"]
				break
			}
		}
	}
	data["status"] = 1
	data["employ_way"] = 1
	data["source_id"] = id
	data["create_time"] = time.Now().Format(date.Date_Full_Layout)
	return data
}

// GetBuyerIdByName 通过采购单位名称获取id
func getBuyerIdByName(buyerName string) (buyerId string) {
	r := elastic.Get(BuyerIndex, BuyerType, fmt.Sprintf(`{"query":{"bool":{"must":[{"term":{"buyer_name":"%s"}}]}},"size":1,"_source":["city","id"]}`, buyerName))
	if r == nil || len(*r) == 0 {
		return
	}
	buyerId, _ = (*r)[0]["_id"].(string)
	return
}

// IdFormat id解密
func idFormat(encryptionId string, employType int64) string {
	decryptId := ""
	switch employType {
	case 1, 2:
		decryptId = util.DecodeId(encryptionId)
	case 3:
		if len(encryptionId) > 0 && len([]rune(encryptionId)) == len(encryptionId) {
			//此数据是id
			//获取中文名字
			decryptId = util.DecodeId(encryptionId)
		} else {
			//次数据传的名字
			decryptId = getBuyerIdByName(encryptionId)
		}
	}
	return decryptId
}

// employKeyFormat 根据收录类型 字段处理 table,findKey,employKey,source
func employKeyFormat(employType int64, id string) (string, string, string, int64) {
	table := ""
	employKey := "company_id"
	findKey := "employ_custom_id"
	source := int64(0)
	switch employType {
	case 1:
		table = entity.EMPLOY_INFO
		employKey = "source_id"
		findKey = "employ_info_id"
		source = 1
		obj := map[string]interface{}{}
		brobj, ok := MC.Mgo.Find("bidding_rec", bson.M{"s_id": id}, `{"l_recoverydate":-1}`, public.MgoBiddingFields, false, 0, 1)
		if ok && (*brobj) != nil && len(*brobj) == 1 && (*brobj)[0] != nil {
			obj = (*brobj)[0]
		} else {
			aobj, ok := MC.MgoBidding.FindById("bidding", id, public.MgoBiddingFields)
			if ok && (aobj == nil || *aobj == nil || len(*aobj) == 0) {
				aobj, ok = MC.MgoBidding.FindById("bidding_back", id, public.MgoBiddingFields)
			}
			obj = *aobj
		}
		if ok && obj != nil && len(obj) > 0 {
			subtype := common.InterfaceToStr(obj["subtype"])
			if subtype == "拟建" {
				source = 2
			}
		}
	case 2:
		table = entity.EMPLOY_CUSTOM
		source = 1
	case 3:
		table = entity.EMPLOY_CUSTOM
		source = 2
	case 4:
		table = entity.EMPLOY_INFO
		source = 2
		employKey = "source_id"
		findKey = "employ_info_id"
	}
	return table, findKey, employKey, source

}

// IgnoreOperate 忽略操作
func (e *EmPloyService) IgnoreOperate(in *types.IgnoreOperateReq) (bool, string) {
	//
	table := ""
	switch in.EmployType {
	case 1, 4:
		table = entity.EMPLOY_INFO
	case 2, 3:
		table = entity.EMPLOY_CUSTOM
	}
	msg := ""
	fool := true
	for _, v := range strings.Split(in.IdArr, ",") {
		id := idFormat(v, in.EmployType)
		if id == "" {
			log.Println(v, id, in.EmployType, "忽略处理，id解析失败")
			break
		}
		findMap := map[string]interface{}{
			"position_id": in.PositionId,
		}
		if in.EmployType == 3 || in.EmployType == 2 {
			findMap["company_id"] = id
		} else {
			findMap["source_id"] = id
		}
		if MC.CrmMysql.Count(table, findMap) == 0 {
			log.Println(v, id, in.EmployType, "忽略处理，id所对应数据不存在")
			fool = false
			if in.EmployType == 1 {
				if in.IsIgnore {
					msg = "该资讯已有员工收录，不可忽略。"
				} else {
					msg = "该资讯已有员工收录，不可取消忽略。"
				}
			} else {
				if in.IsIgnore {
					msg = "该企业已有员工收录，不可忽略。"
				} else {
					msg = "该企业已有员工收录，不可取消忽略。"
				}
			}
			break
		}
		if in.IsIgnore {
			MC.CrmMysql.Update(table, findMap, map[string]interface{}{
				"is_ignore": 1,
				"is_handle": 1,
			})
		} else {
			MC.CrmMysql.Update(table, findMap, map[string]interface{}{
				"is_ignore": 0,
				"is_handle": 1,
			})
		}
	}
	return fool, msg
}

// DistributePerson 分发操作
func (e *EmPloyService) DistributePerson(in *types.EmployDistributeReq) bool {
	return MC.CrmMysql.ExecTx("收录操作", func(tx *sql.Tx) bool {
		personMap := map[int64]PersonSmmary{}
		//人员信息处理
		for _, person := range in.Person {
			personMap[person.PositionId] = PersonSmmary{
				EntUserId: person.EntUserId,
				EntDeptId: person.EntDeptId,
			}
		}
		//分发数据处理
		infoMap := map[string]map[string]interface{}{}
		onlyMap := map[string]interface{}{}
		//标讯数据处理
		for _, v := range strings.Split(in.EmployIdArr, ",") {
			if onlyMap[v] == nil {
				//查询信息
				infoDataList := MC.CrmMysql.SelectBySqlByTx(tx, "SELECT * FROM employ_info WHERE  id=?", v)
				if infoDataList != nil && len(*infoDataList) > 0 {
					source_id := common.InterfaceToStr((*infoDataList)[0]["source_id"])
					if onlyMap[source_id] == nil {
						(*infoDataList)[0]["dis_id"] = v
						(*infoDataList)[0]["is_handle"] = 0
						(*infoDataList)[0]["is_ignore"] = 0
						(*infoDataList)[0]["is_dis"] = 0
						(*infoDataList)[0]["is_create_clue"] = 0
						(*infoDataList)[0]["is_create_chance"] = 0
						(*infoDataList)[0]["is_create_custom"] = 0
						(*infoDataList)[0]["create_time"] = time.Now().Format(date.Date_Full_Layout)
						delete((*infoDataList)[0], "id")
						if common.IntAll((*infoDataList)[0]["is_dis"]) != 1 {
							MC.CrmMysql.UpdateByTx(tx, entity.EMPLOY_INFO, map[string]interface{}{
								"id": v,
							}, map[string]interface{}{
								"is_dis":    1,
								"is_handle": 1,
							})
						}
						infoMap[source_id] = (*infoDataList)[0]
						onlyMap[source_id] = v
					} else {
						log.Println(v, "查询不到改收录信息")
						continue
					}
				}
			}
		}
		//普通人员分发
		for positionId, personSmmary := range personMap {
			//查询时候收录过
			for _, v := range infoMap {
				//查询时候收录过
				findMap := map[string]interface{}{
					"position_id": positionId,
					"source_id":   v["source_id"],
					"source":      v["source"],
				}
				if MC.CrmMysql.Count(entity.EMPLOY_INFO, findMap) == 0 {
					//新增
					v["ent_user_id"] = personSmmary.EntUserId
					v["ent_dept_id"] = personSmmary.EntDeptId
					v["position_id"] = positionId
					MC.CrmMysql.InsertByTx(tx, entity.EMPLOY_INFO, v)
				}
			}
		}
		if in.EntNicheDis > 0 {
			//查询分配员有没有该数据
			for _, v := range infoMap {
				//查询时候收录过
				findMap := map[string]interface{}{
					"position_id": in.PositionId,
					"source_id":   v["source_id"],
					"source":      v["source"],
				}
				employPositionId := v["position_id"]
				if employPositionId != in.PositionId {
					//分发消息不属于自己
					if MC.CrmMysql.Count(entity.EMPLOY_INFO, findMap) == 0 {
						v["ent_user_id"] = in.EntUserId
						v["position_id"] = in.PositionId
						v["is_handle"] = 1
						v["ent_dept_id"] = in.EntDeptId
						MC.CrmMysql.InsertByTx(tx, entity.EMPLOY_INFO, v)
					} else {
						MC.CrmMysql.UpdateByTx(tx, entity.EMPLOY_INFO, findMap, map[string]interface{}{
							"is_handle": 1,
						})
					}
				}
			}
		}
		return true
	})
}

func niJianHandle(infoId string, data map[string]interface{}, v1, pushTime string) map[string]interface{} {
	project, _ := MC.MgoProject.FindOne("projectset_proposed", map[string]interface{}{
		"ids": infoId,
	})
	if project != nil && len(*project) > 0 {
		city := common.InterfaceToStr((*project)["city"])
		area := common.InterfaceToStr((*project)["area"])
		category_code := common.InterfaceToStr((*project)["category_code"])
		lasttime := common.Int64All((*project)["lasttime"])
		ownerclass_code := common.InterfaceToStr((*project)["ownerclass_code"])
		total_investment := common.InterfaceToStr((*project)["total_investment"])
		project_stage_code := common.InterfaceToStr((*project)["project_stage_code"])
		infoList := gconv.SliceMap((*project)["list"])
		title := ""
		projectname := ""
		if len(infoList) > 0 {
			title = common.InterfaceToStr(infoList[0]["title"])
			projectname = common.InterfaceToStr(infoList[0]["projectname"])
		}
		category := ""
		ownerclass := ""
		project_stage := ""
		if category_code != "" {
			categoryData := MC.NiJian.FindOne("d_nzj_category_code", map[string]interface{}{
				"code": category_code,
			}, "name", "")
			category = common.InterfaceToStr((*categoryData)["name"])
		}
		if ownerclass_code != "" {
			ownerclassData := MC.NiJian.FindOne("d_nzj_ownerclass_code", map[string]interface{}{
				"code": ownerclass_code,
			}, "name", "")
			ownerclass = common.InterfaceToStr((*ownerclassData)["name"])
		}
		if project_stage_code != "" {
			projectStageData := MC.NiJian.FindOne("d_nzj_project_stage_code", map[string]interface{}{
				"code": project_stage_code,
			}, "name", "")
			project_stage = common.InterfaceToStr((*projectStageData)["name"])
		}
		data["title"] = title
		data["city"] = city
		data["area"] = area
		data["projectname"] = projectname
		data["ownerclass"] = ownerclass
		data["project_stage"] = project_stage
		data["project_category"] = category
		data["project_investment"] = total_investment
		lastTimeStr := common.If(lasttime == 0, pushTime, time.Unix(lasttime, 0).Format(date.Date_Full_Layout))
		data["project_evolve"] = lastTimeStr
		data["show_time"] = lastTimeStr
		data["publishtime"] = lastTimeStr
		pid := common.InterfaceToStr((*project)["_id"])
		data["jybx_url"] = "/succbi/nzj/app/nzj.app/nzj_detail_0.spg?pid=" + pid
	} else {
		data["jybx_url"] = "/article/content/" + v1 + ".html"
	}
	data["source"] = 2
	return data
}

// 个人收录情况查询
func employInfoPerson(v, table, id, employKey, findKey string, source int64, in *types.InfoEmployinfoReq) map[string]interface{} {
	valueMap := map[string]interface{}{
		"id": v,
	}
	employData := MC.CrmMysql.FindOne(table, map[string]interface{}{
		employKey:     id,
		"position_id": in.PositionId,
		"source":      source,
	}, "is_ignore,status,id", "")
	employId := int64(0)
	if employData != nil && len(*employData) > 0 {
		valueMap["isIgnore"] = common.If(common.Int64All((*employData)["is_ignore"]) == 1, true, false)
		valueMap["isEmploy"] = common.If(common.Int64All((*employData)["status"]) == 1, true, false)
		employId = common.Int64All((*employData)["id"])
		valueMap["employId"] = employId
	} else {
		valueMap["isIgnore"] = false
		valueMap["isEmploy"] = false
	}
	if len(strings.Split(in.IdArr, ",")) == 1 && employId > 0 {
		//列表查询
		//是否忽略处理
		//客户数量
		customCount := MC.CrmMysql.Count(entity.CUSTOM, map[string]interface{}{
			findKey:       employId,
			"position_id": in.PositionId,
		})
		valueMap["customCount"] = customCount
		if in.EmployType == 1 || in.EmployType == 4 {
			//销售线索数量
			clueCount := MC.CrmMysql.Count(entity.SALE_CLUE, map[string]interface{}{
				"employ_info_id": employId,
				"position_id":    in.PositionId,
			})
			valueMap["clueCount"] = clueCount
			//销售机会数量
			chanceCount := MC.CrmMysql.Count(entity.SALE_CHANCE, map[string]interface{}{
				"employ_info_id": id,
				"position_id":    in.PositionId,
			})
			valueMap["chanceCount"] = chanceCount
		}
	} else {
		valueMap["chanceCount"] = 0
		valueMap["clueCount"] = 0
		valueMap["isIgnore"] = false
		valueMap["customCount"] = 0
	}
	return valueMap
}

// 企业收录情况查询
func employInfoEnt(v, table, id, employKey, findKey, depIDArr string, source int64, in *types.InfoEmployinfoReq) map[string]interface{} {
	valueMap := map[string]interface{}{
		"id":          v,
		"isIgnore":    false,
		"isEmploy":    false,
		"customCount": 0,
		"clueCount":   0,
		"chanceCount": 0,
	}
	employData := MC.CrmMysql.SelectBySql(fmt.Sprintf("select  GROUP_CONCAT(id) as  employIdArr,MAX(is_ignore) as is_ignore,MAX(status) as status from  %s   where %s=? and  source=? and    ent_dept_id in  (%s) and  status=1", table, employKey, depIDArr), id, source)
	employArrStr := ""
	if employData != nil && len(*employData) > 0 {
		valueMap["isIgnore"] = common.If(common.Int64All((*employData)[0]["is_ignore"]) == 1, true, false)
		valueMap["isEmploy"] = common.If(common.Int64All((*employData)[0]["status"]) == 1, true, false)
		employArrStr = common.InterfaceToStr((*employData)[0]["employIdArr"])
		valueMap["employId"] = strings.Split(employArrStr, ",")[0]
	} else {
		valueMap["isIgnore"] = false
		valueMap["isIgnore"] = false
	}
	if len(strings.Split(in.IdArr, ",")) == 1 && employArrStr != "" {
		//列表查询
		//是否忽略处理
		//客户数量
		customSql := fmt.Sprintf("select  COUNT(1) from  %s where  %s  in (%s)", entity.CUSTOM, findKey, employArrStr)
		customCount := MC.CrmMysql.CountBySql(customSql)
		valueMap["customCount"] = customCount
		if in.EmployType == 1 || in.EmployType == 4 {
			//销售线索数量
			clueSql := fmt.Sprintf("select  COUNT(1) from  %s where employ_info_id  in (%s)", entity.SALE_CLUE, employArrStr)
			clueCount := MC.CrmMysql.CountBySql(clueSql)
			valueMap["clueCount"] = clueCount
			//销售机会数量
			chanceSql := fmt.Sprintf("select  COUNT(1) from  %s where employ_info_id  in (%s)", entity.SALE_CHANCE, employArrStr)
			chanceCount := MC.CrmMysql.CountBySql(chanceSql)
			valueMap["chanceCount"] = chanceCount
		}
	} else {
		valueMap["chanceCount"] = 0
		valueMap["clueCount"] = 0
		valueMap["customCount"] = 0
	}
	return valueMap
}