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 static java.util.Objects.requireNonNull;
  11
  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.mdsal.binding.api.DataBroker;
  17 import org.opendaylight.mdsal.binding.api.WriteTransaction;
  18 import org.opendaylight.mdsal.common.api.CommitInfo;
  19 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  20 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
  21 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService.CapabilitiesListener;
  22 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService.SessionsListener;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfState;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Sessions;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.sessions.Session;
  28 import org.opendaylight.yangtools.concepts.Registration;
  29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  30 import org.osgi.service.component.annotations.Activate;
  31 import org.osgi.service.component.annotations.Component;
  32 import org.osgi.service.component.annotations.Deactivate;
  33 import org.osgi.service.component.annotations.Reference;
  34 import org.slf4j.Logger;
  35 import org.slf4j.LoggerFactory;
  36
  37 /**
  38  * Writes netconf server state changes received from NetconfMonitoringService to netconf-state datastore subtree.
  39  */
  40 @Component(service = { })
  41 public final class MonitoringToMdsalWriter implements AutoCloseable, CapabilitiesListener, SessionsListener {
  42     private static final Logger LOG = LoggerFactory.getLogger(MonitoringToMdsalWriter.class);
  43
  44     private static final InstanceIdentifier<Capabilities> CAPABILITIES_INSTANCE_IDENTIFIER =
  45             InstanceIdentifier.create(NetconfState.class).child(Capabilities.class);
  46     private static final InstanceIdentifier<Schemas> SCHEMAS_INSTANCE_IDENTIFIER =
  47             InstanceIdentifier.create(NetconfState.class).child(Schemas.class);
  48     private static final InstanceIdentifier<Sessions> SESSIONS_INSTANCE_IDENTIFIER =
  49             InstanceIdentifier.create(NetconfState.class).child(Sessions.class);
  50
  51     private final DataBroker dataBroker;
  52     private final Registration capabilitiesReg;
  53     private final Registration sessionsReg;
  54
  55     @Activate
  56     public MonitoringToMdsalWriter(@Reference final DataBroker dataBroker,
  57             @Reference(target = "(type=netconf-server-monitoring)")
  58             final NetconfMonitoringService serverMonitoringDependency) {
  59         this.dataBroker = requireNonNull(dataBroker);
  60
  61         capabilitiesReg = serverMonitoringDependency.registerCapabilitiesListener(this);
  62         sessionsReg = serverMonitoringDependency.registerSessionsListener(this);
  63     }
  64
  65     @Deactivate
  66     @Override
  67     public void close() {
  68         sessionsReg.close();
  69         capabilitiesReg.close();
  70         runTransaction(tx -> tx.delete(LogicalDatastoreType.OPERATIONAL,
  71                 InstanceIdentifier.create(NetconfState.class)));
  72     }
  73
  74     @Override
  75     public void onSessionStarted(final Session session) {
  76         runTransaction(tx -> tx.put(LogicalDatastoreType.OPERATIONAL,
  77             SESSIONS_INSTANCE_IDENTIFIER.child(Session.class, session.key()), session));
  78     }
  79
  80     @Override
  81     public void onSessionEnded(final Session session) {
  82         runTransaction(tx -> tx.delete(LogicalDatastoreType.OPERATIONAL,
  83             SESSIONS_INSTANCE_IDENTIFIER.child(Session.class, session.key())));
  84     }
  85
  86     @Override
  87     public void onSessionsUpdated(final Collection<Session> sessions) {
  88         runTransaction(tx -> updateSessions(tx, sessions));
  89     }
  90
  91     @Override
  92     public void onCapabilitiesChanged(final Capabilities capabilities) {
  93         runTransaction(tx -> tx.put(LogicalDatastoreType.OPERATIONAL, CAPABILITIES_INSTANCE_IDENTIFIER,
  94                 capabilities));
  95     }
  96
  97     @Override
  98     public void onSchemasChanged(final Schemas schemas) {
  99         runTransaction(tx -> tx.put(LogicalDatastoreType.OPERATIONAL, SCHEMAS_INSTANCE_IDENTIFIER, schemas));
 100     }
 101
 102     private void runTransaction(final Consumer<WriteTransaction> txUser) {
 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             tx.put(LogicalDatastoreType.OPERATIONAL, SESSIONS_INSTANCE_IDENTIFIER.child(Session.class, session.key()),
 121                 session);
 122         }
 123     }
 124 }