panic recover

This commit is contained in:
2023-09-15 18:22:21 +08:00
parent 7c53581ff2
commit 638feaf381

17
main.go
View File

@@ -100,6 +100,12 @@ func main() {
IP: c.ClientIP(), IP: c.ClientIP(),
CreatedAt: time.Now(), CreatedAt: time.Now(),
} }
defer func() {
if err := recover(); err != nil {
log.Println("Error:", err)
c.AbortWithError(500, fmt.Errorf("%s", err))
}
}()
// check authorization header // check authorization header
if !*noauth { if !*noauth {
@@ -221,7 +227,16 @@ func main() {
log.Println("response is", r.Response) log.Println("response is", r.Response)
} }
proxy.ServeHTTP(c.Writer, c.Request)
func() {
defer func() {
if err := recover(); err != nil {
log.Println("Panic recover :", err)
}
}()
proxy.ServeHTTP(c.Writer, c.Request)
}()
resp, err := io.ReadAll(io.NopCloser(&buf)) resp, err := io.ReadAll(io.NopCloser(&buf))
if err != nil { if err != nil {
record.Response = "failed to read response from upstream " + err.Error() record.Response = "failed to read response from upstream " + err.Error()