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
|
/*
* This file is open source software, licensed to you under the terms
* of the Apache License, Version 2.0 (the "License"). See the NOTICE file
* distributed with this work for additional information regarding copyright
* ownership. You may not use this file except in compliance with the License.
*
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
/*
* Copyright (C) 2024 Scylladb, Ltd.
*/
#include <memory>
#include <fmt/core.h>
#include <seastar/http/exception.hh>
#include <seastar/http/httpd.hh>
#include <seastar/http/handlers.hh>
#include <seastar/http/function_handlers.hh>
#include <seastar/http/file_handler.hh>
#include <seastar/core/seastar.hh>
#include <seastar/core/app-template.hh>
#include <seastar/http/api_docs.hh>
#include <seastar/core/thread.hh>
#include <seastar/net/inet_address.hh>
#include <seastar/util/defer.hh>
#include "../../apps/lib/stop_signal.hh"
#include "api.json.hh"
namespace bpo = boost::program_options;
using namespace seastar;
using namespace httpd;
using namespace api_json::ns_hello_world;
void set_routes(routes& r) {
api_json::hello_world.set(r, [] (const_req req) -> json::json_return_type {
api_json::my_object obj;
obj.var1 = req.param.at("var1");
obj.var2 = req.param.at("var2");
query_enum v = str2query_enum(req.query_parameters.at("query_enum"));
// This demonstrate enum conversion
obj.enum_var = v;
std::vector<int> vec123{1, 2, 3};
obj.array_var = vec123;
obj.chunked_array_var = vec123;
auto use_streaming = req.query_parameters.at("use_streaming");
if (use_streaming != "false" && use_streaming != "true") {
throw bad_param_exception("param use_streaming must be true or false");
}
if (use_streaming == "false") {
return obj;
} else {
return stream_object(std::move(obj));
}
});
}
int main(int ac, char** av) {
app_template app;
app.add_options()("port", bpo::value<uint16_t>()->default_value(10000), "HTTP Server port");
return app.run(ac, av, [&] {
return seastar::async([&] {
seastar_apps_lib::stop_signal stop_signal;
auto&& config = app.configuration();
uint16_t port = config["port"].as<uint16_t>();
auto server = std::make_unique<http_server_control>();
auto rb = make_shared<api_registry_builder>("apps/httpd/");
server->start().get();
auto stop_server = defer([&] () noexcept {
std::cout << "Stopping HTTP server" << std::endl; // This can throw, but won't.
server->stop().get();
});
server->set_routes(set_routes).get();
server->set_routes([rb](routes& r){rb->set_api_doc(r);}).get();
server->set_routes([rb](routes& r) {rb->register_function(r, "demo", "rest api test");}).get();
server->listen(port).get();
fmt::print("{}\n", port);
fflush(stdout);
stop_signal.wait().get();
return 0;
});
});
}
|
