b791b115d4f9eb1601a98bcfe894ff63c7bf56b8
[yangtools.git] / third-party / triemap / src / main / java / org / opendaylight / yangtools / triemap / MainNode.java
1 /*
2  * (C) Copyright 2016 Pantheon Technologies, s.r.o. and others.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.opendaylight.yangtools.triemap;
17
18 import java.util.concurrent.atomic.AtomicReferenceFieldUpdater;
19
20 abstract class MainNode<K, V> extends BasicNode {
21     static final int NO_SIZE = -1;
22
23     @SuppressWarnings("rawtypes")
24     private static final AtomicReferenceFieldUpdater<MainNode, MainNode> PREV_UPDATER =
25         AtomicReferenceFieldUpdater.newUpdater(MainNode.class, MainNode.class, "prev");
26
27     private volatile MainNode<K, V> prev;
28
29     MainNode() {
30         this.prev = null;
31     }
32
33     MainNode(final MainNode<K, V> prev) {
34         this.prev = prev;
35     }
36
37     /**
38      * Return the number of entries in this node, or {@link #NO_SIZE} if it is not known.
39      */
40     abstract int trySize();
41
42     /**
43      * Return the number of entries in this node, traversing it if need be. This method should be invoked only
44      * on immutable snapshots.
45      *
46      * @param ct TrieMap reference
47      * @return The actual number of entries.
48      */
49     abstract int size(ImmutableTrieMap<?, ?> ct);
50
51     final boolean casPrev(final MainNode<K, V> oldval, final MainNode<K, V> nval) {
52         return PREV_UPDATER.compareAndSet(this, oldval, nval);
53     }
54
55     final void writePrev(final MainNode<K, V> nval) {
56         prev = nval;
57     }
58
59     final MainNode<K, V> readPrev() {
60         return prev;
61     }
62 }