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
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.document.select.rule;
import com.yahoo.collections.BobHash;
import com.yahoo.document.*;
import com.yahoo.document.datatypes.FieldPathIteratorHandler;
import com.yahoo.document.datatypes.FieldValue;
import com.yahoo.document.select.*;
import java.util.ArrayList;
import java.util.List;
/**
* @author Simon Thoresen Hult
*/
public class AttributeNode implements ExpressionNode {
private ExpressionNode value;
private final List<Item> items = new ArrayList<Item>();
public AttributeNode(ExpressionNode value, List items) {
this.value = value;
for (Object obj : items) {
if (obj instanceof Item) {
this.items.add((Item)obj);
} else {
throw new IllegalStateException("Can not add an instance of " + obj.getClass().getName() +
" as a function item.");
}
}
}
public ExpressionNode getValue() {
return value;
}
public AttributeNode setValue(ExpressionNode value) {
this.value = value;
return this;
}
public List<Item> getItems() {
return items;
}
// Inherit doc from ExpressionNode.
public BucketSet getBucketSet(BucketIdFactory factory) {
return null;
}
// Inherit doc from ExpressionNode.
public Object evaluate(Context context) {
String pos = value.toString();
Object obj = value.evaluate(context);
StringBuilder builder = new StringBuilder();
for (Item item : items) {
if (obj == null) {
throw new IllegalStateException("Can not invoke '" + item + "' on '" + pos + "' because that term " +
"evaluated to null.");
}
if (item.getType() != Item.FUNCTION) {
if (builder.length() > 0) {
builder.append(".");
}
builder.append(item.getName());
} else {
if (builder.length() > 0) {
obj = evaluateFieldPath(builder.toString(), obj);
builder = new StringBuilder();
}
obj = evaluateFunction(item.getName(), obj);
}
pos = pos + "." + item;
}
if (builder.length() > 0) {
obj = evaluateFieldPath(builder.toString(), obj);
}
return obj;
}
public static class VariableValueList extends ArrayList<ResultList.VariableValue> {
}
static class IteratorHandler extends FieldPathIteratorHandler {
VariableValueList values = new VariableValueList();
@Override
public void onPrimitive(FieldValue fv) {
values.add(new ResultList.VariableValue((VariableMap)getVariables().clone(), fv));
}
}
private static Object applyFunction(String function, Object value) {
if (function.equalsIgnoreCase("abs")) {
if (Number.class.isInstance(value)) {
Number nValue = (Number)value;
if (value instanceof Double) {
return nValue.doubleValue() * (nValue.doubleValue() < 0 ? -1 : 1);
} else if (value instanceof Float) {
return nValue.floatValue() * (nValue.floatValue() < 0 ? -1 : 1);
} else if (value instanceof Long) {
return nValue.longValue() * (nValue.longValue() < 0 ? -1 : 1);
} else if (value instanceof Integer) {
return nValue.intValue() * (nValue.intValue() < 0 ? -1 : 1);
}
}
throw new IllegalStateException("Function 'abs' is only available for numerical values.");
} else if (function.equalsIgnoreCase("hash")) {
return BobHash.hash(value.toString());
} else if (function.equalsIgnoreCase("lowercase")) {
return value.toString().toLowerCase();
} else if (function.equalsIgnoreCase("uppercase")) {
return value.toString().toUpperCase();
}
throw new IllegalStateException("Function '" + function + "' is not supported.");
}
private static Object evaluateFieldPath(String fieldPth, Object value) {
if (value instanceof DocumentPut) {
final Document doc = ((DocumentPut) value).getDocument();
FieldPath fieldPath = doc.getDataType().buildFieldPath(fieldPth);
IteratorHandler handler = new IteratorHandler();
doc.iterateNested(fieldPath, 0, handler);
if (handler.values.isEmpty()) {
return null;
}
return handler.values;
} else if (value instanceof DocumentUpdate) {
return Result.INVALID;
} else if (value instanceof DocumentRemove) {
return Result.INVALID;
} else if (value instanceof DocumentGet) {
return Result.INVALID;
}
return Result.FALSE;
}
private static Object evaluateFunction(String function, Object value) {
if (value instanceof VariableValueList) {
VariableValueList retVal = new VariableValueList();
for (ResultList.VariableValue val : ((VariableValueList)value)) {
retVal.add(new ResultList.VariableValue(
(FieldPathIteratorHandler.VariableMap)val.getVariables().clone(),
applyFunction(function, val.getValue())));
}
return retVal;
}
return applyFunction(function, value);
}
public void accept(Visitor visitor) {
visitor.visit(this);
}
@Override
public String toString() {
StringBuilder ret = new StringBuilder();
ret.append(value);
for (Item item : items) {
ret.append(".").append(item);
}
return ret.toString();
}
public OrderingSpecification getOrdering(int order) {
return null;
}
public static class Item {
public static final int ATTRIBUTE = 0;
public static final int FUNCTION = 1;
private String name;
private int type = ATTRIBUTE;
public Item(String name) {
this.name = name;
}
public String getName() {
return name;
}
public Item setName(String name) {
this.name = name;
return this;
}
public int getType() {
return type;
}
public Item setType(int type) {
this.type = type;
return this;
}
@Override public String toString() {
return name + (type == FUNCTION ? "()" : "");
}
}
}
|
