org.opendaylight.jsonrpc.provider.common.BuiltinSchemaContextProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jsonrpc-provider-common Show documentation
The 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.common;

import com.google.common.collect.Maps;
import java.util.Collection;
import java.util.HashSet;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;
import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.jsonrpc.model.SchemaContextProvider;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.YangIdentifier;
import org.opendaylight.yang.gen.v1.urn.opendaylight.jsonrpc.rev161201.Peer;
import org.opendaylight.yangtools.yang.common.QNameModule;
import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
import org.opendaylight.yangtools.yang.model.api.Module;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
import org.opendaylight.yangtools.yang.model.api.stmt.ModuleEffectiveStatement;
import org.opendaylight.yangtools.yang.model.spi.SimpleSchemaContext;

/**
 * {@link SchemaContextProvider} which uses global {@link SchemaContext}. This implementation only validates that
 * required models are present in global {@link EffectiveModelContext}.Implementation is fail-fast, so any missing
 * model will cause error.
 *
 * @author Richard Kosegi
 */
public class BuiltinSchemaContextProvider implements SchemaContextProvider {
    private final EffectiveModelContext schemaContext;

    public BuiltinSchemaContextProvider(@NonNull final EffectiveModelContext schemaContext) {
        this.schemaContext = Objects.requireNonNull(schemaContext);
    }

    @Override
    public EffectiveModelContext createSchemaContext(Peer peer) {
        Set moduleIds = peer.getModules()
                .stream()
                .map(YangIdentifier::getValue)
                .map(m -> schemaContext.findModules(m)
                        .stream()
                        .findFirst()
                        .orElseThrow(() -> new IllegalStateException(
                                String.format("No model '%s' in global schema context", m))))
                .collect(Collectors.toSet());

        return buildSchemaContext(moduleIds);
    }

    private EffectiveModelContext buildSchemaContext(Set modules) {
        final Set resolved = new HashSet<>();
        resolved.addAll(modules);
        modules.stream()
                .forEach(m -> resolved.addAll(m.getImports()
                        .stream()
                        .map(mi -> schemaContext.findModule(mi.getModuleName().getLocalName(), mi.getRevision())
                                .orElseThrow())
                        .collect(Collectors.toSet())));
        return new EffectiveModelContextAdapter(resolved);
    }

    private static class EffectiveModelContextAdapter extends SimpleSchemaContext implements EffectiveModelContext {
        EffectiveModelContextAdapter(Collection modules) {
            super(modules);
        }

        @Override
        public Map getModuleStatements() {
            return Maps.transformValues(getModuleMap(), Module::asEffectiveStatement);
        }
    }
}