f9486695b8ebb11f0349423be2abca53cf805f27
[yangtools.git] / yang / yang-parser-spi / src / main / java / org / opendaylight / yangtools / yang / parser / spi / meta / AbstractInternedStatementSupport.java
1 /*
2  * Copyright (c) 2020 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.yang.parser.spi.meta;
9
10 import com.google.common.annotations.Beta;
11 import com.google.common.cache.CacheBuilder;
12 import com.google.common.cache.CacheLoader;
13 import com.google.common.cache.LoadingCache;
14 import com.google.common.collect.ImmutableList;
15 import org.eclipse.jdt.annotation.NonNull;
16 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
17 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
18 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
19 import org.opendaylight.yangtools.yang.model.parser.api.YangParserConfiguration;
20 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
21
22 /**
23  * A {@link AbstractStatementSupport} specialized for global interning. This base class is useful when the argument can
24  * be reasonably interned and it dominates the {@link EffectiveStatement} implementation. Typical examples include
25  * {@code position} and {@code value} statements, which typically do not have substatements and are based on simple
26  * types.
27  *
28  * <p>
29  * Note: use of this base class implies context-independence.
30  */
31 @Beta
32 public abstract class AbstractInternedStatementSupport<A, D extends DeclaredStatement<A>,
33         E extends EffectiveStatement<A, D>> extends AbstractStatementSupport<A, D, E> {
34     private final LoadingCache<A, D> declaredCache = CacheBuilder.newBuilder().weakValues()
35             .build(new CacheLoader<A, D>() {
36                 @Override
37                 public D load(final A key) {
38                     return createEmptyDeclared(key);
39                 }
40             });
41     private final LoadingCache<D, E> effectiveCache = CacheBuilder.newBuilder().weakValues()
42             .build(new CacheLoader<D, E>() {
43                 @Override
44                 public E load(final D key) {
45                     return createEmptyEffective(key);
46                 }
47             });
48
49     protected AbstractInternedStatementSupport(final StatementDefinition publicDefinition,
50             final StatementPolicy<A, D> policy, final YangParserConfiguration config) {
51         super(publicDefinition, policy, config);
52     }
53
54     @Override
55     protected final D createDeclared(final StmtContext<A, D, ?> ctx,
56             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
57         final A argument = ctx.getArgument();
58         return substatements.isEmpty() ? declaredCache.getUnchecked(ctx.getArgument())
59             : createDeclared(argument, substatements);
60     }
61
62     protected abstract @NonNull D createDeclared(@NonNull A argument,
63             @NonNull ImmutableList<? extends DeclaredStatement<?>> substatements);
64
65     protected abstract @NonNull D createEmptyDeclared(@NonNull A argument);
66
67     @Override
68     protected final E createEffective(final Current<A, D> stmt,
69             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
70         return substatements.isEmpty() ? effectiveCache.getUnchecked(stmt.declared())
71             : createEffective(stmt.declared(), substatements);
72     }
73
74     protected abstract @NonNull E createEffective(@NonNull D declared,
75         @NonNull ImmutableList<? extends EffectiveStatement<?, ?>> substatements);
76
77     protected abstract @NonNull E createEmptyEffective(@NonNull D declared);
78 }