mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-10 01:25:21 +00:00
54 lines
1.6 KiB
C++
54 lines
1.6 KiB
C++
#include "ODBCHandlerFactory.h"
|
|
#include "PingHandler.h"
|
|
#include "ColumnInfoHandler.h"
|
|
#include "config.h"
|
|
#include <Poco/URI.h>
|
|
#include <Poco/Net/HTTPServerRequest.h>
|
|
#include <Common/logger_useful.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
ODBCBridgeHandlerFactory::ODBCBridgeHandlerFactory(const std::string & name_, ContextPtr context_)
|
|
: WithContext(context_), log(getLogger(name_)), name(name_)
|
|
{
|
|
}
|
|
|
|
std::unique_ptr<HTTPRequestHandler> ODBCBridgeHandlerFactory::createRequestHandler(const HTTPServerRequest & request)
|
|
{
|
|
Poco::URI uri{request.getURI()};
|
|
LOG_TRACE(log, "Request URI: {}", uri.toString());
|
|
|
|
if (uri.getPath() == "/ping" && request.getMethod() == Poco::Net::HTTPRequest::HTTP_GET)
|
|
return std::make_unique<PingHandler>();
|
|
|
|
if (request.getMethod() == Poco::Net::HTTPRequest::HTTP_POST)
|
|
{
|
|
|
|
if (uri.getPath() == "/columns_info")
|
|
#if USE_ODBC
|
|
return std::make_unique<ODBCColumnsInfoHandler>(getContext());
|
|
#else
|
|
return nullptr;
|
|
#endif
|
|
else if (uri.getPath() == "/identifier_quote")
|
|
#if USE_ODBC
|
|
return std::make_unique<IdentifierQuoteHandler>(getContext());
|
|
#else
|
|
return nullptr;
|
|
#endif
|
|
else if (uri.getPath() == "/schema_allowed")
|
|
#if USE_ODBC
|
|
return std::make_unique<SchemaAllowedHandler>(getContext());
|
|
#else
|
|
return nullptr;
|
|
#endif
|
|
else if (uri.getPath() == "/write")
|
|
return std::make_unique<ODBCHandler>(getContext(), "write");
|
|
else
|
|
return std::make_unique<ODBCHandler>(getContext(), "read");
|
|
}
|
|
return nullptr;
|
|
}
|
|
}
|