diff --git a/internal/router/logger.go b/internal/router/logger.go index 461fcf442..721b2fb46 100644 --- a/internal/router/logger.go +++ b/internal/router/logger.go @@ -1 +1,65 @@ -package router \ No newline at end of file +/* + GoToSocial + Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see . +*/ + +package router + +import ( + "fmt" + "net/http" + "time" + + "github.com/gin-gonic/gin" + "github.com/sirupsen/logrus" +) + +var skipPaths = map[string]interface{}{ + "/api/v1/streaming": nil, +} + +// LoggerWithConfig instance a Logger middleware with config. +func LoggerWithConfig(log *logrus.Logger) gin.HandlerFunc { + logHandler := func(c *gin.Context) { + start := time.Now() + path := c.Request.URL.Path + raw := c.Request.URL.RawQuery + + // Process request + c.Next() + + // Log only when path is not being skipped + if _, ok := skipPaths[path]; !ok { + latency := time.Now().Sub(start) + clientIP := c.ClientIP() + method := c.Request.Method + statusCode := c.Writer.Status() + errorMessage := c.Errors.ByType(gin.ErrorTypePrivate).String() + bodySize := c.Writer.Size() + if raw != "" { + path = path + "?" + raw + } + + l := log.WithFields(logrus.Fields{ + "latency": fmt.Sprintf("%13v", latency), + }) + + http.StatusText(statusCode) + + } + } + return logHandler +} diff --git a/internal/router/router.go b/internal/router/router.go index df08a6b5e..3ef987ce4 100644 --- a/internal/router/router.go +++ b/internal/router/router.go @@ -122,12 +122,6 @@ func New(ctx context.Context, cfg *config.Config, db db.DB, logger *logrus.Logge // create the actual engine here -- this is the core request routing handler for gts engine := gin.New() - // paths to not log requests to - dontLog := []string{ - // don't print requests to the streaming API to avoid leaking tokens - "/api/v1/streaming", - } - // instruct gin to write out to our logger loggingConfig := gin.LoggerConfig{ Formatter: logFormatter,