opendaylight/md-sal/sal-akka-raft/src/main/java/org/opendaylight/controller/cluster/raft/RaftActorDelegatingPersistentDataProvider.java

   1 /*
   2  * Copyright (c) 2015 Brocade Communications Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.cluster.raft;
   9
  10 import akka.japi.Procedure;
  11 import com.google.common.base.Preconditions;
  12 import org.opendaylight.controller.cluster.DataPersistenceProvider;
  13 import org.opendaylight.controller.cluster.DelegatingPersistentDataProvider;
  14 import org.opendaylight.controller.cluster.PersistentDataProvider;
  15 import org.opendaylight.controller.cluster.raft.protobuff.client.messages.PersistentPayload;
  16
  17 /**
  18  * The DelegatingPersistentDataProvider used by RaftActor to override the configured persistent provider to
  19  * persist ReplicatedLogEntry's based on whether or not the payload is a PersistentPayload instance.
  20  *
  21  * @author Thomas Pantelis
  22  */
  23 class RaftActorDelegatingPersistentDataProvider extends DelegatingPersistentDataProvider {
  24     private final PersistentDataProvider persistentProvider;
  25
  26     RaftActorDelegatingPersistentDataProvider(DataPersistenceProvider delegate,
  27             PersistentDataProvider persistentProvider) {
  28         super(delegate);
  29         this.persistentProvider = Preconditions.checkNotNull(persistentProvider);
  30     }
  31
  32     @Override
  33     public <T> void persist(T o, Procedure<T> procedure) {
  34         if(getDelegate().isRecoveryApplicable()) {
  35             super.persist(o, procedure);
  36         } else {
  37             boolean isPersistentPayload = false;
  38             if(o instanceof ReplicatedLogEntry) {
  39                 isPersistentPayload = ((ReplicatedLogEntry)o).getData() instanceof PersistentPayload;
  40             }
  41
  42             if(isPersistentPayload) {
  43                 persistentProvider.persist(o, procedure);
  44             } else {
  45                 super.persist(o, procedure);
  46             }
  47         }
  48     }
  49 }