netconf/mdsal-netconf-monitoring/src/main/java/org/opendaylight/controller/config/yang/netconf/mdsal/monitoring/MonitoringToMdsalWriter.java

   1 /*
   2  * Copyright (c) 2015 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.config.yang.netconf.mdsal.monitoring;
  10
  11 import com.google.common.base.Preconditions;
  12 import com.google.common.util.concurrent.FutureCallback;
  13 import com.google.common.util.concurrent.Futures;
  14 import com.google.common.util.concurrent.MoreExecutors;
  15 import java.util.Collection;
  16 import java.util.function.Consumer;
  17 import javax.annotation.Nullable;
  18 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  19 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  20 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  21 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfState;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.Session;
  27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  28 import org.slf4j.Logger;
  29 import org.slf4j.LoggerFactory;
  30
  31 /**
  32  * Writes netconf server state changes received from NetconfMonitoringService to netconf-state datastore subtree.
  33  */
  34 public final class MonitoringToMdsalWriter implements AutoCloseable, NetconfMonitoringService.CapabilitiesListener,
  35         NetconfMonitoringService.SessionsListener {
  36
  37     private static final Logger LOG = LoggerFactory.getLogger(MonitoringToMdsalWriter.class);
  38
  39     private static final InstanceIdentifier<Capabilities> CAPABILITIES_INSTANCE_IDENTIFIER =
  40             InstanceIdentifier.create(NetconfState.class).child(Capabilities.class);
  41     private static final InstanceIdentifier<Schemas> SCHEMAS_INSTANCE_IDENTIFIER =
  42             InstanceIdentifier.create(NetconfState.class).child(Schemas.class);
  43     private static final InstanceIdentifier<Sessions> SESSIONS_INSTANCE_IDENTIFIER =
  44             InstanceIdentifier.create(NetconfState.class).child(Sessions.class);
  45
  46     private final NetconfMonitoringService serverMonitoringDependency;
  47     private final DataBroker dataBroker;
  48
  49     public MonitoringToMdsalWriter(final NetconfMonitoringService serverMonitoringDependency,
  50                                    final DataBroker dataBroker) {
  51         this.serverMonitoringDependency = serverMonitoringDependency;
  52         this.dataBroker = dataBroker;
  53     }
  54
  55     /**
  56      * Invoked using blueprint.
  57      */
  58     @Override
  59     public void close() {
  60         runTransaction((tx) -> tx.delete(LogicalDatastoreType.OPERATIONAL,
  61                 InstanceIdentifier.create(NetconfState.class)));
  62     }
  63
  64     @Override
  65     public void onSessionStarted(final Session session) {
  66         final InstanceIdentifier<Session> sessionPath =
  67                 SESSIONS_INSTANCE_IDENTIFIER.child(Session.class, session.key());
  68         runTransaction((tx) -> tx.put(LogicalDatastoreType.OPERATIONAL, sessionPath, session));
  69     }
  70
  71     @Override
  72     public void onSessionEnded(final Session session) {
  73         final InstanceIdentifier<Session> sessionPath =
  74                 SESSIONS_INSTANCE_IDENTIFIER.child(Session.class, session.key());
  75         runTransaction((tx) -> tx.delete(LogicalDatastoreType.OPERATIONAL, sessionPath));
  76     }
  77
  78     @Override
  79     public void onSessionsUpdated(final Collection<Session> sessions) {
  80         runTransaction((tx) -> updateSessions(tx, sessions));
  81     }
  82
  83     @Override
  84     public void onCapabilitiesChanged(final Capabilities capabilities) {
  85         runTransaction((tx) -> tx.put(LogicalDatastoreType.OPERATIONAL, CAPABILITIES_INSTANCE_IDENTIFIER,
  86                 capabilities));
  87     }
  88
  89     @Override
  90     public void onSchemasChanged(final Schemas schemas) {
  91         runTransaction((tx) -> tx.put(LogicalDatastoreType.OPERATIONAL, SCHEMAS_INSTANCE_IDENTIFIER, schemas));
  92     }
  93
  94     /**
  95      * Invoked using blueprint.
  96      */
  97     public void start() {
  98         serverMonitoringDependency.registerCapabilitiesListener(this);
  99         serverMonitoringDependency.registerSessionsListener(this);
 100     }
 101
 102     private void runTransaction(final Consumer<WriteTransaction> txUser) {
 103         Preconditions.checkState(dataBroker != null);
 104         final WriteTransaction tx = dataBroker.newWriteOnlyTransaction();
 105         txUser.accept(tx);
 106         Futures.addCallback(tx.submit(), new FutureCallback<Void>() {
 107             @Override
 108             public void onSuccess(@Nullable final Void result) {
 109                 LOG.debug("Netconf state updated successfully");
 110             }
 111
 112             @Override
 113             public void onFailure(final Throwable throwable) {
 114                 LOG.warn("Unable to update netconf state", throwable);
 115             }
 116         }, MoreExecutors.directExecutor());
 117     }
 118
 119     private static void updateSessions(final WriteTransaction tx, final Collection<Session> sessions) {
 120         for (Session session : sessions) {
 121             final InstanceIdentifier<Session> sessionPath =
 122                     SESSIONS_INSTANCE_IDENTIFIER.child(Session.class, session.key());
 123             tx.put(LogicalDatastoreType.OPERATIONAL, sessionPath, session);
 124         }
 125     }
 126 }