mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-06 15:42:39 +00:00
1ccb333ac5
Since I saw the following:
0. DB::WriteBufferFromOStream::nextImpl()
1. DB::WriteBufferFromHTTPServerResponse::nextImpl()
2. DB::WriteBufferFromHTTPServerResponse::finalize()
3. DB::WriteBufferFromHTTPServerResponse::~WriteBufferFromHTTPServerResponse()
4. DB::StaticRequestHandler::handleRequest(Poco::Net::HTTPServerRequest&, Poco::Net::HTTPServerResponse&)
5. Poco::Net::HTTPServerConnection::run()
6. Poco::Net::TCPServerConnection::start()
54 lines
1.6 KiB
C++
54 lines
1.6 KiB
C++
#include <Server/PrometheusRequestHandler.h>
|
|
|
|
#include <IO/HTTPCommon.h>
|
|
#include <Server/HTTP/WriteBufferFromHTTPServerResponse.h>
|
|
#include <Server/HTTPHandlerFactory.h>
|
|
#include <Server/IServer.h>
|
|
#include <Common/CurrentMetrics.h>
|
|
#include <Common/Exception.h>
|
|
#include <Common/ProfileEvents.h>
|
|
|
|
#include <Poco/Util/LayeredConfiguration.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
void PrometheusRequestHandler::handleRequest(HTTPServerRequest & request, HTTPServerResponse & response)
|
|
{
|
|
try
|
|
{
|
|
const auto & config = server.config();
|
|
unsigned keep_alive_timeout = config.getUInt("keep_alive_timeout", 10);
|
|
|
|
setResponseDefaultHeaders(response, keep_alive_timeout);
|
|
|
|
response.setContentType("text/plain; version=0.0.4; charset=UTF-8");
|
|
|
|
WriteBufferFromHTTPServerResponse wb(response, request.getMethod() == Poco::Net::HTTPRequest::HTTP_HEAD, keep_alive_timeout);
|
|
try
|
|
{
|
|
metrics_writer.write(wb);
|
|
wb.finalize();
|
|
}
|
|
catch (...)
|
|
{
|
|
wb.finalize();
|
|
}
|
|
}
|
|
catch (...)
|
|
{
|
|
tryLogCurrentException("PrometheusRequestHandler");
|
|
}
|
|
}
|
|
|
|
HTTPRequestHandlerFactoryPtr
|
|
createPrometheusHandlerFactory(IServer & server, AsynchronousMetrics & async_metrics, const std::string & config_prefix)
|
|
{
|
|
auto factory = std::make_shared<HandlingRuleHTTPHandlerFactory<PrometheusRequestHandler>>(
|
|
server, PrometheusMetricsWriter(server.config(), config_prefix + ".handler", async_metrics));
|
|
factory->addFiltersFromConfig(server.config(), config_prefix);
|
|
return factory;
|
|
}
|
|
|
|
}
|