main.go 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. package main
  2. import (
  3. "mongodb"
  4. "qfw/util"
  5. "regexp"
  6. )
  7. var (
  8. Sysconfig map[string]interface{}
  9. MgoMix, Mgo *mongodb.MongodbSim
  10. dbname string
  11. CollQy, CollSave string
  12. lastId int64
  13. ChangeMap []map[string]interface{}
  14. timeReg, _ = regexp.Compile(`^[\d]{4}-[\d]{1,2}-[\d]{1,2}`)
  15. )
  16. func init() {
  17. util.ReadConfig(&Sysconfig)
  18. dbname := util.ObjToString(Sysconfig["dbName"])
  19. MgoMix = &mongodb.MongodbSim{
  20. MongodbAddr: util.ObjToString(Sysconfig["dbServer"]),
  21. Size: util.IntAll(Sysconfig["dbSize"]),
  22. DbName: dbname,
  23. UserName: util.ObjToString(Sysconfig["uname"]),
  24. Password: util.ObjToString(Sysconfig["upwd"]),
  25. }
  26. MgoMix.InitPool()
  27. Mgo = &mongodb.MongodbSim{
  28. MongodbAddr: "172.17.4.181:27001", // 172.17.4.181:27001
  29. Size: 10,
  30. DbName: "mixdata",
  31. }
  32. Mgo.InitPool()
  33. CollQy = Sysconfig["coll_qy"].(string)
  34. CollSave = Sysconfig["coll_change"].(string)
  35. lastId = util.Int64All(Sysconfig["lastId"])
  36. ChangeMap = util.ObjArrToMapArr(Sysconfig["changeType"].([]interface{}))
  37. initChangeMap()
  38. }
  39. func initChangeMap() {
  40. for _, v := range ChangeMap {
  41. list := v["change_keyword"].([]interface{})
  42. var regList []string
  43. if len(list) > 0 {
  44. for _, v1 := range list {
  45. reg := ".*" + util.ObjToString(v1) + ".*"
  46. regList = append(regList, reg)
  47. }
  48. v["change_key_reg"] = regList
  49. } else {
  50. v["change_key_reg"] = []string{".*"}
  51. }
  52. }
  53. }
  54. func main() {
  55. //go SaveData()
  56. //go updateMethod()
  57. //go TimeTask()
  58. go GetData()
  59. //go TaskAll()
  60. //ch := make(chan bool, 1)
  61. //<-ch
  62. //var id string
  63. //flag.StringVar(&id, "id", "", "公司id")
  64. //flag.Parse()
  65. //if id == "" {
  66. // flag.PrintDefaults()
  67. // log.Println("参数错误.")
  68. //} else {
  69. // taskInfo(id)
  70. //}
  71. }