U dR…fo ã@s(ddlmZmZmZGdd„deƒZdS)é)ÚTApplicationExceptionÚTTypeÚ TMessageTypec@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)ÚTAsyncProcessorcCs||_||_dS©N)Ú_serviceÚ_handler)ÚselfZserviceÚhandler©r úC/tmp/pip-unpacked-wheel-jqs7l_7o/thriftpy2/contrib/aio/processor.pyÚ__init__szTAsyncProcessor.__init__cƒsÊ| ¡IdH\‰}}ˆˆjjkrR| tj¡IdH| ¡IdHˆ|ttjƒdfSt ˆjˆdƒƒ‰|  ˆ¡IdH| ¡IdHt ˆjˆdƒƒ}‡fdd„t ˆj ƒDƒ‰‡‡‡‡fdd„}ˆ|||fS)NÚ_argsZ_resultcsg|]}ˆj|d‘qS)é)Ú thrift_spec©Ú.0Úk©Úargsr r Ú sz.TAsyncProcessor.process_in..c“s(tˆjˆƒ}|‡fdd„ˆDƒŽIdHS)Nc3s|]}ˆj|VqdSr)Ú__dict__rrr r Ú sz;TAsyncProcessor.process_in..call..)Úgetattrr)Úf©ÚapiZapi_argsrr r r Úcalls z(TAsyncProcessor.process_in..call) Zread_message_beginrZthrift_servicesÚskiprZSTRUCTZread_message_endrZUNKNOWN_METHODrZ read_structÚsortedr)r ÚiprotÚtypeÚseqidÚresultrr rr Ú process_in s zTAsyncProcessor.process_incÃs6| |tj|¡| |¡| ¡|j ¡IdHdSr)Úwrite_message_beginrZ EXCEPTIONÚwriteÚwrite_message_endÚtransÚflush)r ÚoprotrÚexcr"r r r Úsend_exception s zTAsyncProcessor.send_exceptioncÃs6| |tj|¡| |¡| ¡|j ¡IdHdSr)r%rZREPLYZ write_structr'r(r))r r*rr#r"r r r Ú send_result&s zTAsyncProcessor.send_resultcCsVt|jƒD]F}|j|ddkr"q |j|\}}}}t||ƒr t|||ƒdSq dS)NrÚsuccessTF)rrÚ isinstanceÚsetattr)r Úer#rÚ_Zexc_nameZexc_clsr r r Úhandle_exception,s  z TAsyncProcessor.handle_exceptionc Ãsœ| |¡IdH\}}}}t|tƒr8| ||||¡IdHSz|ƒIdH|_Wn0tk rz}z| ||¡sj‚W5d}~XYnX|js˜| ||||¡IdHdSr) r$r/rr,r.Ú Exceptionr3Zonewayr-)r r r*rr"r#rr1r r r Úprocess7s  zTAsyncProcessor.processN) Ú__name__Ú __module__Ú __qualname__r r$r,r-r3r5r r r r rs  rN)Zthriftpy2.thriftrrrÚobjectrr r r r Ús