hlyj.go 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253
  1. package main
  2. import (
  3. "bytes"
  4. // "encoding/json"
  5. "io/ioutil"
  6. "log"
  7. "net/http"
  8. "strings"
  9. "sync"
  10. "time"
  11. "app.yhyue.com/moapp/jybase/date"
  12. "app.yhyue.com/moapp/jybase/common"
  13. "github.com/tealeg/xlsx"
  14. )
  15. func getToken() string {
  16. url := "https://a1.7x24cc.com/accessToken?account=N000000029739&appid=w4w2ex0bnt1n61or&secret=3c8f7dd04d2c11edb786132b38c4d48a"
  17. bs, err := doGet(url)
  18. if err != nil {
  19. log.Println("token生成失败", err)
  20. return ""
  21. }
  22. tokenMap := common.StringToMap(string(bs))
  23. if tokenMap["success"].(bool) {
  24. return common.ObjToString(tokenMap["accessToken"])
  25. }
  26. return ""
  27. }
  28. func hlyjSync() {
  29. token := getToken()
  30. count := 0
  31. if token != "" {
  32. TiDbData.SelectByBath(500, func(l *[]map[string]interface{}) bool {
  33. for _, v := range *l {
  34. count++
  35. log.Println("第", count, "条")
  36. unique_id := common.ObjToString(v["unique_id"])
  37. if count%20000 == 0 {
  38. token = getToken()
  39. }
  40. url := `https://a1.7x24cc.com/commonInte?flag=1008&account=N000000029739&accessToken=` + token + `&cusObj={"unique_id":"` + unique_id + `","belongTo":"直销组"}&dbType=0001`
  41. // updateData := map[string]interface{}{"belongTo": "直销组", "unique_id": unique_id}
  42. // update_data := map[string]interface{}{
  43. // "flag": "1008",
  44. // "account": "N000000029739",
  45. // "dbType": "0001",
  46. // "accessToken": token,
  47. // "cusObj": updateData,
  48. // }
  49. // update_data_byte, _ := json.Marshal(&update_data)
  50. bs, err := doGet(url)
  51. if err != nil {
  52. log.Println("调用接口失败", unique_id, err)
  53. }
  54. resMap := common.StringToMap(string(bs))
  55. if resMap["success"].(bool) {
  56. ok := TiDbData.Update("customer", map[string]interface{}{"unique_id": unique_id}, map[string]interface{}{"belongTo": "直销组"})
  57. if ok {
  58. log.Println("更新成功", unique_id)
  59. } else {
  60. log.Println("更新失败", unique_id)
  61. }
  62. } else {
  63. log.Println("调用接口失败!!", unique_id)
  64. }
  65. }
  66. return true
  67. }, `SELECT unique_id from customer where (empNo = "" or empNo is null) and (belongTo = "" or belongTo is null)`)
  68. }
  69. }
  70. func hlyjSync2() {
  71. token := getToken()
  72. nameG := map[string]string{}
  73. count := 0
  74. if token != "" {
  75. saleData := TiDb.SelectBySql("select * from jy_salesperson_info")
  76. if saleData != nil {
  77. for _, v := range *saleData {
  78. nameG[common.ObjToString(v["seatNumber"])] = common.ObjToString(v["name"]) + "," + common.ObjToString(v["groupname"])
  79. }
  80. }
  81. if len(nameG) > 0 {
  82. TiDbData.SelectByBath(500, func(l *[]map[string]interface{}) bool {
  83. for _, v := range *l {
  84. count++
  85. log.Println("第", count, "条")
  86. if count%20000 == 0 {
  87. token = getToken()
  88. }
  89. unique_id := common.ObjToString(v["unique_id"])
  90. empNo := common.ObjToString(v["empNo"])
  91. belongTo := "市场部"
  92. log.Println("nameG[empNo]", nameG[empNo])
  93. if strings.Contains(nameG[empNo], "电销") {
  94. belongTo = "直销组"
  95. } else if strings.Contains(nameG[empNo], "孙振杰") || strings.Contains(nameG[empNo], "陈兆标") {
  96. belongTo = "大客户"
  97. } else if nameG[empNo] == "" {
  98. belongTo = "直销组"
  99. }
  100. // updateData := map[string]interface{}{"belongTo": belongTo, "unique_id": unique_id}
  101. // update_data := map[string]interface{}{
  102. // "flag": "1008",
  103. // "account": "N000000029739",
  104. // "dbType": "0001",
  105. // "accessToken": token,
  106. // "cusObj": updateData,
  107. // }
  108. // update_data_byte, _ := json.Marshal(&updateData)
  109. url := `https://a1.7x24cc.com/commonInte?flag=1008&account=N000000029739&accessToken=` + token + `&cusObj={"unique_id":"` + unique_id + `","belongTo":"` + belongTo + `"}&dbType=0001`
  110. bs, err := doGet(url)
  111. if err != nil {
  112. log.Println("调用接口失败", unique_id, err)
  113. return false
  114. }
  115. resMap := common.StringToMap(string(bs))
  116. if resMap["success"] != nil && resMap["success"].(bool) {
  117. ok := TiDbData.Update("customer", map[string]interface{}{"unique_id": unique_id}, map[string]interface{}{"belongTo": belongTo})
  118. if ok {
  119. log.Println("更新成功", unique_id)
  120. } else {
  121. log.Println("更新失败", unique_id)
  122. }
  123. } else {
  124. log.Println("调用接口失败!!", unique_id)
  125. }
  126. }
  127. return true
  128. }, `SELECT unique_id,empNo from customer where (empNo != "" or empNo is not null) and (belongTo = "" or belongTo is null)`)
  129. }
  130. }
  131. }
  132. func hlyjSync3() {
  133. count := 0
  134. xlFile := xlsx.NewFile()
  135. sheet1, _ := xlFile.AddSheet("市场创建时间在前")
  136. sheet2, _ := xlFile.AddSheet("电销创建时间在前")
  137. l := TiDbData.SelectBySql(`SELECT phone,createTime from customer where belongTo = "市场部"`)
  138. // TiDbData.SelectByBath(100, func(l *[]map[string]interface{}) bool {
  139. wg := new(sync.WaitGroup)
  140. ch := make(chan bool, 20)
  141. for _, v := range *l {
  142. count++
  143. log.Println("第", count, "条")
  144. wg.Add(1)
  145. ch <- true
  146. go func(v map[string]interface{}) {
  147. defer func() {
  148. wg.Done()
  149. <-ch
  150. }()
  151. // unique_id := common.ObjToString(v["unique_id"])
  152. // empNo := common.ObjToString(v["empNo"])
  153. // belongTo := common.ObjToString(v["belongTo"])
  154. phone := common.ObjToString(v["phone"])
  155. createTime := common.ObjToString(v["createTime"])
  156. clueData := TiDb.FindOne("dwd_f_crm_clue_info", map[string]interface{}{"phone": phone}, "createtime", "")
  157. if clueData != nil {
  158. clueCreateTime := common.ObjToString((*clueData)["createtime"])
  159. createTimes1, _ := time.ParseInLocation(date.Date_Full_Layout, createTime, time.Local)
  160. clueCreateTimes1, _ := time.ParseInLocation(date.Date_Full_Layout, clueCreateTime, time.Local)
  161. createTimes := createTimes1.Unix()
  162. clueCreateTimes := clueCreateTimes1.Unix()
  163. if clueCreateTimes >= createTimes {
  164. row := sheet1.AddRow()
  165. cluename := common.ObjToString((*clueData)["cluename"])
  166. trailstatus := common.ObjToString((*clueData)["trailstatus"])
  167. seatNumber := common.ObjToString((*clueData)["seatNumber"])
  168. row.AddCell().SetValue(cluename)
  169. row.AddCell().SetValue(phone)
  170. row.AddCell().SetValue(trailstatus)
  171. row.AddCell().SetValue(seatNumber)
  172. row.AddCell().SetValue(clueCreateTime)
  173. row.AddCell().SetValue(createTime)
  174. log.Println("存入市场", phone)
  175. } else {
  176. company := common.ObjToString(v["company"])
  177. status999 := common.ObjToString(v["status999"])
  178. empNo := common.ObjToString(v["empNo"])
  179. row := sheet2.AddRow()
  180. row.AddCell().SetValue(company)
  181. row.AddCell().SetValue(phone)
  182. row.AddCell().SetValue(status999)
  183. row.AddCell().SetValue(empNo)
  184. row.AddCell().SetValue(createTime)
  185. row.AddCell().SetValue(clueCreateTime)
  186. log.Println("存入电销", phone)
  187. }
  188. } else {
  189. row := sheet1.AddRow()
  190. row.AddCell().SetValue("")
  191. row.AddCell().SetValue(phone)
  192. row.AddCell().SetValue("")
  193. row.AddCell().SetValue("")
  194. row.AddCell().SetValue("")
  195. row.AddCell().SetValue(createTime)
  196. log.Println("无线索存入市场", phone)
  197. }
  198. }(v)
  199. }
  200. wg.Wait()
  201. // return true
  202. // }, `SELECT phone,createTime from customer where belongTo = "市场部"`)
  203. xlFile.Save("./线索统计.xlsx")
  204. }
  205. func doGet(url string) ([]byte, error) {
  206. req, err := http.NewRequest("GET", url, nil)
  207. if err != nil {
  208. return nil, err
  209. }
  210. resp, err := http.DefaultClient.Do(req)
  211. if err != nil {
  212. return nil, err
  213. }
  214. bs, err := ioutil.ReadAll(resp.Body)
  215. if err != nil {
  216. return nil, err
  217. }
  218. defer func() {
  219. _ = resp.Body.Close()
  220. }()
  221. log.Println(url, "调用结果 ", string(bs))
  222. return bs, nil
  223. }
  224. func doPost(url string, body []byte) ([]byte, error) {
  225. req, err := http.NewRequest("POST", url, bytes.NewReader(body))
  226. if err != nil {
  227. return nil, err
  228. }
  229. req.Header.Add("Content-Type", "application/json;charset=utf-8")
  230. resp, err := http.DefaultClient.Do(req)
  231. if err != nil {
  232. return nil, err
  233. }
  234. bs, err := ioutil.ReadAll(resp.Body)
  235. if err != nil {
  236. return nil, err
  237. }
  238. defer func() {
  239. _ = resp.Body.Close()
  240. }()
  241. log.Println(url, "调用结果 ", string(bs))
  242. return bs, nil
  243. }