198 lines
7.5 KiB
Python
198 lines
7.5 KiB
Python
import socket
|
||
import json
|
||
import threading
|
||
import os
|
||
import sys
|
||
from datetime import datetime
|
||
|
||
# 设置标准输出编码为 UTF-8（解决 Windows 控制台编码问题）
|
||
if sys.platform.startswith('win'):
|
||
import codecs
|
||
sys.stdout = codecs.getwriter('utf-8')(sys.stdout.detach())
|
||
sys.stderr = codecs.getwriter('utf-8')(sys.stderr.detach())
|
||
|
||
class JSONSocketServer:
|
||
def __init__(self, host='localhost', port=8888):
|
||
self.host = host
|
||
self.port = port
|
||
self.socket = None
|
||
self.clients = []
|
||
|
||
def start_server(self):
|
||
"""启动服务器"""
|
||
try:
|
||
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||
self.socket.bind((self.host, self.port))
|
||
self.socket.listen(5)
|
||
|
||
print(f"🚀 服务器启动成功！")
|
||
print(f"📍 监听地址: {self.host}:{self.port}")
|
||
print(f"⏰ 启动时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
||
print(f"✅ 等待客户端连接...")
|
||
|
||
while True:
|
||
client_socket, address = self.socket.accept()
|
||
print(f"🔗 客户端连接: {address}")
|
||
|
||
# 为每个客户端创建独立线程
|
||
client_thread = threading.Thread(
|
||
target=self.handle_client,
|
||
args=(client_socket, address)
|
||
)
|
||
client_thread.daemon = True
|
||
client_thread.start()
|
||
|
||
except Exception as e:
|
||
print(f"❌ 服务器启动失败: {e}")
|
||
finally:
|
||
if self.socket:
|
||
self.socket.close()
|
||
|
||
def handle_client(self, client_socket, address):
|
||
"""处理客户端请求"""
|
||
try:
|
||
while True:
|
||
# 接收客户端消息
|
||
data = client_socket.recv(4096).decode('utf-8')
|
||
if not data:
|
||
break
|
||
|
||
print(f"📨 收到来自 {address} 的消息: {data}")
|
||
|
||
# 解析客户端请求
|
||
try:
|
||
request = json.loads(data)
|
||
response = self.process_request(request, address)
|
||
|
||
# 发送响应
|
||
response_json = json.dumps(response, ensure_ascii=False, indent=2)
|
||
client_socket.send(response_json.encode('utf-8'))
|
||
|
||
except json.JSONDecodeError:
|
||
# 如果不是 JSON 格式，当作普通消息处理
|
||
response = {
|
||
"status": "success",
|
||
"message": f"服务器收到消息: {data}",
|
||
"timestamp": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
response_json = json.dumps(response, ensure_ascii=False, indent=2)
|
||
client_socket.send(response_json.encode('utf-8'))
|
||
|
||
except Exception as e:
|
||
print(f"❌ 处理客户端 {address} 时出错: {e}")
|
||
finally:
|
||
client_socket.close()
|
||
print(f"🔌 客户端 {address} 已断开连接")
|
||
|
||
def process_request(self, request, address):
|
||
"""处理客户端的 JSON 请求"""
|
||
command = request.get('command', 'unknown')
|
||
|
||
if command == 'get_file':
|
||
# 发送文件内容
|
||
filename = request.get('filename', 'test_data.json')
|
||
return self.send_file(filename, address)
|
||
|
||
elif command == 'save_file':
|
||
# 保存文件
|
||
filename = request.get('filename', 'received_data.json')
|
||
content = request.get('content', {})
|
||
return self.save_file(filename, content, address)
|
||
|
||
elif command == 'list_files':
|
||
# 列出当前目录的 JSON 文件
|
||
json_files = [f for f in os.listdir('.') if f.endswith('.json')]
|
||
return {
|
||
"status": "success",
|
||
"command": "list_files",
|
||
"files": json_files,
|
||
"count": len(json_files),
|
||
"timestamp": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
|
||
elif command == 'ping':
|
||
# 心跳测试
|
||
return {
|
||
"status": "success",
|
||
"command": "ping",
|
||
"message": "pong",
|
||
"server_time": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
|
||
else:
|
||
# 未知命令
|
||
return {
|
||
"status": "error",
|
||
"message": f"未知命令: {command}",
|
||
"available_commands": ["get_file", "save_file", "list_files", "ping"],
|
||
"timestamp": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
|
||
def send_file(self, filename, address):
|
||
"""发送文件内容"""
|
||
try:
|
||
if os.path.exists(filename):
|
||
with open(filename, 'r', encoding='utf-8') as f:
|
||
content = json.load(f)
|
||
|
||
return {
|
||
"status": "success",
|
||
"command": "get_file",
|
||
"filename": filename,
|
||
"content": content,
|
||
"size": os.path.getsize(filename),
|
||
"timestamp": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
else:
|
||
return {
|
||
"status": "error",
|
||
"command": "get_file",
|
||
"message": f"文件 {filename} 不存在",
|
||
"timestamp": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
except Exception as e:
|
||
return {
|
||
"status": "error",
|
||
"command": "get_file",
|
||
"message": f"读取文件失败: {str(e)}",
|
||
"timestamp": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
|
||
def save_file(self, filename, content, address):
|
||
"""保存文件"""
|
||
try:
|
||
with open(filename, 'w', encoding='utf-8') as f:
|
||
json.dump(content, f, ensure_ascii=False, indent=2)
|
||
|
||
return {
|
||
"status": "success",
|
||
"command": "save_file",
|
||
"filename": filename,
|
||
"message": f"文件 {filename} 保存成功",
|
||
"size": os.path.getsize(filename),
|
||
"timestamp": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
except Exception as e:
|
||
return {
|
||
"status": "error",
|
||
"command": "save_file",
|
||
"message": f"保存文件失败: {str(e)}",
|
||
"timestamp": datetime.now().isoformat(),
|
||
"client_address": f"{address[0]}:{address[1]}"
|
||
}
|
||
|
||
if __name__ == "__main__":
|
||
server = JSONSocketServer()
|
||
try:
|
||
server.start_server()
|
||
except KeyboardInterrupt:
|
||
print("\n⛔ 服务器已停止") |