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