build: 节点功能构建

src/node.py

@@ -0,0 +1,70 @@
+from fastapi import FastAPI
+import requests
+from contextlib import asynccontextmanager
+import socket
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Load the ML model
+    init()
+    yield
+    # Clean up the ML models and release the resources
+    clear()
+
+app = FastAPI(lifespan=lifespan)
+server_address ="http://中心服务器IP地址:端口号/ip" 
+id = 0
+
+# 向中心服务器发送自己的IP地址,并获取自己的id
+def send_ip(ip: str):
+    url = server_address
+    # ip = get_local_ip # type: ignore
+    data = {"ip": ip}
+    response = requests.post(url, data=data)
+    data = response.json()
+    id = data['id']
+    return id
+
+# 用socket获取本机ip
+def get_local_ip():
+    # 创建一个套接字对象
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    # 连接到一个外部的服务器，这将自动绑定到本地IP地址
+    s.connect(("8.8.8.8", 80))
+    # 获取本地IP地址
+    local_ip = s.getsockname()[0]
+    s.close()
+    return local_ip
+  
+
+id  = int
+def init():
+    ip = get_local_ip()
+    global id
+    id = send_ip(ip)
+
+def clear():
+
+    pass
+
+@app.post("/heartbeat/")
+async def receive_heartbeat():
+    return {"status": "received"}
+
+
+
+
+
+
+
+
+# 接收用户发来的消息，经过处理之后，再将消息发送给其他用户
+@app.post("/send_message")
+async def send_message(message: str):
+    # 处理消息
+    processed_message = message.upper()
+    # 发送消息给其他用户
+    url = "http://其他用户IP地址:端口号/receive_message"
+    data = {"message": processed_message}
+    response = requests.post(url, data=data)
+    return response.json()