
    9h                    b   S SK Jr  S SKJrJr  S SKJrJrJrJ	r	J
r
Jr  S SKJrJr  SSKJrJrJrJrJrJrJrJr   " S S\	5      r " S	 S
\	5      r " S S\S   \	5      r\ " S S\	5      5       r " S S\	5      r " S S\	5      r " S S\S   \	5      r\ " S S\	5      5       r " S S\	5      r g)    )annotations)IterableIterator)AnyClassVarOptionalProtocolUnionruntime_checkable)LiteralNever   )
CodecGzipT	CodecLz4T
CodecMaskT
CodecNoneTCodecSnappyT
CodecZstdTDefaultCompressionTypeTLegacyCompressionTypeTc                  0   \ rS rSr              SS jr            SS jrSS jrSS jr            SS jr\	        SS j5       r
\	        SS j5       r        SS	 jr\SS
 j5       r\SS j5       r\SS j5       rSrg)!DefaultRecordBatchBuilderProtocol   c                    g N )selfmagiccompression_typeis_transactionalproducer_idproducer_epochbase_sequence
batch_sizes           _C:\Suresh\moveshuttle\MDcreated\moveengine\venv\Lib\site-packages\aiokafka/record/_protocols.py__init__*DefaultRecordBatchBuilderProtocol.__init__   s         c                    g r   r   r   offset	timestampkeyvalueheaderss         r%   append(DefaultRecordBatchBuilderProtocol.append&   s     36r(   c                    g r   r   r   s    r%   build'DefaultRecordBatchBuilderProtocol.build.   s    #r(   c                    g r   r   r3   s    r%   size&DefaultRecordBatchBuilderProtocol.size/   s    3r(   c                    g r   r   r*   s         r%   size_in_bytes/DefaultRecordBatchBuilderProtocol.size_in_bytes0   s     r(   c                    g r   r   clsr-   r.   r/   s       r%   size_of)DefaultRecordBatchBuilderProtocol.size_of8        r(   c                    g r   r   r=   s       r%   estimate_size_in_bytes8DefaultRecordBatchBuilderProtocol.estimate_size_in_bytes?   rA   r(   c                    g r   r   )r   r!   r"   r#   s       r%   set_producer_state4DefaultRecordBatchBuilderProtocol.set_producer_stateF   s    r(   c                    g r   r   r3   s    r%   r!   -DefaultRecordBatchBuilderProtocol.producer_idI       "%r(   c                    g r   r   r3   s    r%   r"   0DefaultRecordBatchBuilderProtocol.producer_epochK       %(r(   c                    g r   r   r3   s    r%   r#   /DefaultRecordBatchBuilderProtocol.base_sequenceM       $'r(   r   N)r   intr   r   r    rQ   r!   rQ   r"   rQ   r#   rQ   r$   rQ   )r+   rQ   r,   Optional[int]r-   Optional[bytes]r.   rS   r/   !list[tuple[str, Optional[bytes]]]returnz'Optional[DefaultRecordMetadataProtocol]rU   	bytearrayrU   rQ   )r+   rQ   r,   rQ   r-   rS   r.   rS   r/   rT   rU   rQ   )r-   rS   r.   rS   r/   rT   rU   rQ   )r!   rQ   r"   rQ   r#   rQ   rU   None)__name__
__module____qualname____firstlineno__r&   r0   r4   r7   r:   classmethodr?   rC   rF   propertyr!   r"   r#   __static_attributes__r   r(   r%   r   r      s   		 2	 		
 	 	 	 	66 !6 	6
 6 36 
16 &  	
  3 
   3	
 
    3	
 
 03DG	 % %( (' 'r(   r   c                  n    \ rS rSrS	S jr\S
S j5       r\SS j5       r\S
S j5       r\S
S j5       r	Sr
g)DefaultRecordMetadataProtocolQ   c                    g r   r   )r   r+   r7   r,   s       r%   r&   &DefaultRecordMetadataProtocol.__init__R   s    r(   c                    g r   r   r3   s    r%   r+   $DefaultRecordMetadataProtocol.offsetS        r(   c                    g r   r   r3   s    r%   crc!DefaultRecordMetadataProtocol.crcU       r(   c                    g r   r   r3   s    r%   r7   "DefaultRecordMetadataProtocol.sizeW   rl   r(   c                    g r   r   r3   s    r%   r,   'DefaultRecordMetadataProtocol.timestampY        #r(   r   N)r+   rQ   r7   rQ   r,   rQ   rU   rY   rX   rU   rY   rZ   r[   r\   r]   r&   r_   r+   rj   r7   r,   r`   r   r(   r%   rb   rb   Q   s=    K     # #r(   rb   c                     \ rS rSr% S\S'   S\S'   S\S'   S\S	'   S
\S'   S\S'   S!S jr\S"S j5       r\S"S j5       r\S"S j5       r	\S"S j5       r
\S"S j5       r\S"S j5       r\S#S j5       r\S#S j5       r\S"S j5       r\S"S j5       r\S"S j5       r\S"S j5       r\S"S j5       r\S"S j5       r\S"S j5       rS#S jrSrg )$DefaultRecordBatchProtocol]   ClassVar[CodecMaskT]
CODEC_MASKzClassVar[CodecNoneT]
CODEC_NONEClassVar[CodecGzipT]
CODEC_GZIPClassVar[CodecSnappyT]CODEC_SNAPPYClassVar[CodecLz4T]	CODEC_LZ4zClassVar[CodecZstdT]
CODEC_ZSTDc                    g r   r   )r   buffers     r%   r&   #DefaultRecordBatchProtocol.__init__e   s    Sr(   c                    g r   r   r3   s    r%   base_offset&DefaultRecordBatchProtocol.base_offsetf   rJ   r(   c                    g r   r   r3   s    r%   r    DefaultRecordBatchProtocol.magich   s    r(   c                    g r   r   r3   s    r%   rj   DefaultRecordBatchProtocol.crcj       r(   c                    g r   r   r3   s    r%   
attributes%DefaultRecordBatchProtocol.attributesl   s    !$r(   c                    g r   r   r3   s    r%   r   +DefaultRecordBatchProtocol.compression_typen   s    '*r(   c                    g r   r   r3   s    r%   timestamp_type)DefaultRecordBatchProtocol.timestamp_typep   rM   r(   c                    g r   r   r3   s    r%   r    +DefaultRecordBatchProtocol.is_transactionalr       (+r(   c                    g r   r   r3   s    r%   is_control_batch+DefaultRecordBatchProtocol.is_control_batcht   r   r(   c                    g r   r   r3   s    r%   last_offset_delta,DefaultRecordBatchProtocol.last_offset_deltav   r   r(   c                    g r   r   r3   s    r%   first_timestamp*DefaultRecordBatchProtocol.first_timestampx       &)r(   c                    g r   r   r3   s    r%   max_timestamp(DefaultRecordBatchProtocol.max_timestampz   rP   r(   c                    g r   r   r3   s    r%   r!   &DefaultRecordBatchProtocol.producer_id|   rJ   r(   c                    g r   r   r3   s    r%   r"   )DefaultRecordBatchProtocol.producer_epoch~   rM   r(   c                    g r   r   r3   s    r%   r#   (DefaultRecordBatchProtocol.base_sequence   rP   r(   c                    g r   r   r3   s    r%   next_offset&DefaultRecordBatchProtocol.next_offset   rJ   r(   c                    g r   r   r3   s    r%   validate_crc'DefaultRecordBatchProtocol.validate_crc       Cr(   r   N)r   #Union[bytes, bytearray, memoryview]rU   rY   rX   rU   bool)rZ   r[   r\   r]   __annotations__r&   r_   r   r   rj   r   r   r   r    r   r   r   r   r!   r"   r#   r   r   r`   r   r(   r%   ru   ru   ]   s    $$$$$$((""$$P% %  $ $* *( (+ ++ ++ +) )' '% %( (' '% %'r(   ru   DefaultRecordProtocolc                      \ rS rSr              SS jr\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS j5       r\SS	 j5       rS
rg)r      c                    g r   r   )r   r+   r,   r   r-   r.   r/   s          r%   r&   DefaultRecordProtocol.__init__        r(   c                    g r   r   r3   s    r%   r+   DefaultRecordProtocol.offset   rh   r(   c                    gzEpoch millisecondsNr   r3   s    r%   r,   DefaultRecordProtocol.timestamp       r(   c                    gz CREATE_TIME(0) or APPEND_TIME(1)Nr   r3   s    r%   r   $DefaultRecordProtocol.timestamp_type   r   r(   c                    gzBytes key or NoneNr   r3   s    r%   r-   DefaultRecordProtocol.key   r   r(   c                    gzBytes value or NoneNr   r3   s    r%   r.   DefaultRecordProtocol.value   r   r(   c                    g r   r   r3   s    r%   r/   DefaultRecordProtocol.headers   s    <?r(   c                    g r   r   r3   s    r%   checksumDefaultRecordProtocol.checksum   rq   r(   r   N)r+   rQ   r,   rQ   r   rQ   r-   rS   r.   rS   r/   rT   rU   rY   rX   rU   rS   )rU   rT   rr   rZ   r[   r\   r]   r&   r_   r+   r,   r   r-   r.   r/   r   r`   r   r(   r%   r   r      s      	
   3 
    ! ! / /     " " ? ?# #r(   c                      \ rS rSr        S
S jr S           SS jjrSS jrSS jr          SS jr\	SS j5       r
S	rg) LegacyRecordBatchBuilderProtocol   c                    g r   r   )r   r   r   r$   s       r%   r&   )LegacyRecordBatchBuilderProtocol.__init__   s    
 r(   Nc                    g r   r   r*   s         r%   r0   'LegacyRecordBatchBuilderProtocol.append   s     25r(   c                    g)z#Compress batch to be ready for sendNr   r3   s    r%   r4   &LegacyRecordBatchBuilderProtocol.build   r   r(   c                    g)z-Return current size of data written to bufferNr   r3   s    r%   r7   %LegacyRecordBatchBuilderProtocol.size   r   r(   c                    g)zActual size of message to addNr   )r   r+   r,   r-   r.   s        r%   r:   .LegacyRecordBatchBuilderProtocol.size_in_bytes   r   r(   c                    g r   r   )r>   r   s     r%   record_overhead0LegacyRecordBatchBuilderProtocol.record_overhead   s    14r(   r   )r   zLiteral[0, 1]r   r   r$   rQ   rU   rY   r   )r+   rQ   r,   rR   r-   rS   r.   rS   r/   r   rU   z&Optional[LegacyRecordMetadataProtocol]rV   rX   )
r+   rQ   r,   rQ   r-   rS   r.   rS   rU   rQ   )r   rQ   rU   rQ   )rZ   r[   r\   r]   r&   r0   r4   r7   r:   r^   r   r`   r   r(   r%   r   r      s     1 	
 
 55 !5 	5
 5 5 
052<,, , 	,
 , 
, 4 4r(   r   c                  n    \ rS rSrS	S jr\S
S j5       r\S
S j5       r\S
S j5       r\S
S j5       r	Sr
g)LegacyRecordMetadataProtocol   c                    g r   r   )r   r+   rj   r7   r,   s        r%   r&   %LegacyRecordMetadataProtocol.__init__   s    RUr(   c                    g r   r   r3   s    r%   r+   #LegacyRecordMetadataProtocol.offset   rh   r(   c                    g r   r   r3   s    r%   rj    LegacyRecordMetadataProtocol.crc   r   r(   c                    g r   r   r3   s    r%   r7   !LegacyRecordMetadataProtocol.size   rl   r(   c                    g r   r   r3   s    r%   r,   &LegacyRecordMetadataProtocol.timestamp   rq   r(   r   N)
r+   rQ   rj   rQ   r7   rQ   r,   rQ   rU   rY   rX   rs   r   r(   r%   r   r      s=    U     # #r(   r   c                      \ rS rSr% S\S'   S\S'   S\S'   S\S	'   S
\S'   S
\S'   S\S'   SS jr\SS j5       rSS jrSr	g)LegacyRecordBatchProtocol   rw   rx   rz   r{   r|   r}   r~   r   r   r   r    rR   r!   c                    g r   r   )r   r   r   s      r%   r&   "LegacyRecordBatchProtocol.__init__   s    QTr(   c                    g r   r   r3   s    r%   r   %LegacyRecordBatchProtocol.next_offset   rJ   r(   c                    g r   r   r3   s    r%   r   &LegacyRecordBatchProtocol.validate_crc   r   r(   r   N)r   r   r   rQ   rX   r   )
rZ   r[   r\   r]   r   r&   r_   r   r   r`   r   r(   r%   r   r      s@    $$$$((""T% %'r(   r   LegacyRecordProtocolc                      \ rS rSr              SS jr\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS j5       r\SS	 j5       rS
rg)r      c                    g r   r   )r   r+   r,   r   r-   r.   rj   s          r%   r&   LegacyRecordProtocol.__init__   r   r(   c                    g r   r   r3   s    r%   r+   LegacyRecordProtocol.offset   rh   r(   c                    gr   r   r3   s    r%   r,   LegacyRecordProtocol.timestamp   r   r(   c                    gr   r   r3   s    r%   r   #LegacyRecordProtocol.timestamp_type   r   r(   c                    gr   r   r3   s    r%   r-   LegacyRecordProtocol.key   r   r(   c                    gr   r   r3   s    r%   r.   LegacyRecordProtocol.value  r   r(   c                    g r   r   r3   s    r%   r/   LegacyRecordProtocol.headers  r   r(   c                    g r   r   r3   s    r%   r   LegacyRecordProtocol.checksum  s    "r(   r   N)r+   rQ   r,   rR   r   Optional[Literal[0, 1]]r-   rS   r.   rS   rj   rQ   rU   rY   rX   )rU   rR   )rU   r
  r   )rU   zlist[Never]r   r   r(   r%   r   r      s     ! 0	
    
    ! ! / /     " " ) )" "r(   c                  @    \ rS rSrSS jrS	S jrS
S jr  SS jrSrg)MemoryRecordsProtocoli  c                    g r   r   )r   
bytes_datas     r%   r&   MemoryRecordsProtocol.__init__  s    3r(   c                    g r   r   r3   s    r%   r:   #MemoryRecordsProtocol.size_in_bytes  r   r(   c                    g r   r   r3   s    r%   has_nextMemoryRecordsProtocol.has_next  s    r(   c                    g r   r   r3   s    r%   
next_batch MemoryRecordsProtocol.next_batch  s    QTr(   r   N)r  bytesrU   rY   rX   r   )rU   zFOptional[Union[DefaultRecordBatchProtocol, LegacyRecordBatchProtocol]])	rZ   r[   r\   r]   r&   r:   r  r  r`   r   r(   r%   r  r    s    6'#U	OUr(   r  N)!
__future__r   collections.abcr   r   typingr   r   r   r	   r
   r   typing_extensionsr   r   _typesr   r   r   r   r   r   r   r   r   rb   ru   r   r   r   r   r   r  r   r(   r%   <module>r     s    " .  -	 	 	3( 3(l	$H 	$'(*A!BH '(T $H $ $D5x 5D	$8 	$()? @( (  #8 # #DUH Ur(   