jR&ddlZddlZddlZddlZddlZddlmZmZmZddl m Z ddl m Z m Z mZddlmZddlmZmZmZddlmZejeZdZd efd ZGd d eZGd dZGddeZeZ dS)N)CallableTupleUnion)parse)is_legal_address is_legal_host is_legal_port) GrpcHandler)ConnectionConfigExceptionConnectionNotExistExceptionExceptionsMessage)ConfigifunccBtj_fd}|S)zD Decorator in order to achieve thread-safe singleton class. cVj5|i|cdddS#1swxYwYdSN)__lock__)argskwargsrs b/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/pymilvus/orm/connections.py lock_funczsynchronized..lock_func)s ] ) )4((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s "") threadingLockr)rrs` r synchronizedr#s5N$$DM))))) cDeZdZdZdfd ZdZefdZxZS)SingleInstanceMetaClassNreturnc:tj|i|dSr)super__init__clsrr __class__s rr!z SingleInstanceMetaClass.__init__3s%$)&)))))rc|jr|jS|||_|jj|i||jSr)instance__new__r!)r#rrs r__call__z SingleInstanceMetaClass.__call__6sI < < {{3''  t.v...|rc>tj|g|Ri|Sr)r r'r"s rr'zSingleInstanceMetaClass.__new__=s)uwws4T444V444rrN) __name__ __module__ __qualname__r&r!r(rr' __classcell__)r$s@rrr0svH******5555\55555rrc8eZdZdedededdfdZdZdefd ZdS) ReconnectHandlerconnsconnection_namerrNcn||_||_||_d|_t j|_dS)NF)r2r1_kwargs is_idle_staterrreconnect_lock)selfr1r2rs rr!zReconnectHandler.__init__Cs6.  "'n..rc td}td|dtj||jstddS|j5tdd|_ td|j|j n*#t$rt dYnwxYwd}|s td |jj |j fi|j d }nL#t$r?}t d |d |dtj|Yd}~nd}~wwxYw|n#d}|s td |jj |j fi|j d }nL#t$r?}t d |d |dtj|Yd}~nd}~wwxYw|wxYwtd ddddS#1swxYwYdS)Nz%state is idle, schedule reconnect in z secondsz"idle state changed, skip reconnectzreconnect on idle stateFz#try disconnecting old connection...zdisconnect failed: {e}ztry reconnecting...Tzreconnect failed: z, try again after reconnected)loggerdebugtimesleepr5r6infor1 disconnectr2 Exceptionwarningconnectr4)r7check_after_secondsr:es rcheck_state_and_reconnect_laterz0ReconnectHandler.check_state_and_reconnect_laterJs Z=PZZZ[[[ &'''!  LL= > > > F   ' ' KK1 2 2 2!&D  8 BCCC %%d&:;;;; 9 9 9788888 9$ % 88 %:;;;* *4+?PP4<PPP&* $888cccFYccc #677777777 8 & 8$ % 88 %:;;;* *4+?PP4<PPP&* $888cccFYccc #677777777 8 & 8 8 8 8 8 KK & & &) ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's"H-9B<;E4<$C# E4"C##E4&H-+9D%$H-% E./5E)$H-)E..H-4H:9F43H4 G= >5G8 3H8G= =HH--H14H1statec2td||j5|jddkrd|_ ddddSd|_t j|jddddS#1swxYwYdS)Nzstate change to: idleFT)target) r;r<r6valuer5rThreadrFstart)r7rGs rreconnect_on_idlez"ReconnectHandler.reconnect_on_idlegs 000111   R R{1~''%*" R R R R R R R R"&D   D$H I I I O O Q Q Q  R R R R R R R R R R R R R R R R R RsB  3B  BB)r+r,r-objectstrr!rFrOrrr0r0Bsu/f/s/F/t////''':RvRRRRRRrr0c &eZdZdZddZdedeeeffdZdedee j ffd Z d Z dd ededededee j ff d Z defdZdefdZejd d dd fdedededededdf dZdefdZdefdZdedefdZejfdedefdZdS) ConnectionszRClass for managing all connections of milvus. Used as a singleton in this module.rNc<i|_i|_d|_tjdkr?|tj\}}||f|_|j|jnd|d}ndtjdtjd}|j ditj |idS)alConstructs a default milvus alias config default config will be read from env: MILVUS_URI and MILVUS_CONN_ALIAS with default value: default="localhost:19530" Read default connection config from environment variable: MILVUS_URI. Format is: [scheme://][@]host[:] scheme is one of: http, https, or Examples: localhost localhost:19530 test_user@localhost:19530 http://test_userlocalhost:19530 https://test_user:password@localhost:19530 N)useraddress:rR) _alias_connected_alias_env_urir MILVUS_URI$_Connections__parse_address_from_uriusername DEFAULT_HOST DEFAULT_PORTadd_connectionMILVUS_CONN_ALIAS)r7rX parsed_uridefault_conn_configs rr!zConnections.__init__ts( "   " ""&"?"?@Q"R"R GZ$j1DM0:/B/N ++TV"##   $1IIF4GII##  NNv79LMNNNNNrhostportct|sttjt |sttjdt |cxkrdksnd|d}t|dS)Nmessageriz port number z% out of range, valid range [0, 65535))rr r HostTyper PortTypeint)r7rfrgmsgs r__verify_host_portzConnections.__verify_host_portsT"" P+4E4NOOO OT"" P+4E4NOOO OCII%%%%%%%%LLLLC+C888 8&%ruric zd} tj|}nQ#t$rD}t||dt |jd|ddd}~wwxYwt|jdkr%t||d|j |j n tj }|j dkrdn tj }|j|jn|}|d |}|||t!|s#t||||fS) NzPIllegal uri: [{}], expected form 'http[s]://[user:password@]example.com[:12345]'z: rirhttps443rY)rurlparserAr formattyper+lennetlochostnamerr`schemerarg_Connections__verify_host_portr) r7rpillegal_uri_msgrdrErf default_portrgaddrs r__parse_address_from_uriz$Connections.__parse_address_from_urisp ^  ,,JJ   +*11#66SS477;KSSqSSS   z ! !Q & &+7M7Mc7R7R4TUUU[_ _&0&9&Ez""6K^ * 1W < $\\^^ . .ME6#66 9b)) 5"%% 62&& 62&&    D*---$+e2D2H2H2S2SW[2[2[/8I8VY^8^____  62..L %**;w*F*F)- X&!-DK  ' . .rrVrXc|dkr't|std|d|dfS|dkrJt|tr|dr|dfS||\}}||fS|dkr|n t j}|dkr|n t j}| |||d|}t|std|d|d |dfS) NrVzIllegal address: z%, should be in form 'localhost:19530'rizunix:rYzIllegal host: z or port: z+, should be in form of '111.1.1.1', '19530') rr isinstancerQ startswithr^rr`rar|) r7rXrprfrgparsed_host_portrs r__get_full_addresszConnections.__get_full_addresssA b==#G,, /^^^^D= "99#s## !w(?(? !Dy ";;C@@OGVF? " (; (; u---!!%!!%% +jjjjjj Tzrrct|ts*ttjt |z||jvr.|j|dSdS)zDisconnects connection from the registry. :param alias: The name of milvus connection :type alias: str riN) rrQr r AliasTyperwr[popcloser7rs rr@zConnections.disconnectsz %%% _+4E4ORVW\R]R]4]^^^ ^ D) ) )  ! % %e , , 2 2 4 4 4 4 4 * )rct|ts*ttjt |z|||j|ddS)z|Removes connection from the registry. :param alias: The name of milvus connection :type alias: str riN) rrQr r rrwr@rZrrs rremove_connectionzConnections.remove_connectionsi %%% _+4E4ORVW\R]R]4]^^^ ^  t$$$$$rdefaultrWpassworddb_nametokenc |dr tj|djdvr|ddst d|ddtj|dj }| st d|dd d l m }td |dd ||d} | t d| |d<t!j||d<|d<|d<|d<fd} dt$dt&fd} t)t*s*t t,jt1z|dd|dd|dd|ddf} t+|pdt+|pdt+|pd}}}| | rj| \} }| |d<rAjd| krt t,jz|Y|jp|}|jp|}|j !d}tE|dkr|dn|}|jdkrd|d<| d#i|||||d d Sj#Oj#\} }| |d<|j|jnd}|j|jnd}|jdkrd|d<| d#i||||d!d SjvrBtIj%}||d<| d#i|||d"|d St t,j&z)$a Constructs a milvus connection and register it under given alias. :param alias: The name of milvus connection :type alias: str :param kwargs: * *address* (``str``) -- Optional. The actual address of Milvus instance. Example address: "localhost:19530" * *uri* (``str``) -- Optional. The uri of Milvus instance. Example uri: "http://localhost:19530", "tcp:localhost:19530", "https://ok.s3.south.com:19530". * *host* (``str``) -- Optional. The host of Milvus instance. Default at "localhost", PyMilvus will fill in the default host if only port is provided. * *port* (``str/int``) -- Optional. The port of Milvus instance. Default at 19530, PyMilvus will fill in the default port if only host is provided. * *secure* (``bool``) -- Optional. Default is false. If set to true, tls will be enabled. * *user* (``str``) -- Optional. Use which user to connect to Milvus instance. If user and password are provided, we will add related header in every RPC call. * *password* (``str``) -- Optional and required when user is provided. The password corresponding to the user. * *token* (``str``) -- Optional. Serving as the key for identification and authentication purposes. Whenever a token is furnished, we shall supplement the corresponding header to each RPC call. * *keep_alive* (``bool``) -- Optional. Default is false. If set to true, client will keep an alive connection. * *db_name* (``str``) -- Optional. default database name of this connection * *client_key_path* (``str``) -- Optional. If use tls two-way authentication, need to write the client.key path. * *client_pem_path* (``str``) -- Optional. If use tls two-way authentication, need to write the client.pem path. * *ca_pem_path* (``str``) -- Optional. If use tls two-way authentication, need to write the ca.pem path. * *server_pem_path* (``str``) -- Optional. If use tls one-way authentication, need to write the server.pem path. * *server_name* (``str``) -- Optional. If use tls, need to write the common name. :raises NotImplementedError: If handler in connection parameters is not GRPC. :raises ParamError: If pool in connection parameters is not supported. :raises Exception: If server specified in parameters is not ready, we cannot connect to server. :example: >>> from pymilvus import connections >>> connections.connect("test", host="localhost", port="19530") rp)unixhttprstcpz.dbzuri: zU is illegal, needs start with [unix, http, https, tcp] or a local file endswith [.db]rizOpen local milvus failed, dir: z is not existsr)server_manager_instancezPass in the local path z, and run it using milvus-liteNzOpen local milvus failedrWrrrctd i|}|d}t|ttfr|n t j}|||ddr)|tj | d| dd| dd|j <tj|j<dS) Ntimeout)r keep_aliveFrrrrVrR)r rrrmfloatrMILVUS_CONN_TIMEOUT_wait_for_channel_readyregister_state_change_callbackr0rOrr[copydeepcopyrZ)rghtrr kwargs_copyr7s rconnect_milvusz+Connections.connect..connect_milvuss&&v&&B 9%%A%a#u66VaaF>P JJz " " " JJw % % % JJy" % % %+-D !% (!%v!6!6DK   rrrc4td|DS)Nc3"K|] }|dkV dS)rVNrR).0cs r z;Connections.connect..with_config..s&//1qBw//////r)any)rs r with_configz(Connections.connect..with_configs/////// /rrXrVrfrg/rIrsTr)rWrrr)rWrr)rrrR)'rrrur{lowerendswithr pathlibPathparentis_dirmilvus_lite.server_managerrr;r?start_and_get_urirrrboolrrQr rrwrrhas_connectionrZrr_rpathsplitrxr\dictr ConnLackConf)r7rrWrrrr parent_pathr local_urirrrrrdgroup connect_aliasrs`` @rrCzConnections.connect+s| ::e   &u !>!>!E!K!K!M!MV " " %=))%00 /IF5MIII",ve}55>> from pymilvus import connections >>> connections.connect("test", host="localhost", port="19530") >>> connections.list_connections() cJg|]}|j|df Sr)r[r)rkr7s r z0Connections.list_connections..s1MMMAD)--a667MMMr)rZ)r7s`rlist_connectionszConnections.list_connectionss"NMMMMMMMrct|ts*ttjt |z|j|iS)aC Retrieves connection configure by alias. :param alias: The name of milvus connection :type alias: str :return dict: The connection configure which of the name is alias. If alias does not exist, return empty dict. :example: >>> from pymilvus import connections >>> connections.connect("test", host="localhost", port="19530") >>> connections.list_connections() >>> connections.get_connection_addr('test') {'host': 'localhost', 'port': '19530'} ri)rrQr r rrwrZrrs rget_connection_addrzConnections.get_connection_addrsQ$%%% _+4E4ORVW\R]R]4]^^^ ^{ub)))rct|ts*ttjt |z||jvS)aCheck if connection named alias exists. :param alias: The name of milvus connection :type alias: str :return bool: if the connection of name alias exists. :example: >>> from pymilvus import connections >>> connections.connect("test", host="localhost", port="19530") >>> connections.list_connections() >>> connections.get_connection_addr('test') {'host': 'localhost', 'port': '19530'} ri)rrQr r rrwr[rs rrzConnections.has_connectionsI %%% _+4E4ORVW\R]R]4]^^^ ^---rct|ts*ttjt |z|j|d}|ttj |S)z!Retrieves a GrpcHandler by alias.riN) rrQr r rrwr[rr ConnectFirst)r7rconns r_fetch_handlerzConnections._fetch_handlersp%%% _+4E4ORVW\R]R]4]^^^ ^$((55 <-6G6TUUU U rr*)rVrVrVrV)r+r,r-__doc__r!rQrrmr|r ParseResultr^rbrr@rrrcrClistrrrrr rrRrrrTrTqs7\\&O&O&O&OP9s9%S/9999 C S%:K4L    42.2.2.l     u ! @ 5 5 5 5 5 %s % % % %-  pXpXpXpX pX  pX  pX pXpXpXpXd N$ N N N N*****..C.D....(+1*B  C {      rrT) metaclass)!rloggingrrr=typingrrrurllibrpymilvus.client.checkrrr pymilvus.client.grpc_handlerr pymilvus.exceptionsr r r pymilvus.settingsr getLoggerr+r; VIRTUAL_PORTrrwrr0rT connectionsrRrrrs  ))))))))))PPPPPPPPPP444444 %$$$$$  8 $ $  x    55555d555$,R,R,R,R,R,R,R,R^mmmmm3mmmmb kmm r