All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.opendaylight.jsonrpc.provider.single.JsonRPCProvider Maven / Gradle / Ivy

There is a newer version: 1.17.0
Show newest version
/*
 * Copyright (c) 2017 Brocade Communications Systems, Inc. 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.jsonrpc.provider.single;

import com.google.common.base.Strings;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Objects;
import java.util.Optional;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock.WriteLock;
import java.util.function.Function;
import java.util.stream.Collectors;
import org.checkerframework.checker.lock.qual.GuardedBy;
import org.opendaylight.jsonrpc.bus.messagelib.TransportFactory;
import org.opendaylight.jsonrpc.model.CombinedSchemaContextProvider;
import org.opendaylight.jsonrpc.model.GovernanceProvider;
import org.opendaylight.jsonrpc.provider.common.AbstractPeerContext;
import org.opendaylight.jsonrpc.provider.common.FailedPeerContext;
import org.opendaylight.jsonrpc.provider.common.MappedPeerContext;
import org.opendaylight.jsonrpc.provider.common.ProviderDependencies;
import org.opendaylight.mdsal.binding.api.ClusteredDataTreeChangeListener;
import org.opendaylight.mdsal.binding.api.DataBroker;
import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
import org.opendaylight.mdsal.binding.api.ReadTransaction;
import org.opendaylight.mdsal.binding.api.RpcProviderService;
import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
import org.opendaylight.mdsal.dom.api.DOMDataBroker;
import org.opendaylight.mdsal.dom.api.DOMMountPointService;
import org.opendaylight.mdsal.dom.api.DOMNotificationPublishService;
import org.opendaylight.mdsal.dom.api.DOMRpcService;
import org.opendaylight.mdsal.dom.api.DOMSchemaService;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.Config;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.ConfigBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.ForceRefreshInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.ForceRefreshOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.ForceRefreshOutputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.ForceReloadInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.ForceReloadOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.ForceReloadOutputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.JsonrpcService;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.Peer;
import org.opendaylight.yangtools.concepts.Registration;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.opendaylight.yangtools.yang.common.RpcResult;
import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
import org.opendaylight.yangtools.yang.xpath.api.YangXPathParserFactory;
import org.osgi.service.component.annotations.Activate;
import org.osgi.service.component.annotations.Component;
import org.osgi.service.component.annotations.Deactivate;
import org.osgi.service.component.annotations.Reference;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Component(service = { })
public final class JsonRPCProvider implements JsonrpcService, AutoCloseable {
    private static final String ME = "JSON RPC Provider";
    private static final Logger LOG = LoggerFactory.getLogger(JsonRPCProvider.class);
    private static final InstanceIdentifier GLOBAL_CFG_II = InstanceIdentifier.create(Config.class);
    private static final DataTreeIdentifier CFG_DTI =
        DataTreeIdentifier.create(LogicalDatastoreType.CONFIGURATION, GLOBAL_CFG_II);
    private final Map peerState = new ConcurrentHashMap<>();
    private final ReentrantReadWriteLock changeLock = new ReentrantReadWriteLock();
    private final ProviderDependencies dependencies;
    private final GovernanceProvider governance;
    private final Registration dtclReg;
    private Registration rpcReg;
    private volatile boolean closed = false;

    @Activate
    public JsonRPCProvider(@Reference YangXPathParserFactory yangXPathParserFactory,
            @Reference DataBroker dataBroker, @Reference RpcProviderService rpcProviderService,
            @Reference DOMDataBroker domDataBroker, @Reference DOMMountPointService domMountPointService,
            @Reference DOMSchemaService schemaService, @Reference DOMRpcService domRpcService,
            @Reference DOMNotificationPublishService domNotificationPublishService,
            @Reference TransportFactory transportFactory, @Reference GovernanceProvider governance) {
        this(new ProviderDependencies(transportFactory, dataBroker, domMountPointService, domDataBroker, schemaService,
            domNotificationPublishService, domRpcService, yangXPathParserFactory), governance);
        rpcReg = rpcProviderService.registerRpcImplementation(JsonrpcService.class, this);
    }

    public JsonRPCProvider(ProviderDependencies dependencies, GovernanceProvider governance) {
        this.dependencies = Objects.requireNonNull(dependencies);
        this.governance = Objects.requireNonNull(governance);
        dtclReg = dependencies.getDataBroker().registerDataTreeChangeListener(CFG_DTI,
                    (ClusteredDataTreeChangeListener) changes -> processNotification());
        processNotification();
    }

    /**
     * Get current configuration or operational state. Configuration can be set or deleted by the user via restconf.
     * Operational state will reflect actual result of configuration via restconf and other sources.
     */
    private Optional getConfig(LogicalDatastoreType store) {
        try (ReadTransaction roTrx = dependencies.getDataBroker().newReadOnlyTransaction()) {
            return Futures.getUnchecked(roTrx.read(store, GLOBAL_CFG_II));
        }
    }

    private static Map generateCache(Collection arg) {
        return Optional.ofNullable(arg)
                .orElse(List.of())
                .stream()
                .collect(Collectors.toMap(Peer::getName, Function.identity()));
    }

    private boolean processNotificationInternal() {
        LOG.debug("Processing notification");
        if (closed) {
            LOG.debug("{} was closed already, ignoring configuration change", ME);
            return false;
        }

        final Optional optPeersConfState = getConfig(LogicalDatastoreType.CONFIGURATION);
        if (optPeersConfState.isEmpty()) {
            // in case entire config was wiped, we still need to unconfigure
            // existing peers, hence supply empty list
            unmountPeers(new ConfigBuilder().setConfiguredEndpoints(Map.of()).build());
            LOG.info("{} configuration absent", ME);
            return false;
        }
        final Config peersConfState = optPeersConfState.orElseThrow();
        boolean result = mountPeers(peersConfState);
        result &= unmountPeers(peersConfState);
        return result;
    }

    private boolean unmountPeers(final Config peersConfState) {
        boolean result = true;
        final Map cache = generateCache(peersConfState.nonnullConfiguredEndpoints().values());
        final List toUnmountList = peerState.entrySet()
                .stream()
                .filter(e -> !cache.containsKey(e.getKey()))
                .map(Entry::getKey)
                .collect(Collectors.toList());
        for (final String toUnmount : toUnmountList) {
            result &= doUnmount(toUnmount);
        }
        return result;
    }

    private boolean mountPeers(final Config peersConfState) {
        boolean result = true;
        if (peersConfState.getConfiguredEndpoints() != null) {
            for (final Peer confPeer : peersConfState.nonnullConfiguredEndpoints().values()) {
                LOG.debug("Processing peer from conf {}", confPeer.getName());
                if (!peerState.containsKey(confPeer.getName())) {
                    result &= doMountDevice(confPeer);
                }
            }
        } else {
            LOG.debug("No configured endpoints");
        }
        return result;
    }

    /**
     * Performs reconciliation between our internal mount state and the datastore upon startup and after receiving a
     * DCN. This method ensures serialized access to underlying datastructures via lock.
     */
    @GuardedBy("changeLock")
    private boolean processNotification() {
        final WriteLock wLock = changeLock.writeLock();
        try {
            wLock.lock();
            return processNotificationInternal();
        } finally {
            wLock.unlock();
        }
    }

    @Deactivate
    @Override
    public void close() {
        if (!closed) {
            closed = true;
            if (rpcReg != null) {
                rpcReg.close();
            }
            dtclReg.close();
            peerState.values().forEach(AbstractPeerContext::close);
            peerState.clear();
        }
    }

    /*
     * Do the heavy lifting required in order to mount.
     */
    @SuppressWarnings("checkstyle:IllegalCatch")
    private boolean doMountDevice(Peer peer) {
        try {
            LOG.debug("Creating mapping context for peer {}", peer.getName());
            final CombinedSchemaContextProvider schemaProvider = new CombinedSchemaContextProvider(governance,
                    dependencies);
            final MappedPeerContext ctx = new MappedPeerContext(peer, dependencies.getTransportFactory(),
                    dependencies.getSchemaService(), dependencies.getDataBroker(),
                    dependencies.getDomMountPointService(), governance.get().orElse(null), schemaProvider);
            peerState.put(peer.getName(), ctx);
            LOG.info("Peer mounted : {}", ctx);
            return true;
        } catch (RuntimeException | URISyntaxException e) {
            LOG.error("Mount failed for peer '{}'", peer.getName(), e);
            peerState.put(peer.getName(), new FailedPeerContext(peer, dependencies.getDataBroker(), e));
            return false;
        }
    }

    /*
     * unMount heavy lifter - does the appropriate cleanup when unmounting a device
     */
    @SuppressWarnings("checkstyle:IllegalCatch")
    public boolean doUnmount(String deviceName) {
        if (Strings.isNullOrEmpty(deviceName)) {
            return false;
        }
        final AbstractPeerContext toDelete = peerState.remove(deviceName);
        if (toDelete == null) {
            LOG.error("Device '{}' did not complete mount, cannot remove", deviceName);
            return false;
        } else {
            try {
                LOG.debug("Destroying mapping context of peer '{}'", toDelete);
                toDelete.close();
                LOG.debug("Device '{}' unmounted successfully", deviceName);
                return true;
            } catch (Exception e) {
                LOG.error("Device '{}' unmount, failed", deviceName, e);
                return false;
            }
        }
    }

    /*
     * JSON Rpc Force refresh
     */
    @Override
    public ListenableFuture> forceRefresh(ForceRefreshInput input) {
        LOG.debug("Refreshing json rpc state");
        return Futures.immediateFuture(RpcResultBuilder
                .success(new ForceRefreshOutputBuilder().setResult(processNotification()).build())
                .build());
    }

    /*
     * JSON Rpc Force reload
     */
    @Override
    public ListenableFuture> forceReload(ForceReloadInput input) {
        final ForceReloadOutputBuilder outputBuilder = new ForceReloadOutputBuilder();
        LOG.debug("Remounting all json rpc peers");
        boolean result = true;
        final List toUnmountList = new ArrayList<>();
        toUnmountList.addAll(peerState.keySet());
        for (final String toUnmount : toUnmountList) {
            result &= doUnmount(toUnmount);
        }
        result &= processNotification();
        outputBuilder.setResult(result);
        return Futures.immediateFuture(RpcResultBuilder.success(outputBuilder.build()).build());
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy