NIFI-15055 Add FLOW_REGISTRY_CLIENT to ExtensionType to NiFi API (#21)

Signed-off-by: David Handermann <[email protected]>

Author: pvillard31

src/main/java/org/apache/nifi/documentation/AbstractDocumentationWriter.java

@@ -47,6 +47,7 @@
 import org.apache.nifi.controller.ControllerService;
 import org.apache.nifi.documentation.init.DocumentationControllerServiceInitializationContext;
 import org.apache.nifi.documentation.init.DocumentationFlowAnalysisRuleInitializationContext;
+import org.apache.nifi.documentation.init.DocumentationFlowRegistryClientInitializationContext;
 import org.apache.nifi.documentation.init.DocumentationParameterProviderInitializationContext;
 import org.apache.nifi.documentation.init.DocumentationProcessorInitializationContext;
 import org.apache.nifi.documentation.init.DocumentationReportingInitializationContext;
@@ -56,6 +57,7 @@
 import org.apache.nifi.processor.Relationship;
 import org.apache.nifi.reporting.InitializationException;
 import org.apache.nifi.reporting.ReportingTask;
+import org.apache.nifi.registry.flow.FlowRegistryClient;
 
 import java.io.IOException;
 import java.util.ArrayList;
@@ -95,6 +97,8 @@ public void initialize(final ConfigurableComponent component) {
                 initialize((FlowAnalysisRule) component);
             } else if (component instanceof ParameterProvider) {
                 initialize((ParameterProvider) component);
+            } else if (component instanceof FlowRegistryClient) {
+                initialize((FlowRegistryClient) component);
             }
         } catch (final InitializationException ie) {
             throw new RuntimeException("Failed to initialize " + component, ie);
@@ -121,6 +125,10 @@ protected void initialize(final ParameterProvider parameterProvider) throws Init
         parameterProvider.initialize(new DocumentationParameterProviderInitializationContext());
     }
 
+    protected void initialize(final FlowRegistryClient flowRegistryClient) {
+        flowRegistryClient.initialize(new DocumentationFlowRegistryClientInitializationContext());
+    }
+
     @Override
     public final void write(final ConfigurableComponent component) throws IOException {
         write(component, Collections.emptyList(), Collections.emptyMap());
@@ -296,6 +304,9 @@ protected ExtensionType getExtensionType(final ConfigurableComponent component)
         if (component instanceof ParameterProvider) {
             return ExtensionType.PARAMETER_PROVIDER;
         }
+        if (component instanceof FlowRegistryClient) {
+            return ExtensionType.FLOW_REGISTRY_CLIENT;
+        }
         throw new AssertionError("Encountered unknown Configurable Component Type for " + component);
     }
 

src/main/java/org/apache/nifi/documentation/ExtensionType.java

@@ -25,5 +25,7 @@ public enum ExtensionType {
 
     FLOW_ANALYSIS_RULE,
 
-    PARAMETER_PROVIDER;
+    PARAMETER_PROVIDER,
+
+    FLOW_REGISTRY_CLIENT;
 }

src/main/java/org/apache/nifi/documentation/init/DocumentationFlowRegistryClientInitializationContext.java

@@ -0,0 +1,44 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License 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 org.apache.nifi.documentation.init;
+
+import org.apache.nifi.logging.ComponentLog;
+import org.apache.nifi.registry.flow.FlowRegistryClientInitializationContext;
+
+import javax.net.ssl.SSLContext;
+import java.util.Optional;
+import java.util.UUID;
+
+public class DocumentationFlowRegistryClientInitializationContext implements FlowRegistryClientInitializationContext {
+    private final String identifier = UUID.randomUUID().toString();
+    private final ComponentLog componentLog = new NopComponentLog();
+
+    @Override
+    public String getIdentifier() {
+        return identifier;
+    }
+
+    @Override
+    public ComponentLog getLogger() {
+        return componentLog;
+    }
+
+    @Override
+    public Optional<SSLContext> getSystemSslContext() {
+        return Optional.empty();
+    }
+}