mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-18 05:32:52 +00:00
76 lines
2.3 KiB
C++
76 lines
2.3 KiB
C++
#pragma once
|
||
|
||
#include <Poco/Net/HTTPServerResponse.h>
|
||
|
||
#include <DB/Common/Exception.h>
|
||
#include <DB/Core/ErrorCodes.h>
|
||
|
||
#include <DB/IO/WriteBuffer.h>
|
||
#include <DB/IO/BufferWithOwnMemory.h>
|
||
#include <DB/Common/NetException.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
|
||
/** Отличается от WriteBufferFromOStream тем, что инициализируется не std::ostream, а Poco::Net::HTTPServerResponse.
|
||
* При первом сбросе данных, получает из него std::ostream (с помощью метода send).
|
||
* Это нужно в HTTP серверах, чтобы после передачи в какой-нибудь метод WriteBuffer-а,
|
||
* но до вывода первых данных клиенту, можно было изменить какие-нибудь HTTP заголовки (например, код ответа).
|
||
* (После вызова Poco::Net::HTTPServerResponse::send() изменить заголовки уже нельзя.)
|
||
* То есть, суть в том, чтобы вызывать метод Poco::Net::HTTPServerResponse::send() не сразу.
|
||
*/
|
||
class WriteBufferFromHTTPServerResponse : public BufferWithOwnMemory<WriteBuffer>
|
||
{
|
||
private:
|
||
Poco::Net::HTTPServerResponse & response;
|
||
std::ostream * ostr = nullptr;
|
||
|
||
void nextImpl()
|
||
{
|
||
if (!ostr)
|
||
ostr = &response.send();
|
||
|
||
if (!offset())
|
||
return;
|
||
|
||
ostr->write(working_buffer.begin(), offset());
|
||
ostr->flush();
|
||
|
||
if (!ostr->good())
|
||
throw NetException("Cannot write to ostream", ErrorCodes::CANNOT_WRITE_TO_OSTREAM);
|
||
}
|
||
|
||
public:
|
||
WriteBufferFromHTTPServerResponse(Poco::Net::HTTPServerResponse & response_, size_t size = DBMS_DEFAULT_BUFFER_SIZE)
|
||
: BufferWithOwnMemory<WriteBuffer>(size), response(response_) {}
|
||
|
||
/** Если данные ещё не были отправлены - отправить хотя бы HTTP заголовки.
|
||
* Используйте эту функцию после того, как данные, возможно, были отправлены,
|
||
* и не было ошибок (вы не планируете поменять код ответа).
|
||
*/
|
||
void finalize()
|
||
{
|
||
if (!ostr)
|
||
ostr = &response.send();
|
||
}
|
||
|
||
~WriteBufferFromHTTPServerResponse()
|
||
{
|
||
if (!offset())
|
||
return;
|
||
|
||
try
|
||
{
|
||
next();
|
||
}
|
||
catch (...)
|
||
{
|
||
tryLogCurrentException(__PRETTY_FUNCTION__);
|
||
}
|
||
}
|
||
};
|
||
|
||
}
|