75 lines
1.9 KiB
Go
Raw Normal View History

2017-06-16 23:12:28 +00:00
package middleware
2017-06-05 18:21:06 +00:00
2017-06-10 18:45:30 +00:00
import (
2017-06-15 13:50:39 +00:00
"net/http"
"os"
2017-06-10 18:45:30 +00:00
"strconv"
"strings"
"time"
2017-06-05 18:21:06 +00:00
2017-06-10 18:45:30 +00:00
"github.com/aerogo/aero"
"github.com/aerogo/log"
2017-06-22 20:26:52 +00:00
"github.com/animenotifier/notify.moe/utils"
2017-06-10 18:45:30 +00:00
)
2017-06-05 18:21:06 +00:00
2017-06-23 13:51:26 +00:00
var request = log.New()
var err = log.New()
// Initialize log files
func init() {
2017-06-16 23:34:54 +00:00
request.AddOutput(log.File("logs/request.log"))
2017-06-10 18:45:30 +00:00
err.AddOutput(log.File("logs/error.log"))
2017-06-15 13:50:39 +00:00
err.AddOutput(os.Stderr)
2017-06-23 13:51:26 +00:00
}
2017-06-05 18:21:06 +00:00
2017-06-23 13:51:26 +00:00
// Log middleware logs every request into logs/request.log and errors into logs/error.log.
func Log() aero.Middleware {
2017-06-16 23:12:28 +00:00
return func(ctx *aero.Context, next func()) {
2017-06-10 18:45:30 +00:00
start := time.Now()
next()
responseTime := time.Since(start)
2017-06-23 13:51:26 +00:00
go logRequest(ctx, responseTime)
}
}
// Logs a single request
func logRequest(ctx *aero.Context, responseTime time.Duration) {
responseTimeString := strconv.Itoa(int(responseTime.Nanoseconds()/1000000)) + " ms"
responseTimeString = strings.Repeat(" ", 8-len(responseTimeString)) + responseTimeString
user := utils.GetUser(ctx)
ip := ctx.RealIP()
hostName := "<unknown host>"
hostNames := GetHostsForIP(ip)
if len(hostNames) != 0 {
hostName = hostNames[0]
2017-06-23 13:53:22 +00:00
hostName = strings.TrimSuffix(hostName, ".")
2017-06-23 13:51:26 +00:00
}
// Log every request
if user != nil {
request.Info(user.Nick, ip, hostName, responseTimeString, ctx.StatusCode, ctx.URI())
} else {
request.Info("[guest]", ip, hostName, responseTimeString, ctx.StatusCode, ctx.URI())
}
// Log all requests that failed
switch ctx.StatusCode {
case http.StatusOK, http.StatusFound, http.StatusMovedPermanently, http.StatusPermanentRedirect, http.StatusTemporaryRedirect:
// Ok.
default:
err.Error(http.StatusText(ctx.StatusCode), ip, hostName, responseTimeString, ctx.StatusCode, ctx.URI())
}
// Notify us about long requests.
// However ignore requests under /auth/ because those depend on 3rd party servers.
if responseTime >= 200*time.Millisecond && !strings.HasPrefix(ctx.URI(), "/auth/") {
err.Error("Long response time", ip, hostName, responseTimeString, ctx.StatusCode, ctx.URI())
2017-06-16 23:12:28 +00:00
}
2017-06-10 18:45:30 +00:00
}