QueryLineageRequest.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/sagemaker/SageMakerRequest.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/Direction.h>
#include <aws/sagemaker/model/QueryFilters.h>
 
#include <utility>
 
namespace Aws {
namespace SageMaker {
namespace Model {
 
class QueryLineageRequest : public SageMakerRequest {
 public:
  AWS_SAGEMAKER_API QueryLineageRequest() = default;
 
  // Service request name is the Operation name which will send this request out,
  // each operation should has unique request name, so that we can get operation's name from this request.
  // Note: this is not true for response, multiple operations may have the same response name,
  // so we can not get operation's name from response.
  inline virtual const char* GetServiceRequestName() const override { return "QueryLineage"; }
 
  AWS_SAGEMAKER_API Aws::String SerializePayload() const override;
 
  AWS_SAGEMAKER_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
 
  inline const Aws::Vector<Aws::String>& GetStartArns() const { return m_startArns; }
  inline bool StartArnsHasBeenSet() const { return m_startArnsHasBeenSet; }
  template <typename StartArnsT = Aws::Vector<Aws::String>>
  void SetStartArns(StartArnsT&& value) {
    m_startArnsHasBeenSet = true;
    m_startArns = std::forward<StartArnsT>(value);
  }
  template <typename StartArnsT = Aws::Vector<Aws::String>>
  QueryLineageRequest& WithStartArns(StartArnsT&& value) {
    SetStartArns(std::forward<StartArnsT>(value));
    return *this;
  }
  template <typename StartArnsT = Aws::String>
  QueryLineageRequest& AddStartArns(StartArnsT&& value) {
    m_startArnsHasBeenSet = true;
    m_startArns.emplace_back(std::forward<StartArnsT>(value));
    return *this;
  }
 
 
  inline Direction GetDirection() const { return m_direction; }
  inline bool DirectionHasBeenSet() const { return m_directionHasBeenSet; }
  inline void SetDirection(Direction value) {
    m_directionHasBeenSet = true;
    m_direction = value;
  }
  inline QueryLineageRequest& WithDirection(Direction value) {
    SetDirection(value);
    return *this;
  }
 
 
  inline bool GetIncludeEdges() const { return m_includeEdges; }
  inline bool IncludeEdgesHasBeenSet() const { return m_includeEdgesHasBeenSet; }
  inline void SetIncludeEdges(bool value) {
    m_includeEdgesHasBeenSet = true;
    m_includeEdges = value;
  }
  inline QueryLineageRequest& WithIncludeEdges(bool value) {
    SetIncludeEdges(value);
    return *this;
  }
 
 
  inline const QueryFilters& GetFilters() const { return m_filters; }
  inline bool FiltersHasBeenSet() const { return m_filtersHasBeenSet; }
  template <typename FiltersT = QueryFilters>
  void SetFilters(FiltersT&& value) {
    m_filtersHasBeenSet = true;
    m_filters = std::forward<FiltersT>(value);
  }
  template <typename FiltersT = QueryFilters>
  QueryLineageRequest& WithFilters(FiltersT&& value) {
    SetFilters(std::forward<FiltersT>(value));
    return *this;
  }
 
 
  inline int GetMaxDepth() const { return m_maxDepth; }
  inline bool MaxDepthHasBeenSet() const { return m_maxDepthHasBeenSet; }
  inline void SetMaxDepth(int value) {
    m_maxDepthHasBeenSet = true;
    m_maxDepth = value;
  }
  inline QueryLineageRequest& WithMaxDepth(int value) {
    SetMaxDepth(value);
    return *this;
  }
 
 
  inline int GetMaxResults() const { return m_maxResults; }
  inline bool MaxResultsHasBeenSet() const { return m_maxResultsHasBeenSet; }
  inline void SetMaxResults(int value) {
    m_maxResultsHasBeenSet = true;
    m_maxResults = value;
  }
  inline QueryLineageRequest& WithMaxResults(int value) {
    SetMaxResults(value);
    return *this;
  }
 
 
  inline const Aws::String& GetNextToken() const { return m_nextToken; }
  inline bool NextTokenHasBeenSet() const { return m_nextTokenHasBeenSet; }
  template <typename NextTokenT = Aws::String>
  void SetNextToken(NextTokenT&& value) {
    m_nextTokenHasBeenSet = true;
    m_nextToken = std::forward<NextTokenT>(value);
  }
  template <typename NextTokenT = Aws::String>
  QueryLineageRequest& WithNextToken(NextTokenT&& value) {
    SetNextToken(std::forward<NextTokenT>(value));
    return *this;
  }
 private:
  Aws::Vector<Aws::String> m_startArns;
 
  Direction m_direction{Direction::NOT_SET};
 
  bool m_includeEdges{false};
 
  QueryFilters m_filters;
 
  int m_maxDepth{0};
 
  int m_maxResults{0};
 
  Aws::String m_nextToken;
  bool m_startArnsHasBeenSet = false;
  bool m_directionHasBeenSet = false;
  bool m_includeEdgesHasBeenSet = false;
  bool m_filtersHasBeenSet = false;
  bool m_maxDepthHasBeenSet = false;
  bool m_maxResultsHasBeenSet = false;
  bool m_nextTokenHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws