libminifi/include/core/controller/ControllerServiceProvider.h

/**
 *
 * 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.
 */

#pragma once

#include <memory>
#include <string>
#include <string_view>
#include <utility>
#include <future>
#include <vector>

#include "minifi-cpp/core/controller/ControllerServiceProvider.h"
#include "core/Core.h"
#include "ControllerServiceLookup.h"
#include "core/ConfigurableComponentImpl.h"
#include "ControllerServiceNode.h"
#include "ControllerServiceNodeMap.h"
#include "core/ClassLoader.h"
#include "utils/Monitors.h"

namespace org::apache::nifi::minifi::core::controller {

class ControllerServiceProviderImpl : public CoreComponentImpl, public ConfigurableComponentImpl, public virtual ControllerServiceLookup, public virtual ControllerServiceProvider {
 public:
  explicit ControllerServiceProviderImpl(std::string_view name)
      : CoreComponentImpl(name),
        controller_map_{std::make_unique<ControllerServiceNodeMap>()} {
  }

  explicit ControllerServiceProviderImpl(std::unique_ptr<ControllerServiceNodeMap> services)
      : CoreComponentImpl(core::className<ControllerServiceProvider>()),
        controller_map_(std::move(services)) {
  }

  explicit ControllerServiceProviderImpl(std::string_view name, std::unique_ptr<ControllerServiceNodeMap> services)
      : CoreComponentImpl(name),
        controller_map_(std::move(services)) {
  }

  ControllerServiceProviderImpl(const ControllerServiceProviderImpl &other) = delete;
  ControllerServiceProviderImpl(ControllerServiceProviderImpl &&other) = delete;

  ControllerServiceProviderImpl& operator=(const ControllerServiceProviderImpl &other) = delete;
  ControllerServiceProviderImpl& operator=(ControllerServiceProviderImpl &&other) = delete;

  ~ControllerServiceProviderImpl() override = default;

  ControllerServiceNode* getControllerServiceNode(const std::string &id) const override {
    return controller_map_->get(id);
  }

  ControllerServiceNode* getControllerServiceNode(const std::string &id, const utils::Identifier &processor_or_controller_uuid) const override {
    return controller_map_->get(id, processor_or_controller_uuid);
  }

  void clearControllerServices() override = 0;

  std::vector<std::shared_ptr<core::controller::ControllerServiceNode>> getAllControllerServices() override {
    return controller_map_->getAllControllerServices();
  }

  std::shared_ptr<ControllerService> getControllerService(const std::string &identifier) const override;
  std::shared_ptr<ControllerService> getControllerService(const std::string &identifier, const utils::Identifier &processor_uuid) const override;

  void putControllerServiceNode(const std::string& identifier, const std::shared_ptr<ControllerServiceNode>& controller_service_node, ProcessGroup* process_group) override;

  bool isControllerServiceEnabled(const std::string &identifier) override {
    const ControllerServiceNode* const node = getControllerServiceNode(identifier);
    if (nullptr != node) {
      return linkedServicesAre(ENABLED, node);
    } else {
      return false;
    }
  }

  bool isControllerServiceEnabling(const std::string &identifier) override {
    const ControllerServiceNode* const node = getControllerServiceNode(identifier);
    if (nullptr != node) {
      return linkedServicesAre(ENABLING, node);
    } else {
      return false;
    }
  }

  const std::string getControllerServiceName(const std::string &identifier) const override {
    std::shared_ptr<ControllerService> node = getControllerService(identifier);
    if (nullptr != node) {
      return node->getName();
    } else {
      return "";
    }
  }

  bool supportsDynamicProperties() const final {
    return false;
  }

  bool supportsDynamicRelationships() const final {
    return false;
  }

 protected:
  inline bool linkedServicesAre(ControllerServiceState state, const ControllerServiceNode* node) {
    if (node->getControllerServiceImplementation()->getState() == state) {
      for (auto child_service : node->getLinkedControllerServices()) {
        if (child_service->getControllerServiceImplementation()->getState() != state) {
          return false;
        }
      }
      return true;
    } else {
      return false;
    }
  }

  bool canEdit() override {
    return true;
  }

  std::vector<core::controller::ControllerServiceNode*> findLinkedComponents(core::controller::ControllerServiceNode* referenceNode) {
    std::vector<core::controller::ControllerServiceNode*> references;

    for (auto* linked_node : referenceNode->getLinkedControllerServices()) {
      references.push_back(linked_node);
      std::vector<core::controller::ControllerServiceNode*> linked_references = findLinkedComponents(linked_node);

      auto removal_predicate = [&linked_references](core::controller::ControllerServiceNode* key) ->bool {
        return std::find(linked_references.begin(), linked_references.end(), key) != linked_references.end();
      };

      references.erase(std::remove_if(references.begin(), references.end(), removal_predicate), references.end());

      references.insert(std::end(references), linked_references.begin(), linked_references.end());
    }
    return references;
  }

  std::unique_ptr<ControllerServiceNodeMap> controller_map_;
};

}  // namespace org::apache::nifi::minifi::core::controller