Small fix after rebase

apache · May 20, 2019 · 36b8587 · 36b8587
1 parent b4b910b
commit 36b8587
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/...-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessContext.java b/...-stateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessContext.java
@@ -27,6 +27,7 @@
 import org.apache.nifi.controller.ControllerService;
 import org.apache.nifi.controller.ControllerServiceInitializationContext;
 import org.apache.nifi.controller.ControllerServiceLookup;
+import org.apache.nifi.controller.NodeTypeProvider;
 import org.apache.nifi.logging.ComponentLog;
 import org.apache.nifi.processor.Processor;
 import org.apache.nifi.processor.Relationship;
@@ -392,6 +393,20 @@ public StateManager getStateManager() {
         return stateManager;
     }
 
+    @Override
+    public NodeTypeProvider getNodeTypeProvider() {
+        return new NodeTypeProvider() {
+            @Override
+            public boolean isClustered() {
+                return false;
+            }
+            @Override
+            public boolean isPrimary() {
+                return true;
+            }
+        };
+    }
+
     @Override
     public String getName() {
         return componentName;

diff --git a/...tateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessorWrapper.java b/...tateless-core/src/main/java/org/apache/nifi/stateless/core/StatelessProcessorWrapper.java
@@ -66,7 +66,7 @@ public class StatelessProcessorWrapper extends AbstractStatelessComponent implem
     private volatile boolean initialized = false;
 
 
-    StatelessProcessorWrapper(final String id, final Processor processor, final StatelessProcessorWrapper parent, final StatelessControllerServiceLookup lookup, final VariableRegistry registry,
+    public StatelessProcessorWrapper(final String id, final Processor processor, final StatelessProcessorWrapper parent, final StatelessControllerServiceLookup lookup, final VariableRegistry registry,
                               final boolean materializeContent, final ClassLoader classLoader) throws InvocationTargetException, IllegalAccessException {
 
         this.processor = processor;