diff options
-rw-r--r-- | infra/northbound/netconf/src/main/java/io/fd/honeycomb/northbound/netconf/NetconfNotificationMapperProvider.java | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/infra/northbound/netconf/src/main/java/io/fd/honeycomb/northbound/netconf/NetconfNotificationMapperProvider.java b/infra/northbound/netconf/src/main/java/io/fd/honeycomb/northbound/netconf/NetconfNotificationMapperProvider.java index 49e94bbb4..f64e4b020 100644 --- a/infra/northbound/netconf/src/main/java/io/fd/honeycomb/northbound/netconf/NetconfNotificationMapperProvider.java +++ b/infra/northbound/netconf/src/main/java/io/fd/honeycomb/northbound/netconf/NetconfNotificationMapperProvider.java @@ -64,13 +64,16 @@ public class NetconfNotificationMapperProvider extends ProviderTrait<NetconfOper writer.start(); LOG.trace("Initializing CapabilityChangeNotificationProducer"); - new CapabilityChangeNotificationProducer(notificationCollector, dataBroker); + final CapabilityChangeNotificationProducer capabilityChangeNotificationProducer = + new CapabilityChangeNotificationProducer(notificationCollector, dataBroker); LOG.trace("Providing NetconfNotificationOperationServiceFactory"); - NetconfNotificationOperationServiceFactory netconfNotificationOperationServiceFactory = + final NetconfNotificationOperationServiceFactory netconfNotificationOperationServiceFactory = new NetconfNotificationOperationServiceFactory(notificationRegistry, aggregator); shutdownHandler.register("netconf-notification-service-factory", netconfNotificationOperationServiceFactory); + shutdownHandler.register("capability-change-notification-producer", + capabilityChangeNotificationProducer::close); shutdownHandler.register("notification-to-mdsal-writer", writer); return netconfNotificationOperationServiceFactory; } |