a !f[@sdZddlZddlZddlZddlZddlmZmZmZddl Z ej rRddl m Z Gdddej dZejjdd d ZGd d d eZdS) zSchedulers provide means to *schedule* callbacks asynchronously. These are used by the subscriber to call the user-provided callback to process each message. N)CallableListOptional) pubsub_v1c@s\eZdZdZeejejdddZeje ddddZ ejde e d d d d Z dS) SchedulerzgAbstract base class for schedulers. Schedulers are used to schedule callbacks asynchronously. returncCstdS)zQueue: A concurrency-safe queue specific to the underlying concurrency implementation. This queue is used to send messages *back* to the scheduling actor. NNotImplementedErrorselfr l/var/www/html/python-backend/venv/lib/python3.9/site-packages/google/cloud/pubsub_v1/subscriber/scheduler.pyqueue&szScheduler.queueNcallbackrcOstdS)aSchedule the callback to be called asynchronously. Args: callback: The function to call. args: Positional arguments passed to the callback. kwargs: Key-word arguments passed to the callback. Returns: None Nr r rargskwargsr r rschedule0s zScheduler.scheduleF$pubsub_v1.subscriber.message.Messageawait_msg_callbacksrcCstdS)aShuts down the scheduler and immediately end all pending callbacks. Args: await_msg_callbacks: If ``True``, the method will block until all currently executing callbacks are done processing. If ``False`` (default), the method will not wait for the currently running callbacks to complete. Returns: The messages submitted to the scheduler that were not yet dispatched to their callbacks. It is assumed that each message was submitted to the scheduler as the first positional argument to the provided callback. Nr )r rr r rshutdown>szScheduler.shutdown)F)__name__ __module__ __qualname____doc__propertyabcabstractmethodrQueuerrboolrrr r r rr s r) metaclassrcCstjjdddS)N z"ThreadPoolExecutor-ThreadScheduler) max_workersthread_name_prefix) concurrentfuturesThreadPoolExecutorr r r r"_make_default_thread_pool_executorSsr*c@sZeZdZdZdeejjdddZe ddZ e ddd d Z de ed d ddZdS)ThreadSchedulera$A thread pool-based scheduler. It must not be shared across SubscriberClients. This scheduler is useful in typical I/O-bound message processing. Args: executor: An optional executor to use. If not specified, a default one will be created. N)executorcCs&t|_|durt|_n||_dS)N)rr!_queuer* _executor)r r,r r r__init__es  zThreadScheduler.__init__cCs|jS)zfQueue: A thread-safe queue used for communication between callbacks and the scheduling thread.)r-r r r rrnszThreadScheduler.queuercOsFz|jj|g|Ri|Wn"ty@tjdtddYn0dS)a(Schedule the callback to be called asynchronously in a thread pool. Args: callback: The function to call. args: Positional arguments passed to the callback. kwargs: Key-word arguments passed to the callback. Returns: None z.Scheduling a callback after executor shutdown.)category stacklevelN)r.submit RuntimeErrorwarningswarnRuntimeWarningrr r rrts  zThreadScheduler.scheduleFrrcCs\g}z0|jjjdd}|dur q||jdqWntjyHYn0|jj|d|S)aShut down the scheduler and immediately end all pending callbacks. Args: await_msg_callbacks: If ``True``, the method will block until all currently executing executor threads are done processing. If ``False`` (default), the method will not wait for the currently running threads to complete. Returns: The messages submitted to the scheduler that were not yet dispatched to their callbacks. It is assumed that each message was submitted to the scheduler as the first positional argument to the provided callback. F)blockNr)wait)r.Z _work_queuegetappendrrEmptyr)r rZdropped_messagesZ work_itemr r rrszThreadScheduler.shutdown)N)F)rrrrrr'r(r)r/rrrrr"rrr r r rr+Ys   r+)rrconcurrent.futuresr'rtypingrrrr5 TYPE_CHECKINGZ google.cloudrABCMetarr(r)r*r+r r r rs 3