File size: 2,265 Bytes
a00f760 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import aiohttp, asyncio
from App import SERVER_STATE, Node
import aiohttp
async def upload_file(file_path: str, node: str, chunk: int, task: str):
master_node = SERVER_STATE.get_master()
url = f"http://{master_node.SPACE_HOST}/uploadfile/?node={node}&chunk={chunk}&task={task}"
async with aiohttp.ClientSession() as session:
headers = {"Transfer-Encoding": "chunked"}
with open(file_path, "rb") as file:
async with session.post(url, headers=headers, data=file) as response:
if response.status == 200:
print("File uploaded successfully")
else:
print("Failed to upload file")
class WorkerClient:
base_url = SERVER_STATE.DB
async def register_worker(self):
async with aiohttp.ClientSession() as session:
data = {
"WORKER_ID": SERVER_STATE.SPACE_HOST,
"MASTER": SERVER_STATE.MASTER,
"HOST_NAME": SERVER_STATE.SPACE_HOST,
"SPACE_HOST": SERVER_STATE.SPACE_HOST,
}
response = await self.get_node()
if response:
return response
async with session.put(
f"{self.base_url}/nodes/{SERVER_STATE.SPACE_HOST}.json", json=data
) as resp:
return await resp.json()
async def get_node(self):
async with aiohttp.ClientSession() as session:
async with session.get(
f"{self.base_url}/nodes/{SERVER_STATE.SPACE_HOST}.json"
) as resp:
response = await resp.json()
return response
async def delete_node(self):
async with aiohttp.ClientSession() as session:
async with session.delete(
f"{self.base_url}/nodes/{SERVER_STATE.SPACE_HOST}.json"
) as resp:
response = await resp.json()
async def get_all_nodes(self):
async with aiohttp.ClientSession() as session:
async with session.get(f"{self.base_url}/nodes.json") as resp:
response = await resp.json()
SERVER_STATE.NODES = [Node(**value) for value in response.values()]
return SERVER_STATE.NODES
|