o j!:@sdZddlmZmZddlmZddlmZddlmZddl m Z m Z m Z m Z mZmZdd lmZmZmZdd lmZdd lmZmZmZmZmZe e e gZGd d d eZGdddeZGdddeZ dS) )datetimeMINYEAR)linesep)randint)sleep)FIRST ROUND_ROBINRANDOMSEQUENCE_TYPES STRING_TYPESget_config_parameter)LDAPUnknownStrategyErrorLDAPServerPoolErrorLDAPServerPoolExhaustedError)Server)log log_enabledERRORBASICNETWORKc@seZdZddZdS) ServerStatecCs||_||_||_dSN)serverlast_checked_time available)selfrrrr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/ldap3/core/pooling.py__init__(s zServerState.__init__N)__name__ __module__ __qualname__r!rrrr r's rc@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)ServerPoolStatecCsHg|_|j|_||_d|_|t|_tt r"t t d|dSdS)Nrz"instantiated ServerPoolState: <%r>) server_statesstrategy server_poollast_used_serverrefreshrnowinitialize_timerrr)rr(rrr r!/s zServerPoolState.__init__cCs|dt}|jr|jD] }|t|jt7}q n|dt7}|dt|jt7}|d|jdkr1dnt|j|jj7}|S)N servers: NonePool strategy: z - Last used server: )rr&strrr'r))rsstaterrr __str__:s  (zServerPoolState.__str__c CsJg|_|jjD]}|jt|ttdddqtdt|jd|_ dS)NrTr) r&r(serversappendrrrrlenr))rrrrr r*Fs zServerPoolState.refreshcCs|j|jjSr)r&r)rrrrr get_current_serverLsz"ServerPoolState.get_current_servercCs|jr~|jjtkr|jjr|jdd|_nWd|_nS|jjtkr@|jjr-||jd|_n?|jdt|jkr<|jdnd|_n,|jjt kr\|jjrP| |_nt dt|jd|_nt t rhtt d|jjtdt trwttd|j|j|jjSt t rtt d|td) Nr)startingrz$unknown server pooling strategy <%s>zunknown server pooling strategyz&server returned from Server Pool: <%s>zno servers in Server Pool <%s>zno servers in server pool)r&r(r'r activefind_active_serverr)r r7r find_active_random_serverrrrrrrrrr8rrr get_serverOs,  &   zServerPoolState.get_servercCs(|jj}|rttrttd||jdd}|ry|tdt|d}|j sVt |jj t r3|jj s?t |jj|jj krKttrJttd|jqttrVttd|jt |_ttrfttd|j|jrtd|_ |j|Sd|_ |st |jjt s|d8}|sttrttd |td ) Nz4entering loop for finding active server in pool <%s>rrz8server <%s> excluded from checking because it is offlineserver <%s> reinserted in pool%checking server <%s> for availabilityTFzSno random active server available in Server Pool <%s> after maximum number of trieszNno random active server available in server pool after maximum number of tries)r(r;rrrr&poprr7r isinstanceexhaustboolrr+rsecondsrcheck_availabilityindexrr)rcounter temp_list server_staterrr r=qs8 .    z)ServerPoolState.find_active_random_servercCstd}|jj}|t|jkrd}|rttrttd||d}t|j}||dkr|d7}|||kr9||n|||}|j|}|jst |jj t rR|jj s^t |jj|jj krttrt |jj t rqttd|jnttd|j|jj t |jjq%ttrttd|jt |_ttrttd |j|jrd |_|Sd |_||dks+t |jjt s|d8}ttrttd |t||sttrttd |td)NPOOLING_LOOP_TIMEOUTrz@entering loop number <%s> for finding active server in pool <%s>r0rz5server <%s> excluded from checking because is offlinezDserver <%s> excluded from checking because is offline for %d secondsr?r@TFz9waiting for %d seconds before retrying pool servers cyclezLno active server available in Server Pool <%s> after maximum number of trieszGno active server available in server pool after maximum number of tries)rr(r;r7r&rrrrrBrCrDrr+rrErrFrrr)rr:conf_pool_timeoutrHrG pool_sizeoffsetrJrrr r<sN    .$    " z"ServerPoolState.find_active_servercC t|jSr)r7r&r8rrr __len__ zServerPoolState.__len__N) r"r#r$r!r4r*r9r>r=r<rPrrrr r%.s  " ,r%c@speZdZdedddfddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZdS) ServerPoolNTFcCs|tvrttrttd|td|r#|s#ttrttdtdg|_t|_||_ ||_ ||_ d|_ t |ttfrD||n t |trP|t|||_ttr_ttd|dSdS)Nzunknown pooling strategy <%s>zunknown pooling strategyz3cannot instantiate pool with exhaust and not activez)POOLING_STRATEGIESrrrrrr5dict pool_statesr;rCsingle _pool_staterBr raddr r'r)rr5 pool_strategyr;rC single_staterrr r!s,    zServerPool.__init__cCsdt}|jr|jD] }|t|t7}q n|dt7}|dt|j7}|d|jr/t|jnd7}|d|jr=t|jnd7}|S)Nr-r.r/z - active: Falsez - exhaust pool: )rr5r1r'r;rC)rr2rrrr r4s  zServerPool.__str__cCs~d}|jr |d7}|jD] }||d7}q |ddd}n|d7}|d|7}|d|7}|d |7}|d 7}|S) NzServerPool(servers=[z, ]r.z, pool_strategy={0.strategy!r}z, active={0.active!r}z, exhaust={0.exhaust!r}))r5__repr__format)rrrrrr r`s zServerPool.__repr__cCrOr)r7r5r8rrr rPrQzServerPool.__len__cCs |j|Sr)r5)ritemrrr __getitem__rQzServerPool.__getitem__cCs |jSr)r5__iter__r8rrr rerQzServerPool.__iter__cCst|tr||jvr|j|nNt|tr|jt|n@t|trQ|D])}t|tr4|j|q&t|trB|jt|q&ttrLttd|t dnttr[ttd|t d|j rn|j rl|j dSdS|j D] }|j | qqdS)Nz,element must be a server in Server Pool <%s>z%server in ServerPool must be a ServerzLserver must be a Server of a list of Servers when adding to Server Pool <%s>z+server must be a Server or a list of Server)rBrr5r6r r rrrrrVrWr*rU)rr5r connectionrrr rXs4          zServerPool.addcCsr||jvr |j|nttrttd||td|jr*|jr(|jdSdS|j D] }|j |q-dS)Nz/server %s to be removed not in Server Pool <%s>zserver not in server pool) r5removerrrrrVrWr*rU)rrrfrrr rg"s  zServerPool.removecCs8|jr|js t||_|j|j|<dSt||j|<dSr)rVrWr%rUrrfrrr initialize2s  zServerPool.initializecC6||jvr |j|Sttrttd||tdNz-connection <%s> not in Server Pool State <%s>z!connection not in ServerPoolState)rUr>rrrrrhrrr r>; zServerPool.get_servercCrjrk)rUr9rrrrrhrrr r9CrlzServerPool.get_current_server)r"r#r$r r!r4r`rPrdrerXrgrir>r9rrrr rRs"   rRN)!__doc__rrosrrandomrtimerr r r r r r exceptionsrrrrr utils.logrrrrrrSobjectrr%rRrrrr s