o i^B@s UddlmZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZddlmZmZddlZddlmZd ZeeZed d Gd d d Zed d GdddZed d GdddZ ed d GdddZ!GdddZ"GdddeZ#eGdddeZ$eGdddeZ%eGdddeZ&eGdddeZ'ee$e%e&e'fZ(d e)d!<ed d Gd"d#d#eZ*ed d Gd$d%d%Z+Gd&d'd'eZ,eee!e,fZ-d e)d(<Gd)d*d*Z.d6d0d1Z/Gd2d3d3eZ0ed d Gd4d5d5e0Z1dS)7) annotationsN)ABCabstractmethod) dataclass) timedelta)AnyCallableLiteralOptionalProtocolUnionruntime_checkable)Self TypeAlias)WorkerDeploymentVersioniT)frozenc@eZdZUdZded<dS)FixedSizeSlotSupplierzSA fixed-size slot supplier that will never issue more than a fixed number of slots.int num_slotsN__name__ __module__ __qualname____doc____annotations__rr_/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/temporalio/worker/_tuning.pyrs rc@s$eZdZUdZded< ded<dS)ResourceBasedTunerConfigzXOptions for a :py:class:`ResourceBasedTuner` or a :py:class:`ResourceBasedSlotSupplier`.floattarget_memory_usagetarget_cpu_usageNrrrrrrs rc@s:eZdZUdZdZded< dZded< dZded<dS)ResourceBasedSlotConfigzYOptions for a specific slot type being used with a :py:class:`ResourceBasedSlotSupplier`.N Optional[int] minimum_slots maximum_slotszOptional[timedelta] ramp_throttle)rrrrr$rr%r&rrrrr")s    r"c@"eZdZUdZded<ded<dS)ResourceBasedSlotSupplierzYA slot supplier that will dynamically adjust the number of slots based on resource usage.r" slot_configr tuner_configNrrrrrr(:s r(c@seZdZdZdS) SlotPermitzA permit to use a slot for a workflow/activity/local activity task. You can inherit from this class to add your own data to the permit. N)rrrrrrrrr+Dsr+c@sLeZdZUdZded< ded< ded< ded< ded < d ed <d S) SlotReserveContextzCContext for reserving a slot from a :py:class:`CustomSlotSupplier`.z1Literal['workflow', 'activity', 'local-activity'] slot_typestr task_queueworker_identityworker_build_idz!Optional[WorkerDeploymentVersion]worker_deployment_versionbool is_stickyNrrrrrr,Ns r,c@r')WorkflowSlotInfoz&Info about a workflow task slot usage.r. workflow_typer3r4Nrrrrrr5d  r5c@r)ActivitySlotInfoz'Info about an activity task slot usage.r. activity_typeNrrrrrr8m  r8c@r)LocalActivitySlotInfoz,Info about a local activity task slot usage.r.r9Nrrrrrr;ur:r;c@s"eZdZUdZded<ded<dS) NexusSlotInfoz#Info about a nexus task slot usage.r.service operationNrrrrrr<}r7r<rSlotInfoc@$eZdZUdZded< ded<dS)SlotMarkUsedContextzFContext for marking a slot used from a :py:class:`CustomSlotSupplier`.r? slot_infor+permitNrrrrrrA rAc@r@)SlotReleaseContextzCContext for releasing a slot from a :py:class:`CustomSlotSupplier`.zOptional[SlotInfo]rBr+rCNrrrrrrErDrEc@sHeZdZdZedddZedd d Zedd dZedddZdS)CustomSlotSupplierzGThis class can be implemented to provide custom slot supplier behavior.ctxr,returnr+csdS)a]This function is called before polling for new tasks. Your implementation must block until a slot is available then return a permit to use that slot. The only acceptable exception to throw is :py:class:`asyncio.CancelledError`, as invocations of this method may be cancelled. Any other exceptions thrown will be logged and ignored. It is technically possible but rare, during worker shutdown, for this method to be called and return a value, but the Rust Core may not have a chance to _observe_ that value. In such cases the returned permit will not be released. The permit will, however, be forgotten and python will garbage collect it. So if you use the same slot supplier over the lifetime of more than one worker and it is critically important for you to clean up some resources associated all permits you construct, then consider using a finalizer on your returned permits. Args: ctx: The context for slot reservation. Returns: A permit to use the slot which may be populated with your own data. NrselfrGrrr reserve_slotszCustomSlotSupplier.reserve_slotOptional[SlotPermit]cCdS)aThis function is called when trying to reserve slots for "eager" workflow and activity tasks. Eager tasks are those which are returned as a result of completing a workflow task, rather than from polling. Your implementation must not block, and if a slot is available, return a permit to use that slot. Args: ctx: The context for slot reservation. Returns: Maybe a permit to use the slot which may be populated with your own data. NrrIrrrtry_reserve_slots z#CustomSlotSupplier.try_reserve_slotrANonecCrM)aThis function is called once a slot is actually being used to process some task, which may be some time after the slot was reserved originally. For example, if there is no work for a worker, a number of slots equal to the number of active pollers may already be reserved, but none of them are being used yet. This call should be non-blocking. Args: ctx: The context for marking a slot as used. NrrIrrrmark_slot_useds z!CustomSlotSupplier.mark_slot_usedrEcCrM)aXThis function is called once a permit is no longer needed. This could be because the task has finished, whether successfully or not, or because the slot was no longer needed (ex: the number of active pollers decreased). This call should be non-blocking. Args: ctx: The context for releasing a slot. NrrIrrr release_slots zCustomSlotSupplier.release_slotN)rGr,rHr+rGr,rHrLrGrArHrOrGrErHrO) rrrrrrKrNrPrQrrrrrFs    rF SlotSupplierc@s>eZdZdddZdd d ZdddZdddZdddZdS)_BridgeSlotSupplierWrappersupplierrFcCs ||_dSN) _supplier)rJrWrrr__init__s z#_BridgeSlotSupplierWrapper.__init__rGr, reserve_cbCallable[[Any], None]rHr+csXzt|j|}|||IdHWStjyty+tjdddw)Nz,Error in custom slot supplier `reserve_slot`Texc_info)asyncio create_taskrYrKCancelledError Exceptionloggerwarning)rJrGr[ reserve_futrrrrKs  z'_BridgeSlotSupplierWrapper.reserve_slotrLcCs2z|j|WStytjdddYdSw)Nz0Error in custom slot supplier `try_reserve_slot`Tr])rYrNrbrcrdrIrrrrNs z+_BridgeSlotSupplierWrapper.try_reserve_slotrErOcC6z |j|WdStytjdddYdSw)Nz,Error in custom slot supplier `release_slot`Tr])rYrQrbrcrdrIrrrrQ  z'_BridgeSlotSupplierWrapper.release_slotrAcCrf)Nz.Error in custom slot supplier `mark_slot_used`Tr])rYrPrbrcrdrIrrrrP rgz)_BridgeSlotSupplierWrapper.mark_slot_usedN)rWrF)rGr,r[r\rHr+rRrTrS)rrrrZrKrNrQrPrrrrrVs     rV slot_supplierkind:Literal['workflow', 'activity', 'local_activity', 'nexus']rH%temporalio.bridge.worker.SlotSupplierc Cst|tr tjj|jSt|trc|dkrdnd}t}|dkr%tddntdd}|j j dur4|j j }|j j dur>|j j }|j j durH|j j }tjj||t |tddtjj|jj|jjSt|trqtjjt|Std|) Nworkflowr)seconds2) millisecondszUnknown slot supplier type: ) isinstancer temporaliobridgeworkerrr(_DEFAULT_RESOURCE_SLOTS_MAXrr)r$r%r&rrr*r r!rFBridgeCustomSlotSupplierrV TypeError)rhri min_slots max_slotsr&rrr_to_bridge_slot_suppliers6      r{c@seZdZdZedddddd0d dZedddddd1ddZed2ddZed3ddZ ed3d d!Z ed3d"d#Z ed3d$d%Z d4d'd(Z d5d)d*Zd5d+d,Zed6d.d/ZdS)7 WorkerTunerzXWorkerTuners allow for the dynamic customization of some aspects of worker configurationN)workflow_configactivity_configlocal_activity_config nexus_configr rr!r}!Optional[ResourceBasedSlotConfig]r~rrrHc CsXt||}t|p t|}t|pt|} t|pt|} t|p"t|} t|| | | S)z8Create a resource-based tuner with the provided options.)rr(r"_CompositeTuner) clsr r!r}r~rr resource_cfgwfact local_actnexusrrrcreate_resource_based9s&     z!WorkerTuner.create_resource_based)workflow_slotsactivity_slotslocal_activity_slots nexus_slotsrr#rrrcCsBtt|r|ndt|r |ndt|r|ndt|r|SdS)zCreate a fixed-size tuner with the provided number of slots. Any unspecified slot numbers will default to 100. d)rr)rrrrrrrr create_fixedYs   zWorkerTuner.create_fixedworkflow_supplierrUactivity_supplierlocal_activity_suppliernexus_suppliercCst||||S)z7Create a tuner composed of the provided slot suppliers.)r)rrrrrrrrcreate_compositeos  zWorkerTuner.create_compositecCtrXNotImplementedErrorrJrrr _get_workflow_task_slot_supplierz,WorkerTuner._get_workflow_task_slot_suppliercCrrXrrrrr _get_activity_task_slot_supplierrz,WorkerTuner._get_activity_task_slot_suppliercCrrXrrrrr&_get_local_activity_task_slot_supplierrz2WorkerTuner._get_local_activity_task_slot_suppliercCrrXrrrrr_get_nexus_slot_supplierrz$WorkerTuner._get_nexus_slot_supplier$temporalio.bridge.worker.TunerHoldercCs<tjjt|dt|dt|dt|dS)Nrlactivitylocal_activityr) rsrtru TunerHolderr{rrrrrrrr_to_bridge_tuners zWorkerTuner._to_bridge_tunercCt|SrX)r|_get_slot_supplier_maxrrrrr_get_activities_maxszWorkerTuner._get_activities_maxcCrrX)r|rrrrrr_get_nexus_tasks_maxsz WorkerTuner._get_nexus_tasks_maxrhcCs*t|tr|jSt|tr|jjptSdSrX)rrrrr(r)r%rv)rhrrrrs   z"WorkerTuner._get_slot_supplier_max)r rr!rr}rr~rrrrrrHr|) rr#rr#rr#rr#rHr|) rrUrrUrrUrrUrHr|rHrU)rHr)rHr#)rhrUrHr#)rrrr classmethodrrrrrrrrrrr staticmethodrrrrrr|6s:        r|c@sZeZdZUdZded<ded<ded<ded<ddd Zdd d Zdd d ZdddZdS)rzHThis tuner allows for different slot suppliers for different slot types.rUworkflow_slot_supplieractivity_slot_supplierlocal_activity_slot_suppliernexus_slot_supplierrHcC|jSrX)rrrrrrz0_CompositeTuner._get_workflow_task_slot_suppliercCrrX)rrrrrrrz0_CompositeTuner._get_activity_task_slot_suppliercCrrX)rrrrrrrz6_CompositeTuner._get_local_activity_task_slot_suppliercCrrX)rrrrrrrz(_CompositeTuner._get_nexus_slot_supplierNr) rrrrrrrrrrrrrrs    r)rhrUrirjrHrk)2 __future__rr_loggingabcrr dataclassesrdatetimertypingrrr r r r r typing_extensionsrrtemporalio.bridge.workerrstemporalio.commonrrv getLoggerrrcrrr"r(r+r,r5r8r;r<r?rrArErFrUrVr{r|rrrrrs\  $         @  .#{