&GRP@sdZddlZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZmZmZmZddlmZmZddlmZdd lmZmZdd lmZdd lmZGd d d eZdS)z3 Module yoton.context Defines the context class. N)core) connection) basestringbytesstrxrange split_address)Property getErrorMsgUID PackageQueue)Package BUF_MAX_LEN) SLOT_CONTEXT)ConnectionCollection Connection) TcpConnection) ItcConnectionc@s eZdZdZddddZddZdd Zed d Zed d Z eddZ eddZ ddddZ ddddZ dddZdddZdddZd d!Zd"d#Zd$d%Zd&d'ZdS)(Contexta Context(verbose=0, queue_params=None) A context represents a node in the network. It can connect to multiple other contexts (using a yoton.Connection. These other contexts can be in another process on the same machine, or on another machine connected via a network or the internet. This class represents a context that can be used by channel instances to communicate to other channels in the network. (Thus the name.) The context is the entity that queue routes the packages produced by the channels to the other context in the network, where the packages are distributed to the right channels. A context queues packages while it is not connected to any other context. If messages are send on a channel registered at this context while the context is not connected, the messages are stored by the context and will be send to the first connecting context. Example 1 --------- # Create context and bind to a port on localhost context = yoton.Context() context.bind('localhost:11111') # Create a channel and send a message pub = yoton.PubChannel(context, 'test') pub.send('Hello world!') Example 2 --------- # Create context and connect to the port on localhost context = yoton.Context() context.connect('localhost:11111') # Create a channel and receive a message sub = yoton.SubChannel(context, 'test') print(sub.recv() # Will print 'Hello world!' Queue params ------------ The queue_params parameter allows one to specify the package queues used in the system. It is recommended to use the same parameters for every context in the network. The value of queue_params should be a 2-element tuple specifying queue size and discard mode. The latter can be 'old' (default) or 'new', meaning that if the queue is full, either the oldest or newest messages are discarted. rNcCs||_|dkr$tdf}nt|toBt|dksTtdn||_tj|_ i|_ i|_ g|_ t j|_t||_d|_d|_i|_dS)Noldz)queue_params should be a 2-element tuple.r)_verboser isinstancetuplelen ValueError _queue_paramsr get_int_id_sending_channels_receiving_channels _connections threadingRLock_connections_lockr _startupQueue _send_seq _recv_seq _source_map)selfverboseZ queue_paramsr+H/Applications/pyzo2014a/lib/python3.4/site-packages/iep/yoton/context.py__init__Os  !      zContext.__init__cCs/x|jD]}|jdq W|jdS)a close() Close the context in a nice way, by closing all connections and all channels. Closing a connection means disconnecting two contexts. Closing a channel means disasociating a channel from its context. Unlike connections and channels, a Context instance can be reused after closing (although this might not always the best strategy). zClosed by the context.N)connections_allcloseclose_channels)r)cr+r+r,r/psz Context.closecCsadd|jjD}dd|jjD}x"t||D]}|jqIWdS)z close_channels() Close all channels associated with this context. This does not close the connections. See also close(). cSsg|] }|qSr+r+).0r1r+r+r, s z*Context.close_channels..cSsg|] }|qSr+r+)r2r1r+r+r,r3s N)rvaluesr setr/)r)Z channels1Z channels2r1r+r+r,r0s zContext.close_channelsc Cs:|jjzdd|jDSWd|jjXdS)a6 Get a list of all Connection instances currently associated with this context, including pending connections (connections waiting for another end to connect). In addition to normal list indexing, the connections objects can be queried from this list using their name. cSsg|]}|jr|qSr+)is_alive)r2r1r+r+r,r3s z+Context.connections_all..N)r$acquirer!release)r)r+r+r,r.s zContext.connections_allc Cs|jjz~t}g}xC|jD]8}|jsH|j|q)|jr)|j|q)q)Wx|D]}|jj|qlW|SWd|jjXdS)z Get a list of the Connection instances currently active for this context. In addition to normal list indexing, the connections objects can be queried from this list using their name. N) r$r7rr!r6append is_connectedremover8)r)copyZ to_remover1r+r+r, connectionss     zContext.connectionscCs t|jS)z Get the number of connected contexts. Can be used as a boolean to check if the context is connected to any other context. )rr=)r)r+r+r,connection_countszContext.connection_countcCs|jS)z) The 8-byte UID of this context. )r)r)r+r+r,idsz Context.idc Cs|jt|\}}}t||}|j||||jjz@x)t|jry|j|jj qQW|j j |Wd|jj X|S)a bind(address, max_tries=1, name='') Setup a connection with another Context, by being the host. This method starts a thread that waits for incoming connections. Error messages are printed when an attemped connect fails. the thread keeps trying until a successful connection is made, or until the connection is closed. Returns a Connection instance that represents the connection to the other context. These connection objects can also be obtained via the Context.connections property. Parameters ---------- address : str Should be of the shape hostname:port. The port should be an integer number between 1024 and 2**16. If port does not represent a number, a valid port number is created using a hash function. max_tries : int The number of ports to try; starting from the given port, subsequent ports are tried until a free port is available. The final port can be obtained using the 'port' property of the returned Connection instance. name : string The name for the created Connection instance. It can be used as a key in the connections property. Notes on hostname ----------------- The hostname can be: * The IP address, or the string hostname of this computer. * 'localhost': the connections is only visible from this computer. Also some low level networking layers are bypassed, which results in a faster connection. The other context should also connect to 'localhost'. * 'publichost': the connection is visible by other computers on the same network. Optionally an integer index can be appended if the machine has multiple IP addresses (see socket.gethostbyname_ex). N) r=rr_bindr$r7rr%_inject_packagepopr!r9r8)r)address max_triesnameprotocolhostnameportrr+r+r,binds, z Context.bindg?c Cs|jt|\}}}t||}|j||||jjz:x#|jrs|j|jjqQW|j j |Wd|jj Xdj d}t |t|jdddd} |j| |S)a connect(self, address, timeout=1.0, name='') Setup a connection with another context, by connection to a hosting context. An error is raised when the connection could not be made. Returns a Connection instance that represents the connection to the other context. These connection objects can also be obtained via the Context.connections property. Parameters ---------- address : str Should be of the shape hostname:port. The port should be an integer number between 1024 and 2**16. If port does not represent a number, a valid port number is created using a hash function. max_tries : int The number of ports to try; starting from the given port, subsequent ports are tried until a free port is available. The final port can be obtained using the 'port' property of the returned Connection instance. name : string The name for the created Connection instance. It can be used as a key in the connections property. Notes on hostname ----------------- The hostname can be: * The IP address, or the string hostname of this computer. * 'localhost': the connection is only visible from this computer. Also some low level networking layers are bypassed, which results in a faster connection. The other context should also host as 'localhost'. * 'publichost': the connection is visible by other computers on the same network. Optionally an integer index can be appended if the machine has multiple IP addresses (see socket.gethostbyname_ex). NNEW_CONNECTIONzutf-8r)r=rr_connectr$r7r%rCrDr!r9r8encoder rr _send_package) r)rEtimeoutrGrHrIrJrbbpr+r+r,connect$s*  ! zContext.connectg@cCs%x|jD]}|j|q WdS)a> flush(timeout=5.0) Wait until all pending messages are send. This will flush all messages posted from the calling thread. However, it is not guaranteed that no new messages are posted from another thread. Raises an error when the flushing times out. T)r=flush)r)rPr1r+r+r,rTps z Context.flushcCs9||jkr(tdt|n||j|.N)r rkeysrD)r)rUDrZr+r+r,_unregister_channels zContext._unregister_channelc Cs|jd7_|j|_|jjzVd}x0|jD]%}|jr;|j|d}q;q;W|s}|jj|nWd|jj XdS)a  _send_package(package) Used by the channels to send a package into the network. This method routes the package to all currentlt connected connections. If there are none, the packages is queued at the context. r@FTN) r& _source_seqr$r7r!r6rOr%pushr8)r)packageokr1r+r+r,rOs      zContext._send_packagec Cs|j}d}d}|jj|jd}||jkr|j|j|j<|jdkr|jd7_|j|_d\}}q|j|jkr|jd7_|j|_d}qd}n|r/|jj zAx:|j D]/}||ks|j r qn|j |qWWd|jj Xn|r|tkrQ|j|q|jj|d}|dk r|j|qndS)a( _recv_package(package, connection) Used by the connections to receive a package at this context. The package is distributed to all connections except the calling one. The package is also distributed to the right channel (if applicable). Frr@TN)TT)_slotr(get _source_idr^_dest_idr'rr$r7r!r6rOr8r_recv_context_packager _recv_package) r)r`rrVZ send_furtherZ deposit_hereZlast_seqr1rUr+r+r,rgs:        zContext._recv_packagec Cs|jjd}|dkr|jjzIxB|jD]7}|jr8|j|jkr8|jt j dq8q8WWd|jj Xnb|dkrxS|j j D]1}t|drt|dr|jqqWntd|dS) z _recv_context_package(package) Process a package addressed at the context itself. This is how the context handles higher-level connection tasks. zutf-8ZCLOSE_CONNECTIONFNrLZ_current_message send_lastz)Yoton: Received unknown context message: )_datadecoder$r7r=r:id2rdr/rSTOP_CLOSED_FROM_THEREr8rr4hasattrrhprint)r)r`messager1rUr+r+r,rfs    zContext._recv_context_package)__name__ __module__ __qualname____doc__r-r/r0propertyr.r=r>r?rKrSrTrXrYr]rOrgrfr+r+r+r,rs" 0!    IL   >r)rsossystimer"yotonrr yoton.miscrrrrrr r r r yoton.corer rryoton.connectionrryoton.connection_tcprZyoton.connection_itcrobjectrr+r+r+r, s$ ("