add get message method in Record interface (apache#4341)

ouyanghuangzheng · May 24, 2019 · 92f0cee · 92f0cee
1 parent d6c2813
commit 92f0cee
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pulsar-functions/api-java/src/main/java/org/apache/pulsar/functions/api/Record.java b/pulsar-functions/api-java/src/main/java/org/apache/pulsar/functions/api/Record.java
@@ -18,6 +18,8 @@
  */
 package org.apache.pulsar.functions.api;
 
+import org.apache.pulsar.client.api.Message;
+
 import java.util.Collections;
 import java.util.Map;
 import java.util.Optional;
@@ -104,4 +106,8 @@ default void fail() {
     default Optional<String> getDestinationTopic() {
         return Optional.empty();
     }
+
+    default Optional<Message<T>> getMessage() {
+        return Optional.empty();
+    }
 }
diff --git a/pulsar-functions/instance/src/main/java/org/apache/pulsar/functions/source/PulsarRecord.java b/pulsar-functions/instance/src/main/java/org/apache/pulsar/functions/source/PulsarRecord.java
@@ -106,4 +106,9 @@ public void ack() {
     public void fail() {
         this.failFunction.run();
     }
+
+    @Override
+    public Optional<Message<T>> getMessage() {
+        return Optional.of(message);
+    }
 }