tests/honeynode/netconf/src/main/java/io/fd/honeycomb/northbound/netconf/NetconfNotificationMapperProvider.java

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.honeycomb.northbound.netconf;
  18
  19 import com.google.inject.Inject;
  20 import com.google.inject.name.Named;
  21 import io.fd.honeycomb.binding.init.ProviderTrait;
  22 import io.fd.honeycomb.data.init.ShutdownHandler;
  23 import org.opendaylight.controller.config.yang.netconf.mdsal.notification.CapabilityChangeNotificationProducer;
  24 import org.opendaylight.controller.config.yang.netconf.mdsal.notification.NotificationToMdsalWriter;
  25 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  26 import org.opendaylight.controller.md.sal.binding.api.DataTreeChangeListener;
  27 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
  28 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactory;
  29 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactoryListener;
  30 import org.opendaylight.netconf.mdsal.notification.NetconfNotificationOperationServiceFactory;
  31 import org.opendaylight.netconf.notifications.NetconfNotificationCollector;
  32 import org.opendaylight.netconf.notifications.NetconfNotificationRegistry;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfState;
  34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
  35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  36 import org.slf4j.Logger;
  37 import org.slf4j.LoggerFactory;
  38
  39
  40 public class NetconfNotificationMapperProvider extends ProviderTrait<NetconfOperationServiceFactory> {
  41
  42     private static final Logger LOG = LoggerFactory.getLogger(NetconfNotificationMapperProvider.class);
  43
  44     public static final InstanceIdentifier<Capabilities> capabilitiesIdentifier =
  45             InstanceIdentifier.create(NetconfState.class).child(Capabilities.class).builder().build();
  46     @Inject
  47     private NetconfNotificationCollector notificationCollector;
  48     @Inject
  49     private NetconfNotificationRegistry notificationRegistry;
  50     @Inject
  51     @Named(NetconfModule.HONEYCOMB_NETCONF)
  52     private BindingAwareBroker bindingAwareBroker;
  53     @Inject
  54     @Named(NetconfModule.HONEYCOMB_NETCONF)
  55     private DataBroker dataBroker;
  56     @Inject
  57     private NetconfOperationServiceFactoryListener aggregator;
  58     @Inject
  59     private ShutdownHandler shutdownHandler;
  60
  61     @Override
  62     protected NetconfNotificationOperationServiceFactory create() {
  63         LOG.trace("Initializing NotificationToMdsalWriter");
  64         final NotificationToMdsalWriter writer = new NotificationToMdsalWriter(notificationCollector, dataBroker);
  65         writer.start();
  66
  67         LOG.trace("Initializing CapabilityChangeNotificationProducer");
  68         final DataTreeChangeListener<Capabilities> publisher =
  69             new CapabilityChangeNotificationProducer(notificationCollector, dataBroker);
  70
  71         LOG.trace("Providing NetconfNotificationOperationServiceFactory");
  72         NetconfNotificationOperationServiceFactory netconfNotificationOperationServiceFactory =
  73             new NetconfNotificationOperationServiceFactory(notificationRegistry, aggregator);
  74
  75         shutdownHandler.register("netconf-notification-service-factory", netconfNotificationOperationServiceFactory);
  76         shutdownHandler.register("notification-to-mdsal-writer", writer);
  77         return netconfNotificationOperationServiceFactory;
  78     }
  79 }