main.go 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. package main
  2. import (
  3. "fmt"
  4. util "jygit.jydev.jianyu360.cn/data_processing/common_utils"
  5. "jygit.jydev.jianyu360.cn/data_processing/common_utils/elastic"
  6. "jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
  7. "jygit.jydev.jianyu360.cn/data_processing/common_utils/redis"
  8. "net"
  9. "net/http"
  10. "net/rpc"
  11. "os"
  12. "regexp"
  13. )
  14. var (
  15. Sysconfig map[string]interface{}
  16. FileType []interface{}
  17. MongoTool *mongodb.MongodbSim
  18. Dbname string
  19. MgoColl string
  20. MgoColl1 string
  21. Es *elastic.Elastic
  22. Index string
  23. RpcServer string
  24. ClientAddr string
  25. )
  26. var (
  27. _ent = regexp.MustCompile(".{2,100}(集团|公司|学校|中心|家具城|门诊|\\\\[大中小\\\\]学|部|院|局|厂|店|所|队|社|室|厅|段|会|场|行)$")
  28. down_url = "https://jy-datafile.oss-cn-beijing.aliyuncs.com/"
  29. )
  30. func init() {
  31. util.ReadConfig(&Sysconfig)
  32. FileType = Sysconfig["file_type"].([]interface{})
  33. Dbname = Sysconfig["mongodbName"].(string)
  34. MgoColl = Sysconfig["mongodbColl"].(string)
  35. MgoColl1 = Sysconfig["mongodbColl1"].(string)
  36. MongoTool = &mongodb.MongodbSim{
  37. MongodbAddr: Sysconfig["mongodbServers"].(string),
  38. Size: util.IntAllDef(Sysconfig["mongodbPoolSize"], 5),
  39. DbName: Dbname,
  40. UserName: Sysconfig["uname"].(string),
  41. Password: Sysconfig["upwd"].(string),
  42. }
  43. MongoTool.InitPool()
  44. es := Sysconfig["es"].(map[string]interface{})
  45. Es = &elastic.Elastic{
  46. S_esurl: util.ObjToString(es["addr"]),
  47. I_size: util.IntAllDef(es["pool"], 10),
  48. Username: util.ObjToString(es["user"]),
  49. Password: util.ObjToString(es["password"]),
  50. }
  51. Index = util.ObjToString(es["index"])
  52. Es.InitElasticSize()
  53. ClientAddr = Sysconfig["serverCenter"].(string)
  54. RpcServer = Sysconfig["rpcServer"].(string)
  55. //初始化redis
  56. redis.InitRedisBySize(util.ObjToString(Sysconfig["redis"]), 50, 30, 240)
  57. }
  58. func initRpc() {
  59. rpc.Register(new(JyService)) // 注册rpc服务
  60. rpc.HandleHTTP() // 采用http协议作为rpc载体
  61. lis, err := net.Listen("tcp", RpcServer)
  62. if err != nil {
  63. util.Debug("fatal error: ", err)
  64. }
  65. _, _ = fmt.Fprintf(os.Stdout, "%s\n", "start connection")
  66. _ = http.Serve(lis, nil)
  67. }
  68. func main() {
  69. initRpc()
  70. }