:j7ddZddlZddlZddlmZmZmZddlmZ edZ Gddee Z y)aF Generic Load Balancer This module provides a generic load balancer for distributing requests across multiple resources (sessions, tokens, user-agents, etc.). Key Features: - Round-robin and random load balancing strategies - Thread-safe resource allocation - Usage statistics tracking - Simple and efficient implementation N)GenericListTypeVar)LoadBalanceStrategyTceZdZdZej fdeedefdZdefdZ defdZ dd Z deeddfd Z de fd Zdefd Zdefd ZdefdZdefdZdefdZdefdZy)Balancerz/Generic load balancer for resource distributionitemsstrategyc|s td|j|_||_d|_t j |_tjtt|jd|_ d|_ y)zInitialize load balancer Args: items: List of items to balance across strategy: Load balancing strategy to use Items list cannot be emptyrN) ValueErrorcopyr r index threadingLocklockdictfromkeysrangelen usage_counttotal_requests)selfr r s 5/root/.openclaw/workspace/harvester/tools/balancer.py__init__zBalancer.__init__sf9: :#jjl    NN$  ==s4::)?Creturnc|j5|xjdz c_|jtjk(r7|j }|j dzt |jz|_n,tjdt |jdz }|j|xxdz cc<|j|cdddS#1swYyxYw)zvGet next item according to load balancing strategy Returns: Any: Next item from the pool rN) rrr Strategy ROUND_ROBINrrr randomrandintr)r item_indexs rgetz Balancer.get/s YY *   1 $ }} 4 44!ZZ "jj1nDJJ? #^^As4::/BC   Z (A - (::j) * * *s B;CCc"|jS)zjAlias for get() method for convenience Returns: Any: Next item from the pool )r&rs rnextz Balancer.nextAs xxzrNc|j5d|_tjt t |j d|_d|_dddy#1swYyxYw)z(Reset load balancer state and statisticsrN) rrrrrrr rrr(s rresetzBalancer.resetIsM YY $DJ#}}U3tzz?-CQGD "#D  $ $ $s AAA%c|s td|j5|j|_d|_t j tt|jd|_ dddy#1swYyxYw)z`Update items list Args: items: New list of items to balance across r rN) rrrr rrrrrr)rr s r update_itemszBalancer.update_itemsPsg 9: : YY HDJDJ#}}U3tzz?-CQGD  H H Hs AA77Bc|j5|jdk(r7dt|j|jj idcdddSi}|j jD],\}}||jz dz}|t|dd|d|<.|jt|j|jj |dcdddS#1swYyxYw)zsGet usage statistics Returns: dict: Usage statistics including counts and percentages r)r items_countr usage_distributionNd)count percentageitem_)rrrr r valuerround)r distributionir3r4s r get_statszBalancer.get_stats^s YY ""a'&'#&tzz? $ 3 3*,   L ,,224 a5#d&9&99S@ 6;5Q[]^K_,` uQC[) a #'"5"5"4:: MM//&2    s=C BC  C)c|j5|jtjk(r"|j|j cdddSt j|jcdddS#1swYyxYw)zGet current item without advancing the index Returns: T: Current item (for round-robin) or random item N)rr r!r"r rr#choicer(s rget_current_itemzBalancer.get_current_itemys^ YY 1}} 4 44zz$**- 1 1}}TZZ0  1 1 1s6A5 A55A>c,t|jSz[Get number of items in the pool Returns: int: Number of items rr r(s rsizez Balancer.size 4::rc2t|jdk(S)z_Check if items pool is empty Returns: bool: True if pool is empty rr@r(s ris_emptyzBalancer.is_emptys 4::!##rc,t|jSr?r@r(s r__len__zBalancer.__len__rBrc|dt|jd|jjd|jdS)zgString representation of the balancer Returns: str: String representation Balancer(items= , strategy=z , requests=))rr r r6rr(s r__str__zBalancer.__str__s= !TZZ 1T]]=P=PrXs8 ))6 CLR]wqzR]r