Merge pull request julien-duponchelle#66 from lxyu/primay_col_info

julien-duponchelle · julien-duponchelle · commit 9cec97196927 · 2014-07-09T17:29:16.000+02:00
expose primay_keys info for row based events
diff --git a/pymysqlreplication/binlogstream.py b/pymysqlreplication/binlogstream.py
@@ -189,7 +189,7 @@ def __get_table_information(self, schema, table):
                 cur.execute("""
                     SELECT
                         COLUMN_NAME, COLLATION_NAME, CHARACTER_SET_NAME,
-                        COLUMN_COMMENT, COLUMN_TYPE
+                        COLUMN_COMMENT, COLUMN_TYPE, COLUMN_KEY
                     FROM
                         columns
                     WHERE
diff --git a/pymysqlreplication/column.py b/pymysqlreplication/column.py
@@ -24,6 +24,7 @@ def __parse_column_definition(self, column_type, column_schema, packet):
         self.data["comment"] = column_schema["COLUMN_COMMENT"]
         self.data["unsigned"] = False
         self.data["type_is_bool"] = False
+        self.data["is_primary"] = column_schema["COLUMN_KEY"] == "PRI"
 
         if column_schema["COLUMN_TYPE"].find("unsigned") != -1:
             self.data["unsigned"] = True
diff --git a/pymysqlreplication/row_event.py b/pymysqlreplication/row_event.py
@@ -21,6 +21,7 @@ def __init__(self, from_packet, event_size, table_map, ctl_connection):
 
         #Header
         self.table_id = self._read_table_id()
+        self.primary_key = table_map[self.table_id].data["primary_key"]
         self.flags = struct.unpack('<H', self.packet.read(2))[0]
 
         #Event V2
diff --git a/pymysqlreplication/table.py b/pymysqlreplication/table.py
@@ -3,12 +3,21 @@
 
 class Table(object):
     def __init__(self, column_schemas, table_id, schema, table, columns):
+        primary_key = [c.data["name"] for c in columns if c.data["is_primary"]]
+        if len(primary_key) == 0:
+            primary_key = ''
+        elif len(primary_key) == 1:
+            primary_key, = primary_key
+        else:
+            primary_key = tuple(primary_key)
+
         self.data = {
             "column_schemas": column_schemas,
             "table_id": table_id,
             "schema": schema,
             "table": table,
-            "columns": columns
+            "columns": columns,
+            "primary_key": primary_key
         }
 
     def __getattr__(self, item):
