3 changed files with 81 additions and 5 deletions
@ -0,0 +1,60 @@ |
|||
""" |
|||
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() |
|||
Loading…
Reference in new issue