From 2334cb68cfedc900130faa58f89aa650fa9aea3d Mon Sep 17 00:00:00 2001
From: ccyj <2384899431@qq.com>
Date: Sat, 21 Oct 2023 20:36:58 +0800
Subject: [PATCH] =?UTF-8?q?build:=20=E8=8A=82=E7=82=B9=E5=8A=9F=E8=83=BD?=
 =?UTF-8?q?=E6=9E=84=E5=BB=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/node.py | 70 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)

diff --git a/src/node.py b/src/node.py
index e69de29..1f08167 100644
--- a/src/node.py
+++ b/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()