/
ForwardingSortedSet.java
158 lines (143 loc) · 5.65 KB
/
ForwardingSortedSet.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
/*
* Copyright (C) 2007 The Guava Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.google.common.collect;
import static com.google.common.collect.ForwardingSortedMap.unsafeCompare;
import com.google.common.annotations.GwtCompatible;
import java.util.Comparator;
import java.util.Iterator;
import java.util.NoSuchElementException;
import java.util.SortedSet;
import javax.annotation.CheckForNull;
import org.checkerframework.checker.nullness.qual.Nullable;
/**
* A sorted set which forwards all its method calls to another sorted set. Subclasses should
* override one or more methods to modify the behavior of the backing sorted set as desired per the
* <a href="http://en.wikipedia.org/wiki/Decorator_pattern">decorator pattern</a>.
*
* <p><b>Warning:</b> The methods of {@code ForwardingSortedSet} forward <i>indiscriminately</i> to
* the methods of the delegate. For example, overriding {@link #add} alone <i>will not</i> change
* the behavior of {@link #addAll}, which can lead to unexpected behavior. In this case, you should
* override {@code addAll} as well, either providing your own implementation, or delegating to the
* provided {@code standardAddAll} method.
*
* <p><b>{@code default} method warning:</b> This class does <i>not</i> forward calls to {@code
* default} methods. Instead, it inherits their default implementations. When those implementations
* invoke methods, they invoke methods on the {@code ForwardingSortedSet}.
*
* <p>Each of the {@code standard} methods, where appropriate, uses the set's comparator (or the
* natural ordering of the elements, if there is no comparator) to test element equality. As a
* result, if the comparator is not consistent with equals, some of the standard implementations may
* violate the {@code Set} contract.
*
* <p>The {@code standard} methods and the collection views they return are not guaranteed to be
* thread-safe, even when all of the methods that they depend on are thread-safe.
*
* @author Mike Bostock
* @author Louis Wasserman
* @since 2.0
*/
@GwtCompatible
@ElementTypesAreNonnullByDefault
public abstract class ForwardingSortedSet<E extends @Nullable Object> extends ForwardingSet<E>
implements SortedSet<E> {
/** Constructor for use by subclasses. */
protected ForwardingSortedSet() {}
@Override
protected abstract SortedSet<E> delegate();
@Override
@CheckForNull
public Comparator<? super E> comparator() {
return delegate().comparator();
}
@Override
@ParametricNullness
public E first() {
return delegate().first();
}
@Override
public SortedSet<E> headSet(@ParametricNullness E toElement) {
return delegate().headSet(toElement);
}
@Override
@ParametricNullness
public E last() {
return delegate().last();
}
@Override
public SortedSet<E> subSet(@ParametricNullness E fromElement, @ParametricNullness E toElement) {
return delegate().subSet(fromElement, toElement);
}
@Override
public SortedSet<E> tailSet(@ParametricNullness E fromElement) {
return delegate().tailSet(fromElement);
}
/**
* A sensible definition of {@link #contains} in terms of the {@code first()} method of {@link
* #tailSet}. If you override {@link #tailSet}, you may wish to override {@link #contains} to
* forward to this implementation.
*
* @since 7.0
*/
@Override
protected boolean standardContains(@CheckForNull Object object) {
try {
// any ClassCastExceptions and NullPointerExceptions are caught
@SuppressWarnings({"unchecked", "nullness"})
SortedSet<@Nullable Object> self = (SortedSet<@Nullable Object>) this;
Object ceiling = self.tailSet(object).first();
return unsafeCompare(comparator(), ceiling, object) == 0;
} catch (ClassCastException | NoSuchElementException | NullPointerException e) {
return false;
}
}
/**
* A sensible definition of {@link #remove} in terms of the {@code iterator()} method of {@link
* #tailSet}. If you override {@link #tailSet}, you may wish to override {@link #remove} to
* forward to this implementation.
*
* @since 7.0
*/
@Override
protected boolean standardRemove(@CheckForNull Object object) {
try {
// any ClassCastExceptions and NullPointerExceptions are caught
@SuppressWarnings({"unchecked", "nullness"})
SortedSet<@Nullable Object> self = (SortedSet<@Nullable Object>) this;
Iterator<?> iterator = self.tailSet(object).iterator();
if (iterator.hasNext()) {
Object ceiling = iterator.next();
if (unsafeCompare(comparator(), ceiling, object) == 0) {
iterator.remove();
return true;
}
}
} catch (ClassCastException | NullPointerException e) {
return false;
}
return false;
}
/**
* A sensible default implementation of {@link #subSet(Object, Object)} in terms of {@link
* #headSet(Object)} and {@link #tailSet(Object)}. In some situations, you may wish to override
* {@link #subSet(Object, Object)} to forward to this implementation.
*
* @since 7.0
*/
protected SortedSet<E> standardSubSet(
@ParametricNullness E fromElement, @ParametricNullness E toElement) {
return tailSet(fromElement).headSet(toElement);
}
}