Add room deletion on room closure

Author: WZWren

collab/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	verify "collab/verify"
+	"context"
 	"encoding/json"
 	"io"
 	"log"
@@ -266,6 +267,20 @@ func handleMessages(
 				RoomID:  client.roomID,
 				Content: []byte("The session has been closed by a user."),
 			}
+			targetId := msgData["userId"].(string)
+			data, err := persistMappings.Conn.HGetAll(context.Background(), targetId).Result()
+			if err != nil {
+				log.Printf("Error retrieving data for userID %s: %v", targetId, err)
+			} else {
+				_, err1 := persistMappings.Conn.Del(context.Background(), targetId).Result()
+				if err1 != nil {
+					log.Printf("Error deleting data for userID %s: %v", targetId, err1);
+				}
+				_, err2 := persistMappings.Conn.Del(context.Background(), data["otherUser"]).Result()
+				if err2 != nil {
+					log.Printf("Error deleting data for other user %s: %v", data["otherUser"], err2);
+				}
+			}
 			hub.broadcast <- closeMessage
 		}