main.go 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. package main
  2. import (
  3. "mongodb"
  4. "qfw/util"
  5. )
  6. var (
  7. Sysconfig map[string]interface{}
  8. MgoJy, MgoQy *mongodb.MongodbSim
  9. JyDb string
  10. CollJy, CollQy, CollSave string
  11. LastTime int64
  12. TaskTime int64
  13. ChangeMap []map[string]interface{}
  14. )
  15. func init() {
  16. util.ReadConfig(&Sysconfig)
  17. focusQy := util.ObjToMap(Sysconfig["qy_focus"])
  18. JyDb = util.ObjToString((*focusQy)["dbName"])
  19. MgoJy = &mongodb.MongodbSim{
  20. MongodbAddr: util.ObjToString((*focusQy)["dbServer"]),
  21. Size: util.IntAll((*focusQy)["dbSize"]),
  22. DbName: JyDb,
  23. }
  24. MgoJy.InitPool()
  25. qyxy := util.ObjToMap(Sysconfig["qy_change"])
  26. MgoQy = &mongodb.MongodbSim{
  27. MongodbAddr: util.ObjToString((*qyxy)["dbServer"]),
  28. Size: util.IntAll((*qyxy)["dbSize"]),
  29. DbName: util.ObjToString((*qyxy)["dbName"]),
  30. }
  31. MgoQy.InitPool()
  32. CollJy = util.ObjToString((*focusQy)["dbColl"])
  33. CollQy = util.ObjToString((*qyxy)["dbCollQy"])
  34. CollSave = util.ObjToString((*qyxy)["dbColl"])
  35. LastTime = util.Int64All(Sysconfig["lasttime"])
  36. TaskTime = util.Int64All(Sysconfig["tasktime"])
  37. ChangeMap = util.ObjArrToMapArr(Sysconfig["changeType"].([]interface{}))
  38. initChangeMap()
  39. }
  40. func initChangeMap() {
  41. for _, v := range ChangeMap {
  42. list := v["change_keyword"].([]interface{})
  43. var regList []string
  44. if len(list) > 0 {
  45. for _, v1 := range list {
  46. reg := ".*" + util.ObjToString(v1) + ".*"
  47. regList = append(regList, reg)
  48. }
  49. v["change_key_reg"] = regList
  50. } else {
  51. v["change_key_reg"] = []string{".*"}
  52. }
  53. }
  54. }
  55. func main() {
  56. go SaveData()
  57. go TimeTask()
  58. ch := make(chan bool, 1)
  59. <-ch
  60. }