metric_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/monitoring/model/PutMetricDataRequest.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_metrics_common/definitions/definitions.h>

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

namespace Aws {
namespace CloudWatchMetrics {

class MetricBatcher :
        public DataBatcher<MetricDatum>,
        public Aws::DataFlow::OutputStage<Aws::FileManagement::TaskPtr<MetricDatumCollection>>
{
public:

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

    MetricBatcher(const MetricBatcher & other) = delete;

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

    ~MetricBatcher() override;

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

    virtual void setMetricFileManager(std::shared_ptr<Aws::FileManagement::FileManager<MetricDatumCollection>> file_manager);

private:
    std::shared_ptr<Aws::FileManagement::FileManager<MetricDatumCollection>> metric_file_manager_;
};

}  // namespace CloudWatchMetrics
}  // namespace Aws