main.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. package main
  2. import (
  3. "fmt"
  4. "github.com/beego/beego/v2/server/web"
  5. _ "hongze/hongze_task/routers"
  6. "hongze/hongze_task/services"
  7. "hongze/hongze_task/services/alarm_msg"
  8. "hongze/hongze_task/utils"
  9. "runtime"
  10. "time"
  11. "github.com/beego/beego/v2/adapter/logs"
  12. "github.com/beego/beego/v2/server/web/context"
  13. )
  14. func main() {
  15. if web.BConfig.RunMode == "dev" {
  16. web.BConfig.WebConfig.DirectoryIndex = true
  17. web.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
  18. }
  19. go services.InitTask()
  20. //services.TaskTest()
  21. web.BConfig.RecoverFunc = Recover
  22. web.Run()
  23. }
  24. func Recover(ctx *context.Context, conf *web.Config) {
  25. if err := recover(); err != nil {
  26. if err == web.ErrAbort {
  27. return
  28. }
  29. if !web.BConfig.RecoverPanic {
  30. panic(err)
  31. }
  32. stack := ""
  33. msg := fmt.Sprintf("The request url is %v", ctx.Input.URL())
  34. stack += msg + "</br>"
  35. logs.Critical(msg)
  36. msg = fmt.Sprintf("The request data is %v", string(ctx.Input.RequestBody))
  37. stack += msg + "</br>"
  38. logs.Critical(msg)
  39. msg = fmt.Sprintf("Handler crashed with error %v", err)
  40. stack += msg + "</br>"
  41. logs.Critical(msg)
  42. for i := 1; ; i++ {
  43. _, file, line, ok := runtime.Caller(i)
  44. if !ok {
  45. break
  46. }
  47. logs.Critical(fmt.Sprintf("%s:%d", file, line))
  48. stack = stack + fmt.Sprintln(fmt.Sprintf("%s:%d</br>", file, line))
  49. }
  50. //go utils.SendEmail(utils.APPNAME+"崩了"+time.Now().Format("2006-01-02 15:04:05"), stack, utils.EmailSendToUsers)
  51. go alarm_msg.SendAlarmMsg(utils.APPNAME+"崩了"+time.Now().Format("2006-01-02 15:04:05")+"\n"+stack, 3)
  52. }
  53. return
  54. }