48373446ecf9ef563b02c360be347da02df09ce9,deeppavlov/utils/socket/socket.py,,handle_connection,#Any#Any#Any#Any#Any#,21

Before Change

    if not lengths:
        log.error("got empty request")
        await loop.sock_sendall(conn, b"[]")
        return
    elif len(lengths) > 1:
        log.error("got several different batch sizes")

After Change

    log.info(f"handling connection from {addr}")
    conn.setblocking(False)
    recv_data = b""
    try:
        while True:
            chunk = await loop.run_in_executor(None, conn.recv, bufsize)
            if chunk:
                recv_data += chunk
            else:
                break
    except BlockingIOError:
        pass
    try:
        data = json.loads(recv_data)
    except ValueError:
        await wrap_error(conn, "request type is not json")

Instances