Handle alter table add index (#193)

Author: bakwc

mysql_ch_replicator/converter.py

@@ -688,13 +688,23 @@ def convert_alter_query(self, mysql_query, db_name):
             if op_name == 'add':
                 if tokens[0].lower() in ('constraint', 'index', 'foreign', 'unique', 'key'):
                     continue
-                self.__convert_alter_table_add_column(db_name, table_name, tokens)
+                # Handle cases where keywords are combined with parentheses: INDEX(col), CONSTRAINT(name), etc.
+                for keyword in ('constraint', 'index', 'foreign', 'unique', 'key'):
+                    if tokens[0].lower().startswith(f'{keyword}('):
+                        break
+                else:
+                    self.__convert_alter_table_add_column(db_name, table_name, tokens)
                 continue
 
             if op_name == 'drop':
                 if tokens[0].lower() in ('constraint', 'index', 'foreign', 'unique', 'key'):
                     continue
-                self.__convert_alter_table_drop_column(db_name, table_name, tokens)
+                # Handle cases where keywords are combined with parentheses: INDEX(col), CONSTRAINT(name), etc.
+                for keyword in ('constraint', 'index', 'foreign', 'unique', 'key'):
+                    if tokens[0].lower().startswith(f'{keyword}('):
+                        break
+                else:
+                    self.__convert_alter_table_drop_column(db_name, table_name, tokens)
                 continue
 
             if op_name == 'modify':

test_mysql_ch_replicator.py

@@ -1439,6 +1439,16 @@ def test_add_column_first_after_and_drop_column(monkeypatch):
     assert_wait(lambda: ch.select(TEST_TABLE_NAME, where="id=45")[0]['c1'] == 1111)
     assert_wait(lambda: ch.select(TEST_TABLE_NAME, where="id=45")[0].get('c2') is None)
 
+    # Test add index to c1 column
+    mysql.execute(
+        f"ALTER TABLE `{TEST_TABLE_NAME}` ADD INDEX(c1)")
+    mysql.execute(
+        f"INSERT INTO `{TEST_TABLE_NAME}` (id, c1) VALUES (47, 5555)",
+        commit=True,
+    )
+    assert_wait(lambda: len(ch.select(TEST_TABLE_NAME, where="id=47")) == 1)
+    assert_wait(lambda: ch.select(TEST_TABLE_NAME, where="id=47")[0]['c1'] == 5555)
+
     db_replicator_runner.stop()
     binlog_replicator_runner.stop()