log_batcher.h

Go to the documentation of this file.

/*
 * Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file 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 <aws/core/Aws.h>
#include <aws/logs/model/InputLogEvent.h>
#include <cloudwatch_logs_common/definitions/ros_cloudwatch_logs_errors.h>

#include <file_management/file_upload/file_upload_streamer.h>
#include <file_management/file_upload/file_manager.h>

#include <dataflow_lite/utils/data_batcher.h>

#include <cloudwatch_logs_common/definitions/definitions.h>

#include <chrono>
#include <list>
#include <memory>

namespace Aws {
namespace CloudWatchLogs {

class LogBatcher :
  public Aws::DataFlow::OutputStage<Aws::FileManagement::TaskPtr<LogCollection>>,
  public DataBatcher<LogType>
{
public:


  explicit LogBatcher(size_t max_allowable_batch_size = DataBatcher::kDefaultMaxBatchSize,
                      size_t publish_trigger_size = DataBatcher::kDefaultTriggerSize);

  LogBatcher(const LogBatcher & other) = delete;

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

  ~LogBatcher() override;

  bool publishBatchedData() override;

  void emptyCollection() override;
  bool start() override;

  virtual void setLogFileManager(std::shared_ptr<Aws::FileManagement::FileManager<LogCollection>> log_file_manager);

private:
  std::shared_ptr<Aws::FileManagement::FileManager<LogCollection>> log_file_manager_;
};

}  // namespace CloudWatchLogs
}  // namespace Aws