61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
|
package middlewares
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
|
||
|
|
"github.com/authorizerdev/authorizer/server/constants"
|
||
|
|
"github.com/authorizerdev/authorizer/server/utils"
|
||
|
|
)
|
||
|
|
|
||
|
|
// GinLogWriteFunc convert func to io.Writer.
|
||
|
|
type GinLogWriteFunc func([]byte) (int, error)
|
||
|
|
|
||
|
|
// GinLog Write function
|
||
|
|
func (fn GinLogWriteFunc) Write(data []byte) (int, error) {
|
||
|
|
return fn(data)
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewGinLogrusWrite logrus writer for gin
|
||
|
|
func NewGinLogrusWrite() io.Writer {
|
||
|
|
return GinLogWriteFunc(func(data []byte) (int, error) {
|
||
|
|
log.Debugf("%s", data)
|
||
|
|
return 0, nil
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// JSONLogMiddleware logs a gin HTTP request in JSON format, with some additional custom key/values
|
||
|
|
func JSONLogMiddleware() gin.HandlerFunc {
|
||
|
|
return func(c *gin.Context) {
|
||
|
|
// Start timer
|
||
|
|
start := time.Now()
|
||
|
|
|
||
|
|
// Process Request
|
||
|
|
c.Next()
|
||
|
|
|
||
|
|
// Stop timer
|
||
|
|
duration := utils.GetDurationInMillseconds(start)
|
||
|
|
|
||
|
|
entry := log.WithFields(log.Fields{
|
||
|
|
"client_ip": utils.GetIP(c.Request),
|
||
|
|
"duration": fmt.Sprintf("%.2f", duration),
|
||
|
|
"method": c.Request.Method,
|
||
|
|
"path": c.Request.RequestURI,
|
||
|
|
"status": c.Writer.Status(),
|
||
|
|
"referrer": c.Request.Referer(),
|
||
|
|
"request_id": c.Writer.Header().Get("Request-Id"),
|
||
|
|
"authorizer_version": constants.VERSION,
|
||
|
|
})
|
||
|
|
|
||
|
|
if c.Writer.Status() >= 500 {
|
||
|
|
entry.Error(c.Errors.String())
|
||
|
|
} else {
|
||
|
|
entry.Info("")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|