thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-30 19:42:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

add options support

Browse Source
This commit is contained in:
Artur 2021-09-22 10:34:48 +00:00
parent 82a849ba8e
commit c8892ec7a7
3 changed files with 19 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/Server/HTTPHandler.cpp
View File

@ -25,6 +25,7 @@
#include <Server/HTTPHandlerFactory.h>
#include <Server/HTTPHandlerRequestFilter.h>
#include <Server/IServer.h>
#include "common/logger_useful.h"
#include <Common/SettingsChanges.h>
#include <Common/StringUtils/StringUtils.h>
#include <Common/escapeForFileName.h>
@ -111,10 +112,11 @@ namespace ErrorCodes
namespace
{
/// Process options request. Usefull for CORS.
/// Process options request. Useful for CORS.
void processOptionsRequest(HTTPServerResponse & response, const Poco::Util::LayeredConfiguration & config)
{
/// If answer for options request was not defined, return 501 to client.
/// If response for options request was not defined, return 501 to client.
/// TODO should it be here?
if (!config.has("http_options_response"))
{
response.setStatusAndReason(HTTPResponse::HTTP_NOT_IMPLEMENTED);
@ -129,12 +131,17 @@ namespace
{
if (config_key == "header" || config_key.starts_with("header["))
{
response.add(config.getString("http_options_response." + config_key + ".name", "Empty header"),
config.getString("http_options_response." + config_key + ".value", ""));
response.setKeepAlive(false);
/// If there is empty header name, it will not be processed and message about it will be in logs
if (config.getString("http_options_response." + config_key + ".name", "").empty())
LOG_WARNING(&Poco::Logger::get("processOptionsRequest"), "Empty header was found in config. It will not be processed.");
else
response.add(config.getString("http_options_response." + config_key + ".name", ""),
config.getString("http_options_response." + config_key + ".value", ""));
}
}
response.setStatusAndReason(HTTPResponse::HTTP_NO_CONTENT);
response.setKeepAlive(false);
response.setStatusAndReason(HTTPResponse::HTTP_OK);
response.send();
}
}

4
src/Server/HTTPHandlerFactory.cpp
View File

@ -123,7 +123,7 @@ static inline HTTPRequestHandlerFactoryPtr createInterserverHTTPHandlerFactory(I
addCommonDefaultHandlersFactory(*factory, server);
auto main_handler = std::make_shared<HandlingRuleHTTPHandlerFactory<InterserverIOHTTPHandler>>(server);
main_handler->allowPostAndGetParamsRequest();
main_handler->allowPostAndGetParamsAndOptionsRequest();
factory->addHandler(main_handler);
return factory;
@ -180,7 +180,7 @@ void addDefaultHandlersFactory(HTTPRequestHandlerFactoryMain & factory, IServer
addCommonDefaultHandlersFactory(factory, server);
auto query_handler = std::make_shared<HandlingRuleHTTPHandlerFactory<DynamicQueryHandler>>(server, "query");
query_handler->allowPostAndGetParamsRequest();
query_handler->allowPostAndGetParamsAndOptionsRequest();
factory.addHandler(query_handler);
/// We check that prometheus handler will be served on current (default) port.

6
src/Server/HTTPHandlerFactory.h
View File

@ -104,11 +104,13 @@ public:
}
/// Handle POST or GET with params
void allowPostAndGetParamsRequest()
void allowPostAndGetParamsAndOptionsRequest()
{
addFilter([](const auto & request)
{
return request.getURI().find('?') != std::string::npos
return (request.getURI().find('?') != std::string::npos
&& request.getMethod() == Poco::Net::HTTPRequest::HTTP_GET)
|| request.getMethod() == Poco::Net::HTTPRequest::HTTP_OPTIONS
|| request.getMethod() == Poco::Net::HTTPRequest::HTTP_POST;
});
}