document/src/vespa/document/select/context.cpp


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

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include "context.h"
#include "variablemap.h"
#include <vespa/document/select/value.h>

namespace document::select {

Context::Context()
    : _doc(NULL),
      _docId(NULL),
      _docUpdate(NULL),
      _variables()
{ }

Context::Context(const Document& doc)
    : _doc(&doc),
      _docId(NULL),
      _docUpdate(NULL),
      _variables()
{ }

Context::Context(const DocumentId& docId)
    : _doc(NULL),
      _docId(&docId),
      _docUpdate(NULL),
      _variables()
{ }

Context::Context(const DocumentUpdate& docUpdate)
    : _doc(NULL),
      _docId(NULL),
      _docUpdate(&docUpdate),
      _variables()
{ }

Context::~Context() = default;

std::unique_ptr<Value>
Context::getValue(const vespalib::string & value) const {
    if (_variables) {
        VariableMap::const_iterator iter = _variables->find(value);

        if (iter != _variables->end()) {
            return std::make_unique<FloatValue>(iter->second);
        } else {
            return std::make_unique<FloatValue>(0.0);
        }
    } else {
        return std::make_unique<FloatValue>(0.0);
    }
}

void
Context::setVariableMap(std::unique_ptr<VariableMap> map) {
    _variables = std::move(map);
}

}