File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/transport/__pycache__/base.cpython-310.pyc
o
�N�g � @ s� d dl Z d dlZd dlZdZdd� Zdd� Zdd� Zd d
� ZG d d � d e �Z
G d
d� d�Z G dd� de �Z G dd� d�Z
G dd� de
�ZG dd� d�ZG dd� de�ZG dd� de �ZdS )� N)�zeromq�tcpc K s� d}d| v r | d }nd| � di �� di �v r | d d d }|dkr.dd l}|jj�| �S |dkr=dd l}|jj�| �S |dkrNdd l}|jj �
| �}d S t d��) Nr � transport�pillar�masterr r �localz,Channels are only defined for ZeroMQ and TCP) �get�salt.transport.zeromqr r �
RequestServer�salt.transport.tcpr Z TCPReqServer�salt.transport.localr ZLocalServerChannel� Exception)�opts�kwargs�ttype�saltr � r �G/opt/saltstack/salt/lib/python3.10/site-packages/salt/transport/base.py�request_server s
r c C s� d}d| v r | d }nd| � di �� di �v r | d d d }|dkr0dd l}|jjj| |d�S |dkrAdd l}|jjj| |d�S td��) Nr r r r r )�io_loopr z)Channels are only defined for tcp, zeromq) r r r r �
RequestClientr r Z TCPReqClientr
�r r r r r r r �request_client& s
r c K s� d}d| v r | d }nd| � di �� di �v r | d d d }|dkr2dd l}|jjj| fi |��S |dkrAdd l}|jj�| �S |dkrTdd l}|jj j
| fi |��S t d|� ���) Nr r r r r r r �Transport type not found: ) r r r r �
PublishServerr r ZTCPPublishServerr r ZLocalPubServerChannelr
)r r r r r r r �publish_server8 s
r c C s� d}d| v r | d }nd| � di �� di �v r | d d d }|dkr/dd l}|jj�| |�S |dkr?dd l}|jj�| |�S td|� ���)Nr r r r r r r ) r r r r �
PublishClientr r Z TCPPubClientr
r r r r �publish_clientP s
r c @ s e Zd ZdZdS )�TransportWarningz
Transport warning.
N)�__name__�
__module__� __qualname__�__doc__r r r r r d s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� Transportc O s@ d� t�� d d� �| _t| d�sd| _t| d�s d| _d S d S )N�
����_closingF�_connect_called)�join� traceback� format_stack�_trace�hasattrr& r' ��self�argsr r r r �__init__k s
�zTransport.__init__c O s
d| _ d S )NT)r' r- r r r �connectr s
zTransport.connectc C s@ t | d�r t | dd�s tjd| �d| j� �t| d� dS dS dS )a�
Warn the user if the transport's close method was never called.
If the _closing attribute is missing we won't raise a warning. This
prevents issues when class's dunder init method is called with improper
arguments, and is later getting garbage collected. Users of this class
should take care to call super() and validate the functionality with a
test.
r' r& TzUnclosed transport! z
)�sourceN)�getattr�warnings�warnr+ r �r. r r r �__del__v s
��zTransport.__del__N)r r r! r0 r1 r7 r r r r r# j s r# c sF e Zd ZdZ� fdd�Zejjjj d dd��Z
dd� Z d d
� Z � Z
S ) r zx
The RequestClient transport is used to make requests and get corresponding
replies from the RequestServer.
c � t � �� d S �N��superr0 �r. r r r �� __class__r r r0 � � zRequestClient.__init__�< c C � t �)zN
Send a request message and return the reply from the server.
��NotImplementedError)r. �load�timeoutr r r �send� s zRequestClient.sendc C rA )z'
Close the connection.
rB r6 r r r �close� � zRequestClient.closec C rA )z1
Connect to the server / broker.
rB r6 r r r r1 � rH zRequestClient.connect)r@ )r r r! r" r0 r �ext�tornado�gen� coroutinerF rG r1 �
__classcell__r r r= r r � s
r c @ s e Zd ZdZdd� Zdd� ZdS )r
z�
The RequestServer transport is responsible for handling requests from
RequestClients and sending replies to those requests.
c C � d S r9 r )r. r r r r r0 � � zRequestServer.__init__c C rA )z:
Close the underlying network connection.
rB r6 r r r rG � rH zRequestServer.closeN)r r r! r" r0 rG r r r r r
� s r
c @ s e Zd Zdd� Zdd� ZdS )�DaemonizedRequestServerc C rA r9 rB �r. Zprocess_managerr r r �pre_fork� rO z DaemonizedRequestServer.pre_forkc C rA )z�
The message handler is a coroutine that will be called called when a
new request comes into the server. The return from the message handler
will be send back to the RequestClient
rB )r. Zmessage_handlerr r r r � post_fork� s z!DaemonizedRequestServer.post_forkN)r r r! rR rS r r r r rP � s rP c @ s e Zd ZdZdd� ZdS )r z\
The PublishServer publishes messages to PublishClients or to a borker
service.
c K rA )z�
Publish "load" to minions. This send the load to the publisher daemon
process with does the actual sending to minions.
:param dict load: A load to be sent across the wire to minions
rB )r. �payloadr r r r �publish� � zPublishServer.publishN)r r r! r" rU r r r r r � s r c @ s& e Zd ZdZdd� Z ddd�ZdS )�DaemonizedPublishServerz=
PublishServer that has a daemon associated with it.
c C rA r9 rB rQ r r r rR � rO z DaemonizedPublishServer.pre_forkNc C rA )a�
If a daemon is needed to act as a broker implement it here.
:param func publish_payload: A method used to publish the payload
:param func presence_callback: If the transport support presence
callbacks call this method to notify the
channel of a client's presence
:param func remove_presence_callback: If the transport support presence
callbacks call this method to
notify the channel a client is no
longer present
rB )r. Zpublish_payloadZpresence_callbackZremove_presence_callbackr r r �publish_daemon� s z&DaemonizedPublishServer.publish_daemon�NN)r r r! r" rR rX r r r r rW � s �rW c sX e Zd ZdZ� fdd�Zdd� Zejjj j
ddd��Z d d
� Z d d � Z
d
d� Z� ZS )r zY
The PublishClient receives messages from the PublishServer and runs a callback.
c r8 r9 r: r< r= r r r0 � r? zPublishClient.__init__c C rA )zX
Add a message handler when we receive a message from the PublishServer
rB )r. �callbackr r r �on_recv� rH zPublishClient.on_recvNc C rA )zQ
Create a network connection to the the PublishServer or broker.
rB )r. Z publish_portZconnect_callbackZdisconnect_callbackr r r r1 � rV zPublishClient.connectc C rA )z9
Close the underlying network connection
rB r6 r r r rG rH zPublishClient.closec C s | S r9 r r6 r r r � __enter__ rO zPublishClient.__enter__c C rN r9 r )r. �exc_typeZexc_valZexc_tbr r r �__exit__ rO zPublishClient.__exit__rY )r r r! r" r0 r[ r rI rJ rK rL r1 rG r\ r^ rM r r r= r r � s
�r )r) r4 Zsalt.ext.tornado.genr Z
TRANSPORTSr r r r �Warningr r# r r
rP r rW r r r r r �<module> s