Cleanup

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())
+		}
+	}
+}