application/api/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/jybase/mongodb"
	"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"
	"go.mongodb.org/mongo-driver/bson"
	"log"
	"regexp"
	"strconv"
	"strings"
	"time"
)

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

type EmPloyService struct{}

// 收录情况查询
func (e *EmPloyService) InfoEmployinfo(in *types.InfoEmployinfoReq) []map[string]interface{} {
	data := []map[string]interface{}{}
	for _, v := range strings.Split(in.IdArr, ",") {
		vint := int64(0)
		table, findKey, employKey, source := EmployKeyFormat(in.EmployType)
		id := ""
		id = IdFormat(v, in.EmployType)
		if id == "" {
			log.Println(v, in.EmployType, "该信息查询不到数据")
			break
		}
		valueMap := map[string]interface{}{
			"id": v,
		}
		//有收录情况
		summaryMap := SummaryFormat(in.PositionId)
		//vint = mongodb.StringTOBsonId(id).Timestamp().Unix()
		vint = extractNumbers(id)
		valueMap["isEmploy"] = summaryMap[in.EmployType].Contains(uint32(vint))
		if len(strings.Split(in.IdArr, ",")) == 1 {
			//列表查询
			//是否忽略处理
			employData := MC.CrmMysql.FindOne(table, map[string]interface{}{
				employKey:     id,
				"position_id": in.PositionId,
				"source":      source,
			}, "", "")
			if employData != nil && len(*employData) > 0 {
				valueMap["isIgnore"] = common.If(common.Int64All((*employData)["is_ignore"]) == 1, true, false)
				//客户数量
				customCount := MC.CrmMysql.Count(entity.CUSTOM, map[string]interface{}{
					findKey:       id,
					"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": id,
						"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
			}
		}
		data = append(data, valueMap)
	}
	return data
}

// 收录操作
func (e *EmPloyService) EmployOperate(in *types.EmployOperateReq) bool {
	//收录汇总表
	table, _, employKey, source := EmployKeyFormat(in.EmployType)
	summaryMap := SummaryFormat(in.PositionId)
	return MC.CrmMysql.ExecTx("收录操作", func(tx *sql.Tx) bool {
		fool := false
		for _, v1 := range strings.Split(in.IdArr, ",") {
			//id转中文
			id := ""
			findMap := map[string]interface{}{}
			findMap["position_id"] = in.PositionId
			id = IdFormat(v1, in.EmployType)
			findMap[employKey] = id
			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, in.EmployType)
						data["employ_way"] = 1
						data["jybx_url"] = "/article/content/" + v1 + ".html"
					} else {
						data = CustomFind(id, in.EmployType)
					}
					data["position_id"] = in.PositionId
					data["ent_id"] = in.EntId
					data["source"] = source
					ok := MC.CrmMysql.InsertByTx(tx, table, data)
					if ok <= 0 {
						log.Println(v1, id, in.EmployType, "收录失败")
						break
					}
				}
			} else {
				//取消收录
				ok := MC.CrmMysql.UpdateByTx(tx, table, findMap, map[string]interface{}{
					"status": -1,
				})
				if !ok {
					log.Println(v1, id, in.EmployType, "取消收录失败")
				}
			}
			//收录汇总表处理
			vint := extractNumbers(id)
			if in.IsEmploy {
				summaryMap[in.EmployType].Add(uint32(vint))
			} else {
				summaryMap[in.EmployType].Remove(uint32(vint))
			}
			fool = true
		}
		if fool {
			return SummarySave(tx, in.PositionId, summaryMap)
		} else {
			return false
		}

	})
}

// 企业信息查询
func CustomFind(id string, employType int64) map[string]interface{} {
	data := map[string]interface{}{}
	//company_id 企业id name户名称 address 地址
	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
}

// 标讯信息查询
func InfoFind(id string, employType int64) 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 标讯详情页
	if employType == 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 {
			titleTmp := common.ObjToString(obj["title"])
			if len([]rune(titleTmp)) > 100 {
				titleTmp = string([]rune(titleTmp)[:100]) + "..."
			}
			titleTmp = public.ClearHtml.ReplaceAllString(titleTmp, "")
			data["title"] = titleTmp
			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
			//类型处理
			toptype := common.InterfaceToStr(obj["toptype"])
			switch toptype {
			case "采购意向", "预告", "招标":
				data["type"] = 2
			case "其他", "结果":
				data["type"] = 3
			default:
				data["type"] = 1
			}
			//data["type"] = in.SourceType
			if obj["projectinfo"] != nil {
				projectinfo := common.ObjToMap(obj["projectinfo"])
				if (*projectinfo)["attachments"] != nil {
					data["annex"] = 1
				}
			}
			data["publishtime"] = time.Unix(common.Int64All(obj["publishtime"]), 0).Format(date.Date_Full_Layout)
			data["show_time"] = time.Unix(common.Int64All(obj["publishtime"]), 0).Format(date.Date_Full_Layout)
			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
}
func getBuyerIdByName(buyeName string) (buyerId string) {
	r := elastic.Get(BuyerIndex, BuyerType, fmt.Sprintf(`{"query":{"bool":{"must":[{"term":{"buyer_name":"%s"}}]}},"size":1,"_source":["city","id"]}`, buyeName))
	if r == nil || len(*r) == 0 {
		return
	}
	buyerId, _ = (*r)[0]["_id"].(string)
	return
}
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
}
func SummaryFormat(positionId int64) map[int64]*roaring.Bitmap {
	byte1 := []byte{}
	byte2 := []byte{}
	byte3 := []byte{}
	byte4 := []byte{}
	MC.CrmMysql.DB.QueryRow("select   search_tencent, search_buyer, search_ent, search_nzj from employ_summary where  position_id=?", positionId).Scan(&byte1, &byte3, &byte2, &byte4)
	rb1 := roaring.NewBitmap()
	rb2 := roaring.NewBitmap()
	rb3 := roaring.NewBitmap()
	rb4 := roaring.NewBitmap()
	data := map[int64]*roaring.Bitmap{}
	if byte1 != nil && len(byte1) > 0 {
		rb1.UnmarshalBinary(byte1)
	}
	if byte2 != nil && len(byte2) > 0 {
		rb2.UnmarshalBinary(byte2)
	}
	if byte3 != nil && len(byte3) > 0 {
		rb3.UnmarshalBinary(byte3)
	}
	if byte4 != nil && len(byte4) > 0 {
		rb4.UnmarshalBinary(byte4)
	}
	data[1] = rb1
	data[2] = rb2
	data[3] = rb3
	data[4] = rb4
	return data
}
func SummarySave(tx *sql.Tx, positionId int64, data map[int64]*roaring.Bitmap) bool {
	employSummaryData := MC.CrmMysql.FindOne(entity.EMPLOY_SUMMARY, map[string]interface{}{
		"position_id": positionId,
	}, "", "")
	if employSummaryData != nil && len(*employSummaryData) > 0 {
		//修改
		updateData := map[string]interface{}{}
		updateData["search_tencent"], _ = data[1].ToBytes()
		updateData["search_buyer"], _ = data[3].ToBytes()
		updateData["search_ent"], _ = data[2].ToBytes()
		updateData["search_nzj"], _ = data[4].ToBytes()
		ok := MC.CrmMysql.UpdateByTx(tx, entity.EMPLOY_SUMMARY, map[string]interface{}{
			"position_id": positionId,
		}, updateData)
		return ok
	} else {
		//新增
		insertData := map[string]interface{}{}
		insertData["position_id"] = positionId
		insertData["search_tencent"], _ = data[1].ToBytes()
		insertData["search_buyer"], _ = data[3].ToBytes()
		insertData["search_ent"], _ = data[2].ToBytes()
		insertData["search_nzj"], _ = data[4].ToBytes()
		ok := MC.CrmMysql.InsertByTx(tx, entity.EMPLOY_SUMMARY, insertData)
		return ok > 0
	}
}

// 根据收录类型 字段处理 table,findKey,employKey,source
func EmployKeyFormat(employType int64) (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
	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

}

// 忽略操作
func (e *EmPloyService) IgnoreOperate(in *types.IgnoreOperateReq) {
	//
	table := ""
	switch in.EmployType {
	case 1, 4:
		table = entity.EMPLOY_INFO
	case 2, 3:
		table = entity.EMPLOY_CUSTOM
	}
	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所对应数据不存在")
			break
		}
		if in.IsIgnore {
			MC.CrmMysql.Update(table, findMap, map[string]interface{}{
				"is_ignore": 1,
			})
		} else {
			MC.CrmMysql.Update(table, findMap, map[string]interface{}{
				"is_ignore": 0,
			})
		}
	}
}

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

// 分发操作
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,
				Summary:   SummaryFormat(person.PositionId),
			}
		}
		//分发数据处理
		infoMap := map[string]map[string]interface{}{}
		for positionId, personSmmary := range personMap {
			fool := false
			for _, v := range strings.Split(in.EmployIdArr, ",") {
				VMap := map[string]interface{}{}
				if infoMap[v] == nil {
					//新增记录
					infoData := MC.CrmMysql.FindOne(entity.EMPLOY_INFO, map[string]interface{}{
						"id": v,
					}, "", "")
					if infoData != nil && len(*infoData) > 0 {
						(*infoData)["dis_id"] = v
						(*infoData)["is_handle"] = 0
						(*infoData)["is_ignore"] = 0
						(*infoData)["is_dis"] = 0
						(*infoData)["is_create_clue"] = 0
						(*infoData)["is_create_chance"] = 0
						(*infoData)["is_create_custom"] = 0
						(*infoData)["create_time"] = time.Now().Format(date.Date_Full_Layout)
						delete((*infoData), "id")
					} else {
						log.Println(v, "查询不到改收录信息")
						break
					}
					//更改已分发状态
					if common.IntAll((*infoData)["is_dis"]) != 1 {
						MC.CrmMysql.UpdateByTx(tx, entity.EMPLOY_INFO, map[string]interface{}{
							"id": v,
						}, map[string]interface{}{
							"is_dis": 1,
						})
					}
					infoMap[v] = *infoData
					VMap = (*infoData)
				} else {
					VMap = infoMap[v]
				}
				//查询时候收录过
				findMap := map[string]interface{}{
					"position_id": positionId,
					"source_id":   VMap["source_id"],
				}
				if MC.CrmMysql.Count(entity.EMPLOY_INFO, findMap) == 0 {
					//新增
					//
					VMap["ent_user_id"] = personSmmary.EntUserId
					VMap["position_id"] = positionId
					ok := MC.CrmMysql.InsertByTx(tx, entity.EMPLOY_INFO, VMap)
					if ok > 0 {
						employType := common.If(common.Int64All(VMap["source"]) == 1, 1, 4)
						personSmmary.Summary[common.Int64All(employType)].Add(uint32(mongodb.StringTOBsonId(common.InterfaceToStr(VMap["source_id"])).Timestamp().Unix()))
						fool = true
					}
				}
				if fool {
					if !SummarySave(tx, positionId, personSmmary.Summary) {
						return false
					}
				}
			}
		}
		return true
	})
}

// 提取字符串中的数字
func extractNumbers(str string) int64 {
	re := regexp.MustCompile(`\d+`)
	matches := re.FindAllString(str, -1)
	numbers := make([]int, len(matches))
	for i, match := range matches {
		number, err := strconv.Atoi(match)
		if err == nil {
			numbers[i] = number
		}
	}
	result := 0
	for _, number := range numbers {
		result = result*10 + number
	}
	return int64(result)
}