org.apache.kafka.connect.runtime.isolation.PluginScanResult Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of connect-runtime Show documentation
The newest version!
/*
 * 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.kafka.connect.runtime.isolation;

import org.apache.kafka.common.config.provider.ConfigProvider;
import org.apache.kafka.connect.connector.policy.ConnectorClientConfigOverridePolicy;
import org.apache.kafka.connect.rest.ConnectRestExtension;
import org.apache.kafka.connect.sink.SinkConnector;
import org.apache.kafka.connect.source.SourceConnector;
import org.apache.kafka.connect.storage.Converter;
import org.apache.kafka.connect.storage.HeaderConverter;
import org.apache.kafka.connect.transforms.Transformation;
import org.apache.kafka.connect.transforms.predicates.Predicate;

import java.util.Arrays;
import java.util.List;
import java.util.SortedSet;
import java.util.TreeSet;
import java.util.function.Consumer;
import java.util.function.Function;

public class PluginScanResult {
    private final SortedSet> sinkConnectors;
    private final SortedSet> sourceConnectors;
    private final SortedSet> converters;
    private final SortedSet> headerConverters;
    private final SortedSet>> transformations;
    private final SortedSet>> predicates;
    private final SortedSet> configProviders;
    private final SortedSet> restExtensions;
    private final SortedSet> connectorClientConfigPolicies;

    private final List>> allPlugins;

    public PluginScanResult(
            SortedSet> sinkConnectors,
            SortedSet> sourceConnectors,
            SortedSet> converters,
            SortedSet> headerConverters,
            SortedSet>> transformations,
            SortedSet>> predicates,
            SortedSet> configProviders,
            SortedSet> restExtensions,
            SortedSet> connectorClientConfigPolicies
    ) {
        this.sinkConnectors = sinkConnectors;
        this.sourceConnectors = sourceConnectors;
        this.converters = converters;
        this.headerConverters = headerConverters;
        this.transformations = transformations;
        this.predicates = predicates;
        this.configProviders = configProviders;
        this.restExtensions = restExtensions;
        this.connectorClientConfigPolicies = connectorClientConfigPolicies;
        this.allPlugins =
            Arrays.asList(sinkConnectors, sourceConnectors, converters, headerConverters, transformations, predicates,
                    configProviders, restExtensions, connectorClientConfigPolicies);
    }

    /**
     * Merge one or more {@link PluginScanResult results} into one result object
     */
    public PluginScanResult(List results) {
        this(
                merge(results, PluginScanResult::sinkConnectors),
                merge(results, PluginScanResult::sourceConnectors),
                merge(results, PluginScanResult::converters),
                merge(results, PluginScanResult::headerConverters),
                merge(results, PluginScanResult::transformations),
                merge(results, PluginScanResult::predicates),
                merge(results, PluginScanResult::configProviders),
                merge(results, PluginScanResult::restExtensions),
                merge(results, PluginScanResult::connectorClientConfigPolicies)
        );
    }

    private static > SortedSet merge(List results, Function> accessor) {
        SortedSet merged = new TreeSet<>();
        for (PluginScanResult element : results) {
            merged.addAll(accessor.apply(element));
        }
        return merged;
    }

    public SortedSet> sinkConnectors() {
        return sinkConnectors;
    }

    public SortedSet> sourceConnectors() {
        return sourceConnectors;
    }

    public SortedSet> converters() {
        return converters;
    }

    public SortedSet> headerConverters() {
        return headerConverters;
    }

    public SortedSet>> transformations() {
        return transformations;
    }

    public SortedSet>> predicates() {
        return predicates;
    }

    public SortedSet> configProviders() {
        return configProviders;
    }

    public SortedSet> restExtensions() {
        return restExtensions;
    }

    public SortedSet> connectorClientConfigPolicies() {
        return connectorClientConfigPolicies;
    }

    public void forEach(Consumer> consumer) {
        allPlugins.forEach(plugins -> plugins.forEach(consumer));
    }

    public boolean isEmpty() {
        boolean isEmpty = true;
        for (SortedSet plugins : allPlugins) {
            isEmpty = isEmpty && plugins.isEmpty();
        }
        return isEmpty;
    }
}