Merge branch 'master' of ../controller
[yangtools.git] / yang / rfc8528-data-util / src / main / java / org / opendaylight / yangtools / rcf8528 / data / util / ImmutableMountPointContext.java
1 /*
2  * Copyright (c) 2019 PANTHEON.tech s.r.o. and others. All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.yangtools.rcf8528.data.util;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.base.MoreObjects.ToStringHelper;
13 import com.google.common.collect.ImmutableMap;
14 import com.google.common.collect.Maps;
15 import java.util.Optional;
16 import java.util.function.Function;
17 import org.opendaylight.yangtools.concepts.Immutable;
18 import org.opendaylight.yangtools.rcf8528.data.util.AbstractMountPointContextFactory.MountPointDefinition;
19 import org.opendaylight.yangtools.rfc8528.data.api.MountPointContext;
20 import org.opendaylight.yangtools.rfc8528.data.api.MountPointContextFactory;
21 import org.opendaylight.yangtools.rfc8528.data.api.MountPointIdentifier;
22 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
23 import org.opendaylight.yangtools.yang.model.util.AbstractSchemaContextProvider;
24
25 final class ImmutableMountPointContext extends AbstractSchemaContextProvider implements Immutable, MountPointContext {
26     private final ImmutableMap<MountPointIdentifier, MountPointDefinition> mountPoints;
27     private final Function<MountPointDefinition, MountPointContextFactory> createFactory;
28
29     ImmutableMountPointContext(final SchemaContext schemaContext, final Iterable<MountPointDefinition> mountPoints,
30             final Function<MountPointDefinition, MountPointContextFactory> createFactory) {
31         super(schemaContext);
32         this.mountPoints = Maps.uniqueIndex(mountPoints, MountPointDefinition::getIdentifier);
33         this.createFactory = requireNonNull(createFactory);
34     }
35
36     @Override
37     public Optional<MountPointContextFactory> findMountPoint(final MountPointIdentifier label) {
38         final MountPointDefinition def = mountPoints.get(requireNonNull(label));
39         return def == null ? Optional.empty() : Optional.of(createFactory.apply(def));
40     }
41
42     @Override
43     protected ToStringHelper addToStringAttributes(final ToStringHelper helper) {
44         return super.addToStringAttributes(helper).add("mountPoints", mountPoints);
45     }
46 }