text: /**
   *
   * 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.
   */
  #ifndef LIBMINIFI_INCLUDE_C2_CONTROLLERSOCKETPROTOCOL_H_
  #define LIBMINIFI_INCLUDE_C2_CONTROLLERSOCKETPROTOCOL_H_
  
  #include "core/Resource.h"
  #include "HeartBeatReporter.h"
  #include "io/StreamFactory.h"
  #include "core/logging/LoggerConfiguration.h"
  
  namespace org {
  namespace apache {
  namespace nifi {
  namespace minifi {
  namespace c2 {
  
  /**
   * Purpose: Creates a reporter that can handle basic c2 operations for a localized environment
   * through a simple TCP socket.
   */
  class ControllerSocketProtocol : public HeartBeatReporter {
   public:
  
    ControllerSocketProtocol(std::string name, utils::Identifier uuid = utils::Identifier())
        : HeartBeatReporter(name, uuid),
          logger_(logging::LoggerFactory<ControllerSocketProtocol>::getLogger()) {
  
    }
  
    /**
     * Initialize the socket protocol.
     * @param controller controller service provider.
     * @param updateSink update mechanism that will be used to stop/clear elements
     * @param configuration configuration class.
     */
    virtual void initialize(const std::shared_ptr<core::controller::ControllerServiceProvider> &controller, const std::shared_ptr<state::StateMonitor> &updateSink,
                            const std::shared_ptr<Configure> &configuration);
  
    /**
     * Handles the heartbeat
     * @param payload incoming payload. From this function we only care about queue metrics.
     */
    virtual int16_t heartbeat(const C2Payload &payload);
  
   protected:
  
    /**
     * Parses content from the content response.
     */
    void parse_content(const std::vector<C2ContentResponse> &content);
  
    std::mutex controller_mutex_;
  
    std::map<std::string, bool> queue_full_;
  
    std::map<std::string, uint64_t> queue_size_;
  
    std::map<std::string, uint64_t> queue_max_;
  
    std::map<std::string, bool> component_map_;
  
    std::unique_ptr<io::BaseServerSocket> server_socket_;
  
    std::shared_ptr<minifi::io::StreamFactory> stream_factory_;
  
   private:
  
    std::shared_ptr<logging::Logger> logger_;
  };
  
  REGISTER_RESOURCE(ControllerSocketProtocol, "Creates a reporter that can handle basic c2 operations for a localized environment through a simple TCP socket.");
  
  } /* namesapce c2 */
  } /* namespace minifi */
  } /* namespace nifi */
  } /* namespace apache */
  } /* namespace org */
  
  #endif /* LIBMINIFI_INCLUDE_C2_CONTROLLERSOCKETPROTOCOL_H_ */
  