63 lines
2.3 KiB
Cython
63 lines
2.3 KiB
Cython
"""0MQ Message related class declarations."""
|
|
|
|
#
|
|
# Copyright (c) 2010-2011 Brian E. Granger & Min Ragan-Kelley
|
|
#
|
|
# This file is part of pyzmq.
|
|
#
|
|
# pyzmq is free software; you can redistribute it and/or modify it under
|
|
# the terms of the Lesser GNU General Public License as published by
|
|
# the Free Software Foundation; either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# pyzmq is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# Lesser GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the Lesser GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
#-----------------------------------------------------------------------------
|
|
# Imports
|
|
#-----------------------------------------------------------------------------
|
|
|
|
from cpython cimport PyBytes_FromStringAndSize
|
|
|
|
from zmq.backend.cython.libzmq cimport zmq_msg_t, zmq_msg_data, zmq_msg_size
|
|
|
|
#-----------------------------------------------------------------------------
|
|
# Code
|
|
#-----------------------------------------------------------------------------
|
|
|
|
cdef class MessageTracker(object):
|
|
|
|
cdef set events # Message Event objects to track.
|
|
cdef set peers # Other Message or MessageTracker objects.
|
|
|
|
|
|
cdef class Frame:
|
|
|
|
cdef zmq_msg_t zmq_msg
|
|
cdef object _data # The actual message data as a Python object.
|
|
cdef object _buffer # A Python Buffer/View of the message contents
|
|
cdef object _bytes # A bytes/str copy of the message.
|
|
cdef bint _failed_init # Flag to handle failed zmq_msg_init
|
|
cdef public object tracker_event # Event for use with zmq_free_fn.
|
|
cdef public object tracker # MessageTracker object.
|
|
cdef public bint more # whether RCVMORE was set
|
|
|
|
cdef Frame fast_copy(self) # Create shallow copy of Message object.
|
|
cdef object _getbuffer(self) # Construct self._buffer.
|
|
|
|
|
|
cdef inline object copy_zmq_msg_bytes(zmq_msg_t *zmq_msg):
|
|
""" Copy the data from a zmq_msg_t """
|
|
cdef char *data_c = NULL
|
|
cdef Py_ssize_t data_len_c
|
|
data_c = <char *>zmq_msg_data(zmq_msg)
|
|
data_len_c = zmq_msg_size(zmq_msg)
|
|
return PyBytes_FromStringAndSize(data_c, data_len_c)
|
|
|
|
|