a

�,�hn�	@s�dZd
dgZ
ddlZddlZddlZzddlZWneyF


dZYn0ddlmZm	Z	m
Z
mZ
erne
�d�

Gdd�dej
�Z
Gd	d�dejj�Zdd
d�
eeee	eejefd�d
d�Ze
e	e	ejd�dd
�ZdS)z
Utilities.�	to_thread�open_connection�N)�
TYPE_CHECKING�Any�Callable�Tuple�StreamWriterc@s"eZ
dZd
Zejdd�dd�ZdS)�StreamReaderProtocolz�Extends asyncio.streams.StreamReaderProtocol for adding start_tls().

    The ``start_tls()`` is based on ``asyncio.streams.StreamWriter`` introduced
    in Python 3.11. It provides the same functionality for older Python versions.
    N)�writer�returncCs&|
j}|
|_
||_|�d
�
du
|_dS)zQReplace stream writer.

        Args:
            writer: Stream Writer.
        �
sslcontextN)�	transport�_stream_writer�
_transport�get_extra_info�	_over_ssl)�selfr
r
�r�L/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/aio/utils.py�_replace_writer:s


z$StreamReaderProtocol._replace_writer)�__name__�
__module__�__qualname__�__doc__�asynciorrrrrrr	3s
r	c@s.eZ
dZd
Zddd�ejeedd�dd�ZdS)rz�Extends asyncio.streams.StreamWriter for adding start_tls().

    The ``start_tls()`` is based on ``asyncio.streams.StreamWriter`` introduced
    in Python 3.11. It provides the same functionality for older Python versions.
    N)�server_hostname�ssl_handshake_timeout)�ssl_contextrrrc�sT|jj
d
u
}|j}|��Id
H
|jj|j||
|||d�Id
H}||_|�|�

d
S)z�Upgrade an existing stream-based connection to TLS.

        Args:
            ssl_context: Configured SSL context.
            server_hostname: Server host name.
            ssl_handshake_timeout: SSL handshake timeout.
        N)�server_siderr)�	_protocol�_client_connected_cb�drain�_loop�	start_tlsrr)rrrrr�protocolZ
new_transportrrrr#Ms







�
�zStreamWriter.start_tls)	rrrr�ssl�
SSLContext�str�intr#rrrrrFs


�

�i
)
�limit)�host�portr)�kwdsrc
	�sbt�
�}tjj||d
�}t||d��|j�f
dd�||
fi|�
�
IdH\}}t|�||�}||fS)aA wrapper for create_connection() returning a (reader, writer) pair.

    This function is based on ``asyncio.streams.open_connection`` and adds a custom
    stream reader.

    MySQL expects TLS negotiation to happen in the middle of a TCP connection, not at
    the start.
    This function in conjunction with ``_StreamReaderProtocol`` and ``_StreamWriter``
    allows the TLS negotiation on an existing connection.

    Args:
        host: Server host name.
        port: Server port.
        limit: The buffer size limit used by the returned ``StreamReader`` instance.
               By default the limit is set to 64 KiB.

    Returns:
        tuple: Returns a pair of reader and writer objects that are instances of
               ``StreamReader`` and ``StreamWriter`` classes.
    )r)�loop)
r-c
s�S)
Nrr�
r$rr�<lambda>��z!open_connection.<locals>.<lambda>N)r�get_running_loop�streams�StreamReaderr	�create_connectionr)	r*r+r)r,r-�readerr
�
_r
rr.rrms


(

)�func�args�kwargsrc
�s>t�
�}t��}tj|j|g|
�
Ri|�
�
}|�d
|�Id
HS)aQ
Asynchronously run function ``func`` in a separate thread.

    This function is based on ``asyncio.to_thread()`` introduced in Python 3.9, which
    provides the same functionality for older Python versions.

    Returns:
        coroutine: A coroutine that can be awaited to get the eventual result of
                   ``func``.
    N)rr1�contextvars�copy_context�	functools�partial�run�run_in_executor)r7r8r9r-�ctx�	func_callrrrr
�s



)NN)r�__all__rr:r<r%�ImportError�typingrrrr�appendr	r2rr'r(r3r�Futurer
rrrr�<module> s*







(�
�

�