o ;i@sddlZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z eeZdZdZeeZGdd d ejZGd d d eZGd d d eZdS)N)OptionalSetCallable Awaitable)DeadlineWrapper)Deadline c@sdeZdZejdefddZejdefddZejdej fddZ ejdej dd fd d Z d S) CheckBasereturncCsdSNselfrrH/home/ubuntu/.local/lib/python3.10/site-packages/grpclib/health/check.py __status__szCheckBase.__status__cdSr rrrrr __check__zCheckBase.__check__crr rrrrr __subscribe__rzCheckBase.__subscribe__eventNcrr rrrrrr__unsubscribe__#rzCheckBase.__unsubscribe__) __name__ __module__ __qualname__abcabstractmethod_StatusrrasyncioEventrrrrrrr sr c @seZdZdZdZdZdZdeedde ge e fde e jdededdf d d Zde fd d Zde fd dZdddZde jfddZde jddfddZdS) ServiceCheckaGPerforms periodic checks Example: .. code-block:: python3 async def db_test(): # raised exceptions are the same as returning False, # except that exceptions will be logged await db.execute('SELECT 1;') return True db_check = ServiceCheck(db_test) N)loop check_ttl check_timeoutfuncr#r$r%r cCsN||_||_||_t|_|rtjdtddt |_ |j t |_ dS)a :param func: callable object which returns awaitable object, where result is one of: ``True`` (healthy), ``False`` (unhealthy), or ``None`` (unknown) :param loop: (deprecated) asyncio-compatible event loop :param check_ttl: how long we can cache result of the previous check :param check_timeout: timeout for this check HThe loop argument is deprecated and scheduled for removal in grpclib 0.5r stacklevelN) _func _check_ttl_check_timeoutset_eventswarningswarnDeprecationWarningr r! _check_lockr_check_wrapper)rr&r#r$r%rrr__init__;s   zServiceCheck.__init__cC|jSr _valuerrrrr[zServiceCheck.__status__csf|jdurt|j|jkr|jS|js#|jIdH|jS|j}|jzYz8t |j }|j ||IdH}Wdn1sKwY|dur`t|ts`td|||_Wntjymty}tdd|_YnwW|jn|jwt|_|j|kr|jrtjntj}|d|j|j|jD]}|q|jS)NzInvalid status type: {!r}zHealth check failedFz$Health check %r status changed to %r) _last_checktime monotonicr+r7r2is_setwaitclearr from_timeoutr,r3startr* isinstancebool TypeErrorformatr CancelledError Exceptionlog exceptionr-infowarningr.)r prev_valuedeadlinevalue log_levelrrrrr^sD            zServiceCheck.__check__cs> |IdH}|rt|jIdHn t|jIdHqr )rr sleepr+)rstatusrrr_pollszServiceCheck._pollcs<|jdurt}|||_t}|j||Sr ) _poll_taskr get_event_loop create_taskrQr!r.add)rr#rrrrrs  zServiceCheck.__subscribe__rcsb|j||js/|jdusJ|j}d|_|z|IdHWdStjy.YdSwdSr )r.discardrRcancelr rE)rrtaskrrrrs zServiceCheck.__unsubscribe__)r N)rrr__doc__r7rRr9DEFAULT_CHECK_TTLDEFAULT_CHECK_TIMEOUTrrrrr AbstractEventLoopfloatr4rrrQr!rrrrrrr"(s0   ' r"c@seZdZdZdddeejddfddZdeddfd d Z defd d Z defd dZ dej fddZ dej ddfddZdS) ServiceStatusaContains status of a proactive check Example: .. code-block:: python3 redis_status = ServiceStatus() # detected that Redis is available redis_status.set(True) # detected that Redis is unavailable redis_status.set(False) N)r#r#r cCs&|r tjdtddd|_t|_dS)zI :param loop: (deprecated) asyncio-compatible event loop r'rr(N)r/r0r1r7r-r.)rr#rrrr4s  zServiceStatus.__init__rMcCs2|j}||_|j|kr|jD]}|qdSdS)zSets current status of a check :param value: ``True`` (healthy), ``False`` (unhealthy), or ``None`` (unknown) N)r7r.r-)rrMrKrrrrr-s   zServiceStatus.setcCr5r r6rrrrrr8zServiceStatus.__status__cs|jSr r6rrrrrszServiceStatus.__check__cst}|j||Sr )r r!r.rUrrrrrs zServiceStatus.__subscribe__rcs|j|dSr )r.rVrrrrrszServiceStatus.__unsubscribe__)rrrrYrr r\r4rr-rrr!rrrrrrr^s  r^)rr:r loggingr/typingrrrrutilsrmetadatar getLoggerrrGrZr[rBrABCr r"r^rrrrs   |