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

   1 /*
   2  * Copyright (c) 2014 Cisco 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
   9 package org.opendaylight.controller.cluster.datastore;
  10
  11 import akka.actor.Props;
  12 import akka.japi.Creator;
  13
  14 import com.google.common.base.Preconditions;
  15
  16 import org.opendaylight.controller.cluster.datastore.messages.DataChanged;
  17 import org.opendaylight.controller.cluster.datastore.messages.DataChangedReply;
  18 import org.opendaylight.controller.cluster.datastore.messages.EnableNotification;
  19 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  20 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener;
  21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  22 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  23
  24 public class DataChangeListener extends AbstractUntypedActor {
  25     private final AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> listener;
  26     private volatile boolean notificationsEnabled = false;
  27
  28     public DataChangeListener(AsyncDataChangeListener<YangInstanceIdentifier,
  29                                                       NormalizedNode<?, ?>> listener) {
  30         this.listener = Preconditions.checkNotNull(listener, "listener should not be null");
  31     }
  32
  33     @Override public void handleReceive(Object message) throws Exception {
  34         if(message instanceof DataChanged){
  35             dataChanged(message);
  36         } else if(message instanceof EnableNotification){
  37             enableNotification((EnableNotification) message);
  38         }
  39     }
  40
  41     private void enableNotification(EnableNotification message) {
  42         notificationsEnabled = message.isEnabled();
  43     }
  44
  45     private void dataChanged(Object message) {
  46
  47         // Do nothing if notifications are not enabled
  48         if(!notificationsEnabled){
  49             return;
  50         }
  51
  52         DataChanged reply = (DataChanged) message;
  53         AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>>
  54             change = reply.getChange();
  55         this.listener.onDataChanged(change);
  56
  57         if(getSender() != null){
  58             getSender().tell(new DataChangedReply(), getSelf());
  59         }
  60     }
  61
  62     public static Props props(final AsyncDataChangeListener<YangInstanceIdentifier,
  63                                                             NormalizedNode<?, ?>> listener) {
  64         return Props.create(new DataChangeListenerCreator(listener));
  65     }
  66
  67     private static class DataChangeListenerCreator implements Creator<DataChangeListener> {
  68         private static final long serialVersionUID = 1L;
  69
  70         final AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> listener;
  71
  72         DataChangeListenerCreator(
  73                 AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> listener) {
  74             this.listener = listener;
  75         }
  76
  77         @Override
  78         public DataChangeListener create() throws Exception {
  79             return new DataChangeListener(listener);
  80         }
  81     }
  82 }