/**
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0.
 */

#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/BatchRebootClusterNodeLogicalIdsError.h>
#include <aws/sagemaker/model/BatchRebootClusterNodesError.h>

#include <utility>

namespace Aws {
template <typename RESULT_TYPE>
class AmazonWebServiceResult;

namespace Utils {
namespace Json {
class JsonValue;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
class BatchRebootClusterNodesResult {
 public:
  AWS_SAGEMAKER_API BatchRebootClusterNodesResult() = default;
  AWS_SAGEMAKER_API BatchRebootClusterNodesResult(const Aws::AmazonWebServiceResult<Aws::Utils::Json::JsonValue>& result);
  AWS_SAGEMAKER_API BatchRebootClusterNodesResult& operator=(const Aws::AmazonWebServiceResult<Aws::Utils::Json::JsonValue>& result);

  ///@{
  /**
   * <p>A list of EC2 instance IDs for which the reboot operation was successfully
   * initiated.</p>
   */
  inline const Aws::Vector<Aws::String>& GetSuccessful() const { return m_successful; }
  template <typename SuccessfulT = Aws::Vector<Aws::String>>
  void SetSuccessful(SuccessfulT&& value) {
    m_successfulHasBeenSet = true;
    m_successful = std::forward<SuccessfulT>(value);
  }
  template <typename SuccessfulT = Aws::Vector<Aws::String>>
  BatchRebootClusterNodesResult& WithSuccessful(SuccessfulT&& value) {
    SetSuccessful(std::forward<SuccessfulT>(value));
    return *this;
  }
  template <typename SuccessfulT = Aws::String>
  BatchRebootClusterNodesResult& AddSuccessful(SuccessfulT&& value) {
    m_successfulHasBeenSet = true;
    m_successful.emplace_back(std::forward<SuccessfulT>(value));
    return *this;
  }
  ///@}

  ///@{
  /**
   * <p>A list of errors encountered for EC2 instance IDs that could not be rebooted.
   * Each error includes the instance ID, an error code, and a descriptive
   * message.</p>
   */
  inline const Aws::Vector<BatchRebootClusterNodesError>& GetFailed() const { return m_failed; }
  template <typename FailedT = Aws::Vector<BatchRebootClusterNodesError>>
  void SetFailed(FailedT&& value) {
    m_failedHasBeenSet = true;
    m_failed = std::forward<FailedT>(value);
  }
  template <typename FailedT = Aws::Vector<BatchRebootClusterNodesError>>
  BatchRebootClusterNodesResult& WithFailed(FailedT&& value) {
    SetFailed(std::forward<FailedT>(value));
    return *this;
  }
  template <typename FailedT = BatchRebootClusterNodesError>
  BatchRebootClusterNodesResult& AddFailed(FailedT&& value) {
    m_failedHasBeenSet = true;
    m_failed.emplace_back(std::forward<FailedT>(value));
    return *this;
  }
  ///@}

  ///@{
  /**
   * <p>A list of errors encountered for logical node IDs that could not be rebooted.
   * Each error includes the logical node ID, an error code, and a descriptive
   * message. This field is only present when <code>NodeLogicalIds</code> were
   * provided in the request.</p>
   */
  inline const Aws::Vector<BatchRebootClusterNodeLogicalIdsError>& GetFailedNodeLogicalIds() const { return m_failedNodeLogicalIds; }
  template <typename FailedNodeLogicalIdsT = Aws::Vector<BatchRebootClusterNodeLogicalIdsError>>
  void SetFailedNodeLogicalIds(FailedNodeLogicalIdsT&& value) {
    m_failedNodeLogicalIdsHasBeenSet = true;
    m_failedNodeLogicalIds = std::forward<FailedNodeLogicalIdsT>(value);
  }
  template <typename FailedNodeLogicalIdsT = Aws::Vector<BatchRebootClusterNodeLogicalIdsError>>
  BatchRebootClusterNodesResult& WithFailedNodeLogicalIds(FailedNodeLogicalIdsT&& value) {
    SetFailedNodeLogicalIds(std::forward<FailedNodeLogicalIdsT>(value));
    return *this;
  }
  template <typename FailedNodeLogicalIdsT = BatchRebootClusterNodeLogicalIdsError>
  BatchRebootClusterNodesResult& AddFailedNodeLogicalIds(FailedNodeLogicalIdsT&& value) {
    m_failedNodeLogicalIdsHasBeenSet = true;
    m_failedNodeLogicalIds.emplace_back(std::forward<FailedNodeLogicalIdsT>(value));
    return *this;
  }
  ///@}

  ///@{
  /**
   * <p>A list of logical node IDs for which the reboot operation was successfully
   * initiated. This field is only present when <code>NodeLogicalIds</code> were
   * provided in the request.</p>
   */
  inline const Aws::Vector<Aws::String>& GetSuccessfulNodeLogicalIds() const { return m_successfulNodeLogicalIds; }
  template <typename SuccessfulNodeLogicalIdsT = Aws::Vector<Aws::String>>
  void SetSuccessfulNodeLogicalIds(SuccessfulNodeLogicalIdsT&& value) {
    m_successfulNodeLogicalIdsHasBeenSet = true;
    m_successfulNodeLogicalIds = std::forward<SuccessfulNodeLogicalIdsT>(value);
  }
  template <typename SuccessfulNodeLogicalIdsT = Aws::Vector<Aws::String>>
  BatchRebootClusterNodesResult& WithSuccessfulNodeLogicalIds(SuccessfulNodeLogicalIdsT&& value) {
    SetSuccessfulNodeLogicalIds(std::forward<SuccessfulNodeLogicalIdsT>(value));
    return *this;
  }
  template <typename SuccessfulNodeLogicalIdsT = Aws::String>
  BatchRebootClusterNodesResult& AddSuccessfulNodeLogicalIds(SuccessfulNodeLogicalIdsT&& value) {
    m_successfulNodeLogicalIdsHasBeenSet = true;
    m_successfulNodeLogicalIds.emplace_back(std::forward<SuccessfulNodeLogicalIdsT>(value));
    return *this;
  }
  ///@}

  ///@{

  inline const Aws::String& GetRequestId() const { return m_requestId; }
  template <typename RequestIdT = Aws::String>
  void SetRequestId(RequestIdT&& value) {
    m_requestIdHasBeenSet = true;
    m_requestId = std::forward<RequestIdT>(value);
  }
  template <typename RequestIdT = Aws::String>
  BatchRebootClusterNodesResult& WithRequestId(RequestIdT&& value) {
    SetRequestId(std::forward<RequestIdT>(value));
    return *this;
  }
  ///@}
 private:
  Aws::Vector<Aws::String> m_successful;

  Aws::Vector<BatchRebootClusterNodesError> m_failed;

  Aws::Vector<BatchRebootClusterNodeLogicalIdsError> m_failedNodeLogicalIds;

  Aws::Vector<Aws::String> m_successfulNodeLogicalIds;

  Aws::String m_requestId;
  bool m_successfulHasBeenSet = false;
  bool m_failedHasBeenSet = false;
  bool m_failedNodeLogicalIdsHasBeenSet = false;
  bool m_successfulNodeLogicalIdsHasBeenSet = false;
  bool m_requestIdHasBeenSet = false;
};

}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws
