/*
 * Copyright (c) 2015 Brocade Communications 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.controller.cluster.raft;

import akka.japi.Procedure;
import com.google.common.base.Preconditions;
import org.opendaylight.controller.cluster.DataPersistenceProvider;
import org.opendaylight.controller.cluster.DelegatingPersistentDataProvider;
import org.opendaylight.controller.cluster.PersistentDataProvider;
import org.opendaylight.controller.cluster.raft.protobuff.client.messages.PersistentPayload;

/**
 * The DelegatingPersistentDataProvider used by RaftActor to override the configured persistent provider to
 * persist ReplicatedLogEntry's based on whether or not the payload is a PersistentPayload instance.
 *
 * @author Thomas Pantelis
 */
class RaftActorDelegatingPersistentDataProvider extends DelegatingPersistentDataProvider {
    private final PersistentDataProvider persistentProvider;

    RaftActorDelegatingPersistentDataProvider(DataPersistenceProvider delegate,
            PersistentDataProvider persistentProvider) {
        super(delegate);
        this.persistentProvider = Preconditions.checkNotNull(persistentProvider);
    }

    @Override
    public <T> void persist(T o, Procedure<T> procedure) {
        if(getDelegate().isRecoveryApplicable()) {
            super.persist(o, procedure);
        } else {
            boolean isPersistentPayload = false;
            if(o instanceof ReplicatedLogEntry) {
                isPersistentPayload = ((ReplicatedLogEntry)o).getData() instanceof PersistentPayload;
            }

            if(isPersistentPayload) {
                persistentProvider.persist(o, procedure);
            } else {
                super.persist(o, procedure);
            }
        }
    }
}