From 7811d86edfc445aff8ca2339e673ecc2ddfe248a Mon Sep 17 00:00:00 2001
From: mfvitale <mvitale@redhat.com>
Date: Tue, 4 Jul 2023 17:07:14 +0200
Subject: [PATCH] DBZ-3925 Add support for FLUSH FIREWALL_RULES for Azure
 Database for MySQL Single Server instance

---
 .../io/debezium/ddl/parser/mysql/generated/MySqlLexer.g4      | 3 +++
 .../io/debezium/ddl/parser/mysql/generated/MySqlParser.g4     | 4 +++-
 .../src/test/resources/mysql/examples/ddl_flush.sql           | 3 +++
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/debezium-ddl-parser/src/main/antlr4/io/debezium/ddl/parser/mysql/generated/MySqlLexer.g4 b/debezium-ddl-parser/src/main/antlr4/io/debezium/ddl/parser/mysql/generated/MySqlLexer.g4
index 8852dbf96..9155d53ab 100644
--- a/debezium-ddl-parser/src/main/antlr4/io/debezium/ddl/parser/mysql/generated/MySqlLexer.g4
+++ b/debezium-ddl-parser/src/main/antlr4/io/debezium/ddl/parser/mysql/generated/MySqlLexer.g4
@@ -770,6 +770,9 @@ CLIENT_STATISTICS:                   'CLIENT_STATISTICS';
 INDEX_STATISTICS:                    'INDEX_STATISTICS';
 TABLE_STATISTICS:                    'TABLE_STATISTICS';
 
+// Azure Database for MySQL Single Server instance:
+FIREWALL_RULES:                      'FIREWALL_RULES';
+
 
 // PRIVILEGES
 
diff --git a/debezium-ddl-parser/src/main/antlr4/io/debezium/ddl/parser/mysql/generated/MySqlParser.g4 b/debezium-ddl-parser/src/main/antlr4/io/debezium/ddl/parser/mysql/generated/MySqlParser.g4
index 85106df95..66a741c1e 100644
--- a/debezium-ddl-parser/src/main/antlr4/io/debezium/ddl/parser/mysql/generated/MySqlParser.g4
+++ b/debezium-ddl-parser/src/main/antlr4/io/debezium/ddl/parser/mysql/generated/MySqlParser.g4
@@ -1998,6 +1998,8 @@ flushStatement
       flushOption (',' flushOption)*
     // MariaDB userstat plugin
     | FLUSH (USER_STATISTICS | CLIENT_STATISTICS | INDEX_STATISTICS | TABLE_STATISTICS)
+    // Specific for Azure Database for MySQL Single Server instance.
+    | FLUSH FIREWALL_RULES
     ;
 
 killStatement
@@ -2825,7 +2827,7 @@ keywordsCanBeId
     | BINLOG_MONITOR | BINLOG_REPLAY | CURRENT_ROLE | CYCLE | ENCRYPTED | ENCRYPTION_KEY_ID | FEDERATED_ADMIN
     | INCREMENT | LASTVAL | LOCKED | MAXVALUE | MINVALUE | NEXTVAL | NOCACHE | NOCYCLE | NOMAXVALUE | NOMINVALUE
     | PERSISTENT | PREVIOUS | READ_ONLY_ADMIN | REPLICA | REPLICATION_MASTER_ADMIN | RESTART | SEQUENCE | SETVAL
-    | SKIP_ | STATEMENT | UUID | VIA | MONITOR | READ_ONLY| REPLAY | USER_STATISTICS | CLIENT_STATISTICS | INDEX_STATISTICS | TABLE_STATISTICS
+    | SKIP_ | STATEMENT | UUID | VIA | MONITOR | READ_ONLY| REPLAY | USER_STATISTICS | CLIENT_STATISTICS | INDEX_STATISTICS | TABLE_STATISTICS | FIREWALL_RULES
     ;
 
 functionNameBase
diff --git a/debezium-ddl-parser/src/test/resources/mysql/examples/ddl_flush.sql b/debezium-ddl-parser/src/test/resources/mysql/examples/ddl_flush.sql
index c99e5dfa6..0a2f2e71b 100644
--- a/debezium-ddl-parser/src/test/resources/mysql/examples/ddl_flush.sql
+++ b/debezium-ddl-parser/src/test/resources/mysql/examples/ddl_flush.sql
@@ -17,5 +17,8 @@ flush TABLE Foo, Bar;
 flush table Foo, Bar for export;
 flush table Foo, Bar with read lock;
 
+-- Azure Database for MySQL Single Server instance. This type of database server is being decommissioned on Sept 16 2024 and is succeeded by their "Flexible Server" offering.
+FLUSH FIREWALL_RULES;
+