B 4\ @sddlmZmZmZddlmZddlZddlZddlm Z ddl m Z ddl m Z mZddlmZmZmZeeZd Zyed d d ZWnek red d ZYnXe eejddZe jdddZe jdddZddZddZdS))print_functiondivisionabsolute_import)ThreadPoolExecutorN)gen)protocol)finalizePY3)get_ipget_ipv6nbytesiz Dask-Offload) max_workersZthread_name_prefix)rcOstj|f||S)N)_offload_executorZsubmit)fnargskwargsr5lib/python3.7/site-packages/distributed/comm/utils.pyoffloadsrmessagec#s6fdd}tr"t|V}n|}t|dS)zI Serialize a message into a list of Distributed protocol frames. c sVyttjdStk rP}ztdt|Wdd}~XYnXdS)N) serializerson_errorcontextzUnserializable Message: %s)listrdumps ExceptionloggerinfoZ exception)e)rmsgrrrr _to_frames&s   zto_frames.._to_framesN)r rrReturn)r!rrrr"resr)rr!rrr to_frames!s   r%Tc#sLtttfdd}r8tkr8t|V}n|}t|dS)z< Unserialize a list of Distributed protocol frames. csLytjdStk rFdkr.d}n}td|YnXdS)N) deserialize deserializersiz[too large to display]z$truncated data stream (%d bytes): %s)rloadsEOFErrorrerror)Zdatastr)r&r'framessizerr _from_frames@sz!from_frames.._from_framesN)summapr FRAME_OFFLOAD_THRESHOLDrrr#)r+r&r'r-r$r)r&r'r+r,r from_frames9s   r1cs`t|js td|ffdd}|tj}|dkrH|tj}|dkrXtd|S)z? Get the bound address of a started Tornado TCPServer. zTCP Server %r not started yet?cs xD]}|j|kr|SqWdS)N)Zfamily)Zfamsock)socketsrr_look_for_family_s  z0get_tcp_server_address.._look_for_familyNz'No Internet socket found on TCPServer??)rZ_socketsvalues RuntimeErrorsocketZAF_INETZAF_INET6Z getsockname)Z tcp_serverr4r2r)r3rget_tcp_server_addressWs   r8cCs$|dkrtS|dkrtS|SdS)zm Ensure the given host string (or IP) denotes a concrete host, not a wildcard listening address. )z0.0.0.0z::N)r r )Zhostrrrensure_concrete_hostps r:)NrN)TN) Z __future__rrrZconcurrent.futuresrZloggingr7Ztornadorr9rZ compatibilityr r Zutilsr r r Z getLogger__name__rr0r TypeErrorZshutdownr coroutiner%r1r8r:rrrrs*