Function main

GPU-Virtual-Service/gpu-remoting/proxy.py:217–249

()

            if clientId2gpuInfos[clientID]['usedNum'] == 0:
                del clientId2gpuInfos[clientID]
                logging.info(f"Client#{clientID} released all GPU resources")

def main():
    config = load_runtime_config()

    dpcIp = config['DispatcherConfig']['dpcIp_']
    dpcPort = config['DispatcherConfig']['dpcPort_']
    listenIp = config['ClientConfig']['proxyIp_']
    listenPort = config['ClientConfig']['proxyPort_']
    
    proxy = connect_to_dispatcher(dpcIp, dpcPort)
    if proxy is None:
        return

    logging.info(f"Connected to dispatcher({dpcIp}:{dpcPort})")
    proxy_lock = threading.Lock()

    listen_proxy = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    listen_proxy.bind((listenIp, listenPort))
    listen_proxy.listen()
    logging.info(f"Listening on {listenIp}:{listenPort}")

    try:
        while True:
            conn, addr = listen_proxy.accept()
            logging.debug(f"Accepted connection from {addr}")
            threading.Thread(target=handle_client_worker, args=(conn, addr, proxy, proxy_lock)).start()
    except KeyboardInterrupt:
        logging.info("Shutting down...")
        listen_proxy.close()
        proxy.close()

    proxy.close()

if __name__ == "__main__":
    main()