From be893c81154fc559faa094d5440941db33d293e6 Mon Sep 17 00:00:00 2001 From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?= Date: Wed, 13 Oct 2010 15:25:51 +0200 Subject: [PATCH] Remove Sone ID from session if the Sone is null. --- src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java b/src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java index 4078585..49bd26d 100644 --- a/src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java +++ b/src/main/java/net/pterodactylus/sone/web/SoneTemplatePage.java @@ -129,7 +129,11 @@ public class SoneTemplatePage extends TemplatePage { */ protected void setCurrentSone(ToadletContext toadletContext, Sone sone) { Session session = getCurrentSession(toadletContext); - session.setAttribute("Sone.CurrentSone", sone.getId()); + if (sone == null) { + session.removeAttribute("Sone.CurrentSone"); + } else { + session.setAttribute("Sone.CurrentSone", sone.getId()); + } } // -- 2.7.4