Uploaded Test files
This commit is contained in:
parent
f584ad9d97
commit
2e81cb7d99
16627 changed files with 2065359 additions and 102444 deletions
279
venv/Lib/site-packages/zmq/backend/cffi/socket.py
Normal file
279
venv/Lib/site-packages/zmq/backend/cffi/socket.py
Normal file
|
|
@ -0,0 +1,279 @@
|
|||
# coding: utf-8
|
||||
"""zmq Socket class"""
|
||||
|
||||
# Copyright (C) PyZMQ Developers
|
||||
# Distributed under the terms of the Modified BSD License.
|
||||
|
||||
import errno as errno_mod
|
||||
|
||||
from ._cffi import (C, ffi, new_uint64_pointer, new_int64_pointer,
|
||||
new_int_pointer, new_binary_data, value_uint64_pointer,
|
||||
value_int64_pointer, value_int_pointer, value_binary_data,
|
||||
IPC_PATH_MAX_LEN)
|
||||
|
||||
from .message import Frame
|
||||
from .constants import RCVMORE
|
||||
from .utils import _retry_sys_call
|
||||
|
||||
import zmq
|
||||
from zmq.error import ZMQError, _check_rc, _check_version
|
||||
from zmq.utils.strtypes import unicode
|
||||
|
||||
|
||||
def new_pointer_from_opt(option, length=0):
|
||||
from zmq.sugar.constants import (
|
||||
int64_sockopts, bytes_sockopts,
|
||||
)
|
||||
if option in int64_sockopts:
|
||||
return new_int64_pointer()
|
||||
elif option in bytes_sockopts:
|
||||
return new_binary_data(length)
|
||||
else:
|
||||
# default
|
||||
return new_int_pointer()
|
||||
|
||||
def value_from_opt_pointer(option, opt_pointer, length=0):
|
||||
from zmq.sugar.constants import (
|
||||
int64_sockopts, bytes_sockopts,
|
||||
)
|
||||
if option in int64_sockopts:
|
||||
return int(opt_pointer[0])
|
||||
elif option in bytes_sockopts:
|
||||
return ffi.buffer(opt_pointer, length)[:]
|
||||
else:
|
||||
return int(opt_pointer[0])
|
||||
|
||||
def initialize_opt_pointer(option, value, length=0):
|
||||
from zmq.sugar.constants import (
|
||||
int64_sockopts, bytes_sockopts,
|
||||
)
|
||||
if option in int64_sockopts:
|
||||
return value_int64_pointer(value)
|
||||
elif option in bytes_sockopts:
|
||||
return value_binary_data(value, length)
|
||||
else:
|
||||
return value_int_pointer(value)
|
||||
|
||||
|
||||
class Socket(object):
|
||||
context = None
|
||||
socket_type = None
|
||||
_zmq_socket = None
|
||||
_closed = None
|
||||
_ref = None
|
||||
_shadow = False
|
||||
copy_threshold = 0
|
||||
|
||||
def __init__(self, context=None, socket_type=None, shadow=None):
|
||||
self.context = context
|
||||
if shadow is not None:
|
||||
if isinstance(shadow, Socket):
|
||||
shadow = shadow.underlying
|
||||
self._zmq_socket = ffi.cast("void *", shadow)
|
||||
self._shadow = True
|
||||
else:
|
||||
self._shadow = False
|
||||
self._zmq_socket = C.zmq_socket(context._zmq_ctx, socket_type)
|
||||
if self._zmq_socket == ffi.NULL:
|
||||
raise ZMQError()
|
||||
self._closed = False
|
||||
|
||||
@property
|
||||
def underlying(self):
|
||||
"""The address of the underlying libzmq socket"""
|
||||
return int(ffi.cast('size_t', self._zmq_socket))
|
||||
|
||||
def _check_closed_deep(self):
|
||||
"""thorough check of whether the socket has been closed,
|
||||
even if by another entity (e.g. ctx.destroy).
|
||||
|
||||
Only used by the `closed` property.
|
||||
|
||||
returns True if closed, False otherwise
|
||||
"""
|
||||
if self._closed:
|
||||
return True
|
||||
try:
|
||||
self.get(zmq.TYPE)
|
||||
except ZMQError as e:
|
||||
if e.errno == zmq.ENOTSOCK:
|
||||
self._closed = True
|
||||
return True
|
||||
else:
|
||||
raise
|
||||
return False
|
||||
|
||||
@property
|
||||
def closed(self):
|
||||
return self._check_closed_deep()
|
||||
|
||||
def close(self, linger=None):
|
||||
rc = 0
|
||||
if not self._closed and hasattr(self, '_zmq_socket'):
|
||||
if self._zmq_socket is not None:
|
||||
if linger is not None:
|
||||
self.set(zmq.LINGER, linger)
|
||||
rc = C.zmq_close(self._zmq_socket)
|
||||
self._closed = True
|
||||
if rc < 0:
|
||||
_check_rc(rc)
|
||||
|
||||
def bind(self, address):
|
||||
if isinstance(address, unicode):
|
||||
address = address.encode('utf8')
|
||||
rc = C.zmq_bind(self._zmq_socket, address)
|
||||
if rc < 0:
|
||||
if IPC_PATH_MAX_LEN and C.zmq_errno() == errno_mod.ENAMETOOLONG:
|
||||
# py3compat: address is bytes, but msg wants str
|
||||
if str is unicode:
|
||||
address = address.decode('utf-8', 'replace')
|
||||
path = address.split('://', 1)[-1]
|
||||
msg = ('ipc path "{0}" is longer than {1} '
|
||||
'characters (sizeof(sockaddr_un.sun_path)).'
|
||||
.format(path, IPC_PATH_MAX_LEN))
|
||||
raise ZMQError(C.zmq_errno(), msg=msg)
|
||||
elif C.zmq_errno() == errno_mod.ENOENT:
|
||||
# py3compat: address is bytes, but msg wants str
|
||||
if str is unicode:
|
||||
address = address.decode('utf-8', 'replace')
|
||||
path = address.split('://', 1)[-1]
|
||||
msg = ('No such file or directory for ipc path "{0}".'.format(
|
||||
path))
|
||||
raise ZMQError(C.zmq_errno(), msg=msg)
|
||||
else:
|
||||
_check_rc(rc)
|
||||
|
||||
def unbind(self, address):
|
||||
_check_version((3,2), "unbind")
|
||||
if isinstance(address, unicode):
|
||||
address = address.encode('utf8')
|
||||
rc = C.zmq_unbind(self._zmq_socket, address)
|
||||
_check_rc(rc)
|
||||
|
||||
def connect(self, address):
|
||||
if isinstance(address, unicode):
|
||||
address = address.encode('utf8')
|
||||
rc = C.zmq_connect(self._zmq_socket, address)
|
||||
_check_rc(rc)
|
||||
|
||||
def disconnect(self, address):
|
||||
_check_version((3,2), "disconnect")
|
||||
if isinstance(address, unicode):
|
||||
address = address.encode('utf8')
|
||||
rc = C.zmq_disconnect(self._zmq_socket, address)
|
||||
_check_rc(rc)
|
||||
|
||||
def set(self, option, value):
|
||||
length = None
|
||||
if isinstance(value, unicode):
|
||||
raise TypeError("unicode not allowed, use bytes")
|
||||
|
||||
if isinstance(value, bytes):
|
||||
if option not in zmq.constants.bytes_sockopts:
|
||||
raise TypeError("not a bytes sockopt: %s" % option)
|
||||
length = len(value)
|
||||
|
||||
c_data = initialize_opt_pointer(option, value, length)
|
||||
|
||||
c_value_pointer = c_data[0]
|
||||
c_sizet = c_data[1]
|
||||
|
||||
_retry_sys_call(C.zmq_setsockopt,
|
||||
self._zmq_socket,
|
||||
option,
|
||||
ffi.cast('void*', c_value_pointer),
|
||||
c_sizet)
|
||||
|
||||
def get(self, option):
|
||||
c_data = new_pointer_from_opt(option, length=255)
|
||||
|
||||
c_value_pointer = c_data[0]
|
||||
c_sizet_pointer = c_data[1]
|
||||
|
||||
_retry_sys_call(C.zmq_getsockopt,
|
||||
self._zmq_socket,
|
||||
option,
|
||||
c_value_pointer,
|
||||
c_sizet_pointer)
|
||||
|
||||
sz = c_sizet_pointer[0]
|
||||
v = value_from_opt_pointer(option, c_value_pointer, sz)
|
||||
if option != zmq.IDENTITY and option in zmq.constants.bytes_sockopts and v.endswith(b'\0'):
|
||||
v = v[:-1]
|
||||
return v
|
||||
|
||||
def send(self, message, flags=0, copy=False, track=False):
|
||||
if isinstance(message, unicode):
|
||||
raise TypeError("Message must be in bytes, not an unicode Object")
|
||||
|
||||
if isinstance(message, Frame):
|
||||
message = message.bytes
|
||||
|
||||
zmq_msg = ffi.new('zmq_msg_t*')
|
||||
if not isinstance(message, bytes):
|
||||
# cast any bufferable data to bytes via memoryview
|
||||
message = memoryview(message).tobytes()
|
||||
|
||||
c_message = ffi.new('char[]', message)
|
||||
rc = C.zmq_msg_init_size(zmq_msg, len(message))
|
||||
_check_rc(rc)
|
||||
C.memcpy(C.zmq_msg_data(zmq_msg), c_message, len(message))
|
||||
_retry_sys_call(C.zmq_msg_send, zmq_msg, self._zmq_socket, flags)
|
||||
rc2 = C.zmq_msg_close(zmq_msg)
|
||||
_check_rc(rc2)
|
||||
|
||||
if track:
|
||||
return zmq.MessageTracker()
|
||||
|
||||
def recv(self, flags=0, copy=True, track=False):
|
||||
zmq_msg = ffi.new('zmq_msg_t*')
|
||||
C.zmq_msg_init(zmq_msg)
|
||||
|
||||
try:
|
||||
_retry_sys_call(C.zmq_msg_recv, zmq_msg, self._zmq_socket, flags)
|
||||
except Exception:
|
||||
C.zmq_msg_close(zmq_msg)
|
||||
raise
|
||||
|
||||
_buffer = ffi.buffer(C.zmq_msg_data(zmq_msg), C.zmq_msg_size(zmq_msg))
|
||||
value = _buffer[:]
|
||||
rc = C.zmq_msg_close(zmq_msg)
|
||||
_check_rc(rc)
|
||||
|
||||
frame = Frame(value, track=track)
|
||||
frame.more = self.getsockopt(RCVMORE)
|
||||
|
||||
if copy:
|
||||
return frame.bytes
|
||||
else:
|
||||
return frame
|
||||
|
||||
def monitor(self, addr, events=-1):
|
||||
"""s.monitor(addr, flags)
|
||||
|
||||
Start publishing socket events on inproc.
|
||||
See libzmq docs for zmq_monitor for details.
|
||||
|
||||
Note: requires libzmq >= 3.2
|
||||
|
||||
Parameters
|
||||
----------
|
||||
addr : str
|
||||
The inproc url used for monitoring. Passing None as
|
||||
the addr will cause an existing socket monitor to be
|
||||
deregistered.
|
||||
events : int [default: zmq.EVENT_ALL]
|
||||
The zmq event bitmask for which events will be sent to the monitor.
|
||||
"""
|
||||
|
||||
_check_version((3,2), "monitor")
|
||||
if events < 0:
|
||||
events = zmq.EVENT_ALL
|
||||
if addr is None:
|
||||
addr = ffi.NULL
|
||||
if isinstance(addr, unicode):
|
||||
addr = addr.encode('utf8')
|
||||
rc = C.zmq_socket_monitor(self._zmq_socket, addr, events)
|
||||
|
||||
|
||||
__all__ = ['Socket', 'IPC_PATH_MAX_LEN']
|
||||
Loading…
Add table
Add a link
Reference in a new issue