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.Payload;
  16 import org.opendaylight.controller.cluster.raft.protobuff.client.messages.PersistentPayload;
  17
  18 /**
  19  * The DelegatingPersistentDataProvider used by RaftActor to override the configured persistent provider to
  20  * persist ReplicatedLogEntry's based on whether or not the payload is a PersistentPayload instance.
  21  *
  22  * @author Thomas Pantelis
  23  */
  24 class RaftActorDelegatingPersistentDataProvider extends DelegatingPersistentDataProvider {
  25     private final PersistentDataProvider persistentProvider;
  26
  27     RaftActorDelegatingPersistentDataProvider(DataPersistenceProvider delegate,
  28             PersistentDataProvider persistentProvider) {
  29         super(delegate);
  30         this.persistentProvider = Preconditions.checkNotNull(persistentProvider);
  31     }
  32
  33     @Override
  34     public <T> void persist(final T object, final Procedure<T> procedure) {
  35         if (getDelegate().isRecoveryApplicable()) {
  36             super.persist(object, procedure);
  37         } else {
  38             if (object instanceof ReplicatedLogEntry) {
  39                 Payload payload = ((ReplicatedLogEntry)object).getData();
  40                 if (payload instanceof PersistentPayload) {
  41                     // We persist the Payload but not the ReplicatedLogEntry to avoid gaps in the journal indexes
  42                     // on recovery if data persistence is later enabled.
  43                     persistentProvider.persist(payload, p -> procedure.apply(object));
  44                 } else {
  45                     super.persist(object, procedure);
  46                 }
  47             } else {
  48                 super.persist(object, procedure);
  49             }
  50         }
  51     }
  52 }