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 static java.util.Objects.requireNonNull;
  11
  12 import akka.japi.Procedure;
  13 import org.opendaylight.controller.cluster.DataPersistenceProvider;
  14 import org.opendaylight.controller.cluster.DelegatingPersistentDataProvider;
  15 import org.opendaylight.controller.cluster.PersistentDataProvider;
  16 import org.opendaylight.controller.cluster.raft.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(final DataPersistenceProvider delegate,
  28             final PersistentDataProvider persistentProvider) {
  29         super(delegate);
  30         this.persistentProvider = requireNonNull(persistentProvider);
  31     }
  32
  33     @Override
  34     public <T> void persist(final T entry, final Procedure<T> procedure) {
  35         doPersist(entry, procedure, false);
  36     }
  37
  38     @Override
  39     public <T> void persistAsync(final T entry, final Procedure<T> procedure) {
  40         doPersist(entry, procedure, true);
  41     }
  42
  43     private <T> void doPersist(final T entry, final Procedure<T> procedure, final boolean async) {
  44         if (!getDelegate().isRecoveryApplicable() && entry instanceof ReplicatedLogEntry replicatedLogEntry
  45             && replicatedLogEntry.getData() instanceof PersistentPayload payload) {
  46             // We persist the Payload but not the ReplicatedLogEntry to avoid gaps in the journal indexes on recovery
  47             // if data persistence is later enabled.
  48             if (async) {
  49                 persistentProvider.persistAsync(payload, p -> procedure.apply(entry));
  50             } else {
  51                 persistentProvider.persist(payload, p -> procedure.apply(entry));
  52             }
  53         } else if (async) {
  54             super.persistAsync(entry, procedure);
  55         } else {
  56             super.persist(entry, procedure);
  57         }
  58     }
  59 }