network.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852
  1. package service
  2. import (
  3. "context"
  4. "database/sql"
  5. "fmt"
  6. "math"
  7. "sort"
  8. "strings"
  9. "sync"
  10. . "app.yhyue.com/moapp/jybase/common"
  11. . "app.yhyue.com/moapp/jybase/date"
  12. "app.yhyue.com/moapp/jybase/encrypt"
  13. . "app.yhyue.com/moapp/jybase/es"
  14. . "app.yhyue.com/moapp/jybase/sort"
  15. . "bp.jydev.jianyu360.cn/CRM/application/api/common"
  16. "bp.jydev.jianyu360.cn/CRM/application/api/internal/types"
  17. "github.com/shopspring/decimal"
  18. "github.com/zeromicro/go-zero/core/logx"
  19. )
  20. var Network = &network{}
  21. type network struct {
  22. }
  23. type networkTree struct {
  24. Count int64 `json:"count"`
  25. Name string `json:"name"`
  26. Children []*networkTreeChild `json:"children"`
  27. }
  28. type networkTreeChild struct {
  29. Count int64 `json:"count"`
  30. Name string `json:"name"`
  31. Id string `json:"id"`
  32. Type int `json:"type"`
  33. }
  34. type projectInfo struct {
  35. BuyerCount int64
  36. ProjectCount int64
  37. ProjectAmount float64
  38. MonitorCount int64
  39. ExportId []string
  40. }
  41. type firstpartyNetwork struct {
  42. CompanyId string
  43. CompanyName string
  44. Name string
  45. }
  46. //人脉库-添加/修改人脉
  47. func (n *network) AddOrUpdate(in *types.AddOrUpdateReq) *types.Reply {
  48. reply := &types.Reply{Data: map[string]interface{}{
  49. "status": 0,
  50. }}
  51. if in.Type != "middleman" && in.Company_id == "" {
  52. return reply
  53. }
  54. nowFormat := NowFormat(Date_Full_Layout)
  55. var saveIntroduce = func(tx *sql.Tx, cid int64, isUpdate bool) bool {
  56. if in.Type != "middleman" {
  57. return true
  58. } else if in.Introduce_owner_id == "" || in.Introduce_project_id == "" {
  59. return false
  60. }
  61. values := []interface{}{}
  62. ioi := strings.Split(in.Introduce_owner_id, ",")
  63. ion := strings.Split(in.Introduce_owner_name, ",")
  64. if len(ioi) != len(ion) {
  65. return false
  66. }
  67. for k, v := range ioi {
  68. values = append(values, in.PositionId, in.EntId, in.EntDeptId, in.EntUserId, cid, v, ion[k], 1, nowFormat)
  69. }
  70. ipi := strings.Split(in.Introduce_project_id, ",")
  71. ipn := strings.Split(in.Introduce_project_name, ",")
  72. if len(ipi) != len(ipn) {
  73. return false
  74. }
  75. for k, v := range ipi {
  76. values = append(values, in.PositionId, in.EntId, in.EntDeptId, in.EntUserId, cid, v, ipn[k], 2, nowFormat)
  77. }
  78. var r2 int64
  79. if isUpdate {
  80. r2 = CrmMysql.UpdateOrDeleteBySqlByTx(tx, `delete from crm.connection_introduce where connection_id=? and position_id=?`, in.Id, in.PositionId)
  81. }
  82. r3, _ := CrmMysql.InsertBatchByTx(tx, "crm.connection_introduce", []string{"position_id", "ent_id", "ent_dept_id", "ent_user_id", "connection_id", "relate_id", "relate_name", "itype", "create_time"}, values)
  83. return r2 >= 0 && r3 > 0
  84. }
  85. itype := n.TypeConvert(in.Type)
  86. if in.Id > 0 {
  87. if CrmMysql.ExecTx("更新人脉", func(tx *sql.Tx) bool {
  88. if in.Company_id != "" {
  89. count := CrmMysql.CountBySql(`select count(1) as count from crm.connection where position_id=? and company_id=? and itype=? and id<>? and status=1`, in.PositionId, in.Company_id, itype, in.Id)
  90. if count == -1 {
  91. return false
  92. } else if count > 0 {
  93. reply.Data = map[string]interface{}{
  94. "status": -1,
  95. }
  96. return false
  97. }
  98. }
  99. r1 := CrmMysql.UpdateOrDeleteBySqlByTx(tx, `update crm.connection set company_name=?,company_id=?,contact_person=?,contact_phone=?,update_time=? where id=? and position_id=?`, in.Company_name, in.Company_id, in.Contact_person, in.Contact_phone, nowFormat, in.Id, in.PositionId)
  100. return r1 >= 0 && saveIntroduce(tx, in.Id, true)
  101. }) {
  102. reply.Data = map[string]interface{}{
  103. "status": 1,
  104. }
  105. }
  106. } else {
  107. if in.Company_id != "" {
  108. count := CrmMysql.CountBySql(`select count(1) as count from crm.connection where position_id=? and company_id=? and itype=? and status=1`, in.PositionId, in.Company_id, itype)
  109. if count == -1 {
  110. return reply
  111. } else if count > 0 {
  112. reply.Data = map[string]interface{}{
  113. "status": -1,
  114. }
  115. return reply
  116. }
  117. }
  118. var r1 int64
  119. if CrmMysql.ExecTx("新增人脉", func(tx *sql.Tx) bool {
  120. _, r1 = CrmMysql.InsertBatchByTx(tx, "crm.connection", []string{"position_id", "ent_id", "ent_dept_id", "ent_user_id", "itype", "company_name", "company_id", "contact_person", "contact_phone", "status", "create_time", "update_time"}, []interface{}{in.PositionId, in.EntId, in.EntDeptId, in.EntUserId, itype, in.Company_name, in.Company_id, in.Contact_person, in.Contact_phone, 1, nowFormat, nowFormat})
  121. return r1 > 0 && saveIntroduce(tx, r1, false)
  122. }) {
  123. reply.Data = map[string]interface{}{
  124. "status": 1,
  125. "id": r1,
  126. }
  127. }
  128. }
  129. return reply
  130. }
  131. //人脉库-业主名称联想
  132. func (n *network) Associate(in *types.AssociateReq) (reply *types.Reply) {
  133. //类型;firstparty:甲方 supplier:供应商 adiffb:同甲异业 middleman:中间人 middleman_owner:中间人-业主 middleman_project:中间人-项目 agency:招标代理机构
  134. res := []map[string]interface{}{}
  135. reply = &types.Reply{Data: res}
  136. in.Name = strings.TrimSpace(in.Name)
  137. if in.Name == "" {
  138. return
  139. }
  140. pageSize := 10
  141. if in.Type == "adiffb" {
  142. probusfors := NetworkCom.GetMyProbusfor(in.EntAccountId)
  143. if len(probusfors) > 0 {
  144. args := []interface{}{in.EntName}
  145. wh, newArgs := NetworkCom.WhArgs(probusfors)
  146. args = append(args, newArgs...)
  147. q := `select DISTINCT b.winner_id,b.winner from information.transaction_info a
  148. inner join information.transaction_info b on (has(a.winner, ?) and a.buyer_id<>'' and a.buyer_id=b.buyer_id and hasAny(b.property_form,[` + wh + `])=0) ORDER BY b.project_id`
  149. rows, err := ClickhouseConn.Query(context.Background(), q, args...)
  150. if err != nil {
  151. logx.Error(err)
  152. } else {
  153. repeat := map[string]bool{}
  154. for rows.Next() {
  155. var (
  156. winner_id []string
  157. winner []string
  158. )
  159. if err := rows.Scan(&winner_id, &winner); err != nil {
  160. logx.Error(err)
  161. continue
  162. }
  163. for k, v := range winner {
  164. if repeat[v] || !strings.Contains(v, in.Name) {
  165. continue
  166. }
  167. repeat[v] = true
  168. if k >= len(winner_id) {
  169. continue
  170. }
  171. res = append(res, map[string]interface{}{
  172. "company_name": v,
  173. "company_id": winner_id[k],
  174. })
  175. }
  176. }
  177. rows.Close()
  178. if err := rows.Err(); err != nil {
  179. logx.Error(err)
  180. }
  181. if len(res) > pageSize {
  182. res = res[:pageSize]
  183. }
  184. }
  185. }
  186. } else {
  187. must := []string{fmt.Sprintf(`{"multi_match":{"query":"%s","type":"phrase","fields":["company_name"]}}`, in.Name)}
  188. switch in.Type {
  189. case "firstparty":
  190. must = append(must, fmt.Sprintf(`{"terms":{"company_label":["%s"]}}`, strings.Join(NetworkCom.GetEntTagSeat(2), `","`)))
  191. must = append(must, `{"terms":{"company_unit_type":[1,2]}}`)
  192. case "supplier":
  193. must = append(must, fmt.Sprintf(`{"terms":{"company_label":["%s"]}}`, strings.Join(NetworkCom.GetEntTagSeat(2), `","`)))
  194. must = append(must, `{"term":{"company_unit_type":3}}`)
  195. case "middleman_owner":
  196. must = append(must, `{"terms":{"company_unit_type":[1,2]}}`)
  197. case "agency":
  198. must = append(must, `{"term":{"company_unit_type":4}}`)
  199. }
  200. q := fmt.Sprintf(`{"query":{"bool":{"must":[%s]}},"size":%d,"_source":["id","company_name"]}`, strings.Join(must, ","), pageSize)
  201. logx.Info("人脉库-业主名称联想", q)
  202. datas := VarEs.Get("ent_info", "ent_info", q)
  203. if datas != nil {
  204. for _, v := range *datas {
  205. res = append(res, map[string]interface{}{
  206. "company_name": ObjToString(v["company_name"]),
  207. "company_id": ObjToString(v["id"]),
  208. })
  209. }
  210. }
  211. }
  212. reply.Data = res
  213. return
  214. }
  215. //人脉库-全部人脉项目
  216. func (n *network) AllProject(in *types.AllprojectReq) (reply *types.Reply) {
  217. pool := make(chan bool, 5)
  218. wait := &sync.WaitGroup{}
  219. lock := &sync.Mutex{}
  220. reply = &types.Reply{}
  221. wh, newArgs := NetworkCom.WhArgs(NetworkCom.GetMyProbusfor(in.EntAccountId))
  222. var count int64
  223. var list []*networkTree
  224. if in.Id != "" {
  225. if in.Type == 1 {
  226. result := n.FirstpartyNetwork(in.Name, []string{in.Id})
  227. if result[in.Id] != nil {
  228. nameIndex := map[string]int{}
  229. for _, v := range result[in.Id] {
  230. if _, ok := nameIndex[v.Name]; !ok {
  231. nameIndex[v.Name] = len(list)
  232. list = append(list, &networkTree{
  233. Name: v.Name,
  234. })
  235. }
  236. pool <- true
  237. wait.Add(1)
  238. go func(cIndex int, cId, cName string) {
  239. defer func() {
  240. <-pool
  241. wait.Done()
  242. }()
  243. ntc := &networkTreeChild{
  244. Name: cName,
  245. Id: cId,
  246. Type: 1,
  247. }
  248. if wh != "" {
  249. thisArgs := []interface{}{ntc.Id}
  250. thisArgs = append(thisArgs, newArgs...)
  251. ntc.Count = NetworkCom.Count(`select count(1) from information.transaction_info where buyer_id=? and hasAny(property_form,[`+wh+`])`, thisArgs...)
  252. }
  253. lock.Lock()
  254. count++
  255. list[cIndex].Count += ntc.Count
  256. list[cIndex].Children = append(list[cIndex].Children, ntc)
  257. lock.Unlock()
  258. }(nameIndex[v.Name], v.CompanyId, v.CompanyName)
  259. }
  260. wait.Wait()
  261. }
  262. }
  263. } else {
  264. q := `SELECT a.company_id,a.company_name,a.itype,COUNT(b.id) AS ipc FROM crm.connection a
  265. LEFT JOIN crm.connection_introduce b ON (b.position_id=? AND b.itype=2 AND a.id=b.connection_id) WHERE a.position_id=?`
  266. args := []interface{}{in.PositionId, in.PositionId}
  267. if in.Name != "" {
  268. q += ` and company_name like ?`
  269. args = append(args, "%"+in.Name+"%")
  270. }
  271. q += ` GROUP BY a.id ORDER BY a.create_time DESC`
  272. datas := CrmMysql.SelectBySql(q, args...)
  273. list = []*networkTree{
  274. &networkTree{
  275. Name: "甲方",
  276. Children: []*networkTreeChild{},
  277. },
  278. &networkTree{
  279. Name: "供应商",
  280. Children: []*networkTreeChild{},
  281. },
  282. &networkTree{
  283. Name: "同甲异业渠道",
  284. Children: []*networkTreeChild{},
  285. },
  286. &networkTree{
  287. Name: "中间人",
  288. Children: []*networkTreeChild{},
  289. },
  290. &networkTree{
  291. Name: "招标代理",
  292. Children: []*networkTreeChild{},
  293. },
  294. }
  295. //
  296. for _, vt := range *datas {
  297. pool <- true
  298. wait.Add(1)
  299. go func(v map[string]interface{}) {
  300. defer func() {
  301. <-pool
  302. wait.Done()
  303. }()
  304. itype := IntAll(v["itype"])
  305. if itype <= 0 || itype > len(list) {
  306. return
  307. }
  308. company_name := ObjToString(v["company_name"])
  309. if company_name == "" {
  310. company_name = "未填写"
  311. }
  312. ntc := &networkTreeChild{
  313. Name: company_name,
  314. Id: ObjToString(v["company_id"]),
  315. Type: IntAll(v["itype"]),
  316. }
  317. if wh != "" {
  318. thisArgs := []interface{}{ntc.Id}
  319. thisArgs = append(thisArgs, newArgs...)
  320. if itype == 1 {
  321. ntc.Count = NetworkCom.Count(`select count(1) from information.transaction_info where buyer_id=? and hasAny(property_form,[`+wh+`])`, thisArgs...)
  322. } else if itype == 2 || itype == 3 {
  323. ntc.Count = NetworkCom.Count(`select count(1) from information.transaction_info where has(winner_id,?) and hasAny(property_form,[`+wh+`])`, thisArgs...)
  324. } else if itype == 4 {
  325. ntc.Count = Int64All(v["ipc"])
  326. } else if itype == 5 {
  327. ntc.Count = NetworkCom.Count(`select count(1) from information.transaction_info where agency_id=? and hasAny(property_form,[`+wh+`])`, thisArgs...)
  328. }
  329. }
  330. lock.Lock()
  331. count += ntc.Count
  332. list[itype-1].Count += ntc.Count
  333. list[itype-1].Children = append(list[itype-1].Children, ntc)
  334. lock.Unlock()
  335. }(vt)
  336. }
  337. wait.Wait()
  338. }
  339. reply = &types.Reply{
  340. Data: map[string]interface{}{
  341. "count": count,
  342. "list": list,
  343. },
  344. }
  345. return reply
  346. }
  347. //人脉库-列表
  348. func (n *network) List(in *types.NetWorkListReq) *types.Reply {
  349. q := `select a.company_id,a.company_name,a.itype,a.contact_person as person,a.contact_phone as phone,count(DISTINCT if(b.itype=1,b.relate_id,null)) as buyer_count,count(DISTINCT if(b.itype=2,b.relate_id,null)) as project_count,GROUP_CONCAT(IF(b.itype=1,b.relate_id,NULL)) AS relate_buyer_id,GROUP_CONCAT(IF(b.itype=2,b.relate_id,NULL)) AS relate_project_id,a.create_time from crm.connection a
  350. left join crm.connection_introduce b on (a.id=b.connection_id) where a.position_id=?`
  351. args := []interface{}{in.PositionId}
  352. if in.Type != "" {
  353. q += ` and a.itype=?`
  354. args = append(args, n.TypeConvert(in.Type))
  355. }
  356. if in.Starttime != "" {
  357. q += ` and a.create_time>=?`
  358. args = append(args, in.Starttime)
  359. }
  360. if in.Endtime != "" {
  361. q += ` and a.create_time<=?`
  362. args = append(args, in.Endtime)
  363. }
  364. if in.Name != "" {
  365. q += ` and a.company_name like ?`
  366. args = append(args, "%"+in.Name+"%")
  367. }
  368. q += ` GROUP BY a.id order by a.create_time desc`
  369. listTemp := CrmMysql.SelectBySql(q, args...)
  370. firstparty_array, supplier_array, adiffb_array, agency_array, middleman_project_array := []string{}, []string{}, []string{}, []string{}, []string{}
  371. for _, v := range *listTemp {
  372. switch Int64All(v["itype"]) {
  373. case 1:
  374. firstparty_array = append(firstparty_array, ObjToString(v["company_id"]))
  375. case 2:
  376. supplier_array = append(supplier_array, ObjToString(v["company_id"]))
  377. case 3:
  378. adiffb_array = append(adiffb_array, ObjToString(v["company_id"]))
  379. case 4:
  380. if relate_project_id := ObjToString(v["relate_project_id"]); relate_project_id != "" {
  381. middleman_project_array = append(middleman_project_array, strings.Split(relate_project_id, ",")...)
  382. }
  383. case 5:
  384. agency_array = append(agency_array, ObjToString(v["company_id"]))
  385. }
  386. }
  387. //
  388. firstparty_count, supplier_count, adiffb_count, middleman_count, agency_count := 0, 0, 0, 0, 0
  389. list := []*map[string]interface{}{}
  390. isGoNextSetp := true
  391. probusfors := []string{}
  392. if in.Project_matchme == 1 {
  393. probusfors = NetworkCom.GetMyProbusfor(in.EntAccountId)
  394. if len(probusfors) == 0 {
  395. isGoNextSetp = false
  396. }
  397. }
  398. if isGoNextSetp {
  399. entMonitor := NetworkCom.EntMonitor(in.UserId)
  400. fpn := n.FirstpartyNetwork("", firstparty_array)
  401. firstparty_project := n.Introduce_Firstparty(fpn, entMonitor, in.Project_matchme, probusfors)
  402. supplier_project := n.Introduce_Supplier(supplier_array, entMonitor, in.Project_matchme, probusfors)
  403. adiffb_project := n.Introduce_Supplier(adiffb_array, entMonitor, in.Project_matchme, probusfors)
  404. agency_project := n.Introduce_Agency(agency_array, entMonitor, in.Project_matchme, probusfors)
  405. middleman_project := n.Introduce_Middleman(middleman_project_array, entMonitor, in.Project_matchme, probusfors)
  406. for _, v := range *listTemp {
  407. itype := ""
  408. buyer_count, project_count, expect_amount, monitor_count := int64(0), int64(0), float64(0), int64(0)
  409. company_id := ObjToString(v["company_id"])
  410. export_id := []string{}
  411. jump_type, jump_id := "", ""
  412. switch Int64All(v["itype"]) {
  413. case 1:
  414. itype = "甲方"
  415. jump_type = "firstparty"
  416. for _, vv := range fpn[company_id] {
  417. if jump_id != "" {
  418. jump_id += ","
  419. }
  420. jump_id += vv.CompanyId
  421. }
  422. firstparty_count++
  423. if firstparty_project[company_id] != nil {
  424. buyer_count = firstparty_project[company_id].BuyerCount
  425. project_count = firstparty_project[company_id].ProjectCount
  426. expect_amount = firstparty_project[company_id].ProjectAmount
  427. monitor_count = firstparty_project[company_id].MonitorCount
  428. export_id = firstparty_project[company_id].ExportId
  429. }
  430. case 2:
  431. itype = "供应商"
  432. jump_type = "supplier"
  433. jump_id = company_id
  434. supplier_count++
  435. if supplier_project[company_id] != nil {
  436. buyer_count = supplier_project[company_id].BuyerCount
  437. project_count = supplier_project[company_id].ProjectCount
  438. expect_amount = supplier_project[company_id].ProjectAmount
  439. monitor_count = supplier_project[company_id].MonitorCount
  440. export_id = supplier_project[company_id].ExportId
  441. }
  442. case 3:
  443. itype = "同甲异业渠道"
  444. jump_type = "adiffb"
  445. jump_id = company_id
  446. adiffb_count++
  447. if adiffb_project[company_id] != nil {
  448. buyer_count = adiffb_project[company_id].BuyerCount
  449. project_count = adiffb_project[company_id].ProjectCount
  450. expect_amount = adiffb_project[company_id].ProjectAmount
  451. monitor_count = adiffb_project[company_id].MonitorCount
  452. export_id = adiffb_project[company_id].ExportId
  453. }
  454. case 4:
  455. itype = "中间人"
  456. jump_type = "middleman"
  457. jump_id = company_id
  458. middleman_count++
  459. buyer_count = Int64All(v["buyer_count"])
  460. project_count = Int64All(v["project_count"])
  461. if relate_buyer_id := ObjToString(v["relate_buyer_id"]); relate_buyer_id != "" {
  462. for _, v := range strings.Split(relate_buyer_id, ",") {
  463. if v == "" {
  464. continue
  465. }
  466. if entMonitor[v] {
  467. monitor_count++
  468. }
  469. }
  470. }
  471. if relate_project_id := ObjToString(v["relate_project_id"]); relate_project_id != "" {
  472. export_id = strings.Split(relate_project_id, ",")
  473. for _, v := range export_id {
  474. if middleman_project[v] != nil {
  475. expect_amount += middleman_project[v].ProjectAmount
  476. }
  477. }
  478. }
  479. case 5:
  480. itype = "招标代理机构"
  481. jump_type = "agency"
  482. jump_id = company_id
  483. agency_count++
  484. if agency_project[company_id] != nil {
  485. buyer_count = agency_project[company_id].BuyerCount
  486. project_count = agency_project[company_id].ProjectCount
  487. expect_amount = agency_project[company_id].ProjectAmount
  488. monitor_count = agency_project[company_id].MonitorCount
  489. export_id = agency_project[company_id].ExportId
  490. }
  491. }
  492. if buyer_count < in.Buyercount_start {
  493. continue
  494. } else if in.Buyercount_end > 0 && buyer_count > in.Buyercount_end {
  495. continue
  496. } else if in.Monitor == 1 && monitor_count <= 0 {
  497. continue
  498. } else if in.Monitor == -1 && monitor_count > 0 {
  499. continue
  500. } else if monitor_count < in.Monitorcount_start {
  501. continue
  502. } else if in.Monitorcount_end > 0 && monitor_count > in.Monitorcount_end {
  503. continue
  504. } else if in.Project_matchme == 1 && project_count == 0 {
  505. continue
  506. }
  507. export_url := ""
  508. if len(export_id) > 0 {
  509. export_url = "/subscribepay/network/projectExport?export_id=" + encrypt.SE.EncodeStringByCheck(strings.Join(export_id, ","))
  510. }
  511. company_name, _ := v["company_name"].(string)
  512. url := "/swordfish/page_big_pc/unit_portrayal/" + company_name
  513. if company_name == "" {
  514. company_name = "未填写"
  515. url = ""
  516. }
  517. list = append(list, &map[string]interface{}{
  518. "company_id": company_id,
  519. "company_name": company_name,
  520. "type": itype,
  521. "jump_type": jump_type,
  522. "jump_id": jump_id,
  523. "person": v["person"],
  524. "phone": v["phone"],
  525. "buyer_count": buyer_count,
  526. "monitor_count": 0,
  527. "expect_amount": RetainDecimal(expect_amount/10000, 2),
  528. "project_count": project_count,
  529. "create_time": v["create_time"],
  530. "export_url": export_url,
  531. "url": url,
  532. })
  533. }
  534. }
  535. csList := &ComSortList{
  536. SortKeys: []*ComSortKey{
  537. &ComSortKey{
  538. Keys: []string{"expect_amount"},
  539. Order: 1,
  540. Type: "float",
  541. },
  542. },
  543. List: list,
  544. }
  545. if in.Order_amount == -1 {
  546. csList.SortKeys[0].Order = -1
  547. }
  548. if in.Order_amount != 0 {
  549. sort.Sort(csList)
  550. }
  551. length := int64(len(csList.List))
  552. if in.Page_size <= 0 {
  553. in.Page_size = 10
  554. }
  555. total_page := int64(math.Ceil(float64(length) / float64(in.Page_size)))
  556. finalList := []*map[string]interface{}{}
  557. if length > 0 {
  558. if in.Current_page <= 0 {
  559. in.Current_page = 1
  560. }
  561. if in.Current_page > total_page {
  562. in.Current_page = total_page
  563. }
  564. start := (in.Current_page - 1) * in.Page_size
  565. end := start + in.Page_size
  566. if end > length {
  567. end = length
  568. }
  569. finalList = csList.List[start:end]
  570. }
  571. return &types.Reply{
  572. Data: map[string]interface{}{
  573. "count": length,
  574. "total_page": total_page,
  575. "firstparty_count": firstparty_count,
  576. "supplier_count": supplier_count,
  577. "adiffb_count": adiffb_count,
  578. "middleman_count": middleman_count,
  579. "agency_count": agency_count,
  580. "list": finalList,
  581. },
  582. }
  583. }
  584. //
  585. func (n *network) FirstpartyNetwork(name string, values []string) map[string][]*firstpartyNetwork {
  586. result := map[string][]*firstpartyNetwork{}
  587. wh, args := NetworkCom.WhArgs(values)
  588. q := `select DISTINCT a.a_id,a.b_id as company_id,a.b_name as company_name,b.name from information.ent_map_code a
  589. inner join information.ent_code b on (a.a_id in (` + wh + `) and b.pcode in ('0100','0200') and a.code=b.code)`
  590. if name != "" {
  591. q += ` where c.company_name like ?`
  592. args = append(args, "%"+name+"%")
  593. }
  594. q += ` order by a.a_id,b.name,a.b_name`
  595. rows, err := ClickhouseConn.Query(context.Background(), q, args...)
  596. if err != nil {
  597. logx.Error(err)
  598. return result
  599. }
  600. for rows.Next() {
  601. var (
  602. a_id string
  603. company_id string
  604. company_name string
  605. name string
  606. )
  607. if err := rows.Scan(&a_id, &company_id, &company_name, &name); err != nil {
  608. logx.Error(err)
  609. continue
  610. }
  611. result[a_id] = append(result[a_id], &firstpartyNetwork{
  612. CompanyId: company_id,
  613. CompanyName: company_name,
  614. Name: name,
  615. })
  616. }
  617. return result
  618. }
  619. //
  620. func (n *network) Introduce_Firstparty(fpn map[string][]*firstpartyNetwork, entMonitor map[string]bool, matchme int64, probusfors []string) map[string]*projectInfo {
  621. values := []string{}
  622. vm := map[string]*projectInfo{}
  623. for _, v := range fpn {
  624. for _, vv := range v {
  625. vm[vv.CompanyId] = &projectInfo{}
  626. values = append(values, vv.CompanyId)
  627. }
  628. }
  629. wh, args := NetworkCom.WhArgs(values)
  630. q := `select buyer_id,count(project_id) AS project_count,sum(project_money) AS project_amount,groupUniqArray(project_id) from information.transaction_info where buyer_id in (` + wh + `)`
  631. if matchme == 1 {
  632. newWh, newArgs := NetworkCom.WhArgs(probusfors)
  633. q += ` hasAny(property_form,[` + newWh + `])`
  634. args = append(args, newArgs...)
  635. }
  636. q += ` group by buyer_id`
  637. rows, err := ClickhouseConn.Query(context.Background(), q, args...)
  638. if err != nil {
  639. logx.Error(err)
  640. return nil
  641. }
  642. for rows.Next() {
  643. var (
  644. buyer_id string
  645. project_count uint64
  646. project_amount decimal.Decimal
  647. project_id []string
  648. )
  649. if err := rows.Scan(&buyer_id, &project_count, &project_amount, &project_id); err != nil {
  650. logx.Error(err)
  651. continue
  652. }
  653. if vm[buyer_id] == nil {
  654. continue
  655. }
  656. vm[buyer_id].ProjectCount += int64(project_count)
  657. pf, _ := project_amount.Float64()
  658. vm[buyer_id].ProjectAmount += pf
  659. vm[buyer_id].ExportId = project_id
  660. }
  661. rows.Close()
  662. if err := rows.Err(); err != nil {
  663. logx.Error(err)
  664. }
  665. result := map[string]*projectInfo{}
  666. for k, v := range fpn {
  667. if result[k] == nil {
  668. result[k] = &projectInfo{}
  669. }
  670. result[k].BuyerCount = int64(len(v))
  671. for _, vv := range v {
  672. if entMonitor[vv.CompanyName] {
  673. result[k].MonitorCount++
  674. }
  675. if vm[vv.CompanyId] == nil {
  676. continue
  677. }
  678. result[k].ProjectCount += vm[vv.CompanyId].ProjectCount
  679. result[k].ProjectAmount += vm[vv.CompanyId].ProjectAmount
  680. }
  681. }
  682. return result
  683. }
  684. //
  685. func (n *network) Introduce_Supplier(values []string, entMonitor map[string]bool, matchme int64, probusfors []string) map[string]*projectInfo {
  686. vm := map[string]*projectInfo{}
  687. for _, v := range values {
  688. vm[v] = &projectInfo{}
  689. }
  690. wh, args := NetworkCom.WhArgs(values)
  691. q := `select a.winner_id,count(DISTINCT b.buyer_id) AS buyer_count,count(DISTINCT b.project_id) AS project_count,sum(b.project_money) AS project_amount,groupUniqArray(b.buyer),groupUniqArray(b.project_id) from information.transaction_info a
  692. inner join information.transaction_info b on (hasAny(a.winner_id,[` + wh + `]) and a.buyer_id<>'' and a.buyer_id=b.buyer_id`
  693. if matchme == 1 {
  694. newWh, newArgs := NetworkCom.WhArgs(probusfors)
  695. q += ` hasAny(b.property_form,[` + newWh + `])`
  696. args = append(args, newArgs...)
  697. }
  698. q += `) group by a.winner_id`
  699. rows, err := ClickhouseConn.Query(context.Background(), q, args...)
  700. if err != nil {
  701. logx.Error(err)
  702. return nil
  703. }
  704. for rows.Next() {
  705. var (
  706. winner_id []string
  707. buyer_count uint64
  708. project_count uint64
  709. project_amount decimal.Decimal
  710. buyers []string
  711. project_id []string
  712. )
  713. if err := rows.Scan(&winner_id, &buyer_count, &project_count, &project_amount, &buyers, &project_id); err != nil {
  714. logx.Error(err)
  715. continue
  716. }
  717. for _, v := range winner_id {
  718. if vm[v] == nil {
  719. continue
  720. }
  721. vm[v].BuyerCount += int64(buyer_count)
  722. vm[v].ProjectCount += int64(project_count)
  723. pf, _ := project_amount.Float64()
  724. vm[v].ProjectAmount += pf
  725. vm[v].ExportId = project_id
  726. for _, v := range buyers {
  727. if entMonitor[v] {
  728. vm[v].MonitorCount++
  729. }
  730. }
  731. }
  732. }
  733. rows.Close()
  734. if err := rows.Err(); err != nil {
  735. logx.Error(err)
  736. }
  737. return vm
  738. }
  739. //
  740. func (n *network) Introduce_Agency(values []string, entMonitor map[string]bool, matchme int64, probusfors []string) map[string]*projectInfo {
  741. vm := map[string]*projectInfo{}
  742. for _, v := range values {
  743. vm[v] = &projectInfo{}
  744. }
  745. wh, args := NetworkCom.WhArgs(values)
  746. q := `select a.agency_id,count(DISTINCT b.buyer_id) AS buyer_count,count(DISTINCT b.project_id) AS project_count,sum(b.project_money) AS project_amount,groupUniqArray(b.buyer),groupUniqArray(b.project_id) from information.transaction_info a
  747. inner join information.transaction_info b on (a.agency_id in (` + wh + `) and a.buyer_id<>'' and a.buyer_id=b.buyer_id`
  748. if matchme == 1 {
  749. newWh, newArgs := NetworkCom.WhArgs(probusfors)
  750. q += ` hasAny(b.property_form,[` + newWh + `])`
  751. args = append(args, newArgs...)
  752. }
  753. q += `) group by a.agency_id`
  754. rows, err := ClickhouseConn.Query(context.Background(), q, args...)
  755. if err != nil {
  756. logx.Error(err)
  757. return nil
  758. }
  759. for rows.Next() {
  760. var (
  761. agency_id string
  762. buyer_count uint64
  763. project_count uint64
  764. project_amount decimal.Decimal
  765. buyers []string
  766. project_id []string
  767. )
  768. if err := rows.Scan(&agency_id, &buyer_count, &project_count, &project_amount, &buyers, &project_id); err != nil {
  769. logx.Error(err)
  770. continue
  771. }
  772. if vm[agency_id] == nil {
  773. continue
  774. }
  775. vm[agency_id].BuyerCount += int64(buyer_count)
  776. vm[agency_id].ProjectCount += int64(project_count)
  777. pf, _ := project_amount.Float64()
  778. vm[agency_id].ProjectAmount += pf
  779. vm[agency_id].ExportId = project_id
  780. for _, v := range buyers {
  781. if entMonitor[v] {
  782. vm[agency_id].MonitorCount++
  783. }
  784. }
  785. }
  786. rows.Close()
  787. if err := rows.Err(); err != nil {
  788. logx.Error(err)
  789. }
  790. return vm
  791. }
  792. //
  793. func (n *network) Introduce_Middleman(values []string, entMonitor map[string]bool, matchme int64, probusfors []string) map[string]*projectInfo {
  794. vm := map[string]*projectInfo{}
  795. wh, newArgs := NetworkCom.WhArgs(values)
  796. rows, err := ClickhouseConn.Query(context.Background(), `select project_id,project_money from information.transaction_info where project_id in (`+wh+`)`, newArgs...)
  797. if err != nil {
  798. logx.Error(err)
  799. return nil
  800. }
  801. for rows.Next() {
  802. var (
  803. project_id string
  804. project_money decimal.Decimal
  805. )
  806. if err := rows.Scan(&project_id, &project_money); err != nil {
  807. logx.Error(err)
  808. continue
  809. }
  810. pf, _ := project_money.Float64()
  811. vm[project_id] = &projectInfo{
  812. ProjectAmount: pf,
  813. }
  814. }
  815. rows.Close()
  816. if err := rows.Err(); err != nil {
  817. logx.Error(err)
  818. }
  819. return vm
  820. }
  821. //
  822. func (n *network) TypeConvert(itype string) int {
  823. //firstparty:甲方 supplier:供应商 adiffb:同甲异业 middleman:中间人 agency:招标代理机构
  824. switch itype {
  825. case "firstparty":
  826. return 1
  827. case "supplier":
  828. return 2
  829. case "adiffb":
  830. return 3
  831. case "middleman":
  832. return 4
  833. case "agency":
  834. return 5
  835. }
  836. return 0
  837. }