Add WorkerInterceptorBase and improve WorkerInterceptor docs to help new interceptor users

Author: Spikhalskiy

temporal-sdk/src/main/java/io/temporal/common/interceptors/WorkerInterceptor.java

@@ -24,7 +24,57 @@
 /**
  * Intercepts workflow and activity executions.
  *
- * <p>TODO(maxim): JavaDoc with sample
+ * <p>Prefer extending {@link WorkerInterceptorBase} and overriding only the methods you need
+ * instead of implementing this interface directly. {@link WorkerInterceptorBase} provides correct
+ * default implementations to all the methods of this interface.
+ *
+ * You may want to start your implementation with this initial structure:
+ *
+ * <pre>{@code
+ *   private static class CustomWorkerInterceptor extends WorkerInterceptorBase {
+ *     // remove if you don't need to have a custom WorkflowInboundCallsInterceptor or WorkflowOutboundCallsInterceptor
+ *     @Override
+ *     public WorkflowInboundCallsInterceptor interceptWorkflow(WorkflowInboundCallsInterceptor next) {
+ *       return new CustomWorkflowInboundCallsInterceptor(next) {
+ *         // remove if you don't need to have a custom WorkflowOutboundCallsInterceptor
+ *         @Override
+ *         public void init(WorkflowOutboundCallsInterceptor outboundCalls) {
+ *           next.init(new CustomWorkflowOutboundCallsInterceptor(outboundCalls));
+ *         }
+ *       };
+ *     }
+ *
+ *     // remove if you don't need to have a custom ActivityInboundCallsInterceptor
+ *     @Override
+ *     public ActivityInboundCallsInterceptor interceptActivity(ActivityInboundCallsInterceptor next) {
+ *       return new CustomActivityInboundCallsInterceptor(next);
+ *     }
+ *   }
+ *
+ *   private static class CustomWorkflowInboundCallsInterceptor extends WorkflowInboundCallsInterceptorBase {
+ *     public CustomWorkflowInboundCallsInterceptor(WorkflowInboundCallsInterceptor next) {
+ *       super(next);
+ *     }
+ *
+ *     // override only the methods you need
+ *   }
+ *
+ *   private static class CustomWorkflowOutboundCallsInterceptor extends WorkflowOutboundCallsInterceptorBase {
+ *     public CustomWorkflowOutboundCallsInterceptor(WorkflowOutboundCallsInterceptor next) {
+ *       super(next);
+ *     }
+ *
+ *     // override only the methods you need
+ *   }
+ *
+ *   private static class CustomActivityInboundCallsInterceptor extends ActivityInboundCallsInterceptorBase {
+ *     public CustomActivityInboundCallsInterceptor(ActivityInboundCallsInterceptor next) {
+ *       super(next);
+ *     }
+ *
+ *     // override only the methods you need
+ *   }
+ *   }</pre>
  */
 @Experimental
 public interface WorkerInterceptor {

temporal-sdk/src/main/java/io/temporal/common/interceptors/WorkerInterceptorBase.java

@@ -0,0 +1,32 @@
+/*
+ *  Copyright (C) 2020 Temporal Technologies, Inc. All Rights Reserved.
+ *
+ *  Copyright 2012-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ *  Modifications copyright (C) 2017 Uber Technologies, Inc.
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License"). You may not
+ *  use this file except in compliance with the License. A copy of the License is
+ *  located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ *  or in the "license" file accompanying this file. This file is distributed on
+ *  an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ *  express or implied. See the License for the specific language governing
+ *  permissions and limitations under the License.
+ */
+
+package io.temporal.common.interceptors;
+
+public class WorkerInterceptorBase implements WorkerInterceptor {
+  @Override
+  public WorkflowInboundCallsInterceptor interceptWorkflow(WorkflowInboundCallsInterceptor next) {
+    return next;
+  }
+
+  @Override
+  public ActivityInboundCallsInterceptor interceptActivity(ActivityInboundCallsInterceptor next) {
+    return next;
+  }
+}

temporal-sdk/src/test/java/io/temporal/workflow/MetricsTest.java

@@ -43,11 +43,7 @@
 import io.temporal.client.WorkflowOptions;
 import io.temporal.client.WorkflowStub;
 import io.temporal.common.RetryOptions;
-import io.temporal.common.interceptors.ActivityInboundCallsInterceptor;
-import io.temporal.common.interceptors.WorkerInterceptor;
-import io.temporal.common.interceptors.WorkflowInboundCallsInterceptor;
-import io.temporal.common.interceptors.WorkflowInboundCallsInterceptorBase;
-import io.temporal.common.interceptors.WorkflowOutboundCallsInterceptor;
+import io.temporal.common.interceptors.*;
 import io.temporal.common.reporter.TestStatsReporter;
 import io.temporal.serviceclient.MetricsTag;
 import io.temporal.serviceclient.WorkflowServiceStubs;
@@ -57,7 +53,6 @@
 import io.temporal.worker.Worker;
 import io.temporal.worker.WorkerFactoryOptions;
 import io.temporal.worker.WorkerMetricsTag;
-import io.temporal.workflow.interceptors.SignalWorkflowOutboundCallsInterceptor;
 import io.temporal.workflow.shared.TestActivities.TestActivitiesImpl;
 import io.temporal.workflow.shared.TestActivities.TestActivity3;
 import io.temporal.workflow.shared.TestActivities.VariousTestActivities;
@@ -69,6 +64,8 @@
 import java.util.Map;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutionException;
+import java.util.function.Function;
+
 import org.junit.After;
 import org.junit.Rule;
 import org.junit.Test;
@@ -261,19 +258,7 @@ public void testCorruptedSignalMetrics() throws InterruptedException {
             .setWorkerInterceptors(
                 new CorruptedSignalWorkerInterceptor(),
                 // Add noop just to test that list of interceptors is working.
-                new WorkerInterceptor() {
-                  @Override
-                  public WorkflowInboundCallsInterceptor interceptWorkflow(
-                      WorkflowInboundCallsInterceptor next) {
-                    return next;
-                  }
-
-                  @Override
-                  public ActivityInboundCallsInterceptor interceptActivity(
-                      ActivityInboundCallsInterceptor next) {
-                    return next;
-                  }
-                })
+                new WorkerInterceptorBase())
             .build());
 
     Worker worker = testEnvironment.newWorker(TASK_QUEUE);
@@ -341,24 +326,7 @@ public void testTemporalFailureMetric() throws InterruptedException {
 
   @Test
   public void testTemporalActivityFailureMetric() throws InterruptedException {
-    setUp(
-        WorkerFactoryOptions.newBuilder()
-            .setWorkerInterceptors(
-                // Add noop just to test that list of interceptors is working.
-                new WorkerInterceptor() {
-                  @Override
-                  public WorkflowInboundCallsInterceptor interceptWorkflow(
-                      WorkflowInboundCallsInterceptor next) {
-                    return next;
-                  }
-
-                  @Override
-                  public ActivityInboundCallsInterceptor interceptActivity(
-                      ActivityInboundCallsInterceptor next) {
-                    return next;
-                  }
-                })
-            .build());
+    setUp(WorkerFactoryOptions.getDefaultInstance());
 
     Worker worker = testEnvironment.newWorker(TASK_QUEUE);
     worker.registerWorkflowImplementationTypes(TestActivityFailureCountersWorkflow.class);
@@ -603,12 +571,10 @@ public void execute() {
   }
 
   public static class Signal {
-
     public String value;
   }
 
-  private static class CorruptedSignalWorkerInterceptor implements WorkerInterceptor {
-
+  private static class CorruptedSignalWorkerInterceptor extends WorkerInterceptorBase {
     @Override
     public WorkflowInboundCallsInterceptor interceptWorkflow(WorkflowInboundCallsInterceptor next) {
       return new WorkflowInboundCallsInterceptorBase(next) {
@@ -627,10 +593,87 @@ public void init(WorkflowOutboundCallsInterceptor outboundCalls) {
         }
       };
     }
+  }
+
+  private static class SignalWorkflowOutboundCallsInterceptor extends WorkflowOutboundCallsInterceptorBase {
+    private final Function<Object[], Object[]> overrideArgs;
+    private final Function<String, String> overrideSignalName;
+
+    public SignalWorkflowOutboundCallsInterceptor(
+        Function<Object[], Object[]> overrideArgs,
+        Function<String, String> overrideSignalName,
+        WorkflowOutboundCallsInterceptor next) {
+      super(next);
+      this.overrideArgs = overrideArgs;
+      this.overrideSignalName = overrideSignalName;
+    }
+
+    @Override
+    public SignalExternalOutput signalExternalWorkflow(SignalExternalInput input) {
+      Object[] args = input.getArgs();
+      if (args != null && args.length > 0) {
+        args = new Object[] {"corrupted signal"};
+      }
+      return super.signalExternalWorkflow(
+          new SignalExternalInput(
+              input.getExecution(),
+              overrideSignalName.apply(input.getSignalName()),
+              overrideArgs.apply(args)));
+    }
+  }
+
+
+
+
+
+
+
+
+
+
+
+
+  private static class CustomWorkerInterceptor extends WorkerInterceptorBase {
+    // remove if you don't need to have a custom WorkflowInboundCallsInterceptor or WorkflowOutboundCallsInterceptor
+    @Override
+    public WorkflowInboundCallsInterceptor interceptWorkflow(WorkflowInboundCallsInterceptor next) {
+      return new CustomWorkflowInboundCallsInterceptor(next) {
+        // remove if you don't need to have a custom WorkflowOutboundCallsInterceptor
+        @Override
+        public void init(WorkflowOutboundCallsInterceptor outboundCalls) {
+          next.init(new CustomWorkflowOutboundCallsInterceptor(outboundCalls));
+        }
+      };
+    }
 
+    // remove if you don't need to have a custom ActivityInboundCallsInterceptor
     @Override
     public ActivityInboundCallsInterceptor interceptActivity(ActivityInboundCallsInterceptor next) {
-      return next;
+      return new CustomActivityInboundCallsInterceptor(next);
     }
   }
+
+  private static class CustomWorkflowInboundCallsInterceptor extends WorkflowInboundCallsInterceptorBase {
+    public CustomWorkflowInboundCallsInterceptor(WorkflowInboundCallsInterceptor next) {
+      super(next);
+    }
+
+    // override only the methods you need
+  }
+
+  private static class CustomWorkflowOutboundCallsInterceptor extends WorkflowOutboundCallsInterceptorBase {
+    public CustomWorkflowOutboundCallsInterceptor(WorkflowOutboundCallsInterceptor next) {
+      super(next);
+    }
+
+    // override only the methods you need
+  }
+
+  private static class CustomActivityInboundCallsInterceptor extends ActivityInboundCallsInterceptorBase {
+    public CustomActivityInboundCallsInterceptor(ActivityInboundCallsInterceptor next) {
+      super(next);
+    }
+
+    // override only the methods you need
+  }
 }