Make ParserNamespace an identifier
[yangtools.git] / parser / yang-parser-reactor / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / reactor / ModifierImpl.java
1 /*
2  * Copyright (c) 2015, 2016 Cisco Systems, Inc. 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.stmt.reactor;
9
10 import static com.google.common.base.Preconditions.checkArgument;
11 import static com.google.common.base.Preconditions.checkState;
12 import static com.google.common.base.Verify.verifyNotNull;
13 import static java.util.Objects.requireNonNull;
14 import static org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase.EFFECTIVE_MODEL;
15 import static org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase.FULL_DECLARATION;
16
17 import com.google.common.base.MoreObjects;
18 import com.google.common.base.MoreObjects.ToStringHelper;
19 import java.util.ArrayList;
20 import java.util.HashSet;
21 import java.util.Iterator;
22 import java.util.List;
23 import java.util.Set;
24 import java.util.function.Function;
25 import org.eclipse.jdt.annotation.NonNull;
26 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
27 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.NamespaceKeyCriterion;
31 import org.opendaylight.yangtools.yang.parser.spi.meta.ParserNamespace;
32 import org.opendaylight.yangtools.yang.parser.spi.meta.StatementNamespace;
33 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
34 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
35 import org.opendaylight.yangtools.yang.parser.stmt.reactor.StatementContextBase.ContextMutation;
36 import org.opendaylight.yangtools.yang.parser.stmt.reactor.StatementContextBase.OnNamespaceItemAdded;
37 import org.opendaylight.yangtools.yang.parser.stmt.reactor.StatementContextBase.OnPhaseFinished;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
40
41 final class ModifierImpl implements ModelActionBuilder {
42     private static final Logger LOG = LoggerFactory.getLogger(ModifierImpl.class);
43
44     private final Set<AbstractPrerequisite<?>> unsatisfied = new HashSet<>(1);
45     private final Set<AbstractPrerequisite<?>> mutations = new HashSet<>(1);
46     private final InferenceContext ctx = new InferenceContext() { };
47
48     private List<Runnable> bootstraps;
49     private InferenceAction action;
50     private boolean actionApplied;
51
52     private <D> AbstractPrerequisite<D> addReq(final AbstractPrerequisite<D> prereq) {
53         LOG.trace("Modifier {} adding prerequisite {}", this, prereq);
54         unsatisfied.add(prereq);
55         return prereq;
56     }
57
58     private <T> @NonNull AbstractPrerequisite<T> addMutation(final @NonNull AbstractPrerequisite<T> mutation) {
59         LOG.trace("Modifier {} adding mutation {}", this, mutation);
60         mutations.add(mutation);
61         return mutation;
62     }
63
64     private void checkNotRegistered() {
65         checkState(action == null, "Action was already registered.");
66     }
67
68     private boolean removeSatisfied() {
69         final Iterator<AbstractPrerequisite<?>> it = unsatisfied.iterator();
70         while (it.hasNext()) {
71             final AbstractPrerequisite<?> prereq = it.next();
72             if (prereq.isDone()) {
73                 // We are removing current prerequisite from list.
74                 LOG.trace("Modifier {} prerequisite {} satisfied", this, prereq);
75                 it.remove();
76             }
77         }
78         return unsatisfied.isEmpty();
79     }
80
81     boolean isApplied() {
82         return actionApplied;
83     }
84
85     void failModifier() {
86         removeSatisfied();
87         action.prerequisiteFailed(unsatisfied);
88         action = null;
89     }
90
91     private <K, C extends StmtContext<?, ?, ?>, N extends StatementNamespace<K, ?, ?>> @NonNull AbstractPrerequisite<C>
92             requiresCtxImpl(final StmtContext<?, ?, ?> context, final N namespace, final K key,
93                     final ModelProcessingPhase phase)  {
94         checkNotRegistered();
95
96         AddedToNamespace<C> addedToNs = new AddedToNamespace<>(phase);
97         addReq(addedToNs);
98         contextImpl(context).onNamespaceItemAddedAction(namespace, key, addedToNs);
99         return addedToNs;
100     }
101
102     private <K, C extends StmtContext<?, ?, ?>, N extends StatementNamespace<K, ?, ?>> @NonNull AbstractPrerequisite<C>
103             requiresCtxImpl(final StmtContext<?, ?, ?> context, final N namespace,
104                     final NamespaceKeyCriterion<K> criterion, final ModelProcessingPhase phase)  {
105         checkNotRegistered();
106
107         AddedToNamespace<C> addedToNs = new AddedToNamespace<>(phase);
108         addReq(addedToNs);
109         contextImpl(context).onNamespaceItemAddedAction(namespace, phase, criterion, addedToNs);
110         return addedToNs;
111     }
112
113     private <C extends StmtContext<?, ?, ?>> @NonNull AbstractPrerequisite<C> requiresCtxImpl(final C context,
114             final ModelProcessingPhase phase) {
115         checkNotRegistered();
116
117         PhaseFinished<C> phaseFin = new PhaseFinished<>();
118         addReq(phaseFin);
119         addBootstrap(() -> contextImpl(context).addPhaseCompletedListener(phase, phaseFin));
120         return phaseFin;
121     }
122
123     @SuppressWarnings({ "rawtypes", "unchecked" })
124     private <K, C extends Mutable<?, ?, ?>> AbstractPrerequisite<C> mutatesCtxImpl(final StmtContext<?, ?, ?> context,
125             final ParserNamespace<K, ? extends StmtContext<?, ?, ?>> namespace, final K key,
126             final ModelProcessingPhase phase) {
127         checkNotRegistered();
128
129         final PhaseModificationInNamespace<C> mod = new PhaseModificationInNamespace<>(EFFECTIVE_MODEL);
130         addReq(mod);
131         addMutation(mod);
132         contextImpl(context).onNamespaceItemAddedAction((ParserNamespace) namespace, key, mod);
133         return mod;
134     }
135
136     private static StatementContextBase<?, ?, ?> contextImpl(final Object value) {
137         checkArgument(value instanceof StatementContextBase, "Supplied context %s is not provided by this reactor.",
138             value);
139         return StatementContextBase.class.cast(value);
140     }
141
142     boolean tryApply() {
143         checkState(action != null, "Action was not defined yet.");
144
145         if (removeSatisfied()) {
146             if (!actionApplied) {
147                 action.apply(ctx);
148                 actionApplied = true;
149             }
150             return true;
151         }
152         return false;
153     }
154
155     @Override
156     public <C extends Mutable<?, ?, ?>, T extends C> Prerequisite<C> mutatesCtx(final T context,
157             final ModelProcessingPhase phase) {
158         return addMutation(new PhaseMutation<>(contextImpl(context), phase));
159     }
160
161     @Override
162     public <A, D extends DeclaredStatement<A>, E extends EffectiveStatement<A, D>>
163             AbstractPrerequisite<StmtContext<A, D, E>> requiresCtx(final StmtContext<A, D, E> context,
164                     final ModelProcessingPhase phase) {
165         return requiresCtxImpl(context, phase);
166     }
167
168     @Override
169     public <K, N extends StatementNamespace<K, ?, ?>> Prerequisite<StmtContext<?, ?, ?>> requiresCtx(
170             final StmtContext<?, ?, ?> context, final N namespace, final K key, final ModelProcessingPhase phase) {
171         return requiresCtxImpl(context, namespace, key, phase);
172     }
173
174     @Override
175     public <K, N extends StatementNamespace<K, ?, ?>> Prerequisite<StmtContext<?, ?, ?>> requiresCtx(
176             final StmtContext<?, ?, ?> context, final N namespace, final NamespaceKeyCriterion<K> criterion,
177             final ModelProcessingPhase phase) {
178         return requiresCtxImpl(context, namespace, criterion, phase);
179     }
180
181     @Override
182     public <K, E extends EffectiveStatement<?, ?>, N extends ParserNamespace<K, ? extends StmtContext<?, ?, ?>>>
183             Prerequisite<StmtContext<?, ?, E>> requiresCtxPath(final StmtContext<?, ?, ?> context,
184                 final N namespace, final Iterable<K> keys, final ModelProcessingPhase phase) {
185         checkNotRegistered();
186
187         final var ret = new PhaseRequirementInNamespacePath<StmtContext<?, ?, E>, K, N>(EFFECTIVE_MODEL, keys);
188         addReq(ret);
189         addBootstrap(() -> ret.hookOnto(context, namespace));
190         return ret;
191     }
192
193     @Override
194     public <D extends DeclaredStatement<?>> Prerequisite<D> requiresDeclared(
195             final StmtContext<?, ? extends D, ?> context) {
196         return requiresCtxImpl(context, FULL_DECLARATION).transform(StmtContext::declared);
197     }
198
199     @Override
200     @Deprecated
201     public <K, D extends DeclaredStatement<?>, N extends StatementNamespace<K, ? extends D, ?>> Prerequisite<D>
202             requiresDeclared(final StmtContext<?, ?, ?> context, final N namespace, final K key) {
203         final AbstractPrerequisite<StmtContext<?, D, ?>> rawContext = requiresCtxImpl(context, namespace, key,
204             FULL_DECLARATION);
205         return rawContext.transform(StmtContext::declared);
206     }
207
208     @Override
209     @Deprecated
210     public <K, D extends DeclaredStatement<?>, N extends StatementNamespace<K, ? extends D, ?>>
211             AbstractPrerequisite<StmtContext<?, D, ?>> requiresDeclaredCtx(final StmtContext<?, ?, ?> context,
212                     final N namespace, final K key) {
213         return requiresCtxImpl(context, namespace, key, FULL_DECLARATION);
214     }
215
216     @Override
217     @Deprecated
218     public <E extends EffectiveStatement<?, ?>> Prerequisite<E> requiresEffective(
219             final StmtContext<?, ?, ? extends E> stmt) {
220         return requiresCtxImpl(stmt, EFFECTIVE_MODEL).transform(StmtContext::buildEffective);
221     }
222
223     @Override
224     @Deprecated
225     public <K, E extends EffectiveStatement<?, ?>, N extends StatementNamespace<K, ?, ? extends E>> Prerequisite<E>
226             requiresEffective(final StmtContext<?, ?, ?> context, final N namespace, final K key) {
227         final AbstractPrerequisite<StmtContext<?, ?, E>> rawContext = requiresCtxImpl(context, namespace, key,
228             EFFECTIVE_MODEL);
229         return rawContext.transform(StmtContext::buildEffective);
230     }
231
232     @Override
233     @Deprecated
234     public <K, E extends EffectiveStatement<?, ?>, N extends StatementNamespace<K, ?, ? extends E>>
235             AbstractPrerequisite<StmtContext<?, ?, E>> requiresEffectiveCtx(final StmtContext<?, ?, ?> context,
236                     final N namespace, final K key) {
237         return requiresCtxImpl(contextImpl(context), namespace, key, EFFECTIVE_MODEL);
238     }
239
240     @Override
241     @Deprecated
242     public <N extends ParserNamespace<?, ?>> Prerequisite<Mutable<?, ?, ?>> mutatesNs(final Mutable<?, ?, ?> context,
243             final N namespace) {
244         return addMutation(new NamespaceMutation<>(contextImpl(context), namespace));
245     }
246
247     @Override
248     public <K, E extends EffectiveStatement<?, ?>, N extends ParserNamespace<K, ? extends StmtContext<?, ?, ?>>>
249             AbstractPrerequisite<Mutable<?, ?, E>> mutatesEffectiveCtx(final StmtContext<?, ?, ?> context,
250                     final N namespace, final K key) {
251         return mutatesCtxImpl(context, namespace, key, EFFECTIVE_MODEL);
252     }
253
254     @Override
255     public <K, E extends EffectiveStatement<?, ?>, N extends ParserNamespace<K, ? extends StmtContext<?, ?, ?>>>
256             AbstractPrerequisite<Mutable<?, ?, E>> mutatesEffectiveCtxPath(final StmtContext<?, ?, ?> context,
257                     final N namespace, final Iterable<K> keys) {
258         checkNotRegistered();
259
260         final var ret = new PhaseModificationInNamespacePath<Mutable<?, ?, E>, K, N>(EFFECTIVE_MODEL, keys);
261         addReq(ret);
262         addMutation(ret);
263         addBootstrap(() -> ret.hookOnto(context, namespace));
264         return ret;
265     }
266
267     @Override
268     @SuppressWarnings("checkstyle:hiddenField")
269     public void apply(final InferenceAction action) {
270         checkState(this.action == null, "Action already defined to %s", this.action);
271         this.action = requireNonNull(action);
272         if (bootstraps != null) {
273             bootstraps.forEach(Runnable::run);
274             bootstraps = null;
275         }
276     }
277
278     private void addBootstrap(final Runnable bootstrap) {
279         if (bootstraps == null) {
280             bootstraps = new ArrayList<>(1);
281         }
282         bootstraps.add(bootstrap);
283     }
284
285     private abstract class AbstractPrerequisite<T> implements Prerequisite<T> {
286         private boolean done = false;
287         private T value;
288
289         @Override
290         @SuppressWarnings("checkstyle:hiddenField")
291         public final T resolve(final InferenceContext ctx) {
292             checkState(done);
293             checkArgument(ctx == ModifierImpl.this.ctx);
294             return verifyNotNull(value, "Attempted to access unavailable prerequisite %s", this);
295         }
296
297         final boolean isDone() {
298             return done;
299         }
300
301         @SuppressWarnings("checkstyle:hiddenField")
302         final boolean resolvePrereq(final T value) {
303             this.value = value;
304             this.done = true;
305             return isApplied();
306         }
307
308         final <O> @NonNull Prerequisite<O> transform(final Function<? super T, O> transformation) {
309             return context -> transformation.apply(resolve(context));
310         }
311
312         @Override
313         public final String toString() {
314             return addToStringAttributes(MoreObjects.toStringHelper(this).omitNullValues()).toString();
315         }
316
317         ToStringHelper addToStringAttributes(final ToStringHelper toStringHelper) {
318             return toStringHelper.add("value", value);
319         }
320     }
321
322     private abstract class AbstractPathPrerequisite<C extends StmtContext<?, ?, ?>, K,
323             N extends ParserNamespace<K, ? extends StmtContext<?, ?, ?>>> extends AbstractPrerequisite<C>
324             implements OnNamespaceItemAdded {
325         private final ModelProcessingPhase modPhase;
326         private final Iterable<K> keys;
327         private final Iterator<K> it;
328
329         AbstractPathPrerequisite(final ModelProcessingPhase phase, final Iterable<K> keys) {
330             this.modPhase = requireNonNull(phase);
331             this.keys = requireNonNull(keys);
332             it = keys.iterator();
333         }
334
335         @Override
336         public final void namespaceItemAdded(final StatementContextBase<?, ?, ?> context,
337                 final ParserNamespace<?, ?> namespace, final Object key, final Object value) {
338             LOG.debug("Action for {} got key {}", keys, key);
339
340             final StatementContextBase<?, ?, ?> target = contextImpl(value);
341             if (!target.isSupportedByFeatures()) {
342                 LOG.debug("Key {} in {} is not supported", key, keys);
343                 resolvePrereq(null);
344                 action.prerequisiteUnavailable(this);
345                 return;
346             }
347
348             nextStep(modPhase, context, target);
349
350             if (!it.hasNext()) {
351                 // Last step: we are done
352                 if (resolvePrereq((C) value)) {
353                     tryApply();
354                 }
355                 return;
356             }
357
358             // Make sure target's storage notifies us when the next step becomes available.
359             hookOnto(target, namespace, it.next());
360         }
361
362         abstract void nextStep(ModelProcessingPhase phase, StatementContextBase<?, ?, ?> current,
363             StatementContextBase<?, ?, ?> next);
364
365         @Override
366         final ToStringHelper addToStringAttributes(final ToStringHelper toStringHelper) {
367             return super.addToStringAttributes(toStringHelper).add("phase", modPhase).add("keys", keys);
368         }
369
370         final void hookOnto(final StmtContext<?, ?, ?> context, final ParserNamespace<?, ?> namespace) {
371             checkArgument(it.hasNext(), "Namespace %s keys may not be empty", namespace);
372             hookOnto(contextImpl(context), namespace, it.next());
373         }
374
375         @SuppressWarnings("unchecked")
376         private void hookOnto(final StatementContextBase<?, ?, ?> context, final ParserNamespace<?, ?> namespace,
377                 final K key) {
378             context.onNamespaceItemAddedAction((ParserNamespace) namespace, requireNonNull(key), this);
379         }
380     }
381
382     private final class PhaseMutation<C> extends AbstractPrerequisite<C> implements ContextMutation {
383         @SuppressWarnings("unchecked")
384         PhaseMutation(final StatementContextBase<?, ?, ?> context, final ModelProcessingPhase phase) {
385             context.addMutation(phase, this);
386             resolvePrereq((C) context);
387         }
388
389         @Override
390         public boolean isFinished() {
391             return isApplied();
392         }
393     }
394
395     private final class PhaseFinished<C extends StmtContext<?, ?, ?>> extends AbstractPrerequisite<C>
396             implements OnPhaseFinished {
397         @SuppressWarnings("unchecked")
398         @Override
399         public boolean phaseFinished(final StatementContextBase<?, ?, ?> context,
400                 final ModelProcessingPhase finishedPhase) {
401             return resolvePrereq((C) context) || tryApply();
402         }
403     }
404
405     private final class NamespaceMutation<N extends ParserNamespace<?, ?>>
406             extends AbstractPrerequisite<Mutable<?, ?, ?>>  {
407         NamespaceMutation(final StatementContextBase<?, ?, ?> ctx, final N namespace) {
408             resolvePrereq(ctx);
409         }
410     }
411
412     private final class AddedToNamespace<C extends StmtContext<?, ?, ?>> extends AbstractPrerequisite<C>
413             implements OnNamespaceItemAdded, OnPhaseFinished {
414         private final ModelProcessingPhase phase;
415
416         AddedToNamespace(final ModelProcessingPhase phase) {
417             this.phase = requireNonNull(phase);
418         }
419
420         @Override
421         public void namespaceItemAdded(final StatementContextBase<?, ?, ?> context,
422                 final ParserNamespace<?, ?> namespace, final Object key, final Object value) {
423             ((StatementContextBase<?, ?, ?>) value).addPhaseCompletedListener(phase, this);
424         }
425
426         @SuppressWarnings("unchecked")
427         @Override
428         public boolean phaseFinished(final StatementContextBase<?, ?, ?> context,
429                 final ModelProcessingPhase finishedPhase) {
430             return resolvePrereq((C) context) || tryApply();
431         }
432
433         @Override
434         ToStringHelper addToStringAttributes(final ToStringHelper toStringHelper) {
435             return super.addToStringAttributes(toStringHelper).add("phase", phase);
436         }
437     }
438
439     private final class PhaseRequirementInNamespacePath<C extends StmtContext<?, ?, ?>, K,
440             N extends ParserNamespace<K, ? extends StmtContext<?, ?, ?>>> extends AbstractPathPrerequisite<C, K, N> {
441         PhaseRequirementInNamespacePath(final ModelProcessingPhase phase, final Iterable<K> keys) {
442             super(phase, keys);
443         }
444
445         @Override
446         void nextStep(final ModelProcessingPhase phase, final StatementContextBase<?, ?, ?> current,
447                 final StatementContextBase<?, ?, ?> next) {
448             // No-op
449         }
450     }
451
452     private final class PhaseModificationInNamespace<C extends Mutable<?, ?, ?>> extends AbstractPrerequisite<C>
453             implements OnNamespaceItemAdded, ContextMutation {
454         private final ModelProcessingPhase modPhase;
455
456         PhaseModificationInNamespace(final ModelProcessingPhase phase) {
457             checkArgument(phase != null, "Model processing phase must not be null");
458             this.modPhase = phase;
459         }
460
461         @SuppressWarnings("unchecked")
462         @Override
463         public void namespaceItemAdded(final StatementContextBase<?, ?, ?> context,
464                 final ParserNamespace<?, ?> namespace, final Object key, final Object value) {
465             StatementContextBase<?, ?, ?> targetCtx = contextImpl(value);
466             targetCtx.addMutation(modPhase, this);
467             resolvePrereq((C) targetCtx);
468         }
469
470         @Override
471         public boolean isFinished() {
472             return isApplied();
473         }
474     }
475
476     /**
477      * This similar to {@link PhaseModificationInNamespace}, but allows recursive descent until it finds the real
478      * target. The mechanics is driven as a sequence of prerequisites along a path: first we hook onto namespace to
479      * give us the first step. When it does, we hook onto the first item to provide us the second step and so on.
480      */
481     private final class PhaseModificationInNamespacePath<C extends Mutable<?, ?, ?>, K,
482             N extends ParserNamespace<K, ? extends StmtContext<?, ?, ?>>> extends AbstractPathPrerequisite<C, K, N>
483             implements ContextMutation {
484         PhaseModificationInNamespacePath(final ModelProcessingPhase phase, final Iterable<K> keys) {
485             super(phase, keys);
486         }
487
488         @Override
489         public boolean isFinished() {
490             return isApplied();
491         }
492
493         @Override
494         void nextStep(final ModelProcessingPhase phase, final StatementContextBase<?, ?, ?> current,
495                 final StatementContextBase<?, ?, ?> next) {
496             // Hook onto target: we either have a modification of the target itself or one of its children.
497             next.addMutation(phase, this);
498             // We have completed the context -> target step, hence we are no longer directly blocking context from
499             // making forward progress.
500             current.removeMutation(phase, this);
501         }
502     }
503 }