sangge/tpre-python
2
1
Fork 1
Code Issues 6 Pull Requests Packages Projects 1 Activity

main #19

Merged
sangge merged 3 commits from ccyj/mimajingsai:main into main 2023-10-23 21:41:41 +08:00
Conversation 0 Commits 3 Files Changed 2 +22 -18
1 changed files with 22 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 3349474f22 - Show all commits

40
src/node.py
View File

@ -9,17 +9,15 @@ from tpre import *
@asynccontextmanager @asynccontextmanager
async def lifespan(app: FastAPI): async def lifespan(app: FastAPI):
# Load the ML model
init() init()
yield yield
# Clean up the ML models and release the resources
clear() clear()
app = FastAPI(lifespan=lifespan) app = FastAPI(lifespan=lifespan)
server_address = "http://中心服务器IP地址/server" server_address = "http://10.20.14.232:8000/server"
id = 0 id = 0
ip = "" ip = "10.16.21.163"
client_ip_src = "" # 发送信息用户的ip client_ip_src = "" # 发送信息用户的ip
client_ip_des = "" # 接收信息用户的ip client_ip_des = "" # 接收信息用户的ip
processed_message = () # 重加密后的数据 processed_message = () # 重加密后的数据
@ -31,8 +29,8 @@ processed_message = () # 重加密后的数据
# 向中心服务器发送自己的IP地址,并获取自己的id # 向中心服务器发送自己的IP地址,并获取自己的id
def send_ip(): def send_ip():
url = server_address + "/get_node?ip = " + ip url = server_address + "/get_node?ip=" + ip
# ip = get_local_ip # type: ignore # ip = get_local_ip() # type: ignore
global id global id
id = requests.get(url) id = requests.get(url)
@ -51,10 +49,11 @@ def get_local_ip():
def init(): def init():
get_local_ip() # get_local_ip()
global id global id
send_ip() send_ip()
task = asyncio.create_task(send_heartbeat_internal()) task = asyncio.create_task(send_heartbeat_internal())
print("Finish init")
def clear(): def clear():
@ -65,12 +64,16 @@ def clear():
async def send_heartbeat_internal() -> None: async def send_heartbeat_internal() -> None:
timeout = 3
global ip
url = server_address + "/heartbeat?ip=" + ip
while True: while True:
# print('successful send my_heart') # print('successful send my_heart')
global ip try:
url = server_address + "/get_node?ip = " + ip folderol = requests.get(url)
folderol = requests.get(url) except:
timeout = 30 print("Central server error")
# 删除超时的节点（假设你有一个异步的数据库操作函数） # 删除超时的节点（假设你有一个异步的数据库操作函数）
await asyncio.sleep(timeout) await asyncio.sleep(timeout)
@ -79,36 +82,37 @@ async def send_heartbeat_internal() -> None:
async def receive_user_src_message(message: Request): async def receive_user_src_message(message: Request):
global client_ip_src, client_ip_des global client_ip_src, client_ip_des
# kfrag , capsule_ct ,client_ip_src , client_ip_des = json_data[] # 看梁俊勇 # kfrag , capsule_ct ,client_ip_src , client_ip_des = json_data[] # 看梁俊勇
''' """
payload = { payload = {
"source_ip": local_ip, "source_ip": local_ip,
"dest_ip": dest_ip, "dest_ip": dest_ip,
"capsule_ct": capsule_ct, "capsule_ct": capsule_ct,
"rk": rk_list[i], "rk": rk_list[i],
} }
''' """
data = await message.json() data = await message.json()
source_ip = data.get("source_ip") source_ip = data.get("source_ip")
dest_ip = data.get("dest_ip") dest_ip = data.get("dest_ip")
capsule_ct = data.get("capsule_ct") capsule_ct = data.get("capsule_ct")
rk = data.get("rk") rk = data.get("rk")
processed_message = ReEncrypt(rk, capsule_ct) processed_message = ReEncrypt(rk, capsule_ct)
await send_user_des_message(source_ip, dest_ip, processed_message) await send_user_des_message(source_ip, dest_ip, processed_message)
return HTTPException(status_code=200, detail="message recieved") return HTTPException(status_code=200, detail="message recieved")
async def send_user_des_message(source_ip: str, dest_ip: str, re_message): # 发送消息给用户2 async def send_user_des_message(source_ip: str, dest_ip: str, re_message): # 发送消息给用户2
data = {"Tuple": re_message, "ip": source_ip} # 类型不匹配 data = {"Tuple": re_message, "ip": source_ip} # 类型不匹配
# 发送 HTTP POST 请求 # 发送 HTTP POST 请求
response = requests.post("http://" + dest_ip + "/receive_messages?message", json=data) response = requests.post(
"http://" + dest_ip + "/receive_messages?message", json=data
)
print(response) print(response)
if __name__ == "__main__": if __name__ == "__main__":
import uvicorn # pylint: disable=e0401 import uvicorn # pylint: disable=e0401
uvicorn.run("node:app", host="0.0.0.0", port=8000, reload=True) uvicorn.run("node:app", host="0.0.0.0", port=8001, reload=True)