timedTaskWinner.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672
  1. package main
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "github.com/garyburd/redigo/redis"
  6. "go.mongodb.org/mongo-driver/bson/primitive"
  7. "gopkg.in/mgo.v2/bson"
  8. "log"
  9. mu "mfw/util"
  10. "net"
  11. "qfw/util"
  12. "sort"
  13. "strconv"
  14. "strings"
  15. "time"
  16. )
  17. //之前main方法,只更新
  18. func TaskWinner(mapinfo *map[string]interface{}) {
  19. defer util.Catch()
  20. //释放
  21. defer func() { <-CPool }()
  22. gtid, lteid := util.ObjToString((*mapinfo)["gtid"]), util.ObjToString((*mapinfo)["lteid"])
  23. if gtid == "" || lteid == "" {
  24. log.Println(gtid, lteid, "参数错误")
  25. return
  26. }
  27. GId, err := primitive.ObjectIDFromHex(gtid)
  28. LtId, err2 := primitive.ObjectIDFromHex(lteid)
  29. if err != nil || err2 != nil {
  30. log.Println(gtid, lteid, "转换_id错误")
  31. return
  32. }
  33. //udp的id区间查询bidding 中标人 中标联系人 中标联系电话
  34. // topscopeclass项目类型-industry行业类型&&topscopeclass联系人项目类型
  35. // (area地区-province省份 city城市-city城市 district区县-district区县)
  36. // winneraddr-company_address企业地址
  37. SourceClientcc := SourceClient.GetMgoConn()
  38. defer SourceClient.DestoryMongoConn(SourceClientcc)
  39. cursor := SourceClientcc.DB(Config["mgodb_bidding"]).C(Config["mgodb_mgoinit_c"]).Find(bson.M{
  40. "_id": bson.M{
  41. "$gte": GId,
  42. "$lte": LtId,
  43. },
  44. }).Select(bson.M{"winner": 1, "winnertel": 1, "winnerperson": 1,
  45. "topscopeclass": 1, "winneraddr": 1}).Iter()
  46. if cursor == nil {
  47. log.Println(cursor)
  48. return
  49. }
  50. //判断是否是存量,是存量走Redis遍历
  51. if v, ok := (*mapinfo)["data_info"].(string); ok && v == "save" {
  52. //存量处理
  53. tmp := map[string]interface{}{}
  54. conn := HisRedisPool.Conn()
  55. defer conn.Close()
  56. //选择redis db
  57. redis_winner_db, _ := strconv.Atoi(Config["redis_winner_db"])
  58. conn.Select(redis_winner_db)
  59. //遍历bidding表保存到redis
  60. // key:_id value:json结构体
  61. for cursor.Next(&tmp) {
  62. if tmp["winner"] == nil || tmp["winner"] == "" {
  63. continue
  64. }
  65. mgoId := tmp["_id"].(primitive.ObjectID).Hex()
  66. delete(tmp, "_id")
  67. bytes, _ := json.Marshal(tmp)
  68. if err := conn.Set(mgoId, string(bytes), 0).Err(); err != nil {
  69. log.Println(err)
  70. }
  71. }
  72. //遍历redis
  73. if scan := conn.Scan(0, "", 100); scan.Err() != nil {
  74. log.Println(scan.Err())
  75. return
  76. } else {
  77. iterator := scan.Iterator()
  78. for iterator.Next() {
  79. redisId := iterator.Val() //redis key
  80. redisvalue := conn.Get(iterator.Val()).Val() //redis val
  81. tmp := make(map[string]interface{})
  82. json.Unmarshal([]byte(redisvalue), &tmp)
  83. //重复增量操作
  84. //redis查询是否存在
  85. rdb := RedisPool.Get()
  86. rdb.Do("SELECT", Config["redis_winner_db"])
  87. if reply, err := redis.String(rdb.Do("GET", tmp["winner"])); err != nil {
  88. //redis不存在,存到临时表,定时任务处理
  89. FClient.DbName = Config["mgodb_extract_kf"]
  90. if tmpid := FClient.Save("winner_new", tmp); tmpid == nil {
  91. log.Println("存量 FClient.Save err", tmpid)
  92. }
  93. //log.Println("get redis id err:定时任务处理", err, tmp)
  94. if err := rdb.Close(); err != nil {
  95. log.Println("存量", err)
  96. }
  97. //删除存量redis
  98. conn.Del(redisId)
  99. continue
  100. } else {
  101. if err := rdb.Close(); err != nil {
  102. log.Println(err)
  103. }
  104. //拿到合并后的qyk
  105. FClient.DbName = Config["mgodb_extract_kf"]
  106. oldTmp := FClient.FindById(Config["mgo_qyk_c"], reply)
  107. if oldTmp == nil {
  108. log.Println("存量 redis id 不存在", reply, tmp["winner"])
  109. continue
  110. }
  111. tmpTopscopeclass := []string{}
  112. tmpTopscopeclassMap := make(map[string]bool)
  113. if oldTmp["industry"] == nil {
  114. if v, ok := tmp["topscopeclass"].(primitive.A); ok {
  115. for _, vv := range v {
  116. if vvv, ok := vv.(string); ok && len(vvv) > 1 {
  117. tmpTopscopeclassMap[vvv[:len(vvv)-1]] = true
  118. }
  119. }
  120. for k := range tmpTopscopeclassMap {
  121. tmpTopscopeclass = append(tmpTopscopeclass, k)
  122. }
  123. }
  124. } else {
  125. if v, ok := oldTmp["industry"].(primitive.A); ok {
  126. for _, vv := range v {
  127. if vvv, ok := vv.(string); ok {
  128. tmpTopscopeclassMap[vvv] = true
  129. }
  130. }
  131. }
  132. if v, ok := tmp["topscopeclass"].(primitive.A); ok {
  133. for _, vv := range v {
  134. if vvv, ok := vv.(string); ok && len(vvv) > 1 {
  135. tmpTopscopeclassMap[vvv[:len(vvv)-1]] = true
  136. }
  137. }
  138. for k := range tmpTopscopeclassMap {
  139. tmpTopscopeclass = append(tmpTopscopeclass, k)
  140. }
  141. }
  142. }
  143. sort.Strings(tmpTopscopeclass)
  144. oldTmp["industry"] = tmpTopscopeclass
  145. esId := oldTmp["_id"].(primitive.ObjectID).Hex()
  146. //更新行业类型
  147. if tmp["winnerperson"] == nil || tmp["winnerperson"] == "" || Reg_xing.MatchString(util.ObjToString(tmp["winnerperson"])) {
  148. oldTmp["updatatime"] = time.Now().Unix()
  149. //mongo更新
  150. FClient.DbName = Config["mgodb_extract_kf"]
  151. if !FClient.UpdateById(Config["mgo_qyk_c"], esId, bson.M{"$set": oldTmp}) {
  152. log.Println("mongo更新err", esId)
  153. }
  154. //es更新
  155. delete(oldTmp, "_id")
  156. if _, err := EsConn.Update().Index(Config["elasticsearch_index"]).Type(Config["elasticsearch_type"]).Id(esId).Doc(oldTmp).Refresh(true).Do(); err != nil {
  157. log.Println("update es err:", err)
  158. }
  159. //删除存量redis
  160. conn.Del(redisId)
  161. continue
  162. }
  163. //联系方式合并
  164. var tmpperson, winnertel string
  165. tmpperson = tmp["winnerperson"].(string)
  166. if tmp["winnertel"] == nil || tmp["winnertel"] == "" {
  167. winnertel = ""
  168. } else {
  169. if Reg_xing.MatchString(util.ObjToString(tmp["winnertel"])) || !Reg_tel.MatchString(util.ObjToString(tmp["winnertel"])) {
  170. winnertel = ""
  171. } else {
  172. winnertel = util.ObjToString(tmp["winnertel"])
  173. }
  174. }
  175. contactMaps := make([]interface{}, 0)
  176. if oldTmp["contact"] == nil {
  177. tmpContact := make(map[string]interface{})
  178. tmpContact["infoid"] = redisId
  179. tmpContact["contact_person"] = tmpperson
  180. tmpContact["contact_type"] = "项目联系人"
  181. tmpContact["phone"] = winnertel
  182. tmpContact["topscopeclass"] = strings.Join(tmpTopscopeclass, ";")
  183. tmpContact["updatetime"] = time.Now().Unix()
  184. contactMaps = append(contactMaps, tmpContact)
  185. } else {
  186. //对比前四项,相等丢弃
  187. /*if v, ok := oldTmp["contact"].(primitive.A); ok {
  188. var isNotUpdate bool
  189. for _, vv := range v {
  190. if vvv, ok := vv.(map[string]interface{}); ok {
  191. if vvv["contact_person"] == tmpperson && vvv["contact_type"] == "项目联系人" &&
  192. vvv["phone"] == winnertel && vvv["topscopeclass"] == strings.Join(tmpTopscopeclass, ";") {
  193. isNotUpdate = true
  194. vvv["updatetime"] = time.Now().Unix()
  195. }
  196. contactMaps = append(contactMaps, vvv)
  197. }
  198. }
  199. if !isNotUpdate {
  200. vvv := make(map[string]interface{})
  201. vvv["infoid"] = redisId
  202. vvv["contact_person"] = tmp["winnerperson"]
  203. vvv["contact_type"] = "项目联系人"
  204. vvv["phone"] = winnertel
  205. vvv["topscopeclass"] = strings.Join(tmpTopscopeclass, ";")
  206. vvv["updatetime"] = time.Now().Unix()
  207. contactMaps = append(contactMaps, vvv)
  208. }
  209. }*/
  210. //直接添加联系人,不再判断
  211. if v, ok := oldTmp["contact"].(primitive.A); ok {
  212. contactMaps = append(contactMaps, v...)
  213. }
  214. vvv := make(map[string]interface{})
  215. vvv["infoid"] = redisId
  216. vvv["contact_person"] = tmp["winnerperson"]
  217. vvv["contact_type"] = "项目联系人"
  218. vvv["phone"] = winnertel
  219. vvv["topscopeclass"] = strings.Join(tmpTopscopeclass, ";")
  220. vvv["updatetime"] = time.Now().Unix()
  221. contactMaps = append(contactMaps, vvv)
  222. }
  223. oldTmp["contact"] = contactMaps
  224. //mongo更新
  225. oldTmp["updatatime"] = time.Now().Unix()
  226. FClient.DbName = Config["mgodb_extract_kf"]
  227. if !FClient.UpdateById(Config["mgo_qyk_c"], esId, bson.M{"$set": oldTmp}) {
  228. log.Println("存量 mongo更新 err", esId, oldTmp)
  229. }
  230. //es更新
  231. delete(oldTmp, "_id")
  232. if _, err := EsConn.Update().Index(Config["elasticsearch_index"]).Type(Config["elasticsearch_type"]).Id(esId).Doc(oldTmp).Refresh(true).Do(); err != nil {
  233. log.Println("存量 EsConn err :", err)
  234. }
  235. //最后删除redis
  236. conn.Del(redisId)
  237. }
  238. }
  239. }
  240. log.Println("存量历史合并执行完成 ok", gtid, lteid)
  241. } else {
  242. //增量处理
  243. overid := gtid
  244. tmp := map[string]interface{}{}
  245. for cursor.Next(&tmp) {
  246. overid = tmp["_id"].(primitive.ObjectID).Hex()
  247. log.Println(tmp["_id"])
  248. if tmp["winner"] == nil || tmp["winner"] == "" {
  249. continue
  250. }
  251. //redis查询是否存在
  252. rdb := RedisPool.Get()
  253. rdb.Do("SELECT", Config["redis_winner_db"])
  254. if reply, err := redis.String(rdb.Do("GET", tmp["winner"])); err != nil {
  255. //redis不存在存到临时表,定时任务处理
  256. FClient.DbName = Config["mgodb_extract_kf"]
  257. if tmpid := FClient.Save("winner_new", tmp); tmpid == nil {
  258. log.Println("FClient.Save err", tmpid)
  259. }
  260. //log.Println("get redis id err:定时任务处理", err, tmp)
  261. if err := rdb.Close(); err != nil {
  262. log.Println(err)
  263. }
  264. continue
  265. } else {
  266. if err := rdb.Close(); err != nil {
  267. log.Println(err)
  268. }
  269. //拿到合并后的qyk
  270. FClient.DbName = Config["mgodb_extract_kf"]
  271. oldTmp := FClient.FindById(Config["mgo_qyk_c"], reply)
  272. if oldTmp == nil {
  273. log.Println("redis id 不存在")
  274. continue
  275. }
  276. //比较合并
  277. //行业类型
  278. tmpTopscopeclass := []string{}
  279. tmpTopscopeclassMap := make(map[string]bool)
  280. if oldTmp["industry"] == nil {
  281. if v, ok := tmp["topscopeclass"].(primitive.A); ok {
  282. for _, vv := range v {
  283. if vvv, ok := vv.(string); ok && len(vvv) > 1 {
  284. tmpTopscopeclassMap[vvv[:len(vvv)-1]] = true
  285. }
  286. }
  287. for k := range tmpTopscopeclassMap {
  288. tmpTopscopeclass = append(tmpTopscopeclass, k)
  289. }
  290. }
  291. } else {
  292. if v, ok := oldTmp["industry"].(primitive.A); ok {
  293. for _, vv := range v {
  294. if vvv, ok := vv.(string); ok {
  295. tmpTopscopeclassMap[vvv] = true
  296. }
  297. }
  298. }
  299. if v, ok := tmp["topscopeclass"].(primitive.A); ok {
  300. for _, vv := range v {
  301. if vvv, ok := vv.(string); ok && len(vvv) > 1 {
  302. tmpTopscopeclassMap[vvv[:len(vvv)-1]] = true
  303. }
  304. }
  305. for k := range tmpTopscopeclassMap {
  306. tmpTopscopeclass = append(tmpTopscopeclass, k)
  307. }
  308. }
  309. }
  310. sort.Strings(tmpTopscopeclass)
  311. oldTmp["industry"] = tmpTopscopeclass
  312. esId := oldTmp["_id"].(primitive.ObjectID).Hex()
  313. //更新行业类型
  314. if tmp["winnerperson"] == nil || tmp["winnerperson"] == "" || Reg_xing.MatchString(util.ObjToString(tmp["winnerperson"])) {
  315. oldTmp["updatatime"] = time.Now().Unix()
  316. //mongo更新
  317. FClient.DbName = Config["mgodb_extract_kf"]
  318. if !FClient.UpdateById(Config["mgo_qyk_c"], esId, bson.M{"$set": oldTmp}) {
  319. log.Println("mongo更新err", esId)
  320. }
  321. //es更新
  322. delete(oldTmp, "_id")
  323. if _, err := EsConn.Update().Index(Config["elasticsearch_index"]).Type(Config["elasticsearch_type"]).Id(esId).Doc(oldTmp).Refresh(true).Do(); err != nil {
  324. log.Println("update es err:", err)
  325. }
  326. continue
  327. }
  328. //联系方式合并
  329. var tmpperson, winnertel string
  330. tmpperson = tmp["winnerperson"].(string)
  331. if tmp["winnertel"] == nil || tmp["winnertel"] == "" {
  332. winnertel = ""
  333. } else {
  334. if Reg_xing.MatchString(util.ObjToString(tmp["winnertel"])) || !Reg_tel.MatchString(util.ObjToString(tmp["winnertel"])) {
  335. winnertel = ""
  336. } else {
  337. winnertel = util.ObjToString(tmp["winnertel"])
  338. }
  339. }
  340. contactMaps := make([]interface{}, 0)
  341. if oldTmp["contact"] == nil {
  342. tmpContact := make(map[string]interface{})
  343. tmpContact["infoid"] = overid
  344. tmpContact["contact_person"] = tmpperson
  345. tmpContact["contact_type"] = "项目联系人"
  346. tmpContact["phone"] = winnertel
  347. tmpContact["topscopeclass"] = strings.Join(tmpTopscopeclass, ";")
  348. tmpContact["updatetime"] = time.Now().Unix()
  349. contactMaps = append(contactMaps, tmpContact)
  350. } else {
  351. //对比前四项,相等丢弃
  352. /*if v, ok := oldTmp["contact"].(primitive.A); ok {
  353. var isNotUpdate bool
  354. for _, vv := range v {
  355. if vvv, ok := vv.(map[string]interface{}); ok {
  356. if vvv["contact_person"] == tmpperson && vvv["contact_type"] == "项目联系人" &&
  357. vvv["phone"] == winnertel && vvv["topscopeclass"] == strings.Join(tmpTopscopeclass, ";") {
  358. isNotUpdate = true
  359. vvv["updatetime"] = time.Now().Unix()
  360. }
  361. contactMaps = append(contactMaps, vvv)
  362. }
  363. }
  364. if !isNotUpdate {
  365. vvv := make(map[string]interface{})
  366. vvv["infoid"] = overid
  367. vvv["contact_person"] = tmp["winnerperson"]
  368. vvv["contact_type"] = "项目联系人"
  369. vvv["phone"] = winnertel
  370. vvv["topscopeclass"] = strings.Join(tmpTopscopeclass, ";")
  371. vvv["updatetime"] = time.Now().Unix()
  372. contactMaps = append(contactMaps, vvv)
  373. }
  374. }*/
  375. //直接添加联系人,不再判断
  376. if v, ok := oldTmp["contact"].(primitive.A); ok {
  377. contactMaps = append(contactMaps, v...)
  378. }
  379. vvv := make(map[string]interface{})
  380. vvv["infoid"] = overid
  381. vvv["contact_person"] = tmp["winnerperson"]
  382. vvv["contact_type"] = "项目联系人"
  383. vvv["phone"] = winnertel
  384. vvv["topscopeclass"] = strings.Join(tmpTopscopeclass, ";")
  385. vvv["updatetime"] = time.Now().Unix()
  386. contactMaps = append(contactMaps, vvv)
  387. }
  388. oldTmp["contact"] = contactMaps
  389. //mongo更新
  390. oldTmp["updatatime"] = time.Now().Unix()
  391. FClient.DbName = Config["mgodb_extract_kf"]
  392. if !FClient.UpdateById(Config["mgo_qyk_c"], esId, bson.M{"$set": oldTmp}) {
  393. log.Println("mongo更新 err", esId, oldTmp)
  394. }
  395. //es更新
  396. delete(oldTmp, "_id")
  397. if _, err := EsConn.Update().Index(Config["elasticsearch_index"]).Type(Config["elasticsearch_type"]).Id(esId).Doc(oldTmp).Refresh(true).Do(); err != nil {
  398. log.Println("EsConn err :", err)
  399. }
  400. }
  401. }
  402. log.Println("增量合并执行完成 ok", gtid, lteid, overid)
  403. }
  404. }
  405. //定时任务 新增
  406. //1.存异常表
  407. //2.合并原始库新增
  408. func TimedTaskWinner() {
  409. //time.Sleep(time.Hour*70)
  410. t2 := time.NewTimer(time.Second * 5)
  411. for range t2.C {
  412. Fcconn := FClient.GetMgoConn()
  413. defer FClient.DestoryMongoConn(Fcconn)
  414. tmpLast := map[string]interface{}{}
  415. if iter := Fcconn.DB(Config["mgodb_extract_kf"]).C("winner_new").Find(bson.M{}).Sort("-_id").Limit(1).Iter(); iter != nil {
  416. if !iter.Next(&tmpLast) {
  417. //临时表无数据
  418. log.Println("临时表无数据:")
  419. t2.Reset(time.Minute * 5)
  420. continue
  421. } else {
  422. log.Println("临时表有数据:", tmpLast)
  423. fconn := FClient.GetMgoConn()
  424. defer FClient.DestoryMongoConn(fconn)
  425. cursor := fconn.DB(Config["mgodb_extract_kf"]).C("winner_new").Find(bson.M{
  426. "_id": bson.M{
  427. "$lte": tmpLast["_id"],
  428. },
  429. }).Sort("_id").Iter()
  430. if cursor == nil {
  431. log.Println("查询失败")
  432. t2.Reset(time.Second * 5)
  433. continue
  434. }
  435. //遍历临时表数据,匹配不到原始库存入异常表
  436. tmp := make(map[string]interface{})
  437. for cursor.Next(&tmp) {
  438. tmpId := tmp["_id"].(primitive.ObjectID).Hex()
  439. //再重新查找redis,存在发udp处理,不存在走新增合并
  440. rdb := RedisPool.Get()
  441. rdb.Do("SELECT", Config["redis_winner_db"])
  442. if _, err := redis.String(rdb.Do("GET", tmp["winner"])); err == nil {
  443. //{"gtid":"57d7ad2f61a0721f152d2ad5","lteid":"5e20968d85a9271abf0ad6c2","stype":""}
  444. //redis存在发送udp进行处理
  445. by, _ := json.Marshal(map[string]interface{}{
  446. "gtid": tmpId,
  447. "lteid": tmpId,
  448. "stype": "",
  449. "data_type": "winner",
  450. "data_info": "add",
  451. })
  452. if e := udpclient.WriteUdp(by, mu.OP_TYPE_DATA, &net.UDPAddr{
  453. IP: net.ParseIP("127.0.0.1"),
  454. Port: Updport,
  455. }); e != nil {
  456. log.Println(e)
  457. }
  458. //存在的话删除tmp mongo表
  459. FClient.DbName = Config["mgodb_extract_kf"]
  460. if DeletedCount := FClient.DeleteById("winner_new", tmpId); DeletedCount == 0 {
  461. log.Println("删除临时表err:", DeletedCount)
  462. }
  463. if err := rdb.Close(); err != nil {
  464. log.Println(err)
  465. }
  466. continue
  467. } else {
  468. if err = rdb.Close(); err != nil {
  469. log.Println(err)
  470. }
  471. }
  472. //查询redis不存在新增
  473. FClient.DbName = Config["mgodb_enterprise"]
  474. resulttmp := FClient.FindOne(Config["mgodb_enterprise_c"], bson.M{"company_name": tmp["winner"]})
  475. if resulttmp["_id"] == nil {
  476. //log.Println(r)
  477. //匹配不到原始库,存入异常表删除临时表
  478. FClient.DbName = Config["mgodb_extract_kf"]
  479. if saveid := FClient.Save("winner_err", tmp); saveid == nil {
  480. log.Println("存入异常表错误", tmp)
  481. }
  482. FClient.DbName = Config["mgodb_extract_kf"]
  483. if deleteNum := FClient.DeleteById("winner_new", tmpId); deleteNum == 0 {
  484. log.Println("删除临时表错误", deleteNum)
  485. }
  486. continue
  487. } else {
  488. //log.Println(123)
  489. //匹配到原始库,新增 resulttmp
  490. if resulttmp["credit_no"] != nil {
  491. if credit_no, ok := resulttmp["credit_no"].(string); ok && strings.TrimSpace(credit_no) != "" &&
  492. len(strings.TrimSpace(credit_no)) > 8 {
  493. dataNo := strings.TrimSpace(credit_no)[2:8]
  494. if Addrs[dataNo] != nil {
  495. if v, ok := Addrs[dataNo].(map[string]interface{}); ok {
  496. if resulttmp["province"] == nil || resulttmp["province"] == "" {
  497. resulttmp["province"] = v["province"]
  498. }
  499. resulttmp["city"] = v["city"]
  500. resulttmp["district"] = v["district"]
  501. }
  502. }
  503. }
  504. }
  505. contacts := make([]map[string]interface{}, 0)
  506. contact := make(map[string]interface{}, 0)
  507. if resulttmp["legal_person"] != nil {
  508. contact["contact_person"] = resulttmp["legal_person"] //联系人
  509. } else {
  510. contact["contact_person"] = "" //联系人
  511. }
  512. contact["contact_type"] = "法定代表人" //法定代表人
  513. //log.Println(1)
  514. if resulttmp["annual_reports"] != nil {
  515. bytes, err := json.Marshal(resulttmp["annual_reports"])
  516. if err != nil {
  517. log.Println("annual_reports err:", err)
  518. }
  519. phonetmp := make([]map[string]interface{}, 0)
  520. err = json.Unmarshal(bytes, &phonetmp)
  521. if err != nil {
  522. log.Println("Unmarshal err:", err)
  523. }
  524. for _, vv := range phonetmp {
  525. if vv["company_phone"] != nil {
  526. if vv["company_phone"] == "" {
  527. continue
  528. } else {
  529. contact["phone"] = vv["company_phone"] //联系电话
  530. break
  531. }
  532. } else {
  533. contact["phone"] = "" //联系电话
  534. }
  535. }
  536. }
  537. //log.Println(k, contact["phone"], resulttmp["_id"])
  538. //time.Sleep(10 * time.Second)
  539. if contact["phone"] == nil {
  540. contact["phone"] = "" //联系电话
  541. }
  542. contact["topscopeclass"] = "企业公示" //项目类型
  543. contact["updatetime"] = time.Now().Unix() //更新时间
  544. contact["infoid"] = "" //招标信息id
  545. contacts = append(contacts, contact)
  546. resulttmp["contact"] = contacts
  547. savetmp := make(map[string]interface{}, 0)
  548. for _, sk := range Fields {
  549. if sk == "establish_date" {
  550. if resulttmp[sk] != nil {
  551. savetmp[sk] = resulttmp[sk].(primitive.DateTime).Time().UTC().Unix()
  552. continue
  553. }
  554. } else if sk == "capital" {
  555. //log.Println(sk, resulttmp[sk])
  556. savetmp[sk] = ObjToMoney([]interface{}{resulttmp[sk], ""})[0]
  557. continue
  558. } else if sk == "partners" {
  559. //log.Println(sk, resulttmp[sk], )
  560. //fmt.Println(reflect.TypeOf(resulttmp[sk]))
  561. if resulttmp[sk] != nil {
  562. if ppms, ok := resulttmp[sk].(primitive.A); ok {
  563. for i, _ := range ppms {
  564. if ppms[i].(map[string]interface{})["stock_type"] != nil {
  565. ppms[i].(map[string]interface{})["stock_type"] = "企业公示"
  566. }
  567. delete(ppms[i].(map[string]interface{}), "identify_type")
  568. }
  569. savetmp[sk] = ppms
  570. }
  571. } else {
  572. savetmp[sk] = []interface{}{}
  573. }
  574. continue
  575. } else if sk == "_id" {
  576. savetmp["tmp"+sk] = resulttmp[sk]
  577. continue
  578. } else if sk == "area_code" {
  579. //行政区划代码
  580. savetmp[sk] = fmt.Sprint(resulttmp[sk])
  581. continue
  582. } else if sk == "report_websites" {
  583. //网址
  584. if resulttmp["report_websites"] == nil {
  585. savetmp["website"] = ""
  586. } else {
  587. report_websitesArr := []string{}
  588. if ppms, ok := resulttmp[sk].(primitive.A); ok {
  589. for _, v := range ppms {
  590. if vvv, ok := v.(map[string]interface{}); ok {
  591. if rv, ok := vvv["website_url"].(string); ok {
  592. report_websitesArr = append(report_websitesArr, rv)
  593. }
  594. }
  595. }
  596. }
  597. sort.Strings(report_websitesArr)
  598. savetmp["website"] = strings.Join(report_websitesArr, ";")
  599. }
  600. continue
  601. } else if sk == "wechat_accounts" {
  602. savetmp[sk] = []interface{}{}
  603. continue
  604. }
  605. if resulttmp[sk] == nil && sk != "history_name" && sk != "wechat_accounts" && sk != "establish_date" && sk != "capital" && sk != "partners" && sk != "contact" && sk != "report_websites" {
  606. savetmp[sk] = ""
  607. } else {
  608. savetmp[sk] = resulttmp[sk]
  609. }
  610. }
  611. //tmps = append(tmps, savetmp)
  612. savetmp["updatatime"] = time.Now().Unix()
  613. //保存mongo
  614. FClient.DbName = Config["mgodb_extract_kf"]
  615. saveid := FClient.Save(Config["mgo_qyk_c"], savetmp)
  616. if saveid != nil {
  617. //保存redis
  618. rc := RedisPool.Get()
  619. rc.Do("SELECT", Config["redis_winner_db"])
  620. var _id string
  621. if v, ok := saveid.(primitive.ObjectID); ok {
  622. _id = v.Hex()
  623. }
  624. if _, err := rc.Do("SET", savetmp["company_name"], _id); err != nil {
  625. log.Println("save redis err:", tmp["_id"], savetmp["_id"], savetmp["company_name"], err)
  626. if err := rc.Close(); err != nil {
  627. log.Println(err)
  628. }
  629. } else {
  630. //保存es
  631. delete(savetmp, "_id")
  632. if err := rc.Close(); err != nil {
  633. log.Println(err)
  634. }
  635. //esConn := elastic.GetEsConn()
  636. //defer elastic.DestoryEsConn(esConn)
  637. if _, err := EsConn.Index().Index(Config["elasticsearch_index"]).Type(Config["elasticsearch_type"]).Id(_id).BodyJson(savetmp).Refresh(true).Do(); err != nil {
  638. log.Println("save es err :", tmp["_id"], savetmp["_id"], err)
  639. } else {
  640. //删除临时表
  641. FClient.DbName = Config["mgodb_extract_kf"]
  642. if deleteNum := FClient.DeleteById("winner_new", tmpId); deleteNum == 0 {
  643. log.Println("删除临时表失败", deleteNum)
  644. }
  645. }
  646. }
  647. } else {
  648. log.Println("save mongo err:", saveid, tmp["_id"])
  649. }
  650. }
  651. }
  652. }
  653. }
  654. t2.Reset(time.Minute)
  655. }
  656. }