121 lines
3.3 KiB
Go
121 lines
3.3 KiB
Go
package wssessions
|
|
|
|
import (
|
|
"clickandjoin.app/managementsystem/modules/cache"
|
|
"clickandjoin.app/managementsystem/modules/structs"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/utils"
|
|
)
|
|
|
|
func AddUserWebSocketSessionId(c *fiber.Ctx) error {
|
|
// swagger:operation POST /wsconnections/:userId/:wsSessionId wsconnections wsconnectionsAddUserWebSocketSessionId
|
|
// ---
|
|
// summary: Add user websocket session
|
|
// parameters:
|
|
// - name: X-Ms-Api
|
|
// in: header
|
|
// required: true
|
|
// responses:
|
|
// '200':
|
|
// description: Added user web socket session
|
|
|
|
userId := utils.CopyString(c.Params("userId"))
|
|
wsSessionId := utils.CopyString(c.Params("wsSessionId"))
|
|
|
|
if webSocketSessions, ok := cache.WebSocketSessions[userId]; ok {
|
|
if isInList(userId, wsSessionId) {
|
|
return c.SendStatus(fiber.StatusUnprocessableEntity)
|
|
}
|
|
|
|
cache.WebSocketSessions[userId] = append(webSocketSessions, wsSessionId)
|
|
} else {
|
|
cache.WebSocketSessions[userId] = []string{wsSessionId}
|
|
}
|
|
|
|
return c.SendStatus(fiber.StatusOK)
|
|
}
|
|
|
|
func isInList(userId string, wsSessionId string) bool {
|
|
for _, item := range cache.WebSocketSessions[userId] {
|
|
if item == wsSessionId {
|
|
return true
|
|
}
|
|
|
|
}
|
|
return false
|
|
}
|
|
|
|
func ExistsUserWebSocketSessionId(c *fiber.Ctx) error {
|
|
// swagger:operation GET /wsconnections/:userId/:wsSessionId wsconnections wsconnectionsExistsUserWebSocketSessionId
|
|
// ---
|
|
// summary: Returns whether the user websocket session already exists
|
|
// produces:
|
|
// - application/json
|
|
// parameters:
|
|
// - name: X-Ms-Api
|
|
// in: header
|
|
// required: true
|
|
// responses:
|
|
// '200':
|
|
// description: Added user web socket session
|
|
|
|
return c.JSON(structs.ExistsUserWebSocketSessionIdResponse{E: isInList(c.Params("userId"), c.Params("wsSessionId"))})
|
|
}
|
|
|
|
func GetUserWebSocketSessionIds(c *fiber.Ctx) error {
|
|
// swagger:operation GET /wsconnections/:userId wsconnections wsconnectionsGetUserWebSocketSessionIds
|
|
// ---
|
|
// summary: List of all websocket sessions of the user that are connected
|
|
// produces:
|
|
// - application/json
|
|
// parameters:
|
|
// - name: X-Ms-Api
|
|
// in: header
|
|
// required: true
|
|
// responses:
|
|
// '200':
|
|
// description: List of all websocket sessions of the user that are connected
|
|
// schema:
|
|
// "$ref": "#/definitions/GetUserWebSocketSessionIdsResponse"
|
|
|
|
var list []string
|
|
|
|
if webSocketSessions, ok := cache.WebSocketSessions[c.Params("userId")]; ok {
|
|
list = webSocketSessions
|
|
}
|
|
|
|
return c.JSON(structs.GetUserWebSocketSessionIdsResponse{WebSocketSessions: list})
|
|
}
|
|
|
|
func RemoveConnection(c *fiber.Ctx) error {
|
|
// swagger:operation DELETE /wsconnections/:userId/:wsSessionId wsconnections wsconnectionsRemoveConnection
|
|
// ---
|
|
// summary: Deletes a websocket session from the user
|
|
// parameters:
|
|
// - name: X-Ms-Api
|
|
// in: header
|
|
// required: true
|
|
// responses:
|
|
// '200':
|
|
// description: Websocket session deleted
|
|
|
|
userId := c.Params("userId")
|
|
wsSessionId := c.Params("wsSessionId")
|
|
|
|
if webSocketSessions, ok := cache.WebSocketSessions[userId]; ok {
|
|
for index, session := range webSocketSessions {
|
|
if session == wsSessionId {
|
|
cache.WebSocketSessions[userId] = remove(webSocketSessions, index)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
return c.SendStatus(fiber.StatusOK)
|
|
}
|
|
|
|
func remove(s []string, i int) []string {
|
|
s[i] = s[0]
|
|
return s[1:]
|
|
}
|