core: recreation of matchmaking, add: matchmaking support multiple modes

matchmaking/consumers.py

@@ -6,8 +6,7 @@ from games.models import GameModel
 
 import json
 
-queue_id: [int] = []
-queue_ws: [WebsocketConsumer] = []
+from .models import Waiter, WaitingRoom, WaitingRoomManager, normal
 
 class MatchMaking(WebsocketConsumer):
 
@@ -24,25 +23,20 @@ class MatchMaking(WebsocketConsumer):
 
 		self.channel_layer.group_add(self.group_name, self.channel_name)
 
-		self.accept()
+		self.mode = int(self.scope['url_route']['kwargs']['mode'])
+		self.group_name = self.mode
 
-		global queue_id, queue_ws
-		queue_id.append(user.pk)
-		queue_ws.append(self)
+		waiting_room: WaitingRoom = normal.get(self.mode)
+		waiting_room.append(Waiter(user.pk, self))
 
-		if len(set(queue_id)) == 2:
-			game_id: int = GameModel().create(set(queue_id))
-			event = {"game_id": game_id}
-			for ws in queue_ws:
-				ws.send(text_data=json.dumps({'game_id': game_id}))
-			queue_id.clear()
-			queue_ws.clear()
+		print(len(waiting_room), "/", self.mode, [len(waiting_room),self.mode])
+		if (len(waiting_room) == self.mode):
+			game_id: int = GameModel().create(waiting_room.get_users_id())
+			waiting_room.broadcast({"detail": "Game found !", "game_id": game_id})
+			waiting_room.clear()
 
-		
 	def disconnect(self, close_code):
-		user: User = self.scope["user"]
-		global queue_id, queue_ws
-		if (user.pk in queue_id):
-			queue_ws.pop(queue_id.index(user.pk))
-			queue_id.remove(user.pk)
-		self.channel_layer.group_discard(self.group_name, self.channel_name)
+		waiting_room: WaitingRoom = normal.get(self.mode)
+		waiter: Waiter = waiting_room.get_waiter_by_socket(self)
+		if (waiter is not None):
+			waiting_room.remove(self.scope["user"].pk)

matchmaking/models.py

@@ -1,11 +1,92 @@
 from django.db import models
 
-# Create your models here.
-in_matchmaking = {
-	"tournaments": {
+from channels.generic.websocket import WebsocketConsumer
+import json
 
-	},
-	"normal": {
-		
-	}
-}
+# Create your models here.
+class Waiter:
+
+	def __init__(self, user_id: int, socket: WebsocketConsumer):
+		self.user_id: int = user_id
+		self.socket: WebsocketConsumer = socket
+	
+	def send(self, data: dict):
+		self.socket.send(text_data=json.dumps(data))
+
+	def accept(self):
+		self.socket.accept()
+
+	def disconnect(self):
+		self.socket.disconnect()
+
+	def __eq__(self, obj):
+		return self.user_id == obj.user_id
+
+class WaitingRoom:
+	
+	def __init__(self, waiting_room_manager, mode: int):
+		self._waiter_list: [Waiter] = [] 
+		self._mode: int = mode
+		self._waiting_room_manager = waiting_room_manager
+
+	def broadcast(self, data: dict):
+		for waiter in self._waiter_list:
+			waiter: Waiter
+			waiter.send(data)
+	
+	def clear(self):
+		self._waiter_list.clear()
+
+	def get_waiter_by_socket(self, socket: WebsocketConsumer):
+		for waiter in self._waiter_list:
+			waiter: Waiter
+			if (waiter.socket is socket):
+				return waiter
+		return None
+
+	def append(self, waiter: Waiter):
+
+		self.remove(waiter)
+		waiter.accept()
+		self._waiter_list.append(waiter)
+
+	def remove(self, users_id):
+		for waiter in self._waiter_list:
+			waiter: Waiter = waiter
+			if (waiter == users_id):
+				waiter.disconnect()
+				if (self.empty()):
+					self._waiting_room_manager.remove(self)
+				return
+
+	def empty(self):
+		for _ in self._waiter_list:
+			return False
+		return True
+
+	def get_users_id(self):
+		return [waiter.user_id for waiter in self._waiter_list]
+
+	def __len__(self):
+		return len(self._waiter_list)
+	
+
+
+class WaitingRoomManager:
+
+	def __init__(self):
+		self._waiting_rooms: [WaitingRoom] = []
+
+	def get(self, mode: int):
+		for waiting_room in self._waiting_rooms:
+			waiting_room: WaitingRoom = waiting_room
+			if (waiting_room._mode == mode):
+				return waiting_room
+		tmp: WaitingRoom = WaitingRoom(self, mode)
+		self._waiting_rooms.append(tmp)
+		return tmp
+	
+	def remove(self, waiting_room: WaitingRoom):
+		self._waiting_rooms.remove(waiting_room)
+
+normal: WaitingRoomManager = WaitingRoomManager()

matchmaking/routing.py

@@ -2,5 +2,5 @@ from django.urls import re_path
 from . import consumers
 
 websocket_urlpatterns = [
-	re_path(r'ws/matchmaking/', consumers.MatchMaking.as_asgi())
+	re_path(r'ws/matchmaking/(?P<mode>\d+)$', consumers.MatchMaking.as_asgi())
 ]