o i/ @sddlZddlZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddl m Z dd l mZdd l mZddlZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(e"e)Z*e+dde&ej,dddieZ-e Z.Gdddej/Z0Gd d!d!e0Z1Gd"d#d#e0Z2e1e._3d$d%Z4d&d'Z5d(d)Z6dS)*N)Iterable)Client) PooledClient) HashClient)MemcacheClientError)MemcacheIllegalInputError)MemcacheServerError)MemcacheUnknownCommandError)MemcacheUnknownError)config)Pin)_SPAN_MEASURED_KEY) SPAN_KIND)SpanKind) SpanTypes)db) memcached)net) COMPONENT) get_logger)schematize_cache_operation)asbool)tracer pymemcachecommand_enabled"DD_TRACE_MEMCACHED_COMMAND_ENABLEDF)defaultcs$eZdZfddZddZZS) _WrapperBasecs`||i|}tt||t|i|}tt}|r$|j|d}nt|d}||dS)N)tags) superr__init___get_address_tagsr get_fromrcloneonto)self wrapped_classargskwargscr parent_pinpin __class__^/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/pymemcache/client.pyr 4s  z_WrapperBase.__init__cOs>t|}|r |s||i|St|||g|Ri|SN)r r"enabled_trace)r%funccmdr'r(pr.r.r/_trace_function_as_commandEs  z'_WrapperBase._trace_function_as_command)__name__ __module__ __qualname__r r6 __classcell__r.r.r,r/r3s rcseZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1ZZS)2 WrappedClientzWrapper providing patched methods of a pymemcache Client. Relevant connection information is obtained during initialization and attached to each span. Keys are tagged in spans for methods that act upon a key. c"tt|jtg|Ri|dSr0)rr;r _Clientr%r'r(r,r.r/r W"zWrappedClient.__init__cO|jdg|Ri|S)Nset _traced_cmdr>r.r.r/rAZzWrappedClient.setcOr@)Nset_manyrBr>r.r.r/rE]rDzWrappedClient.set_manycOr@)NaddrBr>r.r.r/rF`rDzWrappedClient.addcOr@)NreplacerBr>r.r.r/rGcrDzWrappedClient.replacecOr@)NappendrBr>r.r.r/rHfrDzWrappedClient.appendcOr@)NprependrBr>r.r.r/rIirDzWrappedClient.prependcOr@)NcasrBr>r.r.r/rJlrDzWrappedClient.cascOr@NgetrBr>r.r.r/rLorDzWrappedClient.getcOr@)Nget_manyrBr>r.r.r/rMrrDzWrappedClient.get_manycOr@NgetsrBr>r.r.r/rOurDzWrappedClient.getscOr@)N gets_manyrBr>r.r.r/rPxrDzWrappedClient.gets_manycOr@)NdeleterBr>r.r.r/rQ{rDzWrappedClient.deletecOr@)N delete_manyrBr>r.r.r/rR~rDzWrappedClient.delete_manycOr@)NincrrBr>r.r.r/rSrDzWrappedClient.incrcOr@)NdecrrBr>r.r.r/rTrDzWrappedClient.decrcOr@)NtouchrBr>r.r.r/rUrDzWrappedClient.touchcOr@)NstatsrBr>r.r.r/rVrDzWrappedClient.statscOr@)NversionrBr>r.r.r/rWrDzWrappedClient.versioncOr@)N flush_allrBr>r.r.r/rXrDzWrappedClient.flush_allcOr@)NquitrBr>r.r.r/rYrDzWrappedClient.quitcOr@)"set_multi is an alias for set_manyrErBr>r.r.r/ set_multizWrappedClient.set_multicOr@)rZrMrBr>r.r.r/ get_multir\zWrappedClient.get_multics$jfddg|Ri|S)Ncstj|i|Sr0)getattr __wrapped___args_kwargscommandr%r.r/sz+WrappedClient._traced_cmd..)r6)r%rdr'r(r.rcr/rCszWrappedClient._traced_cmd)r7r8r9__doc__r rArErFrGrHrIrJrLrMrOrPrQrRrSrTrUrVrWrXrYr[r]rCr:r.r.r,r/r;Ns4 r;cseZdZdZddZfddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZZS) WrappedHashClientaWrapper that traces HashClient commands This wrapper proxies its command invocations to the underlying HashClient instance. When the use_pooling setting is in use, this wrapper starts a span before doing the proxy call. This is necessary because the use_pooling setting causes Client instances to be created and destroyed dynamically in a manner that isn't affected by the patch() function. cOsP||}t|tr|j||||g|Ri|St|j||g|Ri|S)aJ PooledClient creates Client instances dynamically on request, which means those Client instances aren't affected by the wrappers applied in patch(). We handle this case here by calling trace() before running the command, specifically when the client that will be used for the command is a PooledClient. To avoid double-tracing when the key's client is not a PooledClient, we don't create a span and instead rely on patch(). In this case the underlying Client instance is long-lived and has been patched already. ) _get_client isinstancerrCr^r_)r%r4key default_valr'r(client_for_keyr.r.r/_ensure_traceds z WrappedHashClient._ensure_tracedcr<r0)rrgr _HashClientr>r,r.r/r r?zWrappedHashClient.__init__cO|jd|dg|Ri|S)NrAFrmr%rjr'r(r.r.r/rAzWrappedHashClient.setcOro)NrFFrprqr.r.r/rFrrzWrappedHashClient.addcOro)NrGFrprqr.r.r/rGrrzWrappedHashClient.replacecOro)NrHFrprqr.r.r/rHrrzWrappedHashClient.appendcOro)NrIFrprqr.r.r/rIrrzWrappedHashClient.prependcOro)NrJFrprqr.r.r/rJrrzWrappedHashClient.cascO|jd|dg|Ri|SrKrprqr.r.r/rLrrzWrappedHashClient.getcOrsrNrprqr.r.r/rOrrzWrappedHashClient.getscOro)NrQFrprqr.r.r/rQrrzWrappedHashClient.deletecOro)NrSFrprqr.r.r/rSrrzWrappedHashClient.incrcOro)NrTFrprqr.r.r/rTrrzWrappedHashClient.decrcOro)NrUFrprqr.r.r/rUrrzWrappedHashClient.touchcsHdurSt|}|d|jfddg|Ri|S)Nrcs"jtg|Ri|Sr0)_safely_run_funcr^r`clientrdrkr%r.r/res z/WrappedHashClient._traced_cmd..)listinsertr6)r%rdrvrjrkr'r(r.rur/rCs zWrappedHashClient._traced_cmd)r7r8r9rfrmr rArFrGrHrIrJrLrOrQrSrTrUrCr:r.r.r,r/rgs" rgcOsbi}zt|r|d\}}||tj<||tj<||tj<W|SW|Sty0tdY|Sw)zDAttempt to get host and port from args passed to Client initializer.rz$Error collecting client address tags)lenr TARGET_HOST TARGET_PORTSERVER_ADDRESS Exceptionlogdebug)r'r(rhostportr.r.r/r!s     r!cCsd}|s|S|d}t|turt|}t|tur|}|St|tur*|}|St|turRt|rRt|dturCd|}|St|dturRd|}|S)zReturn the query values given the arguments to a pymemcache command. If there are multiple query values, they are joined together space-separated. r  )typedictrwstrbytesdecoderyjoin)r'keysargr.r.r/_get_query_string s&    rc Ostjttjdd|j|tjd}|t t j j |t jtj|ttj|tdz"||jt j jrOt|}d||rDdnd|}|tj|Wnty^tdYnwzR||i|}|d kso|d kr|t j|rt|t rt!d d |Dnd n!|dkr|t j|rdnd n|dkr|t j|d rdnd |WWdSt"t#t$t%t&fyt'(\} } } |)| | | w1swYdS)zRun and trace the given command. Any pymemcache exception is caught and span error information is set. The exception is then reraised for the application to handle appropriately. Relevant tags are set in the span. r)cache_provider)serviceresource span_typez{}{}{}rrz&Error setting relevant pymemcache tagsrMrPcss|]}|rdVqdS)rNr.).0docr.r.r/ Wsz_trace..rrLrON)*rtracer memcachedxCMDrrCACHE _set_tag_strrr rintegration_namerSYSTEM DBMS_NAMErrCLIENT set_metricr set_tagsrrrformatQUERYr}r~rROWCOUNTrirsumrrr r rsysexc_info set_exc_info) r3r5 method_namer'r(spanvalsqueryresulttypvaltbr.r.r/r2-sZ    &* r2)7osrtypingrrpymemcache.client.baserrpymemcache.client.hashrpymemcache.exceptionsrrrr r wraptddtracer ddtrace._trace.pinr ddtrace.constantsr r ddtrace.extrrrrrrddtrace.internal.constantsrddtrace.internal.loggerrddtrace.internal.schemarddtrace.internal.utils.formatsr ddtrace.tracerr7r~_addgetenvr=rn ObjectProxyrr;rg client_classr!rr2r.r.r.r/sT                        VW