main.go 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  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. UserName: util.ObjToString((*qyxy)["uname"]),
  31. Password: util.ObjToString((*qyxy)["upwd"]),
  32. }
  33. MgoQy.InitPool()
  34. CollJy = util.ObjToString((*focusQy)["dbColl"])
  35. CollQy = util.ObjToString((*qyxy)["dbCollQy"])
  36. CollSave = util.ObjToString((*qyxy)["dbColl"])
  37. LastTime = util.Int64All(Sysconfig["lasttime"])
  38. TaskTime = util.Int64All(Sysconfig["tasktime"])
  39. ChangeMap = util.ObjArrToMapArr(Sysconfig["changeType"].([]interface{}))
  40. initChangeMap()
  41. }
  42. func initChangeMap() {
  43. for _, v := range ChangeMap {
  44. list := v["change_keyword"].([]interface{})
  45. var regList []string
  46. if len(list) > 0 {
  47. for _, v1 := range list {
  48. reg := ".*" + util.ObjToString(v1) + ".*"
  49. regList = append(regList, reg)
  50. }
  51. v["change_key_reg"] = regList
  52. } else {
  53. v["change_key_reg"] = []string{".*"}
  54. }
  55. }
  56. }
  57. func main() {
  58. go SaveData()
  59. go TimeTask()
  60. ch := make(chan bool, 1)
  61. <-ch
  62. }