增加定时任务 处理异常

This commit is contained in:
2024-08-30 11:55:19 +08:00
parent aa3f9e8711
commit bfb284c4cc
30 changed files with 945 additions and 336 deletions

View File

@@ -1,48 +1,2 @@
package middleware
import (
"energy-management-system/utils"
"energy-management-system/utils/exception"
"fmt"
"github.com/gin-gonic/gin"
"net"
"os"
"strings"
"time"
)
func Recovery() gin.HandlerFunc {
return func(c *gin.Context) {
defer func() {
if err := recover(); err != nil {
var brokenPipe bool
if ne, ok := err.(*net.OpError); ok {
if se, ok := ne.Err.(*os.SyscallError); ok {
if strings.Contains(strings.ToLower(se.Error()), "broken pipe") || strings.Contains(strings.ToLower(se.Error()), "connection reset by peer") {
brokenPipe = true
}
}
}
if brokenPipe {
c.Error(err.(error))
} else {
if h, ok := err.(*exception.Exception); ok {
exception.Panic(c, h)
c.Errors = append(c.Errors, &gin.Error{Meta: h})
} else if _, ok = err.(error); ok {
if gin.IsDebugging() {
fmt.Printf("[Recovery] %s : %s", utils.TimeFormat(time.Now()), err)
utils.Stack(3)
}
exception.Unknow(c)
} else {
fmt.Print(err)
exception.Server(c)
}
}
c.Abort()
}
}()
c.Next()
}
}