main.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. package main
  2. import (
  3. . "dataPkgRemoveRepeat/config"
  4. . "dataPkgRemoveRepeat/service"
  5. "fmt"
  6. "log"
  7. "net/http"
  8. "net/rpc"
  9. . "app.yhyue.com/moapp/jybase/common"
  10. "app.yhyue.com/moapp/jybase/logger"
  11. )
  12. func main() {
  13. logger.SetConsole(false)
  14. logger.SetRollingDaily("./logs", "dprr.log")
  15. go func() {
  16. TimeTask.MemUsage()
  17. TimeTask.Start()
  18. }()
  19. rpc.Register(&RemoveRepeat{})
  20. rpc.HandleHTTP()
  21. http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
  22. reqType := Int64All(r.FormValue("reqType"))
  23. if reqType <= 0 || reqType > 3 {
  24. fmt.Fprint(w, "无效参数reqType")
  25. return
  26. }
  27. if reqType == 3 { //内存信息
  28. fmt.Fprint(w, TimeTask.PrintMemUsage())
  29. return
  30. } else if reqType == 2 { //清理
  31. appid := r.FormValue("appid")
  32. if appid == "" {
  33. fmt.Fprint(w, "无效参数appid")
  34. return
  35. }
  36. TimeTask.ClearInvalidData(map[string]bool{appid: true})
  37. } else if reqType == 1 { //全量
  38. appid := r.FormValue("appid")
  39. entId := Int64All(r.FormValue("entid"))
  40. if appid == "" || entId == 0 {
  41. fmt.Fprint(w, "无效参数appid、entid")
  42. return
  43. }
  44. TimeTask.LoadFull([]string{appid}, []int64{entId})
  45. }
  46. fmt.Fprint(w, "success")
  47. })
  48. // 设置监听的端口
  49. log.Println("ListenAndServe", Config.Port)
  50. err := http.ListenAndServe(Config.Port, nil)
  51. if err != nil {
  52. log.Fatalln(err)
  53. }
  54. }