/*
* Copyright 2010-2019 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.
*/
package software.amazon.cloudformation.proxy;

import java.time.Instant;
import java.util.ArrayList;
import java.util.List;
import software.amazon.cloudformation.Action;
import software.amazon.cloudformation.HookInvocationPoint;
import software.amazon.cloudformation.metrics.MetricsPublisher;

public class MetricsPublisherProxy {
    private final List<MetricsPublisher> metricsPublishers = new ArrayList<>();

    public void addMetricsPublisher(final MetricsPublisher metricsPublisher) {
        metricsPublishers.add(metricsPublisher);
    }

    public void publishExceptionMetric(final Instant timestamp,
                                       final Action action,
                                       final Throwable e,
                                       final HandlerErrorCode handlerErrorCode) {
        metricsPublishers.stream()
            .forEach(metricsPublisher -> metricsPublisher.publishExceptionMetric(timestamp, action, e, handlerErrorCode));
    }

    public void publishExceptionMetric(final Instant timestamp,
                                       final HookInvocationPoint invocationPoint,
                                       final Throwable e,
                                       final HandlerErrorCode handlerErrorCode) {
        metricsPublishers.stream().forEach(
            metricsPublisher -> metricsPublisher.publishExceptionMetric(timestamp, invocationPoint, e, handlerErrorCode));
    }

    public void publishExceptionByErrorCodeAndCountBulkMetrics(final Instant timestamp,
                                                               final Action action,
                                                               final HandlerErrorCode handlerErrorCode) {
        metricsPublishers.stream().forEach(metricsPublisher -> metricsPublisher
            .publishExceptionByErrorCodeAndCountBulkMetrics(timestamp, action, handlerErrorCode));
    }

    public void publishExceptionByErrorCodeAndCountBulkMetrics(final Instant timestamp,
                                                               final HookInvocationPoint invocationPoint,
                                                               final HandlerErrorCode handlerErrorCode) {
        metricsPublishers.stream().forEach(metricsPublishers -> metricsPublishers
            .publishExceptionByErrorCodeAndCountBulkMetrics(timestamp, invocationPoint, handlerErrorCode));
    }

    public void publishInvocationMetric(final Instant timestamp, final Action action) {
        metricsPublishers.stream().forEach(metricsPublisher -> metricsPublisher.publishInvocationMetric(timestamp, action));
    }

    public void publishInvocationMetric(final Instant timestamp, final HookInvocationPoint invocationPoint) {
        metricsPublishers.stream()
            .forEach(metricsPublisher -> metricsPublisher.publishInvocationMetric(timestamp, invocationPoint));
    }

    public void publishDurationMetric(final Instant timestamp, final Action action, final long milliseconds) {
        metricsPublishers.stream()
            .forEach(metricsPublisher -> metricsPublisher.publishDurationMetric(timestamp, action, milliseconds));
    }

    public void
        publishDurationMetric(final Instant timestamp, final HookInvocationPoint invocationPoint, final long milliseconds) {
        metricsPublishers.stream()
            .forEach(metricsPublisher -> metricsPublisher.publishDurationMetric(timestamp, invocationPoint, milliseconds));
    }

    public void publishProviderLogDeliveryExceptionMetric(final Instant timestamp, final Throwable exception) {
        metricsPublishers.stream()
            .forEach(metricsPublisher -> metricsPublisher.publishProviderLogDeliveryExceptionMetric(timestamp, exception));
    }
}