project.go 32 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223
  1. package main
  2. import (
  3. "encoding/json"
  4. "log"
  5. "time"
  6. // "log"
  7. "math"
  8. qu "qfw/util"
  9. "sort"
  10. "strings"
  11. //"gopkg.in/mgo.v2/bson"
  12. "go.mongodb.org/mongo-driver/bson"
  13. "go.mongodb.org/mongo-driver/bson/primitive"
  14. )
  15. /**
  16. 项目合并,对比,计算,合并,生成项目
  17. **/
  18. //从对应map中获取对比的项目id
  19. func (p *ProjectTask) getCompareIds(pn, pc, ptc, pb string) (bpn, bpc, bptc, bpb int, res []*Key, idArr []string, IDArr []*ID) {
  20. p.findLock.Lock()
  21. defer p.findLock.Unlock()
  22. // p.ConCurrentLock(n1, n2, n3, n4)
  23. // defer p.ConCurrentUnLock(n1, n2, n3, n4)
  24. p.wg.Add(1)
  25. //查找到id数组
  26. res = []*Key{}
  27. //是否查找到,并标识位置。-1代表值为空或已经存在。
  28. bpn, bpc, bptc, bpb = -1, -1, -1, -1
  29. if pn != "" {
  30. ids := p.mapPn[pn]
  31. if ids == nil {
  32. ids = &Key{Arr: []string{}}
  33. p.mapPn[pn] = ids
  34. bpn = 0
  35. }
  36. ids.Lock.Lock()
  37. res = append(res, ids)
  38. }
  39. if pc != "" {
  40. ids := p.mapPc[pc]
  41. if ids == nil {
  42. ids = &Key{Arr: []string{}}
  43. p.mapPc[pc] = ids
  44. bpc = len(res)
  45. }
  46. ids.Lock.Lock()
  47. res = append(res, ids)
  48. }
  49. if ptc != "" {
  50. ids := p.mapPc[ptc]
  51. if ids == nil {
  52. ids = &Key{Arr: []string{}}
  53. p.mapPc[ptc] = ids
  54. bptc = len(res)
  55. }
  56. ids.Lock.Lock()
  57. res = append(res, ids)
  58. }
  59. if pb != "" {
  60. ids := p.mapPb[pb]
  61. if ids == nil {
  62. ids = &Key{Arr: []string{}}
  63. p.mapPb[pb] = ids
  64. bpb = len(res)
  65. }
  66. ids.Lock.Lock()
  67. res = append(res, ids)
  68. }
  69. repeatId := map[string]bool{}
  70. idArr = []string{} //项目id
  71. IDArr = []*ID{} //项目信息
  72. for _, m := range res {
  73. for _, id := range m.Arr {
  74. if !repeatId[id] {
  75. repeatId[id] = true
  76. //_, _ = strconv.ParseInt(id[0:8], 16, 64)
  77. p.AllIdsMapLock.Lock()
  78. Id := p.AllIdsMap[id]
  79. p.AllIdsMapLock.Unlock()
  80. if Id != nil {
  81. Id.Lock.Lock()
  82. idArr = append(idArr, id)
  83. IDArr = append(IDArr, Id)
  84. }
  85. }
  86. }
  87. }
  88. return
  89. }
  90. func (p *ProjectTask) startProjectMerge(info *Info, tmp map[string]interface{}) {
  91. //只有或没有采购单位的无法合并
  92. //bpn, bpc, bptc, bpb 是否查找到,并标识位置。-1代表未查找到。
  93. //pids 是项目id数组集合
  94. //IDArr,是单个项目ID对象集合
  95. bpn, bpc, bptc, bpb, pids, _, IDArr := p.getCompareIds(info.ProjectName, info.ProjectCode, info.PTC, info.Buyer) //, info.LenPN, info.LenPC, info.LenPTC, len([]rune(info.Buyer)))
  96. defer p.wg.Done()
  97. //map--k为pn,ptn,pc,ptc,buyer值 v为Id数组和lock
  98. for _, m := range pids {
  99. defer m.Lock.Unlock()
  100. }
  101. for _, id := range IDArr {
  102. defer id.Lock.Unlock()
  103. }
  104. bFindProject := false
  105. findPid := ""
  106. //获取完id,进行计算
  107. //定义两组
  108. comRes1 := []*ProjectInfo{} //优先级最高的对比结果数组
  109. comRes2 := []*ProjectInfo{} //优化级其次
  110. comRes3 := []*ProjectInfo{}
  111. for _, v := range IDArr {
  112. comStr := ""
  113. compareProject := v.P
  114. compareProject.score = 0
  115. //问题出地LastTime!!!!!
  116. diffTime := int64(math.Abs(float64(info.Publishtime - compareProject.LastTime)))
  117. if diffTime <= p.validTime {
  118. //代理机构完全不相同,直接新建项目
  119. if CheckContain(compareProject.Agency, info.Agency) == 3 {
  120. continue
  121. }
  122. //地区(省、市、区)不同,直接新建项目
  123. if ComparePlace(compareProject, info) {
  124. continue
  125. }
  126. info.PNBH = 0
  127. info.PCBH = 0
  128. info.PTCBH = 0
  129. compareStr, score := comparePNC(info, compareProject)
  130. resVal, pjVal := Select(compareStr, info, compareProject)
  131. //---------------------------------------
  132. //log.Println(resVal, pjVal, compareProject)
  133. if resVal > 0 {
  134. compareBuyer, compareCity, compareTime, compareAgency, compareBudget, compareBidmount, score2 := p.compareBCTABB(info, compareProject, diffTime, score)
  135. //项目名称、项目编号、标题项目编号、采购单位、省、市、发布时间、代理机构
  136. comStr = compareStr + compareBuyer + compareCity + compareTime + compareAgency + compareBudget + compareBidmount
  137. compareProject.comStr = comStr
  138. compareProject.pjVal = pjVal
  139. compareProject.resVal = resVal
  140. //log.Println(compareProject.comStr)
  141. eqV := compareResult(resVal, pjVal, score2, comStr, compareBuyer, compareCity, compareTime, compareAgency, compareBudget, compareBidmount)
  142. if eqV == 1 {
  143. comRes1 = append(comRes1, compareProject)
  144. } else if eqV == 2 {
  145. comRes2 = append(comRes2, compareProject)
  146. } else if eqV == 3 {
  147. comRes3 = append(comRes3, compareProject)
  148. }
  149. }
  150. }
  151. }
  152. //--------------------------------对比完成-----------------------
  153. //更新数组、更新项目
  154. for kv, resN := range [][]*ProjectInfo{comRes1, comRes2, comRes3} {
  155. if len(resN) > 0 {
  156. if len(resN) > 1 {
  157. sort.Slice(resN, func(i, j int) bool {
  158. return resN[i].score > resN[j].score
  159. })
  160. }
  161. ex := 0
  162. resArr := []*ProjectInfo{}
  163. for i, res := range resN {
  164. choose, e := p.CompareStatus(resN[i], info)
  165. if !choose {
  166. ex = e
  167. resArr = append(resArr, res)
  168. }
  169. }
  170. if len(resArr) > 0 {
  171. bFindProject = true
  172. findPid = resArr[0].Id.Hex()
  173. p.UpdateProject(tmp, info, resArr[0], kv+1, resArr[0].comStr, ex)
  174. for k2, bv := range []int{bpn, bpc, bptc, bpb} {
  175. if bv > -1 {
  176. pids[bv].Arr = append(pids[bv].Arr, findPid)
  177. if k2 == 0 {
  178. if resArr[0].ProjectName == "" {
  179. resArr[0].ProjectName = info.ProjectName
  180. } else {
  181. if resArr[0].MPN == nil {
  182. resArr[0].MPN = []string{info.ProjectName}
  183. } else {
  184. resArr[0].MPN = append(resArr[0].MPN, info.ProjectName)
  185. }
  186. }
  187. } else if k2 < 3 {
  188. if resArr[0].ProjectCode == "" {
  189. resArr[0].ProjectCode = qu.If(k2 == 1, info.ProjectCode, info.PTC).(string)
  190. } else {
  191. if resArr[0].MPC == nil {
  192. resArr[0].MPC = []string{qu.If(k2 == 1, info.ProjectCode, info.PTC).(string)}
  193. } else {
  194. resArr[0].MPC = append(resArr[0].MPC, qu.If(k2 == 1, info.ProjectCode, info.PTC).(string))
  195. }
  196. }
  197. } else {
  198. if resArr[0].Buyer == "" {
  199. resArr[0].Buyer = info.Buyer
  200. }
  201. }
  202. }
  203. }
  204. } else {
  205. bFindProject = false
  206. findPid = ""
  207. }
  208. break
  209. }
  210. }
  211. if !bFindProject {
  212. id, p1 := p.NewProject(tmp, info)
  213. p.AllIdsMapLock.Lock()
  214. p.AllIdsMap[id] = &ID{Id: id, P: p1}
  215. p.AllIdsMapLock.Unlock()
  216. for _, m := range pids {
  217. m.Arr = append(m.Arr, id)
  218. }
  219. }
  220. }
  221. func (p *ProjectTask) compareBCTABB(info *Info, cp *ProjectInfo, diffTime int64, score int) (compareBuyer, compareCity, compareTime, compareAgency, compareBudget, compareBidmount string, score2 int) {
  222. compareBuyer = "D"
  223. if len([]rune(info.Buyer)) > 3 && len([]rune(cp.Buyer)) > 3 {
  224. v := CheckContain(info.Buyer, cp.Buyer)
  225. if v == 1 {
  226. compareBuyer = "A"
  227. score += 3
  228. } else {
  229. //v1 := CosineSimilar(info.Buyer, cp.Buyer)
  230. if v == 2 {
  231. compareBuyer = "B"
  232. score += 1
  233. } else {
  234. compareBuyer = "C"
  235. }
  236. }
  237. }
  238. //---------------------------------------
  239. compareCity = ""
  240. if info.Area != "全国" && info.Area != "" && info.Area == cp.Area {
  241. compareCity += "A"
  242. score += 2
  243. } else if info.Area == "全国" || cp.Area == "全国" {
  244. compareCity += "B"
  245. score += 1
  246. } else {
  247. compareCity += "C"
  248. }
  249. if compareCity != "C" {
  250. if info.City != "" && info.City == cp.City {
  251. compareCity += "A"
  252. score += 2
  253. } else {
  254. if info.Area == "全国" || cp.Area == "全国" {
  255. compareCity += "B"
  256. } else if info.City == compareCity {
  257. compareCity += "B"
  258. } else {
  259. compareCity += "C"
  260. }
  261. }
  262. } else {
  263. compareCity += "C"
  264. }
  265. score2 = 0
  266. if compareCity == "AA" {
  267. if info.District != "" && info.District == cp.District {
  268. score2 = 1
  269. }
  270. }
  271. compareTime = "D"
  272. if diffTime < 45*86400 {
  273. compareTime = "A"
  274. score += 2
  275. } else if diffTime < 90*86400 {
  276. compareTime = "B"
  277. score += 1
  278. }
  279. compareAgency = "D"
  280. if info.Agency != "" {
  281. if info.Agency == cp.Agency {
  282. compareAgency = "A"
  283. score += 2
  284. score2 += 1
  285. } else if cp.Agency != "" {
  286. if strings.Contains(info.Agency, cp.Agency) || strings.Contains(cp.Agency, info.Agency) {
  287. compareAgency = "B"
  288. score += 1
  289. score2 += 1
  290. } else {
  291. compareAgency = "C"
  292. }
  293. }
  294. }
  295. compareBudget = "C"
  296. if info.Budget > 0 && (info.Budget == cp.Budget || (cp.Bidamount > 0 && info.Budget > cp.Bidamount && (info.Budget-cp.Bidamount) < (0.15*info.Budget))) {
  297. compareBudget = "A"
  298. score += 1
  299. score2 += 1
  300. }
  301. // else if info.Budget == 0 && cp.Budget == 0 {
  302. // compareBudget = "B"
  303. // }
  304. compareBidmount = "C"
  305. if info.Bidamount > 0 && (info.Bidamount == cp.Bidamount || (cp.Budget > 0 && cp.Budget > info.Bidamount && (cp.Budget-info.Bidamount) < 0.15*cp.Budget)) {
  306. compareBidmount = "A"
  307. score += 1
  308. score2 += 1
  309. }
  310. // else if info.Bidamount == 0 && cp.Bidamount == 0 {
  311. // compareBidmount = "B"
  312. // }
  313. cp.score = score
  314. return
  315. }
  316. func compareResult(resVal, pjVal, score2 int, comStr, compareBuyer, compareCity, compareTime, compareAgency, compareBudget, compareBidmount string) int {
  317. eqV := 0
  318. switch resVal {
  319. case 3:
  320. if pjVal == 3 && comStr[3:] != "CCCDCCC" {
  321. eqV = 1
  322. } else if compareBuyer < "C" {
  323. if pjVal > 1 {
  324. eqV = 1
  325. } else { //if (compareCity[1:1] != "C" || compareTime != "D") && score2 > 0
  326. eqV = 2
  327. }
  328. } else if compareBuyer == "D" {
  329. if pjVal > 1 && (compareCity[1:1] != "C" || score2 > 0) {
  330. eqV = 2
  331. } else if compareCity[1:1] != "C" && compareTime == "A" && score2 > 0 {
  332. eqV = 3
  333. }
  334. } else {
  335. if pjVal == 3 && (score2 > 0 || compareCity[1:1] != "C") {
  336. eqV = 2
  337. } else if pjVal == 2 && compareCity[1:1] != "C" && compareTime == "A" && score2 > 0 {
  338. eqV = 3
  339. } else if compareCity == "AA" && compareTime == "A" && score2 > 0 {
  340. eqV = 3
  341. }
  342. }
  343. case 2:
  344. if compareBuyer < "C" {
  345. if pjVal > 1 {
  346. eqV = 2
  347. } else if compareCity[1:1] != "C" && compareTime == "A" || score2 > 0 {
  348. eqV = 3
  349. }
  350. } else if compareBuyer == "D" {
  351. if pjVal > 1 && (score2 > 0 || compareCity[1:1] != "C") {
  352. eqV = 2
  353. } else if compareCity[1:1] != "C" && compareTime == "A" && score2 > 0 {
  354. eqV = 3
  355. }
  356. } else {
  357. if pjVal > 1 && compareTime == "A" && (score2 > 0 || compareCity[1:1] != "C") {
  358. eqV = 2
  359. } else if compareCity[1:1] != "C" && compareTime == "A" && (compareAgency == "A" || score2 > 0) && (compareBudget == "A" || compareBidmount == "A") {
  360. eqV = 3
  361. }
  362. }
  363. case 1:
  364. if compareBuyer < "C" {
  365. if pjVal > 1 && (score2 > 0 || compareCity[1:1] != "C") {
  366. eqV = 2
  367. } else if compareCity[1:1] != "C" && compareTime == "A" && (compareAgency == "A" || score2 > 0) && (compareBudget == "A" || compareBidmount == "A") {
  368. eqV = 3
  369. }
  370. } else if compareBuyer == "D" {
  371. if pjVal > 1 && compareTime == "A" && (score2 > 0 || compareCity[1:1] != "C") {
  372. eqV = 2
  373. } else if compareCity[1:1] != "C" && compareTime == "A" && (compareAgency == "A" || score2 > 0) && (compareBudget == "A" || compareBidmount == "A") {
  374. eqV = 3
  375. }
  376. } else {
  377. if pjVal > 1 && compareTime == "A" && score2 > 0 && (compareBudget == "A" || compareBidmount == "A") && compareCity[1:1] != "C" {
  378. eqV = 3
  379. }
  380. }
  381. }
  382. return eqV
  383. }
  384. //项目中的字段
  385. var FIELDS = []string{
  386. "area",
  387. "city",
  388. "district",
  389. "projectname",
  390. "projectcode",
  391. "buyer",
  392. "buyerclass",
  393. "buyerperson",
  394. "buyertel",
  395. "winner",
  396. "agency",
  397. "topscopeclass",
  398. "subscopeclass",
  399. "winnerorder",
  400. "package",
  401. }
  402. var bidtype = map[string]string{
  403. "招标": "招标",
  404. "邀标": "邀标",
  405. "询价": "询价",
  406. "单一": "单一",
  407. "竞价": "竞价",
  408. "竞谈": "竞谈",
  409. }
  410. var bidstatus = map[string]string{
  411. "预告": "预告",
  412. "中标": "中标",
  413. "成交": "成交",
  414. "废标": "废标",
  415. "流标": "流标",
  416. "合同": "合同",
  417. }
  418. //招标时间zbtime、中标时间jgtime、项目状态bidstatus、招标类型bidtype、最后发布时间lasttime、首次发布时间firsttime
  419. func (p *ProjectTask) NewProject(tmp map[string]interface{}, thisinfo *Info) (string, *ProjectInfo) {
  420. pId := primitive.NewObjectID() //NewObjectId()
  421. set := map[string]interface{}{}
  422. set["_id"] = pId
  423. for _, f := range FIELDS {
  424. if tmp[f] != nil {
  425. set[f] = tmp[f]
  426. }
  427. }
  428. bidopentime := qu.Int64All(tmp["bidopentime"])
  429. if bidopentime > 0 {
  430. set["bidopentime"] = bidopentime
  431. }
  432. //异常标记
  433. if thisinfo.TopType != "招标" && thisinfo.TopType != "拟建" && thisinfo.TopType != "预告" {
  434. set["exception"] = 1
  435. }
  436. //projecthref保存
  437. if jsonData, ok := tmp["jsondata"].(map[string]interface{}); ok {
  438. if jsonData != nil && qu.ObjToString(jsonData["projecthref"]) != "" {
  439. set["projecthref"] = jsonData["projecthref"]
  440. }
  441. }
  442. //合同编号
  443. if thisinfo.ContractCode != "" {
  444. set["contractcode"] = thisinfo.ContractCode
  445. }
  446. bt := qu.ObjToString(tmp["toptype"])
  447. bs := qu.ObjToString(tmp["subtype"])
  448. p.mapBidLock.Lock()
  449. if thisinfo.Infoformat == 2 {
  450. set["bidstatus"] = "拟建"
  451. bt = "拟建"
  452. }else {
  453. set["bidtype"] = bidtype[bs]
  454. if bt == "招标" {
  455. set["projectscope"] = qu.ObjToString(tmp["projectscope"])
  456. set["bidstatus"] = bt
  457. } else {
  458. if bidstatus[bs] != "" {
  459. set["bidstatus"] = thisinfo.SubType
  460. bt = thisinfo.SubType
  461. } else if bs == "" {
  462. set["bidstatus"] = ""
  463. bt = ""
  464. } else {
  465. set["bidstatus"] = "其它"
  466. bt = "其它"
  467. }
  468. }
  469. }
  470. p.mapBidLock.Unlock()
  471. pkg := PackageFormat(thisinfo, nil)
  472. p1 := p.NewCachePinfo(pId, thisinfo, bs, bt, pkg)
  473. now := time.Now().Unix()
  474. set["createtime"] = now
  475. set["sourceinfoid"] = thisinfo.Id
  476. set["sourceinfourl"] = tmp["href"]
  477. set["firsttime"] = tmp["publishtime"]
  478. set["lasttime"] = tmp["publishtime"]
  479. set["pici"] = p.pici
  480. set["ids"] = []string{thisinfo.Id}
  481. if thisinfo.TopType == "招标" {
  482. if thisinfo.SubType != "变更" && thisinfo.SubType != "其它" {
  483. set["zbtime"] = tmp["publishtime"]
  484. }
  485. } else if thisinfo.TopType == "结果" || thisinfo.SubType == "合同" {
  486. if thisinfo.Infoformat != 2 {
  487. set["jgtime"] = tmp["publishtime"]
  488. p1.Jgtime = thisinfo.Publishtime
  489. }
  490. }
  491. if len(thisinfo.Subscopeclass) > 0 {
  492. set["s_subscopeclass"] = strings.Join(thisinfo.Subscopeclass, ",")
  493. }
  494. if tmp["budget"] != nil && tmp["budget"] != "" {
  495. set["budget"] = thisinfo.Budget
  496. p1.Budgettag = 0
  497. set["budgettag"] = 0
  498. } else {
  499. p1.Budgettag = 1
  500. set["budgettag"] = 1
  501. }
  502. if tmp["bidamount"] != nil && tmp["bidamount"] != "" {
  503. set["bidamount"] = thisinfo.Bidamount
  504. p1.Bidamounttag = 0
  505. set["bidamounttag"] = 0
  506. } else {
  507. p1.Bidamounttag = 1
  508. set["bidamounttag"] = 1
  509. }
  510. if len(thisinfo.Winners) > 0 {
  511. set["s_winner"] = strings.Join(thisinfo.Winners, ",")
  512. p1.Winners = thisinfo.Winners
  513. }
  514. if thisinfo.HasPackage {
  515. set["multipackage"] = 1
  516. set["package"] = pkg
  517. } else {
  518. set["multipackage"] = 0
  519. }
  520. if thisinfo.Buyer == "" {
  521. set["buyerclass"] = ""
  522. }
  523. p1.InfoFiled = make(map[string]InfoField)
  524. infofiled := InfoField{
  525. Budget: thisinfo.Budget,
  526. Bidamount: thisinfo.Bidamount,
  527. ContractCode: thisinfo.ContractCode,
  528. ProjectName: thisinfo.ProjectName,
  529. ProjectCode: thisinfo.ProjectCode,
  530. Bidstatus: bs,
  531. }
  532. p1.InfoFiled[thisinfo.Id] = infofiled
  533. res := StructToMap(infofiled)
  534. set["infofield"] = map[string]interface{}{
  535. thisinfo.Id: res,
  536. }
  537. push := p.PushListInfo(tmp, thisinfo.Id)
  538. push["s_winner"] = strings.Join(thisinfo.Winners, ",")
  539. set["list"] = []bson.M{
  540. push,
  541. }
  542. //p.savePool <- set
  543. p.updatePool <- []map[string]interface{}{
  544. {
  545. "_id": pId,
  546. },
  547. {
  548. "$set": set,
  549. },
  550. }
  551. //log.Println(set)
  552. return pId.Hex(), &p1
  553. }
  554. //招标信息字段
  555. var INFOFIELDS = []string{
  556. "projectname",
  557. "projectcode",
  558. "projectscope",
  559. "contractcode",
  560. "title",
  561. "href",
  562. "publishtime",
  563. "comeintime",
  564. "bidopentime",
  565. "toptype",
  566. "subtype",
  567. "buyer",
  568. "buyerclass",
  569. "agency",
  570. "winner",
  571. "budget",
  572. "bidamount",
  573. "topscopeclass",
  574. "subscopclass",
  575. "infoformat",
  576. "buyerperson",
  577. "buyertel",
  578. "area",
  579. "city",
  580. "district",
  581. "spidercode",
  582. "site",
  583. }
  584. //项目中list的信息
  585. func (p *ProjectTask) PushListInfo(tmp map[string]interface{}, infoid string) bson.M {
  586. res := bson.M{
  587. "infoid": infoid,
  588. }
  589. for _, k := range INFOFIELDS {
  590. if tmp[k] != nil {
  591. res[k] = tmp[k]
  592. }
  593. }
  594. return res
  595. }
  596. //生成存放在内存中的对象
  597. func (p *ProjectTask) NewCachePinfo(id primitive.ObjectID, thisinfo *Info, bidtype, bidstatus string, pkg map[string]interface{}) ProjectInfo {
  598. p1 := ProjectInfo{
  599. Id: id,
  600. Ids: []string{thisinfo.Id},
  601. ProjectName: thisinfo.ProjectName,
  602. ProjectCode: thisinfo.ProjectCode,
  603. ContractCode: thisinfo.ContractCode,
  604. Buyer: thisinfo.Buyer,
  605. Buyerclass: thisinfo.Buyerclass,
  606. Buyerperson: thisinfo.Buyerperson,
  607. Buyertel: thisinfo.Buyertel,
  608. Topscopeclass: thisinfo.Topscopeclass,
  609. Subscopeclass: thisinfo.Subscopeclass,
  610. Agency: thisinfo.Agency,
  611. Area: thisinfo.Area,
  612. City: thisinfo.City,
  613. District: thisinfo.District,
  614. MPN: []string{},
  615. MPC: []string{},
  616. FirstTime: thisinfo.Publishtime,
  617. LastTime: thisinfo.Publishtime,
  618. Budget: thisinfo.Budget,
  619. Package: pkg,
  620. Bidamount: thisinfo.Bidamount,
  621. Bidstatus: bidstatus,
  622. Bidtype: bidtype,
  623. Winners: thisinfo.Winners,
  624. }
  625. if thisinfo.LenPTC > 5 {
  626. p1.MPC = append(p1.MPC, thisinfo.PTC)
  627. }
  628. return p1
  629. }
  630. //更新项目
  631. func (p *ProjectTask) UpdateProject(tmp map[string]interface{}, thisinfo *Info, pInfo *ProjectInfo, weight int, comStr string, ex int) {
  632. if p.currentType != "ql" && p.currentType != "updateInfo" {
  633. if BinarySearch(pInfo.Ids, thisinfo.Id) > -1 {
  634. log.Println("repeat", thisinfo.Id)
  635. return
  636. }
  637. }
  638. set := map[string]interface{}{}
  639. pInfo.Ids = append(pInfo.Ids, thisinfo.Id)
  640. if len(pInfo.Ids) > 30 {
  641. //异常标记
  642. set["listtag"] = 1
  643. }
  644. //zbtime、lasttime、jgtime
  645. pInfo.LastTime = thisinfo.Publishtime
  646. set["lasttime"] = thisinfo.Publishtime
  647. if thisinfo.TopType == "招标" {
  648. if thisinfo.SubType != "变更" && thisinfo.SubType != "其它" && tmp["zbtime"] == nil {
  649. set["zbtime"] = tmp["publishtime"]
  650. }
  651. if pInfo.Jgtime > 0 {
  652. pInfo.Jgtime = int64(0)
  653. set["jgtime"] = int64(0)
  654. }
  655. } else if thisinfo.TopType == "结果" {
  656. pInfo.Jgtime = thisinfo.Publishtime
  657. set["jgtime"] = thisinfo.Publishtime
  658. } else if thisinfo.SubType == "合同" {
  659. if pInfo.Jgtime <= 0 {
  660. set["jgtime"] = tmp["publishtime"]
  661. pInfo.Jgtime = thisinfo.Publishtime
  662. }
  663. }
  664. if thisinfo.Bidopentime > pInfo.Bidopentime {
  665. pInfo.Bidopentime = thisinfo.Bidopentime
  666. set["bidopentime"] = pInfo.Bidopentime
  667. }
  668. bt := qu.ObjToString(tmp["toptype"])
  669. bs := qu.ObjToString(tmp["subtype"])
  670. p.mapBidLock.Lock()
  671. if bidtype[bs] != "" {
  672. set["bidtype"] = bidtype[bs]
  673. }
  674. if bt == "招标" {
  675. //招标状态,更新projectscope
  676. if tmp["projectscope"] != nil {
  677. set["projectscope"] = qu.ObjToString(tmp["projectscope"])
  678. }
  679. set["bidstatus"] = bt
  680. pInfo.Bidstatus = bt
  681. } else {
  682. if bidstatus[bs] != "" {
  683. set["bidstatus"] = thisinfo.SubType
  684. pInfo.Bidstatus = thisinfo.SubType
  685. } else if thisinfo.Infoformat == 2 {
  686. set["bidstatus"] = "拟建"
  687. pInfo.Bidstatus = "拟建"
  688. } else if bs == "" {
  689. set["bidstatus"] = ""
  690. pInfo.Bidstatus = ""
  691. } else {
  692. set["bidstatus"] = "其它"
  693. pInfo.Bidstatus = "其它"
  694. }
  695. }
  696. p.mapBidLock.Unlock()
  697. //异常标记
  698. if ex > 0 {
  699. set["exception"] = ex
  700. }
  701. //3\4\5--省、市、县
  702. if thisinfo.Area != "全国" {
  703. if pInfo.Area == "全国" {
  704. pInfo.Area = thisinfo.Area
  705. set["area"] = thisinfo.Area
  706. } else if pInfo.Area != thisinfo.Area {
  707. //xt = false
  708. }
  709. if pInfo.City == "" && thisinfo.City != "" {
  710. pInfo.City = thisinfo.City
  711. set["city"] = thisinfo.City
  712. } else if pInfo.City != thisinfo.City {
  713. //xt = false
  714. }
  715. if thisinfo.District != "" && pInfo.District == "" {
  716. pInfo.District = thisinfo.District
  717. set["district"] = thisinfo.District
  718. }
  719. }
  720. //6--项目名称
  721. if (thisinfo.ProjectName != "" && pInfo.ProjectName == "") || (len([]rune(pInfo.ProjectName)) < 6 && thisinfo.LenPN > 6) {
  722. pInfo.ProjectName = thisinfo.ProjectName
  723. set["projectname"] = thisinfo.ProjectName
  724. }
  725. //7--项目编号
  726. if (pInfo.ProjectCode == "" && thisinfo.ProjectCode != "") || (len([]rune(pInfo.ProjectCode)) < 6 && len([]rune(thisinfo.ProjectCode)) > 6) {
  727. pInfo.ProjectCode = thisinfo.ProjectCode
  728. set["projectcode"] = thisinfo.ProjectCode
  729. }
  730. //7--采购单位
  731. if (pInfo.Buyer == "" && thisinfo.Buyer != "") || (len([]rune(pInfo.Buyer)) < 5 && len([]rune(thisinfo.Buyer)) > 5) {
  732. pInfo.Buyer = thisinfo.Buyer
  733. set["buyer"] = thisinfo.Buyer
  734. pInfo.Buyerclass = thisinfo.Buyerclass
  735. set["buyerclass"] = thisinfo.Buyerclass
  736. }
  737. if pInfo.Buyer == "" {
  738. set["buyerclass"] = ""
  739. }
  740. if thisinfo.ContractCode != "" {
  741. set["contractcode"] = pInfo.ContractCode + "," + thisinfo.ContractCode
  742. }
  743. //8--代理机构
  744. if (pInfo.Agency == "" && thisinfo.Agency != "") || (len([]rune(pInfo.Agency)) < 5 && len([]rune(thisinfo.Agency)) > 5) {
  745. pInfo.Agency = thisinfo.Agency
  746. set["agency"] = thisinfo.Agency
  747. }
  748. //9--采购单位联系人
  749. if thisinfo.Buyerperson != "" && strings.Index(pInfo.Buyerperson, thisinfo.Buyerperson) < 0 {
  750. pInfo.Buyerperson = thisinfo.Buyerperson
  751. set["buyerperson"] = pInfo.Buyerperson
  752. }
  753. //10--采购单位電話
  754. if thisinfo.Buyertel != "" && strings.Index(pInfo.Buyertel, thisinfo.Buyertel) < 0 {
  755. pInfo.Buyertel = thisinfo.Buyertel
  756. set["buyertel"] = pInfo.Buyertel
  757. }
  758. if len(thisinfo.Topscopeclass) > 0 {
  759. sort.Strings(pInfo.Topscopeclass)
  760. for _, k := range thisinfo.Topscopeclass {
  761. if BinarySearch(pInfo.Topscopeclass, k) == -1 {
  762. pInfo.Topscopeclass = append(pInfo.Topscopeclass, k)
  763. sort.Strings(pInfo.Topscopeclass)
  764. }
  765. }
  766. set["topscopeclass"] = pInfo.Topscopeclass
  767. }
  768. if len(thisinfo.Subscopeclass) > 0 {
  769. sort.Strings(pInfo.Subscopeclass)
  770. for _, k := range thisinfo.Subscopeclass {
  771. if BinarySearch(pInfo.Subscopeclass, k) == -1 {
  772. pInfo.Subscopeclass = append(pInfo.Subscopeclass, k)
  773. sort.Strings(pInfo.Subscopeclass)
  774. }
  775. }
  776. set["subscopeclass"] = pInfo.Subscopeclass
  777. set["s_subscopeclass"] = strings.Join(pInfo.Subscopeclass, ",")
  778. }
  779. //winner
  780. if len(thisinfo.Winners) > 0 {
  781. if len(pInfo.Winners) <= 0 {
  782. set["winner"] = qu.ObjToString(tmp["winner"])
  783. }
  784. sort.Strings(pInfo.Winners)
  785. for _, k := range thisinfo.Winners {
  786. if BinarySearch(pInfo.Winners, k) == -1 {
  787. pInfo.Winners = append(pInfo.Winners, k)
  788. sort.Strings(pInfo.Winners)
  789. }
  790. }
  791. set["s_winner"] = strings.Join(pInfo.Winners, ",")
  792. }
  793. if thisinfo.HasPackage { //多包处理
  794. set["multipackage"] = 1
  795. pkg := PackageFormat(thisinfo, pInfo)
  796. pInfo.Package = pkg
  797. set["package"] = pInfo.Package
  798. } else {
  799. set["multipackage"] = 0
  800. }
  801. //处理多包后,计算预算金额、中标金额
  802. CountAmount(pInfo, thisinfo, tmp)
  803. if pInfo.Budget >= 0 && pInfo.Budgettag != 1 {
  804. set["budget"] = pInfo.Budget
  805. set["budgettag"] = 0
  806. } else {
  807. set["budgettag"] = 1
  808. }
  809. if pInfo.Bidamount >= 0 && pInfo.Bidamounttag != 1 {
  810. set["bidamount"] = pInfo.Bidamount
  811. set["bidamounttag"] = 0
  812. } else {
  813. set["bidamounttag"] = 1
  814. }
  815. infofiled := InfoField{
  816. Budget: thisinfo.Budget,
  817. Bidamount: thisinfo.Bidamount,
  818. ContractCode: thisinfo.ContractCode,
  819. ProjectName: thisinfo.ProjectName,
  820. ProjectCode: thisinfo.ProjectCode,
  821. Bidstatus: bs,
  822. }
  823. copyMap := Copy(pInfo.InfoFiled).(map[string]InfoField)
  824. copyMap[thisinfo.Id] = infofiled
  825. tmpMap := make(map[string]interface{})
  826. for k, v := range copyMap {
  827. tmpMap[k] = StructToMap(v)
  828. }
  829. tmpMap[thisinfo.Id] = StructToMap(infofiled)
  830. pInfo.InfoFiled = copyMap
  831. set["infofield"] = tmpMap
  832. set["mpn"] = pInfo.MPN
  833. set["mpc"] = pInfo.MPC
  834. set["pici"] = p.pici
  835. update := map[string]interface{}{}
  836. if len(set) > 0 {
  837. update["$set"] = set
  838. }
  839. //保留原数据吧
  840. push := p.PushListInfo(tmp, thisinfo.Id)
  841. push["s_winner"] = strings.Join(thisinfo.Winners, ",")
  842. push["compareStr"] = comStr
  843. push["resVal"] = pInfo.resVal
  844. push["pjVal"] = pInfo.pjVal
  845. update["$push"] = map[string]interface{}{
  846. "list": push,
  847. "ids": thisinfo.Id,
  848. }
  849. if len(update) > 0 {
  850. updateInfo := []map[string]interface{}{
  851. {
  852. "_id": pInfo.Id,
  853. },
  854. update,
  855. }
  856. p.updatePool <- updateInfo
  857. }
  858. }
  859. /**
  860. * 更新项目时,项目状态的处理
  861. * 返回是否新增项目,异常标记
  862. * 1、新增项目时,招标信息的状态(toptype)不是招标、拟建、预告 异常:1
  863. * 异常1是在项目新建的时候才会产生
  864. * 3、项目合并时,项目状态是”流标“/”废标“,招标信息状态不是”招标“ 异常:2
  865. * 4、项目合并时,项目状态是”合同“/”其它“,招标信息类型是”结果“ 异常:3
  866. */
  867. func (p *ProjectTask) CompareStatus(project *ProjectInfo, info *Info) (bool, int) {
  868. if info.TopType == "拟建" || info.TopType == "预告" || info.TopType == "招标" {
  869. if project.Bidstatus == "拟建" || project.Bidstatus == "预告" || project.Bidstatus == "招标" {
  870. return false, 0
  871. } else if project.Bidstatus == "废标" || project.Bidstatus == "流标" {
  872. return false, 0
  873. } else {
  874. return true, 0
  875. }
  876. } else if info.TopType == "结果" {
  877. if project.Bidstatus == "拟建" || project.Bidstatus == "预告" || project.Bidstatus == "招标" {
  878. return false, 0
  879. } else if project.Bidstatus == info.SubType {
  880. //状态一样,根据发布时间判断是否合并
  881. if (info.Publishtime - project.LastTime) > p.statusTime {
  882. return true, 0
  883. } else {
  884. return false, 0
  885. }
  886. } else if project.Bidstatus == "成交" && info.SubType == "中标" {
  887. return true, 0
  888. } else if project.Bidstatus == "流标" || project.Bidstatus == "废标" {
  889. return false, 2
  890. } else if project.Bidstatus == "合同" || project.Bidstatus == "其它" {
  891. return false, 3
  892. } else {
  893. return false, 0
  894. }
  895. } else {
  896. return false, 0
  897. }
  898. }
  899. /*
  900. * 对比地区(省、市、区),存在且不同,不能合并
  901. * 返回是否新建项目
  902. */
  903. func ComparePlace(project *ProjectInfo, info *Info) bool {
  904. if info.Area == "全国" || info.Area == "" {
  905. return false
  906. }
  907. if project.Area == "全国" || project.Area == "" {
  908. return false
  909. }
  910. if info.Area == project.Area {
  911. if info.City == "" || project.City == "" {
  912. return false
  913. } else if info.City == project.City {
  914. if project.District == "" || info.District == "" || info.District == project.District {
  915. return false
  916. } else {
  917. return true
  918. }
  919. } else {
  920. return true
  921. }
  922. } else {
  923. return true
  924. }
  925. }
  926. var PackageEle = []string{
  927. "origin",
  928. "name",
  929. //"text",
  930. "budget",
  931. "winner",
  932. "bidamount",
  933. "bidamounttype",
  934. "currency",
  935. "bidstatus",
  936. }
  937. func packageEle(map1 map[string]interface{}, id string) map[string]interface{} {
  938. p2 := map[string]interface{}{}
  939. for _, k := range PackageEle {
  940. if map1[k] != nil {
  941. p2[k] = map1[k]
  942. }
  943. infoid := p2["infoid"]
  944. if infoid == nil {
  945. p2["infoid"] = id
  946. }
  947. }
  948. return p2
  949. }
  950. func PackageFormat(info *Info, project *ProjectInfo) map[string]interface{} {
  951. p1 := map[string]interface{}{}
  952. if project != nil && project.Package != nil && len(project.Package) > 0 {
  953. packageCopy := Copy(project.Package).(map[string]interface{})
  954. p1 = packageCopy
  955. for k, v := range info.Package {
  956. if v1, ok := v.(map[string]interface{}); ok {
  957. v2 := map[string]interface{}{}
  958. v2 = packageEle(v1, info.Id)
  959. if v2["bidstatus"] == nil {
  960. v2["bidstatus"] = info.SubType
  961. }
  962. addFlag := false
  963. for k1, v3 := range p1 {
  964. if v4, ok := v3.([]map[string]interface{}); ok {
  965. //if qu.ObjToString(v4[0]["origin"]) == qu.ObjToString(v2["origin"]) && qu.ObjToString(v4[0]["name"]) == qu.ObjToString(v2["name"]) {
  966. if k1 == k {
  967. v4 = append(v4, v2)
  968. p1[k1] = v4
  969. addFlag = true
  970. break
  971. }
  972. }
  973. }
  974. if !addFlag {
  975. p1[k] = []map[string]interface{}{v2}
  976. }
  977. }
  978. }
  979. p1 = packageCopy
  980. } else {
  981. for k, v := range info.Package {
  982. v1, _ := v.(map[string]interface{})
  983. p2 := map[string]interface{}{}
  984. p2 = packageEle(v1, info.Id)
  985. if p2["bidstatus"] == nil {
  986. p2["bidstatus"] = info.SubType
  987. }
  988. p1[k] = []map[string]interface{}{p2}
  989. }
  990. }
  991. return p1
  992. }
  993. //计算预算(budget)、中标金额(bidamount)
  994. func CountAmount(project *ProjectInfo, info *Info, tmp map[string]interface{}) {
  995. if info.HasPackage {
  996. budget := 0.0
  997. for _, v := range project.Package {
  998. v1, _ := v.([]map[string]interface{})
  999. for _, v2 := range v1 {
  1000. if v2["budget"] != nil {
  1001. b1 := qu.Float64All(v2["budget"])
  1002. if b1 > 0 {
  1003. budget = budget + b1
  1004. break
  1005. }
  1006. } else {
  1007. project.Budgettag = 1
  1008. }
  1009. }
  1010. }
  1011. if budget > 0 {
  1012. project.Budget = budget
  1013. project.Budgettag = 0
  1014. } else if budget == 0 && info.Budget > 0 {
  1015. project.Budget = info.Budget
  1016. project.Budgettag = 0
  1017. }
  1018. } else {
  1019. //招标没有多包
  1020. k := KeyPackage.FindStringSubmatch(info.ProjectName)
  1021. if len(k) > 0 {
  1022. //招标是单包
  1023. if len(project.Package) > 0 {
  1024. //项目有多包
  1025. flag := false
  1026. for _, v := range project.Package {
  1027. v1, _ := v.([]map[string]interface{})
  1028. if len(v1) > 0 && v1[0]["name"] == info.ProjectName {
  1029. flag = true
  1030. }
  1031. }
  1032. if !flag {
  1033. project.Budget = project.Budget + info.Budget
  1034. project.Budgettag = 0
  1035. }
  1036. } else {
  1037. //项目没有多包
  1038. if info.Budget > 0 {
  1039. project.Budget = project.Budget + info.Budget
  1040. project.Budgettag = 0
  1041. } else if info.Budget == 0 && tmp["budget"] != nil {
  1042. project.Budgettag = 0
  1043. } else {
  1044. project.Budgettag = 1
  1045. }
  1046. }
  1047. } else {
  1048. if project.Budget < info.Budget {
  1049. project.Budget = info.Budget
  1050. project.Budgettag = 0
  1051. }
  1052. if tmp["budget"] == nil {
  1053. project.Budgettag = 1
  1054. }
  1055. }
  1056. }
  1057. if info.SubType == "中标" || info.SubType == "成交" || info.SubType == "合同" {
  1058. if info.HasPackage {
  1059. bidamount := 0.0
  1060. for _, v := range project.Package {
  1061. v1, _ := v.([]map[string]interface{})
  1062. for _, v2 := range v1 {
  1063. if tmp["bidamount"] != nil {
  1064. b1 := qu.Float64All(v2["bidamount"])
  1065. if b1 > 0 {
  1066. bidamount = bidamount + b1
  1067. break
  1068. }
  1069. } else {
  1070. project.Bidamount = 1
  1071. }
  1072. }
  1073. }
  1074. if bidamount > 0 {
  1075. project.Bidamount = bidamount
  1076. project.Bidamounttag = 0
  1077. } else if bidamount == 0 && info.Budget > 0 {
  1078. project.Bidamount = info.Bidamount
  1079. project.Bidamounttag = 0
  1080. }
  1081. } else {
  1082. //招标没有多包
  1083. k := KeyPackage.FindStringSubmatch(info.ProjectName)
  1084. if len(k) > 0 {
  1085. //招标是单包
  1086. if len(project.Package) > 0 {
  1087. //项目有多包
  1088. flag := false
  1089. for _, v := range project.Package {
  1090. v1, _ := v.([]map[string]interface{})
  1091. if len(v1) > 0 && v1[0]["name"] == info.ProjectName {
  1092. flag = true
  1093. }
  1094. }
  1095. if !flag {
  1096. project.Bidamount = project.Bidamount + info.Bidamount
  1097. project.Bidamounttag = 0
  1098. }
  1099. } else {
  1100. //项目没有多包
  1101. if info.Bidamount > 0 {
  1102. project.Bidamount = project.Bidamount + info.Bidamount
  1103. project.Bidamounttag = 0
  1104. } else if info.Bidamount == 0 && tmp["bidamount"] != nil {
  1105. project.Bidamounttag = 0
  1106. } else {
  1107. project.Bidamounttag = 1
  1108. }
  1109. }
  1110. } else {
  1111. if info.SubType == "中标" || info.SubType == "成交" {
  1112. if project.Bidamount < info.Bidamount {
  1113. project.Bidamount = info.Bidamount
  1114. project.Bidamounttag = 0
  1115. } else {
  1116. flag := false
  1117. if project.InfoFiled != nil && len(project.InfoFiled) > 0 {
  1118. for _, res := range project.InfoFiled {
  1119. if res.ContractCode != "" && res.ContractCode == info.ContractCode {
  1120. flag = true
  1121. break
  1122. }
  1123. if res.Bidamount == project.Bidamount {
  1124. flag = true
  1125. break
  1126. }
  1127. }
  1128. if !flag {
  1129. project.Bidamount = project.Bidamount + info.Bidamount
  1130. project.Bidamounttag = 0
  1131. } else {
  1132. if project.Bidamount < info.Bidamount {
  1133. project.Bidamount = info.Bidamount
  1134. project.Bidamounttag = 0
  1135. }
  1136. }
  1137. }
  1138. }
  1139. }
  1140. if tmp["bidamount"] == nil {
  1141. project.Budgettag = 1
  1142. }
  1143. }
  1144. }
  1145. } else {
  1146. project.Bidamounttag = 1
  1147. }
  1148. }
  1149. //结构体转map
  1150. func StructToMap(filed InfoField) map[string]interface{} {
  1151. //先转json
  1152. result, err := json.Marshal(filed)
  1153. if err != nil {
  1154. return nil
  1155. }
  1156. //json转map
  1157. res := make(map[string]interface{})
  1158. err = json.Unmarshal(result, &res)
  1159. return res
  1160. }