/*
 * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.openflowplugin.impl.services.sal;

import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
import com.google.common.util.concurrent.MoreExecutors;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
import org.opendaylight.openflowplugin.impl.services.multilayer.MultiLayerMeterService;
import org.opendaylight.openflowplugin.impl.services.singlelayer.SingleLayerMeterService;
import org.opendaylight.openflowplugin.impl.util.ErrorUtil;
import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.AddMeterOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.RemoveMeterOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.SalMeterService;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.UpdateMeterOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.Meter;
import org.opendaylight.yangtools.yang.common.RpcResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class SalMeterServiceImpl implements SalMeterService {
    private static final Logger LOG = LoggerFactory.getLogger(SalMeterServiceImpl.class);
    private final MultiLayerMeterService<AddMeterInput, AddMeterOutput> addMeter;
    private final MultiLayerMeterService<Meter, UpdateMeterOutput> updateMeter;
    private final MultiLayerMeterService<RemoveMeterInput, RemoveMeterOutput> removeMeter;
    private final SingleLayerMeterService<AddMeterOutput> addMeterMessage;
    private final SingleLayerMeterService<UpdateMeterOutput> updateMeterMessage;
    private final SingleLayerMeterService<RemoveMeterOutput> removeMeterMessage;

    private final DeviceContext deviceContext;

    public SalMeterServiceImpl(final RequestContextStack requestContextStack,
                               final DeviceContext deviceContext,
                               final ConvertorExecutor convertorExecutor) {
        this.deviceContext = deviceContext;
        addMeter = new MultiLayerMeterService<>(requestContextStack,
                deviceContext,
                AddMeterOutput.class,
                convertorExecutor);

        updateMeter = new MultiLayerMeterService<>(requestContextStack,
                deviceContext,
                UpdateMeterOutput.class,
                convertorExecutor);

        removeMeter = new MultiLayerMeterService<>(requestContextStack,
                deviceContext,
                RemoveMeterOutput.class,
                convertorExecutor);

        addMeterMessage = new SingleLayerMeterService<>(requestContextStack, deviceContext, AddMeterOutput.class);
        updateMeterMessage = new SingleLayerMeterService<>(requestContextStack, deviceContext, UpdateMeterOutput.class);
        removeMeterMessage = new SingleLayerMeterService<>(requestContextStack, deviceContext, RemoveMeterOutput.class);
    }

    @Override
    public ListenableFuture<RpcResult<AddMeterOutput>> addMeter(final AddMeterInput input) {
        final ListenableFuture<RpcResult<AddMeterOutput>> resultFuture =
                addMeterMessage.canUseSingleLayerSerialization()
                        ? addMeterMessage.handleServiceCall(input)
                        : addMeter.handleServiceCall(input);

        Futures.addCallback(resultFuture, new FutureCallback<RpcResult<AddMeterOutput>>() {
            @Override
            public void onSuccess(RpcResult<AddMeterOutput> result) {
                if (result.isSuccessful()) {
                    if (LOG.isDebugEnabled()) {
                        LOG.debug("Meter add with id={} finished without error", input.getMeterId());
                    }
                } else {
                    if (LOG.isDebugEnabled()) {
                        LOG.debug("Meter add with id={} failed, errors={}", input.getMeterId(),
                                ErrorUtil.errorsToString(result.getErrors()));
                    }
                }
            }

            @Override
            public void onFailure(Throwable throwable) {
                LOG.warn("Service call for adding meter={} failed", input.getMeterId(), throwable);
            }
        }, MoreExecutors.directExecutor());
        return resultFuture;
    }

    @Override
    public ListenableFuture<RpcResult<UpdateMeterOutput>> updateMeter(final UpdateMeterInput input) {
        final ListenableFuture<RpcResult<UpdateMeterOutput>> resultFuture =
                updateMeterMessage.canUseSingleLayerSerialization()
                        ? updateMeterMessage.handleServiceCall(input.getUpdatedMeter())
                        : updateMeter.handleServiceCall(input.getUpdatedMeter());

        Futures.addCallback(resultFuture, new FutureCallback<RpcResult<UpdateMeterOutput>>() {
            @Override
            public void onSuccess(RpcResult<UpdateMeterOutput> result) {
                if (result.isSuccessful()) {
                    if (LOG.isDebugEnabled()) {
                        LOG.debug("Meter update with id={} finished without error",
                                input.getOriginalMeter().getMeterId());
                    }
                } else {
                    LOG.warn("Meter update with id={} failed, errors={}", input.getOriginalMeter().getMeterId(),
                            ErrorUtil.errorsToString(result.getErrors()));
                    LOG.debug("Meter input={}", input.getUpdatedMeter());
                }
            }

            @Override
            public void onFailure(Throwable throwable) {
                LOG.warn("Service call for updating meter={} failed",
                        input.getOriginalMeter().getMeterId(), throwable);
            }
        }, MoreExecutors.directExecutor());
        return resultFuture;
    }

    @Override
    public ListenableFuture<RpcResult<RemoveMeterOutput>> removeMeter(final RemoveMeterInput input) {
        final ListenableFuture<RpcResult<RemoveMeterOutput>> resultFuture =
                removeMeterMessage.canUseSingleLayerSerialization()
                        ? removeMeterMessage.handleServiceCall(input)
                        : removeMeter.handleServiceCall(input);

        Futures.addCallback(resultFuture, new FutureCallback<RpcResult<RemoveMeterOutput>>() {
            @Override
            public void onSuccess(RpcResult<RemoveMeterOutput> result) {
                if (result.isSuccessful()) {
                    if (LOG.isDebugEnabled()) {
                        LOG.debug("Meter remove with id={} finished without error", input.getMeterId());
                    }
                } else {
                    LOG.warn("Meter remove with id={} failed, errors={}", input.getMeterId(),
                            ErrorUtil.errorsToString(result.getErrors()));
                    LOG.debug("Meter input={}", input);
                }
            }

            @Override
            public void onFailure(Throwable throwable) {
                LOG.warn("Service call for removing meter={} failed", input.getMeterId(), throwable);
            }
        }, MoreExecutors.directExecutor());
        return resultFuture;
    }
}