opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/ShardDataTreeChangePublisherActor.java

   1 /*
   2  * Copyright (c) 2017 Inocybe Technologies 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.datastore;
   9
  10 import static java.util.Objects.requireNonNull;
  11
  12 import akka.actor.Props;
  13 import java.util.Optional;
  14 import java.util.function.Consumer;
  15 import org.opendaylight.mdsal.dom.api.DOMDataTreeChangeListener;
  16 import org.opendaylight.yangtools.concepts.Registration;
  17 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  18 import org.opendaylight.yangtools.yang.data.tree.api.DataTreeCandidate;
  19
  20 /**
  21  * Actor used to generate and publish DataTreeChange notifications.
  22  *
  23  * @author Thomas Pantelis
  24  */
  25 public final class ShardDataTreeChangePublisherActor
  26         extends ShardDataTreeNotificationPublisherActor<DefaultShardDataTreeChangeListenerPublisher> {
  27
  28     private ShardDataTreeChangePublisherActor(final String name, final String logContext) {
  29         super(new DefaultShardDataTreeChangeListenerPublisher(logContext), name, logContext);
  30     }
  31
  32     @Override
  33     protected void handleReceive(final Object message) {
  34         if (message instanceof RegisterListener reg) {
  35             LOG.debug("{}: Received {}", logContext(), reg);
  36             if (reg.initialState.isPresent()) {
  37                 DefaultShardDataTreeChangeListenerPublisher.notifySingleListener(reg.path, reg.listener,
  38                         reg.initialState.orElseThrow(), logContext());
  39             } else {
  40                 reg.listener.onInitialData();
  41             }
  42
  43             publisher().registerTreeChangeListener(reg.path, reg.listener, reg.onRegistration);
  44         } else {
  45             super.handleReceive(message);
  46         }
  47     }
  48
  49     static Props props(final String name, final String logContext) {
  50         return Props.create(ShardDataTreeChangePublisherActor.class, name, logContext);
  51     }
  52
  53     static class RegisterListener {
  54         private final YangInstanceIdentifier path;
  55         private final DOMDataTreeChangeListener listener;
  56         private final Optional<DataTreeCandidate> initialState;
  57         private final Consumer<Registration> onRegistration;
  58
  59         RegisterListener(final YangInstanceIdentifier path, final DOMDataTreeChangeListener listener,
  60                 final Optional<DataTreeCandidate> initialState, final Consumer<Registration> onRegistration) {
  61             this.path = requireNonNull(path);
  62             this.listener = requireNonNull(listener);
  63             this.initialState = requireNonNull(initialState);
  64             this.onRegistration = requireNonNull(onRegistration);
  65         }
  66
  67         @Override
  68         public String toString() {
  69             return "RegisterListener [path=" + path + ", listener=" + listener + ", initialState present="
  70                     + initialState.isPresent() + "]";
  71         }
  72     }
  73 }