관리-도구
편집 파일: protocol.cpython-38.pyc
U )�e�> � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z dd lm Z dd lmZ dd lmZ dd lmZ dd lmZ ddlmZmZ ddlmZ ddl m Z m Z m Z ddl m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& ej'j(ej)fej'j*ej+fej'j,ej-fej'j.ej/fej'j0e j1fej'j2ejfej'j3ej4fej'j5ej6fej'j7ej8fej'j9ej:fej'j;ej<fg Z=ddd�Z>G dd� de?�Z@G dd� de?�ZAdS )z3Implementation of the X protocol for MySQL servers.� N� )� mysqlx_pb2)�mysqlx_session_pb2)�mysqlx_sql_pb2)�mysqlx_notice_pb2)�mysqlx_datatypes_pb2)�mysqlx_resultset_pb2)�mysqlx_crud_pb2)�mysqlx_expr_pb2)�mysqlx_connection_pb2)�ColumnMetaData)� STRING_TYPES� INT_TYPES)�DbDoc)�InterfaceError�OperationalError�ProgrammingError)� ExprParser�build_null_scalar�build_string_scalar�build_bool_scalar�build_double_scalar�build_int_scalar�utf-8c C s t | t�r| S | �|�S �N)� isinstance�bytes�encode)�value�encoding� r �d/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysqlx/protocol.py�encode_to_bytes@ s r" c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �MessageReaderWriterc C s || _ d | _d S r )�_stream�_msg)�selfZ socket_streamr r r! �__init__E s z MessageReaderWriter.__init__c C s | j d k rtd��|| _ d S )NzMessage push slot is full)r% r �r& �msgr r r! � push_messageI s z MessageReaderWriter.push_messagec C s" | j d k r| j }d | _ |S | �� S r )r% � _read_message)r& �mr r r! � read_messageN s z MessageReaderWriter.read_messagec C sp | j �d�}t�d|�\}}| j �|d �}tD ],}|d |kr0|d � }|�|� | S q0td�|���d S )N� �<LBr r zUnknown msg_type: {0})r$ �read�struct�unpack�_SERVER_MESSAGES�ParseFromString� ValueError�format)r& �hdrZmsg_len�msg_type�payloadZ msg_tupler) r r r! r+ U s z!MessageReaderWriter._read_messagec C s8 |� � }t�dt|�d |�}| j�d�||g�� d S )Nr/ r � )ZSerializeToStringr1 �pack�lenr$ �sendall�join)r& Zmsg_idr) Zmsg_str�headerr r r! � write_messageb s z!MessageReaderWriter.write_messageN)�__name__� __module__� __qualname__r' r* r- r+ r@ r r r r! r# D s r# c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1S )2�Protocolc C s || _ || _d | _d S r )�_reader�_writer�_message)r& Z reader_writerr r r! r' i s zProtocol.__init__c C s$ t �� }| j�tjj|� | j�� S r ) �MySQLxConnectionZCapabilitiesGetrF r@ �MySQLx�ClientMessagesZCON_CAPABILITIES_GETrE r- r( r r r! �get_capabilitesn s �zProtocol.get_capabilitesc K s\ t �� }|�� D ]0\}}| �|�}t j||d�}|jj�|g� q| j�t j j |� | � � S )N)�namer ) rH ZCapabilitiesSet�items� _create_anyZ CapabilityZ capabilities�extendrF r@ rI rJ ZCON_CAPABILITIES_SET�read_ok)r& �kwargsr) �keyr Z capabilityr r r! �set_capabilitiest s �zProtocol.set_capabilitiesc C s" t j|d�}| j�tjj|� d S )N)Z mech_name)� MySQLxSessionZAuthenticateStartrF r@ rI rJ ZSESS_AUTHENTICATE_START)r& �methodr) r r r! �send_auth_start� s �zProtocol.send_auth_startc C s$ | j �� }t|tj�s td��|jS )Nz>Unexpected message encountered during authentication handshake)rE r- r rT �AuthenticateContinuer � auth_datar( r r r! �read_auth_continue� s zProtocol.read_auth_continuec C s& t jt|�d�}| j�tjj|� d S )N)rX )rT rW r" rF r@ rI rJ �SESS_AUTHENTICATE_CONTINUE)r& �datar) r r r! �send_auth_continue� s � �zProtocol.send_auth_continuec C s4 | j �� }t|tj�rq0t|tj�r t|j��q d S r ) rE r- r rT �AuthenticateOkrI �Errorr r) r( r r r! � read_auth_ok� s zProtocol.read_auth_okc C sf t |j�}|d g }|jD ]F}|d }||jkr>td�|���|j| }| �|d |j �||< q|S )NrL z-Unable to find placeholder for parameter: {0}r )r< Z _binding_mapZ _bindingsr r6 �arg_object_to_scalar� _doc_based)r& � statement�countZscalarsZbindingrL �posr r r! �get_binding_scalars� s � � z Protocol.get_binding_scalarsc C s� |j r|j�|j� |jr,|j�| �|�� |jrF|j |j _ |j |j _ |jrZ|j�|j� |jrn|j�|j� |jr�|j�|j� d S r )Z _has_where�criteria�CopyFromZ _where_exprZ _has_bindings�argsrO re Z _has_limitZ_limit_row_count�limit� row_countZ _limit_offset�offsetZ _has_sort�orderZ _sort_exprZ _has_group_by�groupingZ _groupingZ _has_havingZgrouping_criteriaZ_having)r& �messagerb r r r! � _apply_filter� s zProtocol._apply_filterc C sd t j|jrt jnt jt j|jj|jjd�d�}|j rB|j � |j � | � ||� | j�tjj|� d S �N�rL �schema�Z data_modelZ collection)� MySQLxCrudZFindra �DOCUMENT�TABLE� Collection�targetrL rr Z_has_projection� projectionrO Z_projection_exprro rF r@ rI rJ Z CRUD_FIND)r& �stmt�findr r r! � send_find� s � �� zProtocol.send_findc C s� t j|jrt jnt jt j|jj|jjd�d�}| � ||� |j D ]H}t j |j |j d�}|jd k rz|j�| �|j|j �� |j�|g� q@| j�tjj|� d S )Nrq rs )� operation�source)rt ZUpdatera ru rv rw rx rL rr ro Z _update_opsZUpdateOperationZ update_typer~ r rg �arg_object_to_exprr} rO rF r@ rI rJ Z CRUD_UPDATE)r& rb �updateZ update_opZopexprr r r! � send_update� s. � �� � ��zProtocol.send_updatec C sP t j|jrt jnt jt j|jj|jjd�d�}| � ||� | j � t j j|� d S rp )rt �Deletera ru rv rw rx rL rr ro rF r@ rI rJ Z CRUD_DELETE)r& rz �deleter r r! � send_delete� s � �� zProtocol.send_deletec C sL t j|t|�dd�}|D ] }| �|�}|j�|g� q| j�tj j |� d S )NF)� namespacerz Zcompact_metadata) � MySQLxSQLZ StmtExecuter" rN rh rO rF r@ rI rJ ZSQL_STMT_EXECUTE)r& r� rz rh �argr r r r! �send_execute_statement� s � �z Protocol.send_execute_statementc C s� t j|jrt jnt jt j|jj|jjd�d�}t |d�r`|j D ] }|j � t ||j ��� g� q>|jD ]n}t j�� }t|t�r�|D ]"}| �||j �}|j� |g� q�n | �||j �}|j� |g� |j� |g� qf| j�tjj|� d S )Nrq rs �_fields)rt ZInsertra ru rv rw rx rL rr �hasattrr� ry rO r Zparse_table_insert_fieldZ_valuesZTypedRowr �listr �field�rowrF r@ rI rJ Z CRUD_INSERT)r& rb �insertr� r r� �val�objr r r! � send_insert� s4 � �� � �zProtocol.send_insertc C st t |t�r8tjjt|�d�}tjd|d�}tjd|d�S t |t�rTtjdt|�d�S t |t �rptjdt |�d�S d S )N)r � )�typeZv_stringr )r� �scalar) r r �MySQLxDatatypesZScalar�Stringr" �Any�boolr r r )r& r� r� r� r r r! rN � s zProtocol._create_anyc C s | � |�}|d k rtd��d S )Nz Expected to close the result)r+ r �r& �rsr) r r r! � close_result s zProtocol.close_resultc C s6 | � |�}|d krd S t|tj�r&|S | j�|� d S r )r+ r �MySQLxResultset�RowrE r* r� r r r! �read_row s zProtocol.read_rowc C s� |j dkr:t�� }|�|j� |j�t|j|j|j �� np|j dkrZt� � }|�|j� nP|j dkr�t� � }|�|j� |j tj j kr�|jj|_n|j tj jkr�|jj|_d S )Nr � � )r� � MySQLxNotice�Warningr4 r9 � _warnings�append�level�coder) ZSessionVariableChangedZSessionStateChanged�paramZ ROWS_AFFECTEDr Zv_unsigned_intZ_rows_affectedZGENERATED_INSERT_IDZ _generated_id)r& r) r� Z warningMsgZ sessVarMsgZ sessStateMsgr r r! �_process_frame s( � � �zProtocol._process_framec C s| | j �� }t|tj�r"t|j��q t|tj�r<| � ||� q t|t j �rLd S t|t j �r`d|_q t|t j�rxd|_q qxq |S )NT)rE r- r rI r^ r r) r� �Framer� r� � StmtExecuteOkr� � FetchDone�_closed�FetchDoneMoreResultsetsZ_has_more_resultsr� r r r! r+ * s zProtocol._read_messagec C s� g }| � |�}|d krq�t|tj�r2| j�|� q�t|tj�sFtd��t|j|j |j |j |j |j |j|j|j|j|j|j� }|�|� q|S )NzUnexpected msg type)r+ r r� r� rE r* r r r� �catalogrr �tableZoriginal_tablerL � original_name�lengthZ collationZfractional_digits�flags� content_typer� )r& r� �columnsr) �colr r r! �get_column_metadata; s* � z Protocol.get_column_metadatac C s |d kr t jt jjt� d�S t|t�r<t jt jjt|�d�S t|t�r\t jt jjt|�d�S t|t �r|t jt jjt |�d�S t|t �r�z4t ||�� � }|�� r�t jt jjt|�d�W S |W S t jt jjt|�d� Y S X n&t|t��rt jt jjtt|��d�S td�t|����d S )N)r� �literalzUnsupported type: {0})� MySQLxExpr�Expr�LITERALr r r� r r r �floatr r r �exprZhas_identifierr r �strr r6 r� )r& r �allow_relationalZ expressionr r r! r N sB � � � � � � �zProtocol.arg_object_to_exprc C s | � ||�jS r )r r� )r& r r� r r r! r` j s z Protocol.arg_object_to_scalarc C s8 | j �� }t|tj�r t|j��t|tj�s4td��d S )Nz Unexpected message encountered)rE r- r rI r^ r r) �Okr( r r r! rP m s zProtocol.read_okc C s t �� }| j�tjj|� d S r )rT �CloserF r@ rI rJ Z SESS_CLOSEr( r r r! � send_closeu s zProtocol.send_closeN)rA rB rC r' rK rS rV rY r\ r_ re ro r| r� r� r� r� rN r� r� r� r+ r� r r` rP r� r r r r! rD h s0 rD )r )B�__doc__r1 �protobufr rI r rT r r� r r� r r� r r� r rt r r� r rH �resultr �compatr r Zdbdocr �errorsr r r r� r r r r r r ZServerMessagesrZ rW ZSESS_AUTHENTICATE_OKr] ZSQL_STMT_EXECUTE_OKr� �ERRORr^ ZNOTICEr� ZRESULTSET_COLUMN_META_DATAZ RESULTSET_ROWr� ZRESULTSET_FETCH_DONEr� Z$RESULTSET_FETCH_DONE_MORE_RESULTSETSr� �OKr� ZCONN_CAPABILITIESZ Capabilitiesr3 r" �objectr# rD r r r r! �<module> sL �� � � � $