a d=ic,-@sdZddlZddlZddlZddlZddlZddlZddlmZddl m Z e Z dZ dZdZdZd ZeejjejjejjejjgZd ZGd d d Zdd dZddZdddZddZddZej Gdddej!Z"dS)z-Utilities for working with python gRPC stubs.N)version) tb_loggingg?g?ztensorboard-versionc@s(eZdZdZddZddZddZdS) AsyncCallFutureaEncapsulates the future value of a retriable async gRPC request. Abstracts over the set of futures returned by a set of gRPC calls comprising a single logical gRPC request with retries. Communicates to the caller the result or exception resulting from the request. Args: completion_event: The constructor should provide a `threding.Event` which will be used to communicate when the set of gRPC requests is complete. cCsd|_t|_||_dSN)_active_grpc_future threadingLock_active_grpc_future_lock_completion_event)selfcompletion_eventrk/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorboard/util/grpc_util.py__init__Ds zAsyncCallFuture.__init__cCs@|durtd|j||_Wdn1s20YdS)Nz1_set_active_future invoked with grpc_future=None.) RuntimeErrorr r )rZ grpc_futurerrr_set_active_futureIs z"AsyncCallFuture._set_active_futurecCsd|j|std|d|j,|jdur8td|jWdS1sV0YdS)aUAnalogous to `grpc.Future.result`. Returns the value or exception. This method will wait until the full set of gRPC requests is complete and then act as `grpc.Future.result` for the single gRPC invocation corresponding to the first successful call or final failure, as appropriate. Args: timeout: How long to wait in seconds before giving up and raising. Returns: The result of the future corresponding to the single gRPC corresponding to the successful call. Raises: * `grpc.FutureTimeoutError` if timeout seconds elapse before the gRPC calls could complete, including waits and retries. * The exception corresponding to the last non-retryable gRPC request in the case that a successful gRPC request was not made. z AsyncCallFuture timed out after z secondsNz*AsyncFuture never had an active future set)r waitgrpcZFutureTimeoutErrorr r rresult)rtimeoutrrrrQs   zAsyncCallFuture.resultN)__name__ __module__ __qualname____doc__rrrrrrrr8s rcsddur ttdttfddfddtjddS) aInitiate an asynchronous call to a gRPC stub, with retry logic. This is similar to the `async_call` API, except that the call is handled asynchronously, and the completion may be handled by another thread. The caller must provide a `done_callback` argument which will handle the result or exception rising from the gRPC completion. Retries are handled with jittered exponential backoff to spread out failures due to request spikes. This only supports unary-unary RPCs: i.e., no streaming on either end. Args: api_method: Callable for the API method to invoke. request: Request protocol buffer to pass to the API method. clock: an interface object supporting `time()` and `sleep()` methods like the standard `time` module; if not passed, uses the normal module. Returns: An `AsyncCallFuture` which will encapsulate the `grpc.Future` corresponding to the gRPC call which either completes successfully or represents the final try. Nz"Async RPC call %s with request: %rcs*jttd}|||dS)zDInvokes the gRPC future and orchestrates it via the AsyncCallFuture.rmetadataN)future_GRPC_DEFAULT_TIMEOUT_SECSversion_metadataradd_done_callback)handlerr) api_method async_futurerequestrr async_calls z+async_call_with_retries..async_callcs|}|durdStd||tvrBdS|tkrVdSt|}|t j |dddS)NRPC call %s got error %s num_attempts) exceptionsetloggerinfocode_GRPC_RETRYABLE_STATUS_CODES_GRPC_RETRY_MAX_ATTEMPTS_compute_backoff_secondssleep functoolspartial)rr+e backoff_secs)r$r'clockr retry_handlerrrr:s   z.async_call_with_retries..retry_handlerr)r*)timer.debugr Eventrr5r6)r$r&r9r)r$r'r%r9rr&r:rasync_call_with_retriespsr>cCsttt}t||}|S)z3Compute appropriate wait time between RPC attempts.)randomuniform_GRPC_RETRY_JITTER_FACTOR_MIN_GRPC_RETRY_JITTER_FACTOR_MAX_GRPC_RETRY_EXPONENTIAL_BASE)r+Z jitter_factorr8rrrr3sr3c Cs|dur t}|jjdd}td||d}|d7}z||ttdWStj y}z2t d||| t vrv|t krWYd}~n d}~00t|}t d |||||q.dS) aCall a gRPC stub API method, with automatic retry logic. This only supports unary-unary RPCs: i.e., no streaming on either end. Streamed RPCs will generally need application-level pagination support, because after a gRPC error one must retry the entire request; there is no "retry-resume" functionality. Retries are handled with jittered exponential backoff to spread out failures due to request spikes. Args: api_method: Callable for the API method to invoke. request: Request protocol buffer to pass to the API method. clock: an interface object supporting `time()` and `sleep()` methods like the standard `time` module; if not passed, uses the normal module. Returns: Response protocol buffer returned by the API method. Raises: grpc.RpcError: if a non-retryable error is returned, or if all retry attempts have been exhausted. NRequestzRPC call %s with request: %rrr)rr(z8RPC call %s attempted %d times, retrying in %.1f seconds)r; __class__rreplacer.r<r r!rZRpcErrorr/r0r1r2r3r4)r$r&r9Zrpc_namer+r7r8rrrcall_with_retriess4 rHcCs ttjffS)aCreates gRPC invocation metadata encoding the TensorBoard version. Usage: `stub.MyRpc(request, metadata=version_metadata())`. Returns: A tuple of key-value pairs (themselves 2-tuples) to be passed as the `metadata` kwarg to gRPC stub API methods. )_VERSION_METADATA_KEYrVERSIONrrrrr!s r!cCst|tS)aExtracts version from invocation metadata. The argument should be the result of a prior call to `metadata` or the result of combining such a result with other metadata. Returns: The TensorBoard version listed in this metadata, or `None` if none is listed. )dictgetrI)rrrrextract_version s rMc@s4eZdZdZdZdZddZeddZdd Z d S) ChannelCredsTypelocalsslZssl_devcCs^g}|tjkrt}n>|tjkr,t}n*|tjkrJt}|dn td|||fS)aNCreate channel credentials and options. Returns: A tuple `(channel_creds, channel_options)`, where `channel_creds` is a `grpc.ChannelCredentials` and `channel_options` is a (potentially empty) list of `(key, value)` tuples. Both results may be passed to `grpc.secure_channel`. )zgrpc.ssl_target_name_override localhostzunhandled ChannelCredsType: %r) rNLOCALrZlocal_channel_credentialsSSLZssl_channel_credentialsSSL_DEVappendAssertionError)roptionscredsrrrchannel_configs        zChannelCredsType.channel_configcCs |jSr) __members__values)clsrrrchoices4szChannelCredsType.choicescCs|jSr)value)rrrr__str__8szChannelCredsType.__str__N) rrrrRrSrTrY classmethodr]r_rrrrrNs rN)N)N)#renumr5r?r r;rZ tensorboardrZtensorboard.utilrZ get_loggerr.r r2rCrArB frozensetZ StatusCodeZABORTEDZDEADLINE_EXCEEDEDZRESOURCE_EXHAUSTED UNAVAILABLEr1rIrr>r3rHr!rMuniqueEnumrNrrrrs<   8 K 7