aboutsummaryrefslogtreecommitdiffstats
path: root/staging_vespalib/src/vespa/vespalib/net/json_handler_repo.cpp
blob: 97bb4a9aac7606d47f1ae78fa167281b2eec2445 (plain) (blame)
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
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include "json_handler_repo.h"
#include <algorithm>

namespace vespalib {

namespace {

template <typename T>
void remove_seq(T &collection, size_t seq) {
    collection.erase(std::remove_if(collection.begin(), collection.end(),
                                    [seq](const typename T::value_type &item)
                                    { return (item.seq == seq); }),
                     collection.end());
}

} // namespace vespalib::<unnamed>

size_t
JsonHandlerRepo::State::bind(const vespalib::stringref &path_prefix,
                             const JsonGetHandler &get_handler)
{
    std::lock_guard<std::mutex> guard(lock);
    size_t my_seq = ++seq;
    hooks.emplace_back(my_seq, path_prefix, get_handler);
    std::sort(hooks.begin(), hooks.end());
    return my_seq;
}

size_t
JsonHandlerRepo::State::add_root_resource(const vespalib::stringref &path)
{
    std::lock_guard<std::mutex> guard(lock);
    size_t my_seq = ++seq;
    root_resources.emplace_back(my_seq, path);
    return my_seq;
}

void
JsonHandlerRepo::State::unbind(size_t my_seq) {
    std::lock_guard<std::mutex> guard(lock);
    remove_seq(hooks, my_seq);
    remove_seq(root_resources, my_seq);
}

//-----------------------------------------------------------------------------

JsonHandlerRepo::Token::UP
JsonHandlerRepo::bind(const vespalib::stringref &path_prefix,
                      const JsonGetHandler &get_handler)
{
    return Token::UP(new Unbinder(_state, _state->bind(path_prefix, get_handler)));
}

JsonHandlerRepo::Token::UP
JsonHandlerRepo::add_root_resource(const vespalib::stringref &path)
{
    return Token::UP(new Unbinder(_state, _state->add_root_resource(path)));
}

std::vector<vespalib::string>
JsonHandlerRepo::get_root_resources() const
{
    std::lock_guard<std::mutex> guard(_state->lock);
    std::vector<vespalib::string> result;
    for (const Resource &resource: _state->root_resources) {
        result.push_back(resource.path);
    }
    return result;
}

vespalib::string
JsonHandlerRepo::get(const vespalib::string &host,
                     const vespalib::string &path,
                     const std::map<vespalib::string,vespalib::string> &params) const
{
    std::lock_guard<std::mutex> guard(_state->lock);
    for (const auto &hook: _state->hooks) {
        if (path.find(hook.path_prefix) == 0) {
            return hook.handler->get(host, path, params);
        }
    }
    return "";
}

} // namespace vespalib