mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-27 03:45:11 +00:00
8fccdb77a0
The route object is the url dispatcher, it uses it matching rules to find what handler should handle a request, perform the call and handle exceptions that hanend during the handling. Signed-off-by: Amnon Heiman <amnon@cloudius-systems.com>
115 lines
3.4 KiB
C++
115 lines
3.4 KiB
C++
/*
|
|
* 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 2015 Cloudius Systems
|
|
*/
|
|
|
|
#include "routes.hh"
|
|
#include "reply.hh"
|
|
#include "exception.hh"
|
|
|
|
namespace httpd {
|
|
|
|
using namespace std;
|
|
|
|
void verify_param(const request& req, const sstring& param) {
|
|
if (req.get_query_param(param) == "") {
|
|
throw missing_param_exception(param);
|
|
}
|
|
}
|
|
|
|
routes::~routes() {
|
|
for (int i = 0; i < NUM_OPERATION; i++) {
|
|
for (auto kv : _map[i]) {
|
|
delete kv.second;
|
|
}
|
|
}
|
|
for (int i = 0; i < NUM_OPERATION; i++) {
|
|
for (auto r : _rules[i]) {
|
|
delete r;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
void routes::handle(const sstring& path, request& req, reply& rep) {
|
|
handler_base* handler = get_handler(str2type(req._method),
|
|
normalize_url(path), req.param);
|
|
if (handler != nullptr) {
|
|
try {
|
|
for (auto& i : handler->_mandatory_param) {
|
|
verify_param(req, i);
|
|
}
|
|
handler->handle(path, &req.param, req, rep);
|
|
} catch (const redirect_exception& _e) {
|
|
rep.add_header("Location", _e.url).set_status(_e.status()).done(
|
|
"json");
|
|
return;
|
|
} catch (const base_exception& _e) {
|
|
json_exception e(_e);
|
|
rep.set_status(_e.status(), e.to_json()).done("json");
|
|
} catch (exception& _e) {
|
|
json_exception e(_e);
|
|
cerr << "exception was caught for " << path << ": " << _e.what()
|
|
<< endl;
|
|
rep.set_status(reply::status_type::internal_server_error,
|
|
e.to_json()).done("json");
|
|
return;
|
|
}
|
|
} else {
|
|
json_exception ex(not_found_exception("Not found"));
|
|
rep.set_status(reply::status_type::not_found, ex.to_json()).done(
|
|
"json");
|
|
}
|
|
}
|
|
|
|
sstring routes::normalize_url(const sstring& url) {
|
|
if (url.length() < 2 || url.at(url.length() - 1) != '/') {
|
|
return url;
|
|
}
|
|
return url.substr(0, url.length() - 1);
|
|
}
|
|
|
|
handler_base* routes::get_handler(operation_type type, const sstring& url,
|
|
parameters& params) {
|
|
handler_base* handler = get_exact_match(type, url);
|
|
if (handler != nullptr) {
|
|
return handler;
|
|
}
|
|
|
|
for (auto rule = _rules[type].cbegin(); rule != _rules[type].cend();
|
|
++rule) {
|
|
handler = (*rule)->get(url, params);
|
|
if (handler != nullptr) {
|
|
return handler;
|
|
}
|
|
params.clear();
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
routes& routes::add(operation_type type, const url& url,
|
|
handler_base* handler) {
|
|
match_rule* rule = new match_rule(handler);
|
|
rule->add_str(url._path);
|
|
rule->add_param(url._param, true);
|
|
return add(rule, type);
|
|
}
|
|
|
|
}
|