91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/aerogo/aero"
|
|
"github.com/aerogo/log"
|
|
"github.com/animenotifier/notify.moe/arn"
|
|
)
|
|
|
|
var (
|
|
requestLog = log.New()
|
|
errorLog = log.New()
|
|
ipLog = log.New()
|
|
)
|
|
|
|
// Initialize log files
|
|
func init() {
|
|
// The request log contains every single request to the server
|
|
requestLog.AddWriter(log.File("logs/request.log"))
|
|
|
|
// The IP log contains the IPs accessing the server
|
|
ipLog.AddWriter(log.File("logs/ip.log"))
|
|
|
|
// The error log contains all failed requests
|
|
errorLog.AddWriter(log.File("logs/error.log"))
|
|
errorLog.AddWriter(os.Stderr)
|
|
}
|
|
|
|
// Log middleware logs every request into logs/request.log and errors into logs/error.log.
|
|
func Log(next aero.Handler) aero.Handler {
|
|
return func(ctx aero.Context) error {
|
|
start := time.Now()
|
|
err := next(ctx)
|
|
responseTime := time.Since(start)
|
|
|
|
logRequest(ctx, responseTime)
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Logs a single request
|
|
func logRequest(ctx aero.Context, responseTime time.Duration) {
|
|
responseTimeString := strconv.Itoa(int(responseTime.Nanoseconds()/1000000)) + " ms"
|
|
repeatSpaceCount := 8 - len(responseTimeString)
|
|
|
|
if repeatSpaceCount < 0 {
|
|
repeatSpaceCount = 0
|
|
}
|
|
|
|
responseTimeString = strings.Repeat(" ", repeatSpaceCount) + responseTimeString
|
|
|
|
user := arn.GetUserFromContext(ctx)
|
|
ip := ctx.IP()
|
|
hostNames, cached := GetHostsForIP(ip)
|
|
|
|
if !cached && len(hostNames) > 0 {
|
|
ipLog.Info("%s = %s", ip, strings.Join(hostNames, ", "))
|
|
}
|
|
|
|
// Log every request
|
|
id := "id"
|
|
nick := "guest"
|
|
|
|
if user != nil {
|
|
id = user.ID
|
|
nick = user.Nick
|
|
}
|
|
|
|
requestLog.Info("%s | %s | %s | %s | %d | %s", nick, id, ip, responseTimeString, ctx.Status(), ctx.Path())
|
|
|
|
// Log all requests that failed
|
|
switch ctx.Status() {
|
|
case http.StatusOK, http.StatusTemporaryRedirect, http.StatusPermanentRedirect:
|
|
// Ok.
|
|
|
|
default:
|
|
errorLog.Error("%s | %s | %s | %s | %d | %s", nick, id, ip, responseTimeString, ctx.Status(), ctx.Path())
|
|
}
|
|
|
|
// Notify us about long requests.
|
|
// However ignore requests under /auth/ because those depend on 3rd party servers.
|
|
if responseTime >= 500*time.Millisecond && !strings.HasPrefix(ctx.Path(), "/auth/") && !strings.HasPrefix(ctx.Path(), "/sitemap/") && !strings.HasPrefix(ctx.Path(), "/api/sse/") {
|
|
errorLog.Error("%s | %s | %s | %s | %d | %s (long response time)", nick, id, ip, responseTimeString, ctx.Status(), ctx.Path())
|
|
}
|
|
}
|