
    x-jt                        d dl Z d dlZ G d d          Zedk    rC ed          ZdddZe                                D ]\  ZZe	                    ee           e
                    d	          Z ee           e                                D ]\  ZZee         ek    sJ e	                    d
d            ee                    d
                     e                    d
          dk    sJ e                    d
            ee                    d                      ee                    d                     e                    d          dk    sJ dS dS )    Nc                   6    e Zd Zd
dZd Zd Zd Zd ZddZd	S )KVClientlocalhost:2379c                 H    |                     d          r|nd| | _        d S )Nzhttp://)
startswithendpoint)selfr   s     i/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/distributed/launch/utils/kv_client.py__init__zKVClient.__init__   s0     ++I66PHH<Ph<P<P 	    c                     |                     d          r|nd| }| j         | }	 t          j        ||d d          }|j        dk    rdS dS #  Y dS xY w)N/T)datatimeoutfollow_redirects   F)r   r   httpxpoststatus_code)r	   keyvalueurs        r
   putzKVClient.put   sx    ^^C((7cci#ii}#c##	
15$NNNA}##tu	55s   #A Ac                     |                     d          r|nd| }| j         | }	 t          j        |d d          }|j        dk    r*|                                }|                    |d          S dS #  Y dS xY w)Nr   Tr   r   r    errorr   r   r   getr   json)r	   r   r   r   rets        r
   r    zKVClient.get&   s    ^^C((7cci#ii}#c##		!TDAAAA}##ffhhwwsB'''w	22s   AA7 7A<c                     |                     d          r|nd| }| j         | }	 t          j        |d d          }|j        dk    r|                                S d S #  Y dS xY w)Nr   Tr   r   r   r   r	   r   r   r   s       r
   
get_prefixzKVClient.get_prefix3   s    ^^C((7cci#ii}#c##		!TDAAAA}##vvxx $#	22s   5A! !A&c                     |                     d          r|nd| }| j         | }	 t          j        |d d          }|j        dk    rdS dS #  Y dS xY w)Nr   Tr   r   F)r   r   r   deleter   r$   s       r
   r'   zKVClient.delete=   sv    ^^C((7cci#ii}#c##	QtDDDA}##tu	55s   "A A   c                     t          j                     |z   }t          j                     |k     r4|                     d          dk    rdS t          j                     |k     2d S d S )N/healthyokT)timer    )r	   r   ends      r
   wait_server_readyzKVClient.wait_server_readyI   s^    ikkG#ikkCxx
##t++t ikkCr   N)r   )r(   )	__name__
__module____qualname__r   r   r    r%   r'   r.    r   r
   r   r      sx        
 
 
 


 
 
    
 
 
     r   r   __main__zhttp://localhost:8090rank1rank2)z
/workers/1z
/workers/2z/workersr   r   z/keyr*   r+   )r,   r   r   r/   clir   itemskvr   r%   xprintr    r'   r2   r   r
   <module>r<      s    9 9 9 9 9 9 9 9x z
(*
+
+C!99D

  11z""A	E!HHH

  1tqyyyyyGGE7	E#''%..775>>W$$$$JJu	E#''&//	E#''*

77:$&&&&!   '&r   