/*
* Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
package org.opendaylight.controller.config.persist.impl;
import com.google.common.base.Optional;
import java.io.Closeable;
import java.io.IOException;
import java.util.Set;
import javax.annotation.concurrent.ThreadSafe;
import javax.management.InstanceNotFoundException;
import javax.management.MBeanServerConnection;
import javax.management.Notification;
import javax.management.NotificationListener;
import javax.management.ObjectName;
import org.opendaylight.controller.config.api.jmx.notifications.CommitJMXNotification;
import org.opendaylight.controller.config.api.jmx.notifications.ConfigJMXNotification;
import org.opendaylight.controller.config.facade.xml.ConfigSubsystemFacadeFactory;
import org.opendaylight.controller.config.facade.xml.Datastore;
import org.opendaylight.controller.config.persist.api.Persister;
import org.opendaylight.controller.config.util.capability.Capability;
import org.opendaylight.controller.config.util.xml.XmlUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Element;
/**
* Responsible for listening for notifications from netconf (via JMX) containing latest
* committed configuration that should be persisted, and also for loading last
* configuration.
*/
@ThreadSafe
public class ConfigPersisterNotificationHandler implements Closeable {
private static final Logger LOG = LoggerFactory.getLogger(ConfigPersisterNotificationHandler.class);
private final MBeanServerConnection mBeanServerConnection;
private final NotificationListener listener;
public ConfigPersisterNotificationHandler(final MBeanServerConnection mBeanServerConnection, final Persister persisterAggregator, final ConfigSubsystemFacadeFactory facade) {
this(mBeanServerConnection, new ConfigPersisterNotificationListener(persisterAggregator, facade));
}
public ConfigPersisterNotificationHandler(final MBeanServerConnection mBeanServerConnection, final NotificationListener notificationListener) {
this.mBeanServerConnection = mBeanServerConnection;
this.listener = notificationListener;
registerAsJMXListener(mBeanServerConnection, listener);
}
private static void registerAsJMXListener(final MBeanServerConnection mBeanServerConnection, final NotificationListener listener) {
LOG.trace("Called registerAsJMXListener");
try {
mBeanServerConnection.addNotificationListener(ConfigJMXNotification.OBJECT_NAME, listener, null, null);
} catch (InstanceNotFoundException | IOException e) {
throw new IllegalStateException("Cannot register as JMX listener to netconf", e);
}
}
@Override
public synchronized void close() {
// unregister from JMX
final ObjectName on = ConfigJMXNotification.OBJECT_NAME;
try {
if (mBeanServerConnection.isRegistered(on)) {
mBeanServerConnection.removeNotificationListener(on, listener);
}
} catch (final Exception e) {
LOG.warn("Unable to unregister {} as listener for {}", listener, on, e);
}
}
}
class ConfigPersisterNotificationListener implements NotificationListener {
private static final Logger LOG = LoggerFactory.getLogger(ConfigPersisterNotificationListener.class);
private final Persister persisterAggregator;
private final ConfigSubsystemFacadeFactory facade;
ConfigPersisterNotificationListener(final Persister persisterAggregator, final ConfigSubsystemFacadeFactory facade) {
this.persisterAggregator = persisterAggregator;
this.facade = facade;
}
@Override
public void handleNotification(final Notification notification, final Object handback) {
if (!(notification instanceof ConfigJMXNotification)) {
return;
}
// Socket should not be closed at this point
// Activator unregisters this as JMX listener before close is called
LOG.trace("Received notification {}", notification);
if (notification instanceof CommitJMXNotification) {
try {
handleAfterCommitNotification();
} catch (final Exception e) {
// log exceptions from notification Handler here since
// notificationBroadcastSupport logs only DEBUG level
LOG.warn("Failed to handle notification {}", notification, e);
throw e;
}
} else {
throw new IllegalStateException("Unknown config registry notification type " + notification);
}
}
private void handleAfterCommitNotification() {
try {
final Set<Capability> currentCapabilities = facade.getCurrentCapabilities();
final Element configSnapshot = facade.createFacade("config-persister").getConfiguration(XmlUtil.newDocument(), Datastore.running, Optional.<String>absent());
persisterAggregator.persistConfig(new CapabilityStrippingConfigSnapshotHolder(configSnapshot,
ConfigPusherImpl.transformCapabilities(currentCapabilities)));
LOG.trace("Configuration persisted successfully");
} catch (final IOException e) {
throw new RuntimeException("Unable to persist configuration snapshot", e);
}
}
}