networkManage/api/internal/service/CoopHistoryService.go

package service

import (
	"context"
	"encoding/json"
	"fmt"
	"strings"
	"time"

	"app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/encrypt"
	elastic "app.yhyue.com/moapp/jybase/es"
	T "bp.jydev.jianyu360.cn/CRM/networkManage/api/common"
	"bp.jydev.jianyu360.cn/CRM/networkManage/api/internal/types"
	"github.com/ClickHouse/clickhouse-go/v2/lib/driver"
	"github.com/zeromicro/go-zero/core/logx"
)

var (
	INDEX_1   = "transaction_info"
	sql_2_0   = `SELECT buyer, buyer_id, agency, agency_id, property_form FROM information.transaction_info WHERE project_id = ?`
	sql_2_1   = `SELECT project_id, project_name, zbtime FROM information.transaction_info WHERE buyer_id = ? AND has(winner, ?) ORDER BY zbtime DESC`
	sql_2_1_1 = `SELECT project_id, project_name, zbtime FROM information.transaction_info WHERE buyer_id = ? AND agency = ? ORDER BY zbtime DESC`
	sql_2_2   = `select DISTINCT  b.company_id,b.company_name,a.relate_id,a.relate_name,b.contact_person   from  connection_introduce a  INNER JOIN connection b on  a.position_id= ? and a.connection_id=b.id  and   a.relate_Id = ? and  a.itype =1  and  b.itype=4 and  b.status=1 `
	sql_2_3   = `select a.a_id as a_id, a.b_id as b_id, a.a_name as a_name, a.b_name as b_name, a.code as code, c.legal_person as c_person, d.legal_person d_person from crm.ent_map_code a left join ent_info c on (a.a_id = ? or a.b_id = ? ) and a.code in('0101', '0201') and a.a_id = c.id left join ent_info d on (a.a_id = ? or a.b_id = ? ) and a.code in('0101', '0201') and a.b_id = d.id where (a.a_id = ? or a.b_id = ? ) and a.code in('0101', '0201')`
	es_query  = `{"query": {"bool": {"must": [{"terms": {"buyer_id": ["%s"]}},{"exists": {"field": "winner_id"}}],"must_not": {"terms": {"property_form": ["%s"]}}}},"aggs": {"winner_count": {"terms": {"field": "winner","size": 1000,"order": {"_count": "desc"}},"aggs": {"amount_all": {"sum": {"field": "project_money"}},"ent_id": {"terms": {"field": "winner_id"}},"buyer_id": {"terms": {"field": "buyer_id"}}}}},"size": 1}`
	es_query1 = `{"query": {"bool": {"must": [{"terms": {"buyer_id": ["%s"]}}],"must_not": [{"term": {"agency_id": {"value": ""}}}]}},"aggs": {"agency_count": {"terms": {"field": "agency","size": 1000,"order": {"_count": "desc"}},"aggs": {"amount_all": {"sum": {"field": "project_money"}},"ent_id": {"terms": {"field": "agency_id"}},"buyer_id": {"terms": {"field": "buyer_id"}}}}},"size": 1}`
)

type ResultData struct {
	SourceType   string  `json:"SourceType"`
	EntName      string  `json:"EntName"`
	EntId        string  `json:"EntId"`
	EntPerson    string  `json:"EntPerson"`
	Middleman    string  `json:"Middleman"`
	ProjectNum   int     `json:"ProjectNum"`
	TotalAmount  float64 `json:"TotalAmount"`
	RecentTime   int64   `json:"RecentTime"`
	NearlyYears  bool    `json:"NearlyYears"`
	SupEnt       string  `json:"SupEnt"`       // 上级
	NextEnt      string  `json:"NextEnt"`      // 下级
	Relationship string  `json:"Relationship"` // 关系
	BuyerId      string  `json:"BuyerId"`
	IsIgnore     bool    `json:"isIgnore"`
}

type ResultDatas []*ResultData

func (n *ResultDatas) Len() int {
	return len(*n)
}

func (n *ResultDatas) Less(i, j int) bool {
	if (*n)[i].RecentTime == (*n)[j].RecentTime {
		return (*n)[i].EntName < (*n)[j].EntName
	}
	return (*n)[i].RecentTime > (*n)[j].RecentTime
}

func (n *ResultDatas) Swap(i, j int) {
	(*n)[i], (*n)[j] = (*n)[j], (*n)[i]
}

type ProjectTmp struct {
	Buyer        string   `ch:"buyer"`
	BuyerId      string   `ch:"buyer_id"`
	Agency       string   `ch:"agency"`
	AgencyId     string   `ch:"agency_id"`
	PropertyForm []string `ch:"property_form"`
}

type Cooperate struct {
	Near   bool
	ZbTime int64
}

func GetPrList(req *types.CoopHistoryReq) (result []*ResultData, size_1, size_2, size_3, size_4 int64) {
	pTmp := ProjectTmp{}
	err := T.ClickhouseConn.QueryRow(context.TODO(), sql_2_0, req.Pid).ScanStruct(&pTmp)
	if err != nil {
		return nil, 0, 0, 0, 0
	}
	propertyForm := T.NetworkCom.GetMyProbusfor(req.EntAccountId)
	// firstparty：甲方 supplier：供应商 adiffb：同甲异业 middleman：中间人 agency：招标代理机构 sup_sub: 上下级
	// 1、同甲异业数据/ 3、招标代理机构渠道
	if pTmp.BuyerId != "" {
		// 中间人可介绍业主
		var r3 []map[string]interface{}
		r3 = FindMiddleman([]string{fmt.Sprintf("'%s'", pTmp.BuyerId)}, req.PositionId, r3)
		if r3 != nil && len(r3) > 0 {
			size_4 = int64(len(r3))
			if req.ChannelType == "0" || req.ChannelType == "4" {
				for _, m := range r3 {
					tmp := ResultData{
						SourceType:   "middleman",
						EntName:      common.ObjToString(m["b_name"]),
						EntId:        common.ObjToString(m["b_id"]),
						EntPerson:    common.ObjToString(m["personName"]),
						Relationship: "业主的关系人",
						BuyerId:      pTmp.BuyerId,
					}
					result = append(result, &tmp)
				}
			}
		}
		// 关联单位
		var r4 []map[string]interface{}
		r4 = Findfirstparty([]string{fmt.Sprintf("'%s'", pTmp.BuyerId)}, r4)
		if r4 != nil && len(r4) > 0 {
			size_2 = int64(len(r4))
			if req.ChannelType == "0" || req.ChannelType == "2" {
				for _, m := range r4 {
					tmp := ResultData{
						SourceType:   "sup_sub",
						EntName:      common.ObjToString(m["b_name"]),
						EntId:        common.ObjToString(m["b_id"]),
						EntPerson:    common.ObjToString(m["personName"]),
						Relationship: common.ObjToString(m["relationship"]),
						BuyerId:      pTmp.BuyerId,
					}
					result = append(result, &tmp)
				}
			}
		}
		rs1, rs2 := GetData(propertyForm, pTmp.BuyerId)
		r1 := rs1[pTmp.BuyerId]
		r2 := rs2[pTmp.BuyerId]
		if r1 != nil && len(r1) > 0 {
			size_1 = int64(len(r1))
			if req.ChannelType == "0" || req.ChannelType == "1" {
				for _, m := range r1 {
					near, zbtime := LastTimeCoop(pTmp.BuyerId, common.ObjToString(m["name"]), "adiffb")
					tmp := ResultData{
						SourceType:  "adiffb",
						EntName:     common.ObjToString(m["name"]),
						ProjectNum:  common.IntAll(m["coop_size"]),
						TotalAmount: common.Float64All(m["coop_amount"]),
						RecentTime:  zbtime,
						NearlyYears: near,
						BuyerId:     pTmp.BuyerId,
					}
					result = append(result, &tmp)
				}
			}
		}
		if r2 != nil && len(r2) > 0 {
			size_3 = int64(len(r2))
			if req.ChannelType == "0" || req.ChannelType == "3" {
				for _, m := range r2 {
					near, zbtime := LastTimeCoop(pTmp.BuyerId, common.ObjToString(m["name"]), "agency")
					tmp := ResultData{
						SourceType:  "agency",
						EntName:     common.ObjToString(m["name"]),
						ProjectNum:  common.IntAll(m["coop_size"]),
						TotalAmount: common.Float64All(m["coop_amount"]),
						RecentTime:  zbtime,
						NearlyYears: near,
						BuyerId:     pTmp.BuyerId,
					}
					result = append(result, &tmp)
				}
			}
		}
	}
	return
}

func GetData(propertyForm []string, bid string) (result1, result2 map[string][]map[string]interface{}) {
	result1, result2 = map[string][]map[string]interface{}{}, map[string][]map[string]interface{}{}
	type AggStruct struct {
		Buckets []struct {
			Key        string `json:"key,omitempty"`
			Doc_count  int64  `json:"doc_count,omitempty"`
			Amount_all struct {
				Value float64 `json:"value,omitempty"`
			} `json:"amount_all"`
			EntId struct {
				Buckets []struct {
					Key       string `json:"key,omitempty"`
					Doc_count int64  `json:"doc_count,omitempty"`
				} `json:"buckets"`
			} `json:"ent_id"`
			BuyerId struct {
				Buckets []struct {
					Key       string `json:"key,omitempty"`
					Doc_count int64  `json:"doc_count,omitempty"`
				} `json:"buckets"`
			} `json:"buyer_id"`
		} `json:"buckets"`
	}
	m1 := make(map[string]interface{}) //采购单位-中标单位
	m2 := make(map[string]interface{}) //采购单位-代理机构
	noIds := []string{}
	if len(propertyForm) > 0 {
		//同甲异态 中标企业
		aggs, _, _ := elastic.GetAggs(INDEX_1, INDEX_1, fmt.Sprintf(es_query, strings.ReplaceAll(bid, ",", `","`), strings.Join(propertyForm, "\",\"")))
		logx.Info("es聚合查询结果:", fmt.Sprintf(es_query, strings.ReplaceAll(bid, ",", `","`), strings.Join(propertyForm, "\",\"")))
		var m1Buckets = AggStruct{}
		if aggs != nil && aggs["winner_count"] != nil {
			bs, err := aggs["winner_count"].MarshalJSON()
			if err != nil {
				logx.Info(err)
			} else {
				if len(bs) == 0 {
					logx.Info(err)
				} else {
					err := json.Unmarshal(bs, &m1Buckets)
					logx.Info(err)
					if len(m1Buckets.Buckets) > 0 {
						for _, v := range m1Buckets.Buckets {
							if v.Key != "" && len(v.EntId.Buckets) > 0 && len(v.BuyerId.Buckets) > 0 {
								mv := map[string]interface{}{"name": v.Key, "amount": v.Amount_all.Value, "count": v.Doc_count, "buyerId": v.BuyerId.Buckets[0].Key}
								if len(v.EntId.Buckets) == 1 && v.EntId.Buckets[0].Key != "" {
									mv["nameId"] = v.EntId.Buckets[0].Key
								} else {
									noIds = append(noIds, v.Key)
								}
								m1[v.Key] = mv
							}
						}
					}
				}
			}
		}
	}
	//代理机构
	aggs1, _, _ := elastic.GetAggs(INDEX_1, INDEX_1, fmt.Sprintf(es_query1, strings.ReplaceAll(bid, ",", `","`)))
	var m2Buckets = AggStruct{}
	if aggs1 != nil && aggs1["agency_count"] != nil {
		bs, err := aggs1["agency_count"].MarshalJSON()
		if err != nil {
			logx.Info(err)
		} else {
			if len(bs) == 0 {
				logx.Info(err)
			} else {
				err = json.Unmarshal(bs, &m2Buckets)
				logx.Info(err)
				if len(m2Buckets.Buckets) > 0 {
					for _, v := range m2Buckets.Buckets {
						if v.Key != "" && len(v.EntId.Buckets) > 0 && len(v.BuyerId.Buckets) > 0 {
							if v.EntId.Buckets[0].Key == "" {
								noIds = append(noIds, v.Key)
							}
							m2[v.Key] = map[string]interface{}{"name": v.Key, "amount": v.Amount_all.Value, "count": v.Doc_count, "nameId": v.EntId.Buckets[0].Key, "buyerId": v.BuyerId.Buckets[0].Key}
						}
					}
				}
			}
		}
	}
	idName := T.NetworkCom.GetEntIdByName(noIds)
	for k, v := range m1 {
		v1 := v.(map[string]interface{})
		tmp := make(map[string]interface{})
		tmp["name"] = k
		nameId, _ := v1["nameId"].(string)
		if nameId == "" {
			nameId = idName[k]
		}
		tmp["nameId"] = nameId
		tmp["coop_size"] = v1["count"]
		tmp["coop_amount"] = v1["amount"]
		buyerId, _ := v1["buyerId"].(string)
		result1[buyerId] = append(result1[buyerId], tmp)
	}
	for k, v := range m2 {
		v1 := v.(map[string]interface{})
		tmp := make(map[string]interface{})
		tmp["name"] = k
		nameId, _ := v1["nameId"].(string)
		if nameId == "" {
			nameId = idName[k]
		}
		tmp["nameId"] = nameId
		tmp["coop_size"] = v1["count"]
		tmp["coop_amount"] = v1["amount"]
		buyerId, _ := v1["buyerId"].(string)
		result2[buyerId] = append(result2[buyerId], tmp)
	}
	return
}

type P_History struct {
	ProjectId   string `ch:"project_id"`
	ProjectName string `ch:"project_name"`
	ZbTime      int64  `ch:"zbtime"`
	Href        string `ch:"href"`
}

// @Author jianghan
// @Description 合作历史
// @Date 2024/4/20
func GetData3(source string, buyerId, winnerId string) (result []*P_History) {
	var rows driver.Rows
	if source == "1" {
		rows, _ = T.ClickhouseConn.Query(context.TODO(), sql_2_1_1, buyerId, winnerId)
	} else if source == "2" {
		rows, _ = T.ClickhouseConn.Query(context.TODO(), sql_2_1, buyerId, winnerId)
	}
	for rows.Next() {
		pHis := P_History{}
		_ = rows.ScanStruct(&pHis)
		pHis.Href = fmt.Sprintf("/article/content/%s.html", encrypt.CommonEncodeArticle("content", pHis.ProjectId))
		result = append(result, &pHis)
	}
	return
}

// @Author jianghan
// @Description 上次合作时间
// @Date 2024/4/24
func LastTimeCoop(buyerId, ent, stype string) (bool, int64) {
	zbtime := int64(0)
	near := false
	if stype == "adiffb" {
		sql := `SELECT zbtime FROM information.transaction_info WHERE buyer_id = ? AND has(winner, ?) ORDER BY zbtime DESC LIMIT 1`
		_ = T.ClickhouseConn.QueryRow(context.TODO(), sql, buyerId, ent).Scan(&zbtime)
	} else if stype == "agency" {
		sql := `SELECT zbtime FROM information.transaction_info WHERE buyer_id = ? AND agency = ? ORDER BY zbtime DESC LIMIT 1`
		_ = T.ClickhouseConn.QueryRow(context.TODO(), sql, buyerId, ent).Scan(&zbtime)
	}
	timestamp := time.Now().AddDate(-3, 0, 0).Unix()
	if timestamp <= zbtime {
		near = true
	}
	return near, zbtime
}

func LastTimeCoopBath(positionId int64, buyerIds, winners, agencys []string) (map[string]map[string]*Cooperate, map[string]map[string]*Cooperate) {
	if len(buyerIds) > 50 {
		buyerIds = buyerIds[:50]
	}
	if len(winners) > 50 {
		winners = winners[:50]
	}
	if len(agencys) > 10 {
		agencys = agencys[:50]
	}
	adiffb, agency := map[string]map[string]*Cooperate{}, map[string]map[string]*Cooperate{}
	if len(buyerIds) == 0 {
		return adiffb, agency
	}
	var toSearch = func(tp int, query string, args []interface{}) {
		logx.Info("LastTimeCoopBath once start ", positionId)
		defer logx.Info("LastTimeCoopBath once over ", positionId)
		rows, err := T.ClickhouseConn.Query(context.Background(), query, args...)
		if err != nil {
			logx.Error(err)
			return
		}
		for rows.Next() {
			var (
				buyer_id string
				one      string
				zbtime   int64
			)
			if err := rows.Scan(&buyer_id, &one, &zbtime); err != nil {
				logx.Error(err)
				continue
			}
			c := &Cooperate{
				ZbTime: zbtime,
			}
			timestamp := time.Now().AddDate(-3, 0, 0).Unix()
			if timestamp <= c.ZbTime {
				c.Near = true
			}
			if tp == 1 {
				if adiffb[buyer_id] == nil {
					adiffb[buyer_id] = map[string]*Cooperate{}
				}
				adiffb[buyer_id][one] = c
			} else {
				if agency[buyer_id] == nil {
					agency[buyer_id] = map[string]*Cooperate{}
				}
				agency[buyer_id][one] = c
			}
		}
		rows.Close()
		if err := rows.Err(); err != nil {
			logx.Error(err)
		}
	}
	if len(winners) > 0 {
		sql := `SELECT buyer_id,winner_one,max(zbtime) FROM information.transaction_info ARRAY JOIN winner AS winner_one WHERE %s group by buyer_id,winner_one`
		text := `(buyer_id=? AND has(winner,?))`
		array := []string{}
		args := []interface{}{}
		for _, v := range buyerIds {
			for _, vv := range winners {
				args = append(args, v, vv)
				array = append(array, text)
				if len(array) == 50 {
					toSearch(1, fmt.Sprintf(sql, strings.Join(array, " or ")), args)
					array = []string{}
					args = []interface{}{}
				}
			}
		}
		if len(array) > 0 {
			toSearch(1, fmt.Sprintf(sql, strings.Join(array, " or ")), args)
		}
	}
	if len(agencys) > 0 {
		sql := `SELECT buyer_id,agency,max(zbtime) FROM information.transaction_info WHERE %s group by buyer_id,agency`
		text := `(buyer_id=? AND agency=?)`
		array := []string{}
		args := []interface{}{}
		for _, v := range buyerIds {
			for _, vv := range winners {
				args = append(args, v, vv)
				array = append(array, text)
				if len(array) == 50 {
					toSearch(2, fmt.Sprintf(sql, strings.Join(array, " or ")), args)
					array = []string{}
					args = []interface{}{}
				}
			}
		}
		if len(array) > 0 {
			toSearch(2, fmt.Sprintf(sql, strings.Join(array, " or ")), args)
		}
	}
	return adiffb, agency
}