[camel-kafka-connector] 01/02: SFTP connector: Add RemoteFileTransforms to the connector directly

[ac...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

acosentino pushed a commit to branch sftp-regen
in repository https://gitbox.apache.org/repos/asf/camel-kafka-connector.git

commit dfc07f74ac54dd4e67c7e73eff58ba7034359c14
Author: Andrea Cosentino <an...@gmail.com>
AuthorDate: Mon Nov 9 08:25:27 2020 +0100

    SFTP connector: Add RemoteFileTransforms to the connector directly
---
 .../transformers/SftpRemoteFileTransforms.java     | 55 ++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/connectors/camel-sftp-kafka-connector/src/main/java/org/apache/camel/kafkaconnector/sftp/transformers/SftpRemoteFileTransforms.java b/connectors/camel-sftp-kafka-connector/src/main/java/org/apache/camel/kafkaconnector/sftp/transformers/SftpRemoteFileTransforms.java
new file mode 100644
index 0000000..d04cf99
--- /dev/null
+++ b/connectors/camel-sftp-kafka-connector/src/main/java/org/apache/camel/kafkaconnector/sftp/transformers/SftpRemoteFileTransforms.java
@@ -0,0 +1,55 @@
+package org.apache.camel.kafkaconnector.sftp.transformers;
+
+import java.util.Map;
+
+import org.apache.camel.component.file.remote.RemoteFile;
+import org.apache.camel.kafkaconnector.utils.SchemaHelper;
+import org.apache.kafka.common.config.ConfigDef;
+import org.apache.kafka.connect.connector.ConnectRecord;
+import org.apache.kafka.connect.transforms.Transformation;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class SftpRemoteFileTransforms <R extends ConnectRecord<R>> implements Transformation<R> {
+    public static final String FIELD_KEY_CONFIG = "key";
+    public static final ConfigDef CONFIG_DEF = new ConfigDef()
+            .define(FIELD_KEY_CONFIG, ConfigDef.Type.STRING, null, ConfigDef.Importance.MEDIUM,
+                    "Transforms Remote File to String");
+
+    private static final Logger LOG = LoggerFactory.getLogger(SftpRemoteFileTransforms.class);
+
+    @Override
+    public R apply(R r) {
+        Object value = r.value();
+
+        if (r.value() instanceof RemoteFile) {
+            LOG.debug("Converting record from RemoteFile to text");
+            RemoteFile message = (RemoteFile) r.value();
+
+            LOG.debug("Received text: {}", message.getBody());
+
+            return r.newRecord(r.topic(), r.kafkaPartition(), null, r.key(),
+                    SchemaHelper.buildSchemaBuilderForType(message.getBody()), message.getBody(), r.timestamp());
+
+        } else {
+            LOG.debug("Unexpected message type: {}", r.value().getClass());
+
+            return r;
+        }
+    }
+
+    @Override
+    public ConfigDef config() {
+        return CONFIG_DEF;
+    }
+
+    @Override
+    public void close() {
+
+    }
+
+    @Override
+    public void configure(Map<String, ?> map) {
+
+    }
+}