jUddlZddlZddlZddlZddlZddlZddlmZmZddl m Z m Z m Z m Z mZmZddlmZddlmZmZmZddlmZe rddlmZej4eZGdd ej:ZGd d ej:Ze rdd l m!Z!m"Z"Gd deZ#Gdde#Z$Gdde#Z%Gdde#Z&Gdde#Z'Gdde#Z(Gdde#Z)Gdde#Z*de+de,fdZ-e$d e%d!e&d"e'd#e(d$e)d%e*d&iZ.e/e0e#e+fe1d'<d(d)d*ee+e#ffd+Z2Gd,d-Z3Gd.d/Z4Gd0d1Z5Gd2d3Z6y)4N)ABCabstractmethod) TYPE_CHECKINGDictListLiteralOptionalUnion) get_pool_name)record_connection_handoff!record_connection_relaxed_timeoutrecord_maint_notification_count)Number)&MaintNotificationsAbstractRedisClusterceZdZdZdZdZy)MaintenanceStatenonemoving maintenanceN)__name__ __module__ __qualname__NONEMOVING MAINTENANCEc/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/redis/maint_notifications.pyrrs D FKrrc*eZdZdZdZdZdZdZdZdZ y) EndpointTypez@Valid endpoint types used in CLIENT MAINT_NOTIFICATIONS command.z internal-ipz internal-fqdnz external-ipz external-fqdnrc|jS)z$Return the string value of the enum.)valueselfs r__str__zEndpointType.__str__'s zzrN) rrr__doc__ INTERNAL_IP INTERNAL_FQDN EXTERNAL_IP EXTERNAL_FQDNrr%rrrr r s#JK#MK#M Drr )$MaintNotificationsAbstractConnection(MaintNotificationsAbstractConnectionPoolcneZdZdZdedefdZdefdZede fdZ edefdZ edefd Z y ) MaintenanceNotificationa Base class for maintenance notifications sent through push messages by Redis server. This class provides common functionality for all maintenance notifications including unique identification and TTL (Time-To-Live) functionality. Attributes: id (int): Unique identifier for this notification ttl (int): Time-to-live in seconds for this notification creation_time (float): Timestamp when the notification was created/read idttlc||_||_tj|_|j|jz|_y)z Initialize a new MaintenanceNotification with unique ID and TTL functionality. Args: id (int): Unique identifier for this notification ttl (int): Time-to-live in seconds for this notification N)r/r0time monotonic creation_time expire_at)r$r/r0s r__init__z MaintenanceNotification.__init__@s7!^^-++dhh6rreturnc^tj|j|jzkDS)z Check if this notification has expired based on its TTL and creation time. Returns: bool: True if the notification has expired, False otherwise )r2r3r4r0r#s r is_expiredz"MaintenanceNotification.is_expiredMs%~~4#5#5#@AArcy)z Return a string representation of the maintenance notification. This method must be implemented by all concrete subclasses. Returns: str: String representation of the notification Nrr#s r__repr__z MaintenanceNotification.__repr__Ws rcy)a Compare two maintenance notifications for equality. This method must be implemented by all concrete subclasses. Notifications are typically considered equal if they have the same id and are of the same type. Args: other: The other object to compare with Returns: bool: True if the notifications are equal, False otherwise Nrr$others r__eq__zMaintenanceNotification.__eq__cs rcy)a Return a hash value for the maintenance notification. This method must be implemented by all concrete subclasses to allow instances to be used in sets and as dictionary keys. Returns: int: Hash value for the notification Nrr#s r__hash__z MaintenanceNotification.__hash__ts rN) rrrr&intr6boolr9rstrr;r?rArrrr.r.3sy  73 7S 7BDB  #     t    #    rr.cdeZdZdZdedeedeedeffd ZdefdZde fd Z defd Z xZ S) NodeMovingNotificationz This notification is received when a node is replaced with a new node during cluster rebalancing or maintenance operations. r/ new_node_host new_node_portr0cBt|||||_||_y)ai Initialize a new NodeMovingNotification. Args: id (int): Unique identifier for this notification new_node_host (str): Hostname or IP address of the new replacement node new_node_port (int): Port number of the new replacement node ttl (int): Time-to-live in seconds for this notification N)superr6rGrH)r$r/rGrHr0 __class__s rr6zNodeMovingNotification.__init__s$ S!**rr7c@|j}td|tjz }|jj d|j d|jd|jd|jd|jd|d|d d |jd S) Nr(id=z, new_node_host='z', new_node_port=, ttl=, creation_time= , expires_at= , remaining=.1f s, expired=)) r5maxr2r3rKrr/rGrHr0r4r9r$ expiry_time remainings rr;zNodeMovingNotification.__repr__snn ;)99: ~~&&'(''"0012!//0188*!//01%'"3(()  rct|tsy|j|jk(xr4|j|jk(xr|j|jk(S)z Two NodeMovingNotification notifications are considered equal if they have the same id, new_node_host, and new_node_port. F) isinstancerFr/rGrHr=s rr?zNodeMovingNotification.__eq__sY %!78 GGuxx  :""e&9&99 :""e&9&99 rc |jrt|jnd}t|jj t|j t|j|fS#t$rd}YVwxYw)a Return a hash value for the notification to allow instances to be used in sets and as dictionary keys. Returns: int: Hash value based on notification type class name, id, new_node_host and new_node_port Nr) rHrB ValueErrorhashrKrr/rDrG)r$ node_ports rrAzNodeMovingNotification.__hash__st 373E3ED../4I''DGG D&&'     I s#A.. A<;A<) rrrr&rBr rDr6r;rCr?rA __classcell__rKs@rrFrFs_ + + }+ } +  +( # "  t   # rrFcPeZdZdZdedeffd ZdefdZdefdZ defdZ xZ S) NodeMigratingNotificationat Notification for when a Redis cluster node is in the process of migrating slots. This notification is received when a node starts migrating its slots to another node during cluster rebalancing or maintenance operations. Args: id (int): Unique identifier for this notification ttl (int): Time-to-live in seconds for this notification r/r0c&t|||yNrJr6r$r/r0rKs rr6z"NodeMigratingNotification.__init__ S!rr7c&|j|jz}td|tjz }|j j d|jd|jd|jd|d|dd|jd S NrrMrNrOrPrQrRrSrT r4r0rUr2r3rKrr/r9rVs rr;z"NodeMigratingNotification.__repr__((4883 ;)99: ~~&&'(''88*!//01%'"3(()  rct|tsy|j|jk(xrt|t|uS)z Two NodeMigratingNotification notifications are considered equal if they have the same id and are of the same type. F)rZrbr/typer=s rr?z NodeMigratingNotification.__eq__s7 %!:;ww%(("@tDzT%['@@rcjt|jjt|jfSz Return a hash value for the notification to allow instances to be used in sets and as dictionary keys. Returns: int: Hash value based on notification type and id r]rKrrBr/r#s rrAz"NodeMigratingNotification.__hash__&T^^,,c$''l;<ww%(("@tDzT%['@@rcjt|jjt|jfSrorpr#s rrAz%OSSNodeMigratingNotification.__hash__rqrrd)rrrr&rxrBr rDr6r;rCr?rAr_r`s@rrrsU K $ }  #  AtA=#=rrc neZdZdZdZdedeeeeeeffffd Z defdZ de fdZ defd Z xZS) OSSNodeMigratedNotificationa Notification for when a Redis OSS API client is used and a node has completed migrating slots. This notification is received when a node has finished migrating all its slots to other nodes during cluster rebalancing or maintenance operations. Args: id (int): Unique identifier for this notification nodes_to_slots_mapping (Dict[str, List[Dict[str, str]]]): Map of source node address to list of destination mappings. Each destination mapping is a dict with the destination node address as key and the slot range as value. Structure example: { "127.0.0.1:6379": [ {"127.0.0.1:6380": "1-100"}, {"127.0.0.1:6381": "101-200"} ], "127.0.0.1:6382": [ {"127.0.0.1:6383": "201-300"} ] } Where: - Key (str): Source node address in "host:port" format - Value (List[Dict[str, str]]): List of destination mappings where each dict contains destination node address as key and slot range as value xr/nodes_to_slots_mappingcPt||tj||_yrd)rJr6rrxr)r$r/rrKs rr6z$OSSNodeMigratedNotification.__init__s# 8DDE&<#rr7c@|j|jz}td|tjz }|j j d|jd|jd|jd|jd|d|dd |jd S) NrrMz, nodes_to_slots_mapping=rNrOrPrQrRrSrT) r4r0rUr2r3rKrr/rr9rVs rr;z$OSSNodeMigratedNotification.__repr__s((4883 ;)99: ~~&&'(''&&*&A&A%BC88*!//01%'"3(()  rct|tsy|j|jk(xrt|t|uS)z Two OSSNodeMigratedNotification notifications are considered equal if they have the same id and are of the same type. F)rZrr/rmr=s rr?z"OSSNodeMigratedNotification.__eq__rrcjt|jjt|jfSrorpr#s rrAz$OSSNodeMigratedNotification.__hash__rqr)rrrr&rxrBrrDrr6r;rCr?rAr_r`s@rrrsc:K= =!%S$tCH~*>%> ?=  #  AtA=#=rrhostr7c|jjd}d|vrygd}|D]}tj||syy)a Determine if an FQDN is likely to be internal/private. This uses heuristics based on RFC 952 and RFC 1123 standards: - .local domains (RFC 6762 - Multicast DNS) - .internal domains (common internal convention) - Single-label hostnames (no dots) - Common internal TLDs Args: host (str): The FQDN to check Returns: bool: True if the FQDN appears to be internal/private .T)z\.local$z \.internal$z\.corp$z\.lan$z \.intranet$z \.private$F)lowerrstripresearch)r host_lowerinternal_patternspatterns r_is_private_fqdnrsQ $$S)J *% 99Wj )% rr MIGRATINGMIGRATED FAILING_OVER FAILED_OVER SMIGRATING SMIGRATEDnotification_types_mapping connectionr+ notificationc <tjtjrdd} |jr|jj nd}|r|dnd}tjd|d|d|jd|yy#t tf$rYBwxYw)Nz#Handling maintenance notification: , with connection: , connected to ip z, local socket port: ) logger isEnabledForloggingDEBUG_sock getsocknameAttributeErrorOSErrordebugget_resolved_ip)rrsocket_addresss radd_debug_log_for_notificationrOs7==) 2<2B2B   ,,. 3A^A.dN  1,@ *|+=j>X>X>Z=[\""0!1 3 *(   s1B BBc zeZdZdZ ddeeedfdedeedee fdZ d e fd Z d efd Z d e d dd e fdZy)MaintNotificationsConfiga# Configuration class for maintenance notifications handling behaviour. Notifications are received through push notifications. This class defines how the Redis client should react to different push notifications such as node moving, migrations, etc. in a Redis cluster. autoNenabledproactive_reconnectrelaxed_timeout endpoint_typec<||_||_||_||_y)a! Initialize a new MaintNotificationsConfig. Args: enabled (bool | "auto"): Controls maintenance notifications handling behavior. - True: The CLIENT MAINT_NOTIFICATIONS command must succeed during connection setup, otherwise a ResponseError is raised. - "auto": The CLIENT MAINT_NOTIFICATIONS command is attempted but failures are gracefully handled - a warning is logged and normal operation continues. - False: Maintenance notifications are completely disabled. Defaults to "auto". proactive_reconnect (bool): Whether to proactively reconnect when a node is replaced. Defaults to True. relaxed_timeout (Number): The relaxed timeout to use for the connection during maintenance. If -1 is provided - the relaxed timeout is disabled. Defaults to 20. endpoint_type (Optional[EndpointType]): Override for the endpoint type to use in CLIENT MAINT_NOTIFICATIONS. If None, the endpoint type will be automatically determined based on the host and TLS configuration. Defaults to None. Raises: ValueError: If endpoint_type is provided but is not a valid endpoint type. N)rrrr)r$rrrrs rr6z!MaintNotificationsConfig.__init__ns#: .#6 *rr7c |jjd|jd|jd|jd|j d S)Nz (enabled=z, proactive_reconnect=z, relaxed_timeout=z, endpoint_type=rT)rKrrrrrr#s rr;z!MaintNotificationsConfig.__repr__s`~~&&'(||n%##'#;#;"<=#3345!//2   rc |jdk7S)aK Check if the relaxed_timeout is enabled. The '-1' value is used to disable the relaxed_timeout. If relaxed_timeout is set to None, it will make the operation blocking and waiting until any response is received. Returns: True if the relaxed_timeout is enabled, False otherwise. )rr#s ris_relaxed_timeouts_enabledz4MaintNotificationsConfig.is_relaxed_timeouts_enableds##r))rrrr+c|j |jS tj|}|j}|rtj Stj S#t$rYnwxYw|j}|rS tj|}|j}|rtjStjS#t$rYnwxYwt|}|rtjStjS)aY Determine the appropriate endpoint type for CLIENT MAINT_NOTIFICATIONS command. Logic: 1. If endpoint_type is explicitly set, use it 2. Otherwise, check the original host from connection.host: - If host is an IP address, use it directly to determine internal-ip vs external-ip - If host is an FQDN, get the resolved IP to determine internal-fqdn vs external-fqdn Args: host: User provided hostname to analyze connection: The connection object to analyze for endpoint type determination Returns: ) r ipaddress ip_address is_privater r'r)r\rr(r*r)r$rrip_addrr resolved_ips rget_endpoint_typez*MaintNotificationsConfig.get_endpoint_types(    )%% % **40G ++J/9<++ W|?W?W W    !002  #..{;$// "!..&33    &d+ -7|))W\=W=WWs/2A A A)(A)?2C2C C C)rT N)rrrr&r rCrr rr r6rDr;rrrrrrrds17$(,.04 +tWV_,- +" +"&) +  - +D #  *T *4X4X%K4X 4XrrcleZdZdddeddfdZddZdZd Zd efd Z d e fd Z dd e e fdZd e fdZy)MaintNotificationsPoolHandlerpoolr,configr7Nc~||_||_t|_t j |_d|_yrd)rrset_processed_notifications threadingRLock_lockr)r$rrs rr6z&MaintNotificationsPoolHandler.__init__s1   (+%__& rc||_yrd)r)r$rs rset_connectionz,MaintNotificationsPoolHandler.set_connections $rct|j|j}|j|_|j|_d|_|Srd)rrrrrrr$copys rget_handler_for_connectionz8MaintNotificationsPoolHandler.get_handler_for_connections=-TYY D(,(E(E%ZZ  rc|j5t|jD].}|js|jj |0 dddy#1swYyxYwrdrtuplerr9remover$rs rremove_expired_notificationsz:MaintNotificationsPoolHandler.remove_expired_notificationsI ZZ %d&C&C D **,1188F!EZZ)AAA'rc|jt|tr|j|Stj d|yNUnhandled notification type: )rrZrFhandle_node_moving_notificationrerrorrs rhandle_notificationz1MaintNotificationsPoolHandler.handle_notifications> ))+ l$: ;77 E E LL8G Hrc 0|jjs|jjsy|j5||jvr dddy|j j5t jd|d|jd|jr|jjnd|jjs|jjr |jr|jjnd}t|j ddr|j jd|j jtjt!||jj"|j$|ddd|jjr\|j$|j'|n>t)j*|j,d z |j&|f j/tjt!|d }|j$|j1d |j$i|jjr<|j1|jj"|jj"d |j j2di|t|j ddr|j jddddt)j*|j,|j4|f j/t7t9|j |jj;|dddy#1swYxYw#1swYyxYw)Nz#Handling node MOVING notification: rrset_in_maintenanceFTconnected_address)statemaintenance_notification_hashr host_addressmatching_addressmatching_patternupdate_notification_hashinclude_free_connections)args)maintenance_staterr)socket_timeoutsocket_connect_timeout pool_namer)rrrrrrrrrr getpeernamegetattrrupdate_connections_settingsrrr]rrGrun_proactive_reconnectrTimerr0startupdateupdate_connection_kwargshandle_node_moved_notificationr r add)r$rmoving_address_srckwargss rrz=MaintNotificationsPoolHandler.handle_node_moving_notifications //KK;;=  ZZt<<< Z 9,H((,'88J<@OOt668QUVX KK33{{>>@:>335d'tyy*>F 44T:II99.556:<6H(, (C(C%1%?%?);)<1515: {{66'55A 889KL%OO , 0 01 4 $ < <&8%:$eg.>-D-D9=l9K$F$11= & (B(B {{>>@ 26++2M2M:>++:U:U 7DII66@@tyy*>F 44U;S!V OO  33"_ eg %' 2   ) ) - -l ;yZ!Zs+N N -J N:A=N N N  Nr c |j5|jj5|jj||jj|ddddddy#1swYxYw#1swYyxYw)z Run proactive reconnect for the pool. Active connections are marked for reconnect after they complete the current command. Inactive connections are disconnected and will be connected on next use. )r N)rr'update_active_connections_for_reconnectdisconnect_free_connections)r$r s rrz5MaintNotificationsPoolHandler.run_proactive_reconnectgsi ZZ AA'9B  55'96!Z Zs"A:9A.A:.A7 3A::Bc t|}|j5tjd|d|jd|jr|jj nd|j jjd|k(r|j jjd}|j jjd}|j jjd}tjd|||d }|j jdi||j j5|jj}|jj}|j jd tjd|d d ||d ddddddy#1swYxYw#1swYyxYw)zN Handle the cleanup after a node moving notification expires. z5Reverting temporary changes related to notification: rrNrorig_host_addressorig_socket_timeoutorig_socket_connect_timeout)rrrrrrnotification_hashT) rrrmatching_notification_hashrrreset_relaxed_timeoutreset_host_addressrr)r]rrrrrrconnection_kwargsgetrrr rrrr) r$rr orig_hostrorig_connect_timeoutr rrs rr z)>59%&9.B  3 22 >| L LL8G Hrc |j|j5||jvs||jvr dddytj d||jj |g}t}|jjD]\}}|jd\}}|jjj||}||j ||D]G} | jD]2} | jd\} } |j| t!| f4I|jjj#d|t|} | j%|jjj&j)} | D]~}|j*|j*j,j5d}||vre|j*j,j/D]}|j1t3t5|j*j,d}nFtj7t8j:r#tj d|j<d ||jjj&j)vrd|j*j,j?D]}|jA|s)t3t5|j*j,ddd|jj ||jjC|dddy#1swYxYw#1swYyxYw) Nz!Handling SMIGRATED notification: :)rr-F)disconnect_startup_nodes_poolsadditional_startup_nodes_inforTzSMIGRATED: Node z9 not affected by maintenance, skipping mark for reconnect)"rrrCrrrr rritemssplitrA nodes_managerget_nodekeysappendrB initializeunion nodes_cachevaluesredis_connectionconnection_pool_get_in_use_connectionsmark_for_reconnectr r rrrname_get_free_connections disconnectr)r$rrKaffected_nodes src_address dest_mappingssrc_hostsrc_portsrc_node dest_mapping dest_address dest_host dest_port all_nodes current_nodehandoff_recordedconns rrGzJOSSMaintNotificationsHandler.handle_oss_maintenance_completed_notificationEs/ ))+ ZZ 1 114#@#@@Z LL<\NK L    ! !, /-/ ) UN44::<%0%6%6s%;"(..<<EE!F'"&&x0$1L(4(9(9(; /;/A/A#/F, 95<<&I7)<%2="    - - 8 8/4.K 9  N+I!##11==DDFI!* 008!22BBHH',$#~5 %1$A$A$Q$Q$i$i$kD 335%l2&3 , = = M M' ,0(!..w}}="LL"2<3D3D2EF>!? %#22@@LLSSUV %1$A$A$Q$Q$g$g$iD OO-%j 05*7$0$A$A$Q$Q+"EIH!*X  ) ) - -l ;    $ $\ 2KZvIHwZs,M>FM>!EM1'AM>1M; 6M>>N) rrrrr6rrr.rrrGrrrr@r@#sL '@ ') '  'G I0GI j37j3rr@)7enumrrrrr2abcrrtypingrrrrr r redis.observability.attributesr redis.observability.recorderr r r redis.typingr redis.clusterr getLoggerrrEnumrr redis.connectionr+r,r.rFrbrurrrrrDrCrrr=rmr>rrrrr@rrrrus   #FF8  D   8 $ tyy 499  L cL ^M 4M `.= 7.=b/=6/=d.="9.=b/=!8/=d6=#:6=rH="9H=V%3%4%RH{j  ,HD&=!>!CD 6 445 *uXuXpOOdrrjL3L3r