forked from sangge/tpre-python
71 lines
1.6 KiB
Python
71 lines
1.6 KiB
Python
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()
|