h78j"UdZddlZddlZddlZddlmZddlZddlmZm Z m Z m Z ddl m Z ddlmZejeZGddZdaedzed <ejZd efd ZdS) zHDHT service for distributed cache - runs in background thread with Trio.N)Any) Libp2pClient ResultCachecompute_query_hashnormalize_query) DEFAULT_TTL)DEFAULT_MAX_HOP_CONNECTIONSc `eZdZdZdeeddedededdfd Zdd Zdd e de fd Z ddZ de de de fdZddededeeee fdzfdZddedeeee fdeddfdZdeee ffdZdeefdZede fdZedefdZedefdZdS) DhtServicezBackground service running Trio event loop with Libp2pClient and ResultCache. Provides thread-safe access to distributed cache DHT from FastAPI. r) listen_port cache_ttlmax_hop_connectionsr r rreturnNc||_||_||_d|_d|_d|_d|_tj|_ tj|_ d|_ dS)NF) r r r_dht_cache_thread_started threadingEvent _start_event _stop_event _trio_token)selfr r rs c/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/ddgs/api_server/dht_service.py__init__zDhtService.__init__se'"#6 )- *. 04  %O--$?,,;?cvdfd } tj|n2#t$r%}td|Yd}~nd}~wwxYwjjdS#jjwxYw)z)Run Trio event loop in background thread.rNc,Ktj_tj_t jdj_ t_ j d{Vtdj jj jjj5tjd{VddddS#1swxYwYdS)NT)r bootstraprz.Network service started: connected=%s, port=%d)triolowlevelcurrent_trio_tokenr CancelScope _cancel_scoperr rrrrastartloggerinfo is_runningportrset sleep_foreverrsr_mainz#DhtService._run_trio.._main-sU#}??AAD !%!1!3!3D $ ,$($<DI &--DK)""$$ $ $ $ $ $ $ $ KK@ $       ! ! # # ## + +(********* + + + + + + + + + + + + + + + + + +s"D  D D zNetwork service loop exited: %rrN)r!run Exceptionr'warningrr+r)rr.exs` r _run_triozDhtService._run_trio*s + + + + + +, # HUOOOO B B B NN._wrappedys$$ ) )!T4[(((((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s 044) trio_token)rrrr! from_threadr0)rrCrDrIs `` r _run_in_triozDhtService._run_in_triotsf}  0 84 ) ) ) ) ) ) ) )##H9I#JJJrtextquerycategoryc^|jr|j|jdSt||}t |}|j|}|rt d||S|jjr | |jj |}|r@t d||j |||||j |S|j |n3#t$r&}t d||Yd}~nd}~wwxYwdS)zGet cached results synchronously (for use from FastAPI). Args: query: The search query. category: The search category. Returns: Cached results or None. NzCache hit (local): %szCache hit (network): %szDHT get failed for %s: %r)rrrrrgetr'debugr)rLagetr+r add_negativer1)rrNrO query_hash normalized local_results dht_resultsr3s r get_cachedzDhtService.get_cachedsM}  1T[5H4'x88 $U++   33  ! LL0* = = = 9  J J;?;L;LTY^]g;h;h 'LL!:JGGGKOOJ HkSWSabbb&& ((4444 J J J 8*bIIIIIIII Jts=A!C:C:: D*D%%D*resultsc|jr|j|jdSt||}t |}|j|||||j|jjry ||jj |||jt d|dS#t$r'}t d||Yd}~dSd}~wwxYwdS)zCache search results synchronously (for use from FastAPI). Args: query: The search query. results: The search results. category: The search category. NzCached to network: %szDHT set failed for %s: %r) rrrrrr+r r)rLasetr'rRr1)rrNrZrOrUrVr3s rcachezDhtService.caches}  1T[5H F'x88 $U++   J'4>RRR 9  J J!!$).*gt~VVV 4jAAAAA J J J 8*bIIIIIIIII J  J Js'AB++ C5CCc|jr|j ddddddidSd|jj|jj|jj|jj|jr|jnd|jr|jndt|j |jj |jj |jj |jjtj|jjdz dd S) z]Get DHT service status. Returns: Dict with status information. NFr)running connectedpeer_idr* cache_size cache_countmetricsT started_at)query_success_rateaverage_query_latency_mskbucket_distributionuptime_seconds) r_r`rar* listen_addrsrbrc peer_countrouting_table_sizerd)rrr)rar*rjr size_bytescountlen find_peersrlrfrgrhtimerdr-s r get_statuszDhtService.get_statuss}  1 "  -y(IN I26:kH$+00222q26+D4;,,...1di224455"&)">&*i&B,0I,N(, (F"&)++ 0A,0O"O    rc|jr|jgS ||jj}|pgS#t$r'}t d|gcYd}~Sd}~wwxYw)zVGet list of connected peers. Returns: List of peer IDs. NzFailed to get peers: %r)rrrL afind_peersr1r'rR)rpeersr3s r get_peerszDhtService.get_peerss}  1I #001FGGE ;B      LL2B 7 7 7IIIIII s5 A&A!A&!A&c:|jo|jduo |jjS)z Check if DHT service is running.N)rrr)r-s rr)zDhtService.is_runnings"}O$!6O49;OOrcF|jr|jndS)zGet number of cached entries.r)rrnr-s rrczDhtService.cache_counts$'+k8t{  """q8rcF|jr|jndS)zGet cache size in bytes.r)rrmr-s rrbzDhtService.cache_sizes$,0;=t{%%'''A=rr/)r5)rM)__name__ __module__ __qualname____doc__rr intrr4floatboolr<rArrLstrlistdictrYr]rrrvpropertyr)rcrbrrr r sD$#> @@@@ @ ! @  @@@@&####BUd,& K KS KS K K K K"""s"T#s(^@TW[@[""""HJJ3Jd38n)=JJZ^JJJJ0" DcN" " " " H49$PDPPPXP9S999X9>C>>>X>>>rr _dht_servicerct5t'tatdddn #1swxYwYtS)zVGet the global DHT service instance. Returns: DhtService singleton. N)_dht_service_lockrr r<rrrget_dht_servicers !!  %<rsNNN  SSSSSSSSSSSS&&&&&&>>>>>>  8 $ $m>m>m>m>m>m>m>m>`#' j4&&&"IN$$       r