From e22a96f53ea88ba0119c1b03db14c34aff706f32 Mon Sep 17 00:00:00 2001
From: Azareal <azareal@users.noreply.github.com>
Date: Sat, 8 Sep 2018 15:29:51 +1000
Subject: [PATCH] Oops, make sure those locks are released.

---
 common/ws_user.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/common/ws_user.go b/common/ws_user.go
index 1107923d..d73e3742 100644
--- a/common/ws_user.go
+++ b/common/ws_user.go
@@ -110,22 +110,22 @@ func (wsUser *WSUser) SetPageForSocket(conn *websocket.Conn, page string) {
 
 func (wsUser *WSUser) InPage(page string) bool {
 	wsUser.Lock()
+	defer wsUser.Unlock()
 	for _, socket := range wsUser.Sockets {
 		if socket.Page == page {
 			return true
 		}
 	}
-	wsUser.Unlock()
 	return false
 }
 
 func (wsUser *WSUser) FinalizePage(page string, handle func()) {
 	wsUser.Lock()
+	defer wsUser.Unlock()
 	for _, socket := range wsUser.Sockets {
 		if socket.Page == page {
 			return
 		}
 	}
 	handle()
-	wsUser.Unlock()
 }