B .l\1ã@s€ddlZddlZddlZddlZdZe d¡jZe d¡jZe d¡j Z e d¡j Z dd„Z dd„Z Gd d „d eƒZd d „ZdS) éNióiódcCst|ƒ|||d…<dS)Né)Ú_pack_double_func)ÚdataÚposÚvalue©r ú:lib/python3.7/site-packages/prometheus_client/mmap_dict.pyÚ _pack_doublesr cCst|ƒ|||d…<dS)Né)Ú_pack_integer_func)rrrr r r Ú _pack_integersrc@sJeZdZdZddd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dS)Ú MmapedDicta…A dict of doubles, backed by an mmapped file. The file starts with a 4 byte int, indicating how much of it is used. Then 4 bytes of padding. There's then a number of entries, consisting of a 4 byte int which is the size of the next field, a utf-8 encoded string key, padding to a 8 byte alignment, and then a 8 byte float which is the value. Not thread safe. FcCsÚt||r dndƒ|_||_t |j ¡¡jdkr<|j t¡t |j ¡¡j|_ t j |j ¡|j |rjt j nt j d|_ i|_t|j dƒd|_|jdkr°d|_t|j d|jƒn&|sÖx | ¡D]\}}}||j|<q¾WdS)NÚrbza+br)Úaccessr)ÚopenÚ_fÚ_fnameÚosÚfstatÚfilenoÚst_sizeÚtruncateÚ_INITIAL_MMAP_SIZEÚ _capacityÚmmapZ ACCESS_READZ ACCESS_WRITEÚ_mÚ _positionsÚ_unpack_integerÚ_usedrÚ_read_all_values)ÚselfÚfilenameZ read_modeÚkeyÚ_rr r r Ú__init__&s & zMmapedDict.__init__cCsä| d¡}|ddt|ƒdd}t d t|ƒ¡ ¡t|ƒ|d¡}xJ|jt|ƒ|jkr’|jd9_|j |j¡t   |j  ¡|j¡|_ qJW||j |j|jt|ƒ…<|jt|ƒ7_t |j d|jƒ|jd|j |<d S) z0Initialize a value. Lock must be held by caller.zutf-8ó rr zi{0}sdgérN)ÚencodeÚlenÚstructÚpackÚformatr rrrrrrrr)r"r$ÚencodedZpaddedrr r r Ú _init_value8s "zMmapedDict._init_valuec cs¸d}|j}|j}tj}xœ||kr²t||ƒd}|||krLd}t||jƒ‚|d7}|d| ¡||ƒd}|d|dd}||7}t||ƒd} |  d¡| |fV|d7}qWdS)z1Yield (key, value, pos). No locking is performed.rrz0Read beyond file size detected, %s is corrupted.r z%sszutf-8N) r rr+Ú unpack_fromrÚ RuntimeErrorrr)Ú_unpack_doubleÚdecode) r"rZusedrr0Z encoded_lenÚmsgr.Z padded_lenrr r r r!Is   zMmapedDict._read_all_valuesccs&x | ¡D]\}}}||fVq WdS)z1Yield (key, value, pos). No locking is performed.N)r!)r"ÚkÚvr%r r r Úread_all_valuesbszMmapedDict.read_all_valuescCs.||jkr| |¡|j|}t|j|ƒdS)Nr)rr/r2r)r"r$rr r r Ú read_valuegs   zMmapedDict.read_valuecCs0||jkr| |¡|j|}t|j||ƒdS)N)rr/r r)r"r$rrr r r Ú write_valuens   zMmapedDict.write_valuecCs*|jr&|j ¡d|_|j ¡d|_dS)N)rrÚclose)r"r r r r:us   zMmapedDict.closeN)F) Ú__name__Ú __module__Ú __qualname__Ú__doc__r&r/r!r7r8r9r:r r r r rs  rcCs"tt||ƒƒ}tj|||gddS)z&Format a key for use in the mmap file.T)Z sort_keys)ÚdictÚzipÚjsonÚdumps)Z metric_nameÚnameZ labelnamesZ labelvaluesÚlabelsr r r Úmmap_key}srE)rArrr+rZStructr,r rr0rr2r rÚobjectrrEr r r r Ús    c