1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.hadoop.hbase.filter;
21
22 import com.google.common.base.Preconditions;
23 import org.apache.hadoop.hbase.classification.InterfaceAudience;
24
25 import org.apache.hadoop.hbase.Cell;
26 import org.apache.hadoop.hbase.classification.InterfaceStability;
27 import org.apache.hadoop.hbase.protobuf.ProtobufUtil;
28 import org.apache.hadoop.hbase.protobuf.generated.FilterProtos;
29 import org.apache.hadoop.hbase.protobuf.generated.HBaseProtos;
30 import org.apache.hadoop.hbase.protobuf.generated.HBaseProtos.CompareType;
31 import org.apache.hadoop.hbase.util.Bytes;
32
33 import java.util.ArrayList;
34 import java.util.Objects;
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51 @InterfaceAudience.Public
52 @InterfaceStability.Stable
53 public abstract class CompareFilter extends FilterBase {
54
55
56 @InterfaceAudience.Public
57 @InterfaceStability.Stable
58 public enum CompareOp {
59
60 LESS,
61
62 LESS_OR_EQUAL,
63
64 EQUAL,
65
66 NOT_EQUAL,
67
68 GREATER_OR_EQUAL,
69
70 GREATER,
71
72 NO_OP,
73 }
74
75 protected CompareOp compareOp;
76 protected ByteArrayComparable comparator;
77
78
79
80
81
82
83 public CompareFilter(final CompareOp compareOp,
84 final ByteArrayComparable comparator) {
85 this.compareOp = compareOp;
86 this.comparator = comparator;
87 }
88
89
90
91
92 public CompareOp getOperator() {
93 return compareOp;
94 }
95
96
97
98
99 public ByteArrayComparable getComparator() {
100 return comparator;
101 }
102
103 protected boolean doCompare(final CompareOp compareOp,
104 final ByteArrayComparable comparator, final byte [] data,
105 final int offset, final int length) {
106 if (compareOp == CompareOp.NO_OP) {
107 return true;
108 }
109 int compareResult = comparator.compareTo(data, offset, length);
110 switch (compareOp) {
111 case LESS:
112 return compareResult <= 0;
113 case LESS_OR_EQUAL:
114 return compareResult < 0;
115 case EQUAL:
116 return compareResult != 0;
117 case NOT_EQUAL:
118 return compareResult == 0;
119 case GREATER_OR_EQUAL:
120 return compareResult > 0;
121 case GREATER:
122 return compareResult >= 0;
123 default:
124 throw new RuntimeException("Unknown Compare op " +
125 compareOp.name());
126 }
127 }
128
129
130
131 @Override
132 public Cell transformCell(Cell v) {
133 return v;
134 }
135
136
137 public static ArrayList<Object> extractArguments(ArrayList<byte []> filterArguments) {
138 Preconditions.checkArgument(filterArguments.size() == 2,
139 "Expected 2 but got: %s", filterArguments.size());
140 CompareOp compareOp = ParseFilter.createCompareOp(filterArguments.get(0));
141 ByteArrayComparable comparator = ParseFilter.createComparator(
142 ParseFilter.removeQuotesFromByteArray(filterArguments.get(1)));
143
144 if (comparator instanceof RegexStringComparator ||
145 comparator instanceof SubstringComparator) {
146 if (compareOp != CompareOp.EQUAL &&
147 compareOp != CompareOp.NOT_EQUAL) {
148 throw new IllegalArgumentException ("A regexstring comparator and substring comparator" +
149 " can only be used with EQUAL and NOT_EQUAL");
150 }
151 }
152 ArrayList<Object> arguments = new ArrayList<Object>();
153 arguments.add(compareOp);
154 arguments.add(comparator);
155 return arguments;
156 }
157
158
159
160
161 FilterProtos.CompareFilter convert() {
162 FilterProtos.CompareFilter.Builder builder =
163 FilterProtos.CompareFilter.newBuilder();
164 HBaseProtos.CompareType compareOp = CompareType.valueOf(this.compareOp.name());
165 builder.setCompareOp(compareOp);
166 if (this.comparator != null) builder.setComparator(ProtobufUtil.toComparator(this.comparator));
167 return builder.build();
168 }
169
170
171
172
173
174
175
176 @Override
177 boolean areSerializedFieldsEqual(Filter o) {
178 if (o == this) return true;
179 if (!(o instanceof CompareFilter)) return false;
180
181 CompareFilter other = (CompareFilter)o;
182 return this.getOperator().equals(other.getOperator()) &&
183 (this.getComparator() == other.getComparator()
184 || this.getComparator().areSerializedFieldsEqual(other.getComparator()));
185 }
186
187 @Override
188 public String toString() {
189 return String.format("%s (%s, %s)",
190 this.getClass().getSimpleName(),
191 this.compareOp.name(),
192 Bytes.toStringBinary(this.comparator.getValue()));
193 }
194
195 @Override
196 public boolean equals(Object obj) {
197 return obj instanceof Filter && areSerializedFieldsEqual((Filter) obj);
198 }
199
200 @Override
201 public int hashCode() {
202 return Objects.hash(this.getComparator(), this.getOperator());
203 }
204 }