File: //lib64/python3.6/site-packages/tornado/__pycache__/tcpclient.cpython-36.pyc
3
��fz/ � @ s� d Z ddlZddlZddlZddlZddlZddlmZmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ ddlmZ dd lmZ ddlZdd
lmZmZmZmZmZmZm Z ej r�dd lm Z m Z d Z G d
d� de!�Z"G dd� de!�Z#dS )z'A non-blocking TCP connection factory.
� N)�Future�future_add_done_callback)�IOLoop)�IOStream)�gen)�Resolver)�set_close_exec)� TimeoutError)�Any�Union�Dict�Tuple�List�Callable�Iterator)�Optional�Setg333333�?c @ sN e Zd ZdZee eejegee df f dd�dd�Z
e ee eeeejef eeejef f d�dd ��Z e
dfeeeejf d
d �d d
�Zeeejef dd�dd�Zeeejef ejeddd�dd�Zedd�dd�Zdd�dd�Zdd�dd�Zeeejf dd �d d �Zdd�d d �Zdd�d!d"�Zdd�d#d$�Z dS )%�
_Connectora� A stateless implementation of the "Happy Eyeballs" algorithm.
"Happy Eyeballs" is documented in RFC6555 as the recommended practice
for when both IPv4 and IPv6 addresses are available.
In this implementation, we partition the addresses by family, and
make the first connection attempt to whichever address was
returned first by ``getaddrinfo``. If that connection fails or
times out, we begin a connection in parallel to the first address
of the other family. If there are additional failures we retry
with other addresses, keeping one connection attempt per family
in flight at a time.
http://tools.ietf.org/html/rfc6555
zFuture[IOStream]N)�addrinfo�connect�returnc C sR t j� | _|| _t� | _d | _d | _d | _t |�| _
| j |�\| _ | _
t� | _d S )N)r �current�io_loopr r �future�timeout�connect_timeout�
last_error�len� remaining�split�
primary_addrs�secondary_addrs�set�streams)�selfr r � r% �!/usr/lib64/python3.6/tcpclient.py�__init__< s
z_Connector.__init__)r r c C sT g }g }| d d }x6| D ].\}}||kr:|j ||f� q|j ||f� qW ||fS )a� Partition the ``addrinfo`` list by address family.
Returns two lists. The first list contains the first entry from
``addrinfo`` and all others with the same family, and the
second list contains all other addresses (normally one list will
be AF_INET and the other AF_INET6, although non-standard resolvers
may return additional families).
r )�append)r ZprimaryZ secondaryZ
primary_af�af�addrr% r% r& r P s z_Connector.splitz2Future[Tuple[socket.AddressFamily, Any, IOStream]])r r r c C s2 | j t| j�� | j|� |d k r,| j|� | jS )N)� try_connect�iterr � set_timeout�set_connect_timeoutr )r$ r r r% r% r& �starti s
z_Connector.start)�addrsr c C s� yt |�\}}W n@ tk
rP | jdkrL| jj� rL| jj| jpHtd�� d S X | j||�\}}| j j
|� t |t j
| j|||�� d S )Nr zconnection failed)�next�
StopIterationr r �done�
set_exceptionr �IOErrorr r# �addr � functools�partial�on_connect_done)r$ r0 r) r* �streamr r% r% r& r+ t s z_Connector.try_connect)r0 r) r* r r c C s� | j d8 _ y |j� }W n^ tk
rx } zB| jj� r:d S || _| j|� | jd k rj| jj | j� | j
� d S d }~X nX | j � | jj� r�|j � n&| j
j|� | jj|||f� | j� d S )N� )r �result� Exceptionr r3 r r+ r r �remove_timeout�
on_timeout�clear_timeouts�closer# �discardZ
set_result�
close_streams)r$ r0 r) r* r r: �er% r% r& r9 � s$
z_Connector.on_connect_done)r r c C s | j j| j j� | | j�| _d S )N)r � add_timeout�timer? r )r$ r r% r% r& r- � s z_Connector.set_timeout)r c C s$ d | _ | jj� s | jt| j�� d S )N)r r r3 r+ r, r! )r$ r% r% r& r? � s
z_Connector.on_timeoutc C s | j d k r| jj| j � d S )N)r r r> )r$ r% r% r& �
clear_timeout� s
z_Connector.clear_timeout)r r c C s | j j|| j�| _d S )N)r rE �on_connect_timeoutr )r$ r r% r% r& r. � s z _Connector.set_connect_timeoutc C s$ | j j� s| j jt� � | j� d S )N)r r3 r4 r rC )r$ r% r% r& rH � s
z _Connector.on_connect_timeoutc C s4 | j d k r| jj| j � | jd k r0| jj| j� d S )N)r r r> r )r$ r% r% r& r@ � s
z_Connector.clear_timeoutsc C s x| j D ] }|j� qW d S )N)r# rA )r$ r: r% r% r& rC � s z_Connector.close_streams) �__name__�
__module__� __qualname__�__doc__r r
r �socket�
AddressFamilyr r' � staticmethodr �_INITIAL_CONNECT_TIMEOUT�floatr �datetime� timedeltar/ r r+ r9 r- r? rG r. rH r@ rC r% r% r% r& r * s6 " r c @ s� e Zd ZdZdedd�dd�Zdd�dd�Zejdddddfe e
ej e e
e ef ejf e
e e
e eejf ed � d
d �Zde
ej ee e
eed f d
�dd�ZdS )� TCPClientz�A non-blocking TCP connection factory.
.. versionchanged:: 5.0
The ``io_loop`` argument (deprecated since version 4.1) has been removed.
N)�resolverr c C s( |d k r|| _ d| _nt� | _ d| _d S )NFT)rU �
_own_resolverr )r$ rU r% r% r& r' � s
zTCPClient.__init__)r c C s | j r| jj� d S )N)rV rU rA )r$ r% r% r& rA � s zTCPClient.close) �host�portr) � ssl_options�max_buffer_size� source_ip� source_portr r c
� s |dk rTt |tj�r&tj� j� | }n.t |tj�rHtj� j� |j� }n t d| ��|dk r|t
j || j j
|||��I dH } n| j j
|||�I dH } t| tj| j|||d��}
|
j|d�I dH \}} } |dk �r |dk r�t
j || jd||d��I dH } n| jd||d�I dH } | S )a� Connect to the given host and port.
Asynchronously returns an `.IOStream` (or `.SSLIOStream` if
``ssl_options`` is not None).
Using the ``source_ip`` kwarg, one can specify the source
IP address to use when establishing the connection.
In case the user needs to resolve and
use a specific interface, it has to be handled outside
of Tornado as this depends very much on the platform.
Raises `TimeoutError` if the input future does not complete before
``timeout``, which may be specified in any form allowed by
`.IOLoop.add_timeout` (i.e. a `datetime.timedelta` or an absolute time
relative to `.IOLoop.time`)
Similarly, when the user requires a certain source port, it can
be specified using the ``source_port`` arg.
.. versionchanged:: 4.5
Added the ``source_ip`` and ``source_port`` arguments.
.. versionchanged:: 5.0
Added the ``timeout`` argument.
NzUnsupported timeout %r)r[ r\ )r F)rY Zserver_hostname)�
isinstance�numbersZRealr r rF rR rS Z
total_seconds� TypeErrorr Z with_timeoutrU Zresolver r7 r8 �_create_streamr/ Z start_tls)
r$ rW rX r) rY rZ r[ r\ r r Z connectorr* r: r% r% r& r � s6 $
zTCPClient.connectzFuture[IOStream])rZ r) r* r[ r\ r c C s� t |t�r|nd}|}|r2| r2|tjkr.dnd}tj|�}t|j� � |sP|r�y|j||f� W n tjk
r� |j� � Y nX yt ||d�} W n6 tjk
r� }
zt
� } | j |
� | | fS d }
~
X nX | | j |�fS d S )Nr z::1z 127.0.0.1)rZ )
r] �intrM ZAF_INET6r �filenoZbind�errorrA r r r4 r ) r$ rZ r) r* r[ r\ Zsource_port_bindZsource_ip_bindZ
socket_objr: rD Zfur% r% r& r` * s&
zTCPClient._create_stream)N)NN)rI rJ rK rL r r' rA rM Z AF_UNSPEC�strra rN r r r
�sslZ
SSLContextrQ rR rS r r r
r` r% r% r% r& rT � s 2G rT )$rL r7 rM r^ rR re Ztornado.concurrentr r Ztornado.ioloopr Ztornado.iostreamr Ztornador Ztornado.netutilr Ztornado.platform.autor Z tornado.genr Ztypingr
r r r
r r r Z
TYPE_CHECKINGr r rP �objectr rT r% r% r% r& �<module> s( $ "