43 lines
969 B
Go
43 lines
969 B
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-gost/gost/pkg/auth"
|
|
"github.com/go-gost/gost/pkg/logger"
|
|
)
|
|
|
|
func mwLogger() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
// start time
|
|
startTime := time.Now()
|
|
// Processing request
|
|
ctx.Next()
|
|
duration := time.Since(startTime)
|
|
|
|
logger.Default().WithFields(map[string]interface{}{
|
|
"kind": "api",
|
|
"method": ctx.Request.Method,
|
|
"uri": ctx.Request.RequestURI,
|
|
"code": ctx.Writer.Status(),
|
|
"client": ctx.ClientIP(),
|
|
"duration": duration,
|
|
}).Infof("| %3d | %13v | %15s | %-7s %s",
|
|
ctx.Writer.Status(), duration, ctx.ClientIP(), ctx.Request.Method, ctx.Request.RequestURI)
|
|
}
|
|
}
|
|
|
|
func mwBasicAuth(auther auth.Authenticator) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
if auther == nil {
|
|
return
|
|
}
|
|
u, p, _ := c.Request.BasicAuth()
|
|
if !auther.Authenticate(u, p) {
|
|
c.AbortWithStatus(http.StatusUnauthorized)
|
|
}
|
|
}
|
|
}
|