
    _j                         d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlm	Z	m
Z
 d dlmZ d dlmZmZ d dlmZ d dlmZ dZd	Z e            d
eddfd            ZdS )    N)	ClientAPI)SealLogRequestMigrateLogRequest)LogServiceStub)resetskip_if_not_cluster)
invariants)wait_for_version_increasei d   clientreturnc           	         t          j                     }t          j        |           t          d|           t	          |            |                     ddddd          }t          j        d           t          d|j                   d}t          d	t          t                    D ]}g }g }|                    d
 t          ||t          z             D                        |                    d t          ||t          z             D                        	 |                    ||           # t          $ r5}t          d|            dt          |          v rd}Y d }~ nY d }~d }~ww xY w|s
J d            d S )NzGenerating data with seed test   )zhnsw:construction_efzhnsw:search_efzhnsw:M)namemetadata   zbackpressuring forFr   c                 ,    g | ]}t          |          S  )str.0xs     l/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/test/distributed/test_log_backpressure.py
<listcomp>z)test_log_backpressure.<locals>.<listcomp>-   s    ===qCFF===    c                 Z    g | ](}t           j                            d d          d         )S )r      r   )nprandomrandr   s     r   r   z)test_log_backpressure.<locals>.<listcomp>.   s-    UUUq29>>!Q//2UUUr   )ids
embeddingszCaught exception:
zYlog needs compaction before accepting more writes; please backoff exponentially and retryTz#Expected an exception to be thrown.)timer    seedprintr   create_collectionsleepidrangeRECORDS
BATCH_SIZEextendadd	Exceptionr   )r   r%   
collectionexceptedir"   r#   r   s           r   test_log_backpressurer3      s    9;;D
K	
&---	&MMM))*-PSTT *  J
 	JqMMM	

...H1gz**  


==E!Q^$<$<===>>>UUE!Q^<T<TUUUVVV	NNszN:::: 	 	 	+++,,,jnqrsntnttt utttt	
 ::::::::s   D11
E0;%E++E0)grpcmathr    r$   numpyr   chromadb.apir   chromadb.proto.logservice_pb2r   r   "chromadb.proto.logservice_pb2_grpcr   chromadb.test.conftestr   r   chromadb.test.propertyr	   -chromadb.test.utils.wait_for_version_increaser
   r+   r,   r3   r   r   r   <module>r=      s           " " " " " " K K K K K K K K = = = = = =        . - - - - - S S S S S S

;;	; ; ; ; ; ;r   