realtime_manager.cpython-312.pyc•7.72 kB
�
�a1h� � � � d Z ddlZddlZddlZddlmZmZmZmZ ddl m
Z
ddl m Z ddl m
Z
ej d� Z G d� d e
� Z G d
� d � Z e� Zy) uN
实时更新管理器
负责WebSocket连接管理和数据变更事件广播
� N)�Dict�Set�Any�Optional)� WebSocket)�datetime)�Enum�realtime_managerc �L � e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdZdZy)� EventTypeu 事件类型枚举� device_added�device_updated�device_deleted�credential_added�credential_updated�credential_deleted�topology_updated�topology_cleared�topology_discovery_started� topology_discovery_completed� scan_started�scan_completed�scan_results_updated�scan_results_cleared�system_status_changedN)�__name__�
__module__� __qualname__�__doc__� DEVICE_ADDED�DEVICE_UPDATED�DEVICE_DELETED�CREDENTIAL_ADDED�CREDENTIAL_UPDATED�CREDENTIAL_DELETED�TOPOLOGY_UPDATED�TOPOLOGY_CLEARED�TOPOLOGY_DISCOVERY_STARTED� TOPOLOGY_DISCOVERY_COMPLETED� SCAN_STARTED�SCAN_COMPLETED�SCAN_RESULTS_UPDATED�SCAN_RESULTS_CLEARED�SYSTEM_STATUS_CHANGED� � �"F:\NetBrainMCP\realtime_manager.pyr r sb � � �!�L�%�N�%�N� *��-��-�� *��)��!=� �#A� � "�L�%�N�1��1�� 4�r0 r c � � e Zd ZdZd� Z ddededeee de
fd�Z defd �Z ded
e
eef de
fd �Z dd e d
e
eef d
eee fd�Zdefd�Zde
eef fd�Zy)�RealtimeUpdateManageru 实时更新管理器c �J � i | _ i | _ t j d� y )Nu& 🔄 实时更新管理器已初始化�� connections�
subscriptions�logger�info��selfs r1 �__init__z RealtimeUpdateManager.__init__/ s � �13���8:���� � �<�=r0 N� websocket�
connection_id� subscribe_to�returnc � �. K � |j � � d{ ��� || j |< |�t t � }|| j |< t
j
d|� d|D �cg c] }|j �� c}� �� | j |d||D �cg c] }|j �� c}t j � j � d�� � d{ ��� y7 ��c c}w c c}w 7 �# t $ r%}t
j d|� d|� �� Y d}~y d}~ww xY w�w)
u5
添加新的WebSocket连接
Args:
websocket: WebSocket连接对象
connection_id: 连接唯一标识符
subscribe_to: 要订阅的事件类型集合，默认订阅所有事件
Returns:
bool: 连接是否成功
Nu ✅ 新的实时更新连接: u , 订阅事件: �connection_established)�typer> �subscribed_events� timestampTu ❌ WebSocket连接失败: u
, 错误: F)�acceptr6 �setr r7 r8 r9 �value�send_to_connectionr �now� isoformat� Exception�error)r; r= r> r? �es r1 �connectz RealtimeUpdateManager.connect7 s � �� � ��"�"�$� $� $� /8�D� � �]� +� �#� "�9�~� �0<�D� � �}� -� �K�K�8� ��GW�jv�Xw�jv�ef�YZ�Y`�Y`�jv�Xw�Wx�y� z� �)�)�-�0�!.�7C�%D�|�!�a�g�g�|�%D� %�\�\�^� 5� 5� 7� :� �
�
� �+
%�� Yx�� &E�
�� � � �L�L�6�}�o�Z�PQ�s�S� T��� �si �D�C$ �C�A
C$ �"C�5C$ �C �#-C$ �C"�C$ �D�C$ � C$ �$ D�-D
�D�
D�Dc � � || j v r
| j |= || j v r
| j |= t j d|� �� y)u 断开WebSocket连接u 🔌 WebSocket连接已断开: Nr5 )r; r> s r1 �
disconnectz RealtimeUpdateManager.disconnect` sN � � �D� ,� ,� ,�� � � �/� �D� .� .� .��"�"�=�1�� � �5�m�_�E�Fr0 �datac � �. K � || j vry | j | }|j t j |d�� � � d{ ��� y7 �# t $ r6}t
j
d|� d|� �� | j |� Y d}~yd}~ww xY w�w)u�
向指定连接发送数据
Args:
connection_id: 连接ID
data: 要发送的数据
Returns:
bool: 发送是否成功
F)� ensure_asciiNTu ❌ 发送数据到连接 u 失败: )r6 � send_text�json�dumpsrL r8 rM rQ )r; r> rR r= rN s r1 rI z(RealtimeUpdateManager.send_to_connectioni s� � �� � �� 0� 0� 0�� � �(�(� �7�I��%�%�d�j�j��E�&J�K� K� K��
L��� � �L�L�5�m�_�I�a�S�Q� R� �O�O�M� *��� �s@ �B�8A � A� A �B�A � B� ,B
�B�
B�B�
event_type�exclude_connectionsc � �� K � |�
t � }d|j |t j � j � d�}d}g }| j
j
� D ]B \ }}||v r� ||v s�| j ||� � d{ ��� } | r|dz
}�2|j |� �D |D ] }
| j |
� � t j d|j � d|� d�� y7 �^�w) u�
广播事件到所有订阅的连接
Args:
event_type: 事件类型
data: 事件数据
exclude_connections: 要排除的连接ID集合
N�realtime_event)rC rX rR rE r � u 📡 广播事件 u 完成: u 个连接接收成功) rG rH r rJ rK r7 �itemsrI �appendrQ r8 r9 ) r; rX rR rY �
event_message�successful_sends�failed_connectionsr> rD �success� failed_ids r1 �broadcast_eventz%RealtimeUpdateManager.broadcast_event� s� � �� � � &�"%�%� � %�$�*�*��!� � � �1�1�3�
�
� �� ��04�0B�0B�0H�0H�0J� ,�M�,� � 3�3�� � .�.� $� 7� 7�
�}� U�U���$��)�$�&�-�-�m�<� 1K� ,�I� �O�O�I� &� ,� � � �(��)9�)9�(:�)�DT�CU�Uj�k�l� V�s �A+C$�.C$�C"�A C$c �, � t | j � S )u 获取当前连接数量)�lenr6 r: s r1 �get_connection_countz*RealtimeUpdateManager.get_connection_count� s � ��4�#�#�$�$r0 c � � | j j � D ���ci c]" \ }}|d|D �cg c] }|j �� c}i��$ c}}}S c c}w c c}}}w )u 获取活跃连接信息rD )r7 r] rH )r; r> �eventsrN s r1 �get_active_connectionsz,RealtimeUpdateManager.get_active_connections� sh � � *.�);�);�)A�)A�)C�
� *D�%�
�v�
�#�v�%>�v�!�a�g�g�v�%>� �
� *D�
�
��%>��
s �A�A � A� A)N)r r r r r< r �strr r r �boolrO rQ r r rI rd �intrg rj r/ r0 r1 r3 r3 , s� � � �>� ?C�'�y� '�� '�#+�C� �N�#;�'�GK�'�RG�� G��c� ��c�3�h� � �TX� �2 GK�*m� � *m��c�3�h� � *m�19�#�c�(�1C�*m�X%�c� %�
��S�#�X��
r0 r3 )r �asynciorV �logging�typingr r r r �fastapir r �enumr � getLoggerr8 r r3 r
r/ r0 r1 �<module>rt s[ ���
� � � +� +� � � � �� � �-� .��4�� 4�6L
� L
�^ )�*� r0