queue_message exposed by api

Author: lsbardel

pq/api.py

@@ -1,4 +1,4 @@
-from .utils.serializers import Message, MessageDict
+from .utils.serializers import Message, MessageDict, queue_message
 from .utils.concurrency import ASYNC_IO, THREAD_IO, CPUBOUND
 from .mq import MessageFuture, MQ, Manager, register_broker
 from .consumer import ConsumerAPI
@@ -17,6 +17,7 @@
  'PulsarQueue',
  #
  'Message',
+ 'queue_message',
  'MessageDict',
  'MessageFuture',
  'ConsumerAPI',

pq/utils/serializers.py

@@ -58,7 +58,7 @@ class Json:
  def decode(cls, data):
  if isinstance(data, bytes):
  data = data.decode('utf-8')
- return json.loads(data, object_hook=object_hook)
+ return json.loads(data, object_hook=queue_message)
 
  @classmethod
  def encode(cls, message):
@@ -72,15 +72,15 @@ class MsgPack:
 
  @classmethod
  def decode(cls, data):
- return msgpack.unpackb(data, object_hook=object_hook,
+ return msgpack.unpackb(data, object_hook=queue_message,
  encoding='utf-8')
 
  @classmethod
  def encode(cls, message):
  return msgpack.packb(message, default=as_message)
 
 
-def object_hook(d):
+def queue_message(d):
  type = d.get('type')
  MsgType = message_types.get(type)
  if MsgType: