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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "addvalueupdate.h"
#include <vespa/document/base/field.h>
#include <vespa/document/datatype/arraydatatype.h>
#include <vespa/document/fieldvalue/fieldvalues.h>
#include <vespa/document/serialization/vespadocumentdeserializer.h>
#include <vespa/document/util/serializableexceptions.h>
#include <vespa/vespalib/objects/nbostream.h>
#include <vespa/vespalib/util/xmlstream.h>
#include <ostream>
using vespalib::IllegalArgumentException;
using vespalib::IllegalStateException;
using vespalib::nbostream;
using vespalib::make_string;
using namespace vespalib::xml;
namespace document {
AddValueUpdate:: AddValueUpdate(std::unique_ptr<FieldValue> value, int weight)
: ValueUpdate(Add),
_value(std::move(value)),
_weight(weight)
{}
AddValueUpdate::~AddValueUpdate() = default;
bool
AddValueUpdate::operator==(const ValueUpdate& other) const
{
if (other.getType() != Add) return false;
const AddValueUpdate& o(static_cast<const AddValueUpdate&>(other));
if (*_value != *o._value) return false;
if (_weight != o._weight) return false;
return true;
}
// Ensure that this update is compatible with given field.
void
AddValueUpdate::checkCompatibility(const Field& field) const
{
const CollectionDataType *ct = field.getDataType().cast_collection();
if (ct != nullptr) {
if (!ct->getNestedType().isValueType(*_value)) {
throw IllegalArgumentException("Cannot add value of type " + _value->getDataType()->toString() +
" to field " + field.getName() + " of container type " +
field.getDataType().toString(), VESPA_STRLOC);
}
} else {
throw IllegalArgumentException("Can not add a value to field of type" + field.getDataType().toString(), VESPA_STRLOC);
}
}
// Print this update in human readable form.
void
AddValueUpdate::print(std::ostream& out, bool, const std::string& indent) const
{
out << indent << "AddValueUpdate(" << *_value << ", " << _weight << ")";
}
// Apply this update to the given document.
bool
AddValueUpdate::applyTo(FieldValue& value) const
{
if (value.isA(FieldValue::Type::ARRAY)) {
ArrayFieldValue& doc(static_cast<ArrayFieldValue&>(value));
doc.add(*_value);
} else if (value.isA(FieldValue::Type::WSET)) {
WeightedSetFieldValue& doc(static_cast<WeightedSetFieldValue&>(value));
doc.add(*_value, _weight);
} else {
vespalib::string err = make_string("Unable to add a value to a \"%s\" field value.", value.className());
throw IllegalStateException(err, VESPA_STRLOC);
}
return true;
}
void
AddValueUpdate::printXml(XmlOutputStream& xos) const
{
xos << XmlTag("add") << XmlAttribute("weight", _weight)
<< *_value
<< XmlEndTag();
}
// Deserialize this update from the given buffer.
void
AddValueUpdate::deserialize(const DocumentTypeRepo& repo, const DataType& type, nbostream& stream)
{
const CollectionDataType *ctype = type.cast_collection();
if (ctype == nullptr) {
throw DeserializeException("Can not perform add operation on non-collection type.");
}
_value.reset(ctype->getNestedType().createFieldValue().release());
VespaDocumentDeserializer deserializer(repo, stream, Document::getNewestSerializationVersion());
deserializer.read(*_value);
stream >> _weight;
}
}
|
