main.go 8.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
  1. package main
  2. import (
  3. "app.yhyue.com/moapp/jybase/common"
  4. elastic "app.yhyue.com/moapp/jybase/es"
  5. "app.yhyue.com/moapp/jybase/mongodb"
  6. "app.yhyue.com/moapp/jybase/mysql"
  7. "app.yhyue.com/moapp/jybase/redis"
  8. "bp.jydev.jianyu360.cn/BaseService/pushpkg/p"
  9. "flag"
  10. "fmt"
  11. "github.com/robfig/cron"
  12. "log"
  13. )
  14. var (
  15. cfg = new(Config)
  16. db = new(DB)
  17. Mysql *mysql.Mysql
  18. TiDb *mysql.Mysql
  19. TiDbData *mysql.Mysql
  20. TiDbPower *mysql.Mysql
  21. WorkOrder *mysql.Mysql
  22. ThirdParty *mysql.Mysql
  23. Jyactivities *mysql.Mysql
  24. debrisProductMysql *mysql.Mysql
  25. Base *mysql.Mysql
  26. BiService *mysql.Mysql
  27. DataAnalysisService *mysql.Mysql
  28. Es elastic.Es
  29. Mgo *mongodb.MongodbSim
  30. MgoLog *mongodb.MongodbSim
  31. MgoQyxy *mongodb.MongodbSim
  32. mode = flag.Int("m", 1, "")
  33. startTime, endTime = "00:00", "23:59"
  34. )
  35. func main() {
  36. flag.Parse()
  37. common.ReadConfig(&cfg)
  38. log.Println("cfg ", cfg)
  39. common.ReadConfig("./db.json", &db)
  40. log.Println("db ", db)
  41. TiDb = &mysql.Mysql{
  42. Address: db.TiDb.Host + ":" + fmt.Sprint(db.TiDb.Port),
  43. UserName: db.TiDb.User,
  44. PassWord: db.TiDb.Password,
  45. DBName: db.TiDb.Database,
  46. MaxOpenConns: db.TiDb.PollSize,
  47. MaxIdleConns: db.TiDb.MaxIdle,
  48. }
  49. TiDb.Init()
  50. TiDbData = &mysql.Mysql{
  51. Address: db.TiDbData.Host + ":" + fmt.Sprint(db.TiDbData.Port),
  52. UserName: db.TiDbData.User,
  53. PassWord: db.TiDbData.Password,
  54. DBName: db.TiDbData.Database,
  55. MaxOpenConns: db.TiDbData.PollSize,
  56. MaxIdleConns: db.TiDbData.MaxIdle,
  57. }
  58. TiDbData.Init()
  59. TiDbPower = &mysql.Mysql{
  60. Address: db.TiDbPower.Host + ":" + fmt.Sprint(db.TiDbPower.Port),
  61. UserName: db.TiDbPower.User,
  62. PassWord: db.TiDbPower.Password,
  63. DBName: db.TiDbPower.Database,
  64. MaxOpenConns: db.TiDbPower.PollSize,
  65. MaxIdleConns: db.TiDbPower.MaxIdle,
  66. }
  67. TiDbPower.Init()
  68. ThirdParty = &mysql.Mysql{
  69. Address: db.ThirdParty.Host + ":" + fmt.Sprint(db.ThirdParty.Port),
  70. UserName: db.ThirdParty.User,
  71. PassWord: db.ThirdParty.Password,
  72. DBName: db.ThirdParty.Database,
  73. MaxOpenConns: db.ThirdParty.PollSize,
  74. MaxIdleConns: db.ThirdParty.MaxIdle,
  75. }
  76. ThirdParty.Init()
  77. Jyactivities = &mysql.Mysql{
  78. Address: db.Jyactivities.Host + ":" + fmt.Sprint(db.Jyactivities.Port),
  79. UserName: db.Jyactivities.User,
  80. PassWord: db.Jyactivities.Password,
  81. DBName: db.Jyactivities.Database,
  82. MaxOpenConns: db.Jyactivities.PollSize,
  83. MaxIdleConns: db.Jyactivities.MaxIdle,
  84. }
  85. Jyactivities.Init()
  86. debrisProductMysql = &mysql.Mysql{
  87. Address: db.DebrisProduct.Host + ":" + fmt.Sprint(db.DebrisProduct.Port),
  88. UserName: db.DebrisProduct.User,
  89. PassWord: db.DebrisProduct.Password,
  90. DBName: db.DebrisProduct.Database,
  91. }
  92. debrisProductMysql.Init()
  93. WorkOrder = &mysql.Mysql{
  94. Address: db.WorderOrder.Host + ":" + fmt.Sprint(db.WorderOrder.Port),
  95. UserName: db.WorderOrder.User,
  96. PassWord: db.WorderOrder.Password,
  97. DBName: db.WorderOrder.Database,
  98. MaxOpenConns: db.WorderOrder.PollSize,
  99. MaxIdleConns: db.WorderOrder.MaxIdle,
  100. }
  101. WorkOrder.Init()
  102. Mysql = &mysql.Mysql{
  103. Address: db.Mysql.Host + ":" + fmt.Sprint(db.Mysql.Port),
  104. UserName: db.Mysql.User,
  105. PassWord: db.Mysql.Password,
  106. DBName: db.Mysql.Database,
  107. MaxOpenConns: db.Mysql.PollSize,
  108. MaxIdleConns: db.Mysql.MaxIdle,
  109. }
  110. Mysql.Init()
  111. Base = &mysql.Mysql{
  112. Address: db.BaseService.Host + ":" + fmt.Sprint(db.BaseService.Port),
  113. UserName: db.BaseService.User,
  114. PassWord: db.BaseService.Password,
  115. DBName: db.BaseService.Database,
  116. MaxOpenConns: db.BaseService.PollSize,
  117. MaxIdleConns: db.BaseService.MaxIdle,
  118. }
  119. Base.Init()
  120. BiService = &mysql.Mysql{
  121. Address: db.BiService.Host + ":" + fmt.Sprint(db.BiService.Port),
  122. UserName: db.BiService.User,
  123. PassWord: db.BiService.Password,
  124. DBName: db.BiService.Database,
  125. MaxOpenConns: db.BiService.PollSize,
  126. MaxIdleConns: db.BiService.MaxIdle,
  127. }
  128. BiService.Init()
  129. DataAnalysisService = &mysql.Mysql{
  130. Address: db.DataAnalysis.Host + ":" + fmt.Sprint(db.DataAnalysis.Port),
  131. UserName: db.DataAnalysis.User,
  132. PassWord: db.DataAnalysis.Password,
  133. DBName: db.DataAnalysis.Database,
  134. MaxOpenConns: db.DataAnalysis.PollSize,
  135. MaxIdleConns: db.DataAnalysis.MaxIdle,
  136. }
  137. DataAnalysisService.Init()
  138. Es = elastic.NewEs(db.Es.Version, db.Es.Address, db.Es.DbSize, db.Es.UserName, db.Es.Password)
  139. redis.InitRedis(db.RedisServer)
  140. Mgo = mongodb.NewMgo(db.Mgo.Address, db.Mgo.DbName, db.Mgo.DbSize)
  141. MgoLog = mongodb.NewMgoWithUser(db.MgoLog.Address, db.MgoLog.DbName, db.MgoLog.User, db.MgoLog.Password, db.MgoLog.DbSize)
  142. MgoQyxy = mongodb.NewMgoWithUser(db.MgoQyxy.Address, db.MgoQyxy.DbName, db.MgoQyxy.User, db.MgoQyxy.Password, db.MgoQyxy.DbSize)
  143. InitArea()
  144. InitProduct(db.ProductArr)
  145. //kcSync()
  146. //refundAuto() //客成移交销售
  147. if *mode == 1 {
  148. go ordersClue() //后台订单进线索
  149. //一秒钟一次
  150. go p.VarTimeTask.RunInSecondTimeLoop("1秒钟定时任务", "", "", db.CornExp8, true, true, nil, func() {
  151. LockHandle() //绑定状态处理
  152. })
  153. //30分钟一次
  154. go p.VarTimeTask.RunInTimeLoop("30分钟定时任务", "", "", db.CornExp1, true, true, nil, func() {
  155. orders() //未支付订单
  156. messageSync() //聊天记录进线索
  157. userbase() //新绑定手机号进线索
  158. })
  159. //5分钟一次
  160. go p.VarTimeTask.RunInTimeSection("5分钟定时任务1", db.CornExp2Start, db.CornExp2End, db.CornExp2, func(dayFirst bool) {
  161. users() //新注册用户进线索
  162. saleLeads() //留资进线索
  163. eventReg() //渠道
  164. bigCustomer() //大客户线索
  165. MarketCustomer() //市场部线索
  166. AdvisoryCommittee() //咨询组线索
  167. SelectionDepartment() //运营部线索
  168. rebind() //取关重新关注处理
  169. activeUsers() //活跃用户处理
  170. inviteUser() //邀请用户
  171. })
  172. //每天8点30
  173. c := cron.New()
  174. c.AddFunc(db.CornExp3, func() {
  175. users() //新注册用户进线索
  176. saleLeads() //留资进线索
  177. activeUsers() //活跃用户处理
  178. })
  179. c.Start()
  180. // 一天一次
  181. // tagAllSync()
  182. d := cron.New()
  183. d.AddFunc(db.CornExp4, func() {
  184. tagAllSync() //用户标签
  185. })
  186. d.Start()
  187. // 5分钟一次
  188. go p.VarTimeTask.RunInTimeLoop("5分钟定时任务2", "", "", db.CornExp5, true, true, nil, func() {
  189. //go p.VarTimeTask.RunInTimeSection("5分钟定时任务2", startTime, endTime, db.CornExp5, func(dayFirst bool) {
  190. everythingSync() //渠道
  191. ordersClue() //后台订单进线索
  192. kcSync() //移交客成
  193. kcAuto() //客成自动加入任务车
  194. tagAddSync() //用户标签增量
  195. autoExitSea() //自动退海
  196. Thaw() //自动解冻处理
  197. subscribeAddSync() //订阅增量
  198. })
  199. //5分钟一次
  200. go p.VarTimeTask.RunInTimeLoop("5分钟定时任务3", "", "", db.CornExp6, true, true, nil, func() {
  201. //go p.VarTimeTask.RunInTimeSection("5分钟定时任务3", startTime, endTime, db.CornExp6, func(dayFirst bool) {
  202. subscribeAddSync() //订阅增量
  203. rderAcceptance() //工单生成
  204. })
  205. //自动进入任务车 1天一次
  206. g := cron.New()
  207. g.AddFunc(db.CornExp7, func() {
  208. readClue() //读取中间表进线索
  209. refundAuto() //客成移交销售
  210. autoTask() //超时未跟进加入任务车
  211. autoTasks() //按照下次跟进时间提前一天加入任务车
  212. ClueToDxTask()
  213. })
  214. g.Start()
  215. log.Println("个人邮件告警查询开始")
  216. WarningPerl() // 个人
  217. log.Println("部门邮件告警查询开始")
  218. WarningDm() // 部门
  219. log.Println("个人/部门邮件告警查询结束")
  220. h := cron.New()
  221. h.AddFunc(db.CornMail, func() {
  222. log.Println("个人邮件告警查询开始")
  223. WarningPerl() // 个人
  224. log.Println("部门邮件告警查询开始")
  225. WarningDm() // 部门
  226. log.Println("个人/部门邮件告警查询结束")
  227. UpperLimitAutoExitSea(db.AllocationCap)
  228. })
  229. h.Start()
  230. select {}
  231. } else {
  232. users()
  233. saleLeads()
  234. }
  235. }