main.go 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. package main
  2. import (
  3. "mongodb"
  4. qu "qfw/util"
  5. es "qfw/util/elastic"
  6. )
  7. var (
  8. Sysconfig map[string]interface{} //配置文件
  9. Mgo *mongodb.MongodbSim
  10. Dbname string
  11. Dbcoll string
  12. Es1, Es2 *es.Elastic
  13. Index string
  14. Itype string
  15. EsFields []string
  16. TaskTime int
  17. UpdateId string
  18. )
  19. var EsSaveCache = make(chan map[string]interface{}, 5000)
  20. var SP = make(chan bool, 5)
  21. func init() {
  22. qu.ReadConfig(&Sysconfig)
  23. Dbname = Sysconfig["dbname"].(string)
  24. Dbcoll = Sysconfig["dbcoll"].(string)
  25. Mgo = &mongodb.MongodbSim{
  26. MongodbAddr: Sysconfig["mgodb"].(string),
  27. Size: qu.IntAllDef(Sysconfig["dbsize"], 5),
  28. DbName: Dbname,
  29. UserName: Sysconfig["uname"].(string),
  30. Password: Sysconfig["upwd"].(string),
  31. }
  32. Mgo.InitPool()
  33. //es
  34. econf := Sysconfig["elastic"].(map[string]interface{})
  35. Index = econf["index"].(string)
  36. Itype = econf["itype"].(string)
  37. //Es1 = &es.Elastic{
  38. // S_esurl: econf["addr"].(string),
  39. // I_size: qu.IntAllDef(econf["pool"], 12),
  40. //}
  41. //Es1.InitElasticSize()
  42. Es2 = &es.Elastic{
  43. S_esurl: econf["addr1"].(string),
  44. I_size: qu.IntAllDef(econf["pool"], 12),
  45. }
  46. Es2.InitElasticSize()
  47. EsFields = qu.ObjArrToStringArr(econf["esfields"].([]interface{}))
  48. //TaskTime = qu.IntAll(Sysconfig["tasktime"])
  49. UpdateId = qu.ObjToString(Sysconfig["updateid"])
  50. }
  51. func main() {
  52. go SaveEs()
  53. go TimeTask()
  54. //go SaveAll()
  55. //go SaveAdd()
  56. ch := make(chan bool, 1)
  57. <-ch
  58. }