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 javax.annotation.Nullable;
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  18 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
  19 import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
  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.DataObject;
  27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  28 import org.slf4j.Logger;
  29 import org.slf4j.LoggerFactory;
  30
  31 final class MonitoringToMdsalWriter implements AutoCloseable, NetconfMonitoringService.MonitoringListener, BindingAwareProvider {
  32
  33     private static final Logger LOG = LoggerFactory.getLogger(MonitoringToMdsalWriter.class);
  34
  35     private static final InstanceIdentifier<Capabilities> CAPABILITIES_INSTANCE_IDENTIFIER =
  36             InstanceIdentifier.create(NetconfState.class).child(Capabilities.class);
  37     private static final InstanceIdentifier<Schemas> SCHEMAS_INSTANCE_IDENTIFIER =
  38             InstanceIdentifier.create(NetconfState.class).child(Schemas.class);
  39     private static final InstanceIdentifier<Sessions> SESSIONS_INSTANCE_IDENTIFIER =
  40             InstanceIdentifier.create(NetconfState.class).child(Sessions.class);
  41
  42     private final NetconfMonitoringService serverMonitoringDependency;
  43     private DataBroker dataBroker;
  44
  45     public MonitoringToMdsalWriter(final NetconfMonitoringService serverMonitoringDependency) {
  46         this.serverMonitoringDependency = serverMonitoringDependency;
  47     }
  48
  49     @Override
  50     public void close() {
  51         deleteFromDatastore(InstanceIdentifier.create(NetconfState.class));
  52     }
  53
  54     @Override
  55     public void onSessionStarted(Session session) {
  56         final InstanceIdentifier<Session> sessionPath =
  57                 SESSIONS_INSTANCE_IDENTIFIER.child(Session.class, session.getKey());
  58         putToDatastore(sessionPath, session);
  59     }
  60
  61     @Override
  62     public void onSessionEnded(Session session) {
  63         final InstanceIdentifier<Session> sessionPath =
  64                 SESSIONS_INSTANCE_IDENTIFIER.child(Session.class, session.getKey());
  65         deleteFromDatastore(sessionPath);
  66     }
  67
  68     @Override
  69     public void onCapabilitiesChanged(Capabilities capabilities) {
  70         putToDatastore(CAPABILITIES_INSTANCE_IDENTIFIER, capabilities);
  71     }
  72
  73     @Override
  74     public void onSchemasChanged(Schemas schemas) {
  75         putToDatastore(SCHEMAS_INSTANCE_IDENTIFIER, schemas);
  76     }
  77
  78     @Override
  79     public void onSessionInitiated(final BindingAwareBroker.ProviderContext providerContext) {
  80         dataBroker = providerContext.getSALService(DataBroker.class);
  81         serverMonitoringDependency.registerListener(this);
  82     }
  83
  84     private <T extends DataObject> void putToDatastore(InstanceIdentifier<T> path, T value) {
  85         Preconditions.checkState(dataBroker != null);
  86         final WriteTransaction tx = dataBroker.newWriteOnlyTransaction();
  87         tx.put(LogicalDatastoreType.OPERATIONAL, path, value);
  88         Futures.addCallback(tx.submit(), new FutureCallback<Void>() {
  89             @Override
  90             public void onSuccess(@Nullable Void result) {
  91                 LOG.debug("Netconf state updated successfully");
  92             }
  93
  94             @Override
  95             public void onFailure(Throwable t) {
  96                 LOG.warn("Unable to update netconf state", t);
  97             }
  98         });
  99     }
 100
 101     private <T extends DataObject> void deleteFromDatastore(InstanceIdentifier<T> path) {
 102         Preconditions.checkState(dataBroker != null);
 103         final WriteTransaction tx = dataBroker.newWriteOnlyTransaction();
 104         tx.delete(LogicalDatastoreType.OPERATIONAL, path);
 105         Futures.addCallback(tx.submit(), new FutureCallback<Void>() {
 106             @Override
 107             public void onSuccess(@Nullable Void result) {
 108                 LOG.debug("Netconf state updated successfully");
 109             }
 110
 111             @Override
 112             public void onFailure(Throwable t) {
 113                 LOG.warn("Unable to update netconf state", t);
 114             }
 115         });
 116     }
 117 }