pep.py/objects/matchList.py

import threading
import time

from common.sentry import sentry
from constants.exceptions import periodicLoopException
from objects import match
from objects import glob
from constants import serverPackets
from common.log import logUtils as log

class matchList:
	def __init__(self):
		"""Initialize a matchList object"""
		self.matches = {}
		self.lastID = 1

	def createMatch(self, matchName, matchPassword, beatmapID, beatmapName, beatmapMD5, gameMode, hostUserID, isTourney=False):
		"""
		Add a new match to matches list

		:param matchName: match name, string
		:param matchPassword: match md5 password. Leave empty for no password
		:param beatmapID: beatmap ID
		:param beatmapName: beatmap name, string
		:param beatmapMD5: beatmap md5 hash, string
		:param gameMode: game mode ID. See gameModes.py
		:param hostUserID: user id of who created the match
		:return: match ID
		"""
		# Add a new match to matches list and create its stream
		matchID = self.lastID
		self.lastID+=1
		self.matches[matchID] = match.match(matchID, matchName, matchPassword, beatmapID, beatmapName, beatmapMD5, gameMode, hostUserID, isTourney)
		return matchID

	def disposeMatch(self, matchID):
		"""
		Destroy match object with id = matchID

		:param matchID: ID of match to dispose
		:return:
		"""
		# Make sure the match exists
		if matchID not in self.matches:
			return

		# Get match and disconnect all players
		_match = self.matches[matchID]
		for slot in _match.slots:
			_token = glob.tokens.getTokenFromUserID(slot.userID, ignoreIRC=True)
			if _token is None:
				continue
			_match.userLeft(_token, disposeMatch=False)	# don't dispose the match twice when we remove all players

		# Delete chat channel
		glob.channels.removeChannel("#multi_{}".format(_match.matchID))

		# Send matchDisposed packet before disposing streams
		glob.streams.broadcast(_match.streamName, serverPackets.disposeMatch(_match.matchID))

		# Dispose all streams
		glob.streams.dispose(_match.streamName)
		glob.streams.dispose(_match.playingStreamName)
		glob.streams.remove(_match.streamName)
		glob.streams.remove(_match.playingStreamName)

		# Send match dispose packet to everyone in lobby
		glob.streams.broadcast("lobby", serverPackets.disposeMatch(matchID))
		del self.matches[matchID]
		log.info("MPROOM{}: Room disposed manually".format(_match.matchID))

	@sentry.capture()
	def cleanupLoop(self):
		"""
		Start match cleanup loop.
		Empty matches that have been created more than 60 seconds ago will get deleted.
		Useful when people create useless lobbies with `!mp make`.
		The check is done every 30 seconds.
		This method starts an infinite loop, call it only once!
		:return:
		"""
		try:
			log.debug("Checking empty matches")
			t = int(time.time())
			emptyMatches = []
			exceptions = []

			# Collect all empty matches
			for key, m in self.matches.items():
				if [x for x in m.slots if x.user is not None]:
					continue
				if t - m.createTime >= 120:
					log.debug("Match #{} marked for cleanup".format(m.matchID))
					emptyMatches.append(m.matchID)

			# Dispose all empty matches
			for matchID in emptyMatches:
				try:
					self.disposeMatch(matchID)
				except Exception as e:
					exceptions.append(e)
					log.error(
						"Something wrong happened while disposing a timed out match. Reporting to Sentry when "
						"the loop ends."
					)

			# Re-raise exception if needed
			if exceptions:
				raise periodicLoopException(exceptions)
		finally:
			# Schedule a new check (endless loop)
			threading.Timer(30, self.cleanupLoop).start()