add query prefix

JNSimba · Dec 26, 2024 · 7e85bed · 7e85bed
1 parent 7e27dc5
commit 7e85bed
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...-connector/src/main/java/org/apache/doris/flink/source/reader/DorisFlightValueReader.java b/...-connector/src/main/java/org/apache/doris/flink/source/reader/DorisFlightValueReader.java
@@ -54,6 +54,8 @@
 
 public class DorisFlightValueReader extends ValueReader implements AutoCloseable {
     private static final Logger LOG = LoggerFactory.getLogger(DorisFlightValueReader.class);
+    private static final String PREFIX = "/* ApplicationName=Flink ArrowFlightSQL Query */";
+
     protected AdbcConnection client;
     protected Lock clientLock = new ReentrantLock();
 
@@ -117,7 +119,8 @@ private String parseFlightSql(
                 Arrays.stream(tableIdentifiers)
                         .map(v -> "`" + v + "`")
                         .collect(Collectors.joining("."));
-        String sql = "SELECT " + readFields + " FROM " + queryTable;
+
+        String sql = PREFIX + " SELECT " + readFields + " FROM " + queryTable;
         if (CollectionUtils.isNotEmpty(partition.getTabletIds())) {
             String tablet =
                     partition.getTabletIds().stream()