o پiY@sdZddlZddlZddlZddlZddlmZddlmZ ej ddkr2ej ddkr2ddl m Z ejjd jZd ZGd d d Zd dZGddde jZdS)zs "z_Watcher.closeN) __name__ __module__ __qualname__rrrrr!r%r&rrrrr "s r csfdd}|S)Ncs"|dur dS|dSr)r&r%)r$watcherrrsend_response_callbackEs zJ_watcher_to_send_response_callback_adapter..send_response_callbackr)r+r,rr*r*_watcher_to_send_response_callback_adapterDs r-c@sFeZdZdZ dddZddZdd Zdd d Zd d ZddZ dS)HealthServicerz,Servicer handling RPCs for service statuses.TNcCs<t|_dtjji|_i|_||jj _ ||jj _ d|_ dS)Nr F) r RLock_lock _health_pb2HealthCheckResponseSERVING_server_status_send_response_callbacksWatch__func__experimental_non_blockingexperimental_thread_pool_gracefully_shutting_down)rr8r9rrrrQs   zHealthServicer.__init__csfdd}|S)NcsBjjWdn1swYddSr)r0r5removerrr,servicerrcallback^s   z3HealthServicer._on_close_callback..callbackr)rr,r=r>rr<r_on_close_callback]sz!HealthServicer._on_close_callbackcCsp|j+|j|j}|dur!|tjjt WdStj |dWdS1s1wYdSNstatus) r0r4getr=set_codegrpc StatusCode NOT_FOUNDr1r2)rrequestcontextrBrrrCheckgs $zHealthServicer.CheckcCsd}|dur t}t|}|j}|j;|j|}|dur"tjj}|tj|d||j vr5t |j |<|j | || | ||Wd|S1sQwY|Sr@)r r-r=r0r4rCr1r2SERVICE_UNKNOWNr5setr% add_callbackr?)rrHrIr,blocking_watcherr=rBrrrr6ps2      zHealthServicer.WatchcCs|j7|jr WddS||j|<||jvr*|j|D]}|tj|dqWddSWddS1s=wYdS)zSets the status of a service. Args: service: string, the name of the service. status: HealthCheckResponse.status enum value indicating the status of the service NrA)r0r:r4r5r1r2)rr=rBr,rrrrLs    "zHealthServicer.setcCsf|j&|jr WddS|jD] }||tjjqd|_WddS1s,wYdS)a0Permanently sets the status of all services to NOT_SERVING. This should be invoked when the server is entering a graceful shutdown period. After this method is invoked, future attempts to set the status of a service will be ignored. This is an EXPERIMENTAL API. NT)r0r:r4rLr1r2 NOT_SERVING)rr=rrrenter_graceful_shutdowns  "z&HealthServicer.enter_graceful_shutdown)TNr) r'r(r)__doc__rr?rJr6rLrPrrrrr.Ns    r.)rQrsysr rEgrpc_health.v1rr1r_health_pb2_grpc version_infor raio DESCRIPTORservices_by_name full_name SERVICE_NAMEOVERALL_HEALTHr r-r.rrrrs    "