org.opendaylight.serviceutils.srm.impl.ServiceRecoveryRegistryImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of srm-impl Show documentation
The newest version!
/*
 * Copyright (c) 2018 Ericsson India Global Services Pvt Ltd. 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.serviceutils.srm.impl;

import java.util.Map;
import java.util.Queue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedQueue;
import javax.inject.Inject;
import javax.inject.Singleton;
import org.opendaylight.serviceutils.srm.RecoverableListener;
import org.opendaylight.serviceutils.srm.ServiceRecoveryInterface;
import org.opendaylight.serviceutils.srm.ServiceRecoveryRegistry;
import org.osgi.service.component.annotations.Activate;
import org.osgi.service.component.annotations.Component;
import org.osgi.service.component.annotations.RequireServiceComponentRuntime;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Singleton
@Component(immediate = true)
@RequireServiceComponentRuntime
public final class ServiceRecoveryRegistryImpl implements ServiceRecoveryRegistry {
    private static final Logger LOG = LoggerFactory.getLogger(ServiceRecoveryRegistryImpl.class);

    private final Map serviceRecoveryRegistry = new ConcurrentHashMap<>();

    private final Map> recoverableListenersMap =
            new ConcurrentHashMap<>();

    @Inject
    @Activate
    public ServiceRecoveryRegistryImpl() {
        // Exposed for DI
    }

    @Override
    public void registerServiceRecoveryRegistry(String entityName, ServiceRecoveryInterface serviceRecoveryHandler) {
        serviceRecoveryRegistry.put(entityName, serviceRecoveryHandler);
        LOG.trace("Registered service recovery handler for {}", entityName);
    }

    @Override
    public synchronized void addRecoverableListener(String serviceName, RecoverableListener recoverableListener) {
        recoverableListenersMap
                .computeIfAbsent(serviceName, k -> new ConcurrentLinkedQueue<>())
                .add(recoverableListener);
    }

    @Override
    public void removeRecoverableListener(String serviceName, RecoverableListener recoverableListener) {
        Queue queue = getRecoverableListeners(serviceName);
        if (queue != null) {
            queue.remove(recoverableListener);
        }
    }

    @Override
    public Queue getRecoverableListeners(String serviceName) {
        return recoverableListenersMap.get(serviceName);
    }

    @Override
    public ServiceRecoveryInterface getRegisteredServiceRecoveryHandler(String entityName) {
        return serviceRecoveryRegistry.get(entityName);
    }
}