Function create_receive_handler

battlecode-manager/server.py:319–667

Create a Class that will be used a receive handler Args: game: The game the receive handler should operate on dockers: A map of the docker files with the key being use_docker: if True sleep and wake with docker otherwise don't use docker. Useful

(game: Game, dockers, use_docker: bool,
                           is_unix_stream: bool)

def create_receive_handler(game: Game, dockers, use_docker: bool,
                           is_unix_stream: bool)  \
                                    -> socketserver.BaseRequestHandler:
    '''
    Create a Class that will be used a receive handler

    Args:
        game: The game the receive handler should operate on
        dockers: A map of the docker files with the key being
        use_docker: if True sleep and wake with docker otherwise don't use
                    docker. Useful for testing the socket server
    Return:
        A ReceiveHandler class
    '''
    class ReceiveHandler(socketserver.BaseRequestHandler):
        '''
        This class overrides the default handling method in socketServer, so it
        calls what we want
        '''

        def __init__(self, *args, **kwargs):
            '''
            Hidden init
            '''
            self.game = game
            self.dockers = dockers
            self.client_id = 0
            self.error = ""
            self.logged_in = False
            self.is_unix_stream = is_unix_stream
            self.buffer_small = b''
            self.buffer_large = []

            super(ReceiveHandler, self).__init__(*args, **kwargs)

        def read_line(self):
            while True:
                if self.buffer_small:
                    pos = self.buffer_small.find(b'\n')
                    if pos != -1:
                        ret = b''.join(self.buffer_large) + self.buffer_small[:pos]
                        self.buffer_small = self.buffer_small[pos+1:]
                        self.buffer_large = []
                        return ret
                    else:
                        self.buffer_large.append(self.buffer_small)
                self.buffer_small = self.request.recv(4096)
                if not self.buffer_small:
                    raise IOError("reached socket EOF before finding newline")

        def get_next_message(self) -> object:
            '''
            Returns the json loaded object of the next string that is sent over the
            socket

            Returns:
                An object, for our purposes this will be a dictionary, of the json
                loaded string