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 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) { //收录汇总表 //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 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{}{} data["source"] = source 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["position_id"] = in.PositionId data["ent_id"] = in.EntId data["ent_dept_id"] = in.EntDeptId 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" //类型处理 /*if subtype == "拟建" { data["type"] = 1 data["type"] = 2 } else */ if subtype == "拟建" { data = niJianHandle(id, data, v1) 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"] = 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 } // 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, } } //分发数据处理 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["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 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 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, nil, 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 }