Function receive_message

GPU-Virtual-Service/gpu-remoting/proxy.py:28–46

(conn)

    except Exception as e:
        logging.error(f"Failed to connect to dispatcher({dpc_ip}:{dpc_port}): {e}")
        return None

def receive_message(conn):
    # 首先接收4字节的长度信息
    length_bytes = conn.recv(8)
    if not length_bytes:
        return None
    
    # 将网络字节序转换回整数
    message_length = socket.ntohl(int.from_bytes(length_bytes, 'big'))
    
    # 接收实际数据
    data = b''
    while len(data) < message_length:
        chunk = conn.recv(min(message_length - len(data), 4096))
        if not chunk:
            return None
        data += chunk
    
    # 解码并解析 JSON 数据为 Python 对象
    return json.loads(data.decode('utf-8'))