dkforest

commit d88a36962ba08d85dcf962ccf1691fc9ee81c3fe
parent b1771a3544e7e8bfae73cfb05de92a27d180fab8
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Tue, 23 May 2023 23:35:06 -0700

fixes

Diffstat:
M
pkg/web/handlers/handlers.go
 | 
6
+++---

1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/pkg/web/handlers/handlers.go b/pkg/web/handlers/handlers.go
@@ -4828,7 +4828,7 @@ func ChessGameHandler(c echo.Context) error {
 	if err := usersStreamsManager.Add(authUser.ID); err != nil {
 		return nil
 	}
-	usersStreamsManager.Remove(authUser.ID)
+	defer usersStreamsManager.Remove(authUser.ID)
 
 	c.Response().Header().Set(echo.HeaderContentType, echo.MIMETextHTMLCharsetUTF8)
 	c.Response().WriteHeader(http.StatusOK)
@@ -4931,7 +4931,7 @@ func NewUsersStreamsManager() *UsersStreamsManager {
 func (m *UsersStreamsManager) Add(userID database.UserID) error {
 	m.Lock()
 	defer m.Unlock()
-	if m.m[userID] > userMaxStream {
+	if m.m[userID] >= userMaxStream {
 		return ErrTooManyStreams
 	}
 	m.m[userID]++
@@ -4982,7 +4982,7 @@ func ChatStreamMessagesHandler(c echo.Context) error {
 	if err := usersStreamsManager.Add(authUser.ID); err != nil {
 		return nil
 	}
-	usersStreamsManager.Remove(authUser.ID)
+	defer usersStreamsManager.Remove(authUser.ID)
 
 	c.Response().Header().Set(echo.HeaderContentType, echo.MIMETextHTMLCharsetUTF8)
 	c.Response().WriteHeader(http.StatusOK)