Merge pull request #5 from Matvey-Kuk/feature/inventory_fix_message_type

Added Message type for inventory purposes

Author: TimEvens

src/openbmp/api/parsed/message/Message.py

@@ -12,6 +12,9 @@ class Message(object):
         Kafka Message class. This class processes header of raw Kafka messages.
     """
 
+    TYPE_PEER = "PEER"
+    TYPE_ROUTER = "ROUTER"
+
     def __init__(self, data):
         """
         Handle the message by parsing header of it.
@@ -23,6 +26,7 @@ def __init__(self, data):
             raise "Invalid data!", data
 
         self.version = float()
+        self.type = str()
         self.collector_hash_id = str()
         self.length = long()
         self.records = long()
@@ -59,6 +63,9 @@ def __parse(self, data):
             elif attr == "C_HASH_ID":
                 self.collector_hash_id = value
 
+            elif attr == "T":
+                self.type = value
+
             elif attr == "L":
                 self.length = long(value)