[FLINK-32772] Add support of SET statements in the SQL Runner example

Author: mehdid93

Diff for: examples/flink-sql-runner-example/src/main/java/org/apache/flink/examples/SqlRunner.java

@@ -27,6 +27,8 @@
 import java.io.File;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 
 /** Main class for executing SQL scripts. */
 public class SqlRunner {
@@ -38,6 +40,9 @@ public class SqlRunner {
 
     private static final String COMMENT_PATTERN = "(--.*)|(((\\/\\*)+?[\\w\\W]+?(\\*\\/)+))";
 
+    private static final Pattern SET_STATEMENT_PATTERN =
+            Pattern.compile("SET\\s+'(\\S+)'\\s+=\\s+'(.*)';", Pattern.CASE_INSENSITIVE);
+
     public static void main(String[] args) throws Exception {
         if (args.length != 1) {
             throw new Exception("Exactly one argument is expected.");
@@ -48,8 +53,17 @@ public static void main(String[] args) throws Exception {
         var tableEnv = TableEnvironment.create(new Configuration());
 
         for (String statement : statements) {
-            LOG.info("Executing:\n{}", statement);
-            tableEnv.executeSql(statement);
+            Matcher setMatcher = SET_STATEMENT_PATTERN.matcher(statement.trim());
+
+            if (setMatcher.matches()) {
+                // Handle SET statements
+                String key = setMatcher.group(1);
+                String value = setMatcher.group(2);
+                tableEnv.getConfig().getConfiguration().setString(key, value);
+            } else {
+                LOG.info("Executing:\n{}", statement);
+                tableEnv.executeSql(statement);
+            }
         }
     }