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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.document;
import com.yahoo.document.datatypes.FieldValue;
import com.yahoo.document.datatypes.MapFieldValue;
import java.util.ArrayList;
import java.util.List;
/**
* Represents a map type.
*
* @author vegardh
*/
public class MapDataType extends DataType {
private DataType keyType;
private DataType valueType;
public MapDataType(DataType keyType, DataType valueType, int id) {
super("Map<"+keyType.getName()+","+valueType.getName()+">", id);
this.keyType=keyType;
this.valueType = valueType;
}
public MapDataType(DataType keyType, DataType valueType) {
this(keyType, valueType, 0);
setId(getName().toLowerCase().hashCode());
}
@Override
public MapDataType clone() {
MapDataType type = (MapDataType)super.clone();
type.keyType = keyType.clone();
type.valueType = valueType.clone();
return type;
}
@Override
protected FieldValue createByReflection(Object arg) { return null; }
@Override
public boolean isValueCompatible(FieldValue value) {
return value.getDataType().equals(this);
}
public DataType getKeyType() {
return keyType;
}
public DataType getValueType() {
return valueType;
}
@Override
public MapFieldValue createFieldValue() {
return new MapFieldValue(this);
}
@Override
public Class getValueClass() {
return MapFieldValue.class;
}
@Override
protected void register(DocumentTypeManager manager,
List<DataType> seenTypes) {
seenTypes.add(this);
if (!seenTypes.contains(getKeyType())) {
getKeyType().register(manager, seenTypes);
}
if (!seenTypes.contains(getValueType())) {
getValueType().register(manager, seenTypes);
}
super.register(manager, seenTypes);
}
public static FieldPath buildFieldPath(String remainFieldName, DataType keyType, DataType valueType) {
if (remainFieldName.length() > 0 && remainFieldName.charAt(0) == '{') {
FieldPathEntry.KeyParseResult result = FieldPathEntry.parseKey(remainFieldName);
String keyValue = result.parsed;
FieldPath path = valueType.buildFieldPath(skipDotInString(remainFieldName, result.consumedChars - 1));
List<FieldPathEntry> tmpPath = new ArrayList<FieldPathEntry>(path.getList());
if (remainFieldName.charAt(1) == '$') {
tmpPath.add(0, FieldPathEntry.newVariableLookupEntry(keyValue.substring(1), valueType));
} else {
FieldValue fv = keyType.createFieldValue();
fv.assign(keyValue);
tmpPath.add(0, FieldPathEntry.newMapLookupEntry(fv, valueType));
}
return new FieldPath(tmpPath);
} else if (remainFieldName.startsWith("key")) {
FieldPath path = keyType.buildFieldPath(skipDotInString(remainFieldName, 2));
List<FieldPathEntry> tmpPath = new ArrayList<FieldPathEntry>(path.getList());
tmpPath.add(0, FieldPathEntry.newAllKeysLookupEntry(keyType));
return new FieldPath(tmpPath);
} else if (remainFieldName.startsWith("value")) {
FieldPath path = valueType.buildFieldPath(skipDotInString(remainFieldName, 4));
List<FieldPathEntry> tmpPath = new ArrayList<FieldPathEntry>(path.getList());
tmpPath.add(0, FieldPathEntry.newAllValuesLookupEntry(valueType));
return new FieldPath(tmpPath);
}
return keyType.buildFieldPath(remainFieldName);
}
@Override
public FieldPath buildFieldPath(String remainFieldName) {
return buildFieldPath(remainFieldName, getKeyType(), getValueType());
}
@Override
public boolean isMultivalue() { return true; }
}
|