You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.1 KiB
60 lines
2.1 KiB
"""
|
|
WebSocket消息序列化器
|
|
类似于.NET的序列化器，负责消息的序列化和反序列化
|
|
"""
|
|
import json
|
|
from typing import Any, Dict
|
|
from app.utils.structured_log import get_structured_logger, LogLevel
|
|
|
|
logger = get_structured_logger(__name__, LogLevel.INFO)
|
|
|
|
class WebSocketMessageSerializer:
|
|
"""WebSocket消息序列化器
|
|
|
|
单一职责：
|
|
- 序列化消息为JSON字符串
|
|
- 反序列化JSON字符串为消息对象
|
|
- 处理消息格式转换
|
|
"""
|
|
|
|
def __init__(self):
|
|
logger.info("WebSocket消息序列化器初始化完成")
|
|
|
|
def serialize(self, message: Dict[str, Any]) -> str:
|
|
"""序列化消息为JSON字符串
|
|
|
|
类似于.NET的 _serializer.Serialize(processedMessage)
|
|
"""
|
|
try:
|
|
# 转换字段名为小写，适配.NET序列化器
|
|
converted_message = self._convert_to_lowercase(message)
|
|
# 使用JSON序列化
|
|
serialized_data = json.dumps(converted_message, ensure_ascii=False)
|
|
logger.debug(f"消息序列化成功: {serialized_data}")
|
|
return serialized_data
|
|
except Exception as e:
|
|
logger.error(f"消息序列化失败: {e}")
|
|
raise
|
|
|
|
def _convert_to_lowercase(self, obj: Any) -> Any:
|
|
"""递归转换字典中的字段名为小写"""
|
|
if isinstance(obj, dict):
|
|
return {key.lower(): self._convert_to_lowercase(value) for key, value in obj.items()}
|
|
elif isinstance(obj, list):
|
|
return [self._convert_to_lowercase(item) for item in obj]
|
|
else:
|
|
return obj
|
|
|
|
def deserialize(self, data: str) -> Dict[str, Any]:
|
|
"""反序列化JSON字符串为消息对象"""
|
|
try:
|
|
# 使用JSON反序列化
|
|
message = json.loads(data)
|
|
logger.debug(f"消息反序列化成功: {message}")
|
|
return message
|
|
except Exception as e:
|
|
logger.error(f"消息反序列化失败: {e}")
|
|
raise
|
|
|
|
# 创建全局序列化器实例
|
|
websocket_serializer = WebSocketMessageSerializer()
|