Add 'car/' from commit 'eee0e8dc445691e600680f4abc77f2814b20b054'
git-subtree-dir: car git-subtree-mainline:1d29a5526cgit-subtree-split:eee0e8dc44
This commit is contained in:
64
car/Messaging/message_factory.py
Normal file
64
car/Messaging/message_factory.py
Normal file
@@ -0,0 +1,64 @@
|
||||
import zmq
|
||||
|
||||
|
||||
class ZmqPubSubStreamer:
|
||||
'''
|
||||
Not thread-safe. Always get this inside the thread/process where you intend
|
||||
to use it.
|
||||
'''
|
||||
|
||||
def __init__(self, port):
|
||||
self._socket = zmq.Context.instance().socket(zmq.PUB)
|
||||
print('Starting socket with address: ' + 'tcp://*:' + str(port))
|
||||
self._socket.bind("tcp://*:" + str(port))
|
||||
|
||||
|
||||
def send_message(self, message):
|
||||
'''
|
||||
Args
|
||||
----
|
||||
message: A message type that has the serialise() method.
|
||||
'''
|
||||
self.send_message_topic("", message)
|
||||
|
||||
def send_message_topic(self, topic, message):
|
||||
self._socket.send_multipart([bytes(topic), message.serialise()])
|
||||
|
||||
|
||||
class BluetoothStreamer:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def send_message(self, message_bytes):
|
||||
pass
|
||||
|
||||
class TestStreamer:
|
||||
def __init__(self):
|
||||
self._listeners = []
|
||||
|
||||
def send_message(self, message_bytes):
|
||||
print('Got a message')
|
||||
|
||||
def send_message_topic(self, topic, message):
|
||||
print('Got a message with topic: ' + str(topic))
|
||||
self._fire_message_received(message)
|
||||
|
||||
def add_message_listener(self, listener):
|
||||
self._listeners.append(listener)
|
||||
|
||||
def _fire_message_received(self, message):
|
||||
for listener in self._listeners:
|
||||
listener(message)
|
||||
|
||||
def getZmqPubSubStreamer(port):
|
||||
'''
|
||||
Not thread-safe. Always get this inside the thread/process where you intend
|
||||
to use it.
|
||||
'''
|
||||
return ZmqPubSubStreamer(port)
|
||||
|
||||
def getTestingStreamer():
|
||||
return TestStreamer()
|
||||
|
||||
# TODO: Create a general get method that will get the streamer based on an
|
||||
# environment variable that is set.
|
||||
Reference in New Issue
Block a user