message_base.py 668 B

1234567891011121314151617181920212223242526272829
  1. from queue import Queue
  2. class MessageBase:
  3. def __init__(self):
  4. self.data = dict()
  5. def add(self, device, data):
  6. if device in self.data:
  7. self.data[device].put(data)
  8. else:
  9. self.data[device] = Queue()
  10. self.data[device].put(data)
  11. def get(self, device):
  12. data_queue: Queue = self.data.get(device)
  13. if not data_queue or data_queue.empty():
  14. return None
  15. data = data_queue.get()
  16. return data
  17. if __name__ == '__main__':
  18. mb = MessageBase()
  19. mb.add("a", "asdasd")
  20. mb.add("a", "11111111")
  21. print(mb.data)
  22. data = mb.get("a")
  23. print(data)