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