fix(ftpfile): 携带路径时去掉多余前缀

chaorongzhi · Sep 3, 2024 · 89494ee · 89494ee
1 parent 6043592
commit 89494ee
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/...va/org/apache/seatunnel/connectors/seatunnel/file/source/reader/AbstractReadStrategy.java b/...va/org/apache/seatunnel/connectors/seatunnel/file/source/reader/AbstractReadStrategy.java
@@ -227,7 +227,17 @@ protected SeaTunnelRow dataCarryFilename(SeaTunnelRow seaTunnelRow, String path)
         } else {
             System.arraycopy(fields, 1, newFields, 1, fields.length - 1);
         }
-        newFields[0] = path;
+        String oriPath = pluginConfig.getString(BaseSourceConfigOptions.FILE_PATH.key());
+        String[] split = path.split(oriPath.replace("\\", "/"));
+        String filePath = oriPath;
+        if (split.length > 1) {
+            if (oriPath.endsWith("/")) {
+                filePath = oriPath + split[1];
+            } else {
+                filePath = oriPath + "/" + split[1];
+            }
+        }
+        newFields[0] = filePath;
         SeaTunnelRow newSeaTunnelRow = new SeaTunnelRow(newFields);
         newSeaTunnelRow.setRowKind(seaTunnelRow.getRowKind());
         newSeaTunnelRow.setTableId(seaTunnelRow.getTableId());