main.go 915 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package main
  2. import (
  3. "customer_project/config"
  4. "fmt"
  5. "github.com/robfig/cron/v3"
  6. util "jygit.jydev.jianyu360.cn/data_processing/common_utils"
  7. "net"
  8. "net/http"
  9. "net/rpc"
  10. "os"
  11. )
  12. var (
  13. updatePool chan []map[string]interface{}
  14. updateSp chan bool
  15. saveSize int
  16. )
  17. func init() {
  18. config.Init("./common.toml")
  19. InitLog()
  20. InitMysql()
  21. }
  22. func main() {
  23. loadData()
  24. go RpcService()
  25. taskProject()
  26. crn := cron.New()
  27. _, _ = crn.AddFunc("@every 10m", func() {
  28. taskProject()
  29. })
  30. crn.Start()
  31. ch := make(chan bool, 1)
  32. <-ch
  33. }
  34. func RpcService() {
  35. rpc.Register(new(JyService)) // 注册rpc服务
  36. rpc.HandleHTTP() // 采用http协议作为rpc载体
  37. util.Debug(config.Conf.Serve.RpcServer)
  38. lis, err := net.Listen("tcp", config.Conf.Serve.RpcServer)
  39. if err != nil {
  40. util.Debug("fatal error: ", err)
  41. }
  42. _, _ = fmt.Fprintf(os.Stdout, "%s\n", "start connection")
  43. _ = http.Serve(lis, nil)
  44. }