60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/getsentry/sentry-go"
|
|
"github.com/gin-gonic/gin"
|
|
"io/ioutil"
|
|
"strconv"
|
|
)
|
|
|
|
func Logging(minLevel sentry.Level) gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
ctx.Next()
|
|
|
|
statusCode := ctx.Writer.Status()
|
|
|
|
var level sentry.Level
|
|
if statusCode >= 500 {
|
|
level = sentry.LevelError
|
|
} else if statusCode >= 400 {
|
|
level = sentry.LevelWarning
|
|
} else {
|
|
level = sentry.LevelInfo
|
|
}
|
|
|
|
if level < minLevel {
|
|
return
|
|
}
|
|
|
|
requestBody, _ := ioutil.ReadAll(ctx.Request.Body)
|
|
|
|
var responseBody []byte
|
|
if statusCode >= 400 && statusCode <= 599 {
|
|
cw, ok := ctx.Writer.(*CustomWriter)
|
|
if ok {
|
|
responseBody = cw.Read()
|
|
}
|
|
}
|
|
|
|
sentry.CaptureEvent(&sentry.Event{
|
|
Extra: map[string]interface{}{
|
|
"status_code": strconv.Itoa(statusCode),
|
|
"method": ctx.Request.Method,
|
|
"path": ctx.Request.URL.Path,
|
|
"guild_id": ctx.Keys["guildid"],
|
|
"user_id": ctx.Keys["userid"],
|
|
"request_body": string(requestBody),
|
|
"response": string(responseBody),
|
|
},
|
|
Level: level,
|
|
Message: fmt.Sprintf("HTTP %d on %s %s", statusCode, ctx.Request.Method, ctx.FullPath()),
|
|
Tags: map[string]string{
|
|
"status_code": strconv.Itoa(statusCode),
|
|
"method": ctx.Request.Method,
|
|
"path": ctx.Request.URL.Path,
|
|
},
|
|
})
|
|
}
|
|
}
|