Cleanup

middleware/request-log.go

@@ -0,0 +1,47 @@
+package middleware
+
+import (
+	"net/http"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/aerogo/aero"
+	"github.com/aerogo/log"
+)
+
+// RequestLog logs every request into logs/request.log.
+func RequestLog() aero.Middleware {
+	err := log.NewLog()
+	err.AddOutput(log.File("logs/error.log"))
+	err.AddOutput(os.Stderr)
+
+	request := log.NewLog()
+	request.AddOutput(log.File("logs/request.log"))
+
+	return func(ctx *aero.Context, next func()) {
+		start := time.Now()
+		next()
+		responseTime := time.Since(start)
+		responseTimeString := strconv.Itoa(int(responseTime.Nanoseconds()/1000000)) + " ms"
+		responseTimeString = strings.Repeat(" ", 8-len(responseTimeString)) + responseTimeString
+
+		// Log every request
+		request.Info(ctx.RealIP(), ctx.StatusCode, responseTimeString, 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), ctx.RealIP(), ctx.StatusCode, responseTimeString, ctx.URI())
+		}
+
+		// Notify us about long requests
+		if responseTime >= 200*time.Millisecond {
+			err.Error("Long response time", ctx.RealIP(), ctx.StatusCode, responseTimeString, ctx.URI())
+		}
+	}
+}

middleware/save-session.go

@@ -0,0 +1,16 @@
+package middleware
+
+import "github.com/aerogo/aero"
+
+// SaveSession saves an existing session if it has been modified.
+func SaveSession() aero.Middleware {
+	return func(ctx *aero.Context, next func()) {
+		// Handle the request first
+		next()
+
+		// Update session if it has been modified
+		if ctx.HasSession() && ctx.Session().Modified() {
+			ctx.App.Sessions.Store.Set(ctx.Session().ID(), ctx.Session())
+		}
+	}
+}