add for test

xwm1992 · Aug 20, 2024 · 8e885bf · 8e885bf
1 parent f3eacde
commit 8e885bf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...va/org/apache/eventmesh/connector/http/sink/handler/impl/HttpSinkHandlerRetryWrapper.java b/...va/org/apache/eventmesh/connector/http/sink/handler/impl/HttpSinkHandlerRetryWrapper.java
@@ -38,6 +38,7 @@
 
 import dev.failsafe.Failsafe;
 import dev.failsafe.RetryPolicy;
+import org.apache.eventmesh.openconnect.offsetmgmt.api.data.ConnectRecord;
 
 
 /**
@@ -75,7 +76,7 @@ public void start() {
      * @return processing chain
      */
     @Override
-    public Future<HttpResponse<Buffer>> deliver(URI url, HttpConnectRecord httpConnectRecord, Map<String, Object> attributes) {
+    public Future<HttpResponse<Buffer>> deliver(URI url, HttpConnectRecord httpConnectRecord, Map<String, Object> attributes, ConnectRecord connectRecord) {
 
         // Build the retry policy
         RetryPolicy<HttpResponse<Buffer>> retryPolicy = RetryPolicy.<HttpResponse<Buffer>>builder()
@@ -104,7 +105,7 @@ public Future<HttpResponse<Buffer>> deliver(URI url, HttpConnectRecord httpConne
 
         // Handle the ConnectRecord with retry policy
         Failsafe.with(retryPolicy)
-            .getStageAsync(() -> sinkHandler.deliver(url, httpConnectRecord, attributes).toCompletionStage());
+            .getStageAsync(() -> sinkHandler.deliver(url, httpConnectRecord, attributes, connectRecord).toCompletionStage());
 
         return null;
     }