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