ClickHouse/dbms/src/Storages/MergeTree/RemoteQueryExecutor.cpp

#include <DB/Storages/MergeTree/RemoteQueryExecutor.h>
#include <DB/Interpreters/executeQuery.h>
#include <DB/IO/ReadBufferFromHTTP.h>
#include <DB/IO/ReadHelpers.h>
#include <DB/IO/WriteHelpers.h>

namespace DB
{

namespace ErrorCodes
{
	extern const int ABORTED;
}

namespace RemoteQueryExecutor
{

namespace
{

std::string getEndpointId(const std::string & node_id)
{
	return "RemoteQueryExecutor:" + node_id;
}

}

Service::Service(Context & context_)
	: context{context_}
{
}

std::string Service::getId(const std::string & node_id) const
{
	return getEndpointId(node_id);
}

void Service::processQuery(const Poco::Net::HTMLForm & params, ReadBuffer & body, WriteBuffer & out)
{
	if (is_cancelled)
		throw Exception{"RemoteQueryExecutor service terminated", ErrorCodes::ABORTED};

	std::string query = params.get("query");

	bool flag = true;

	try
	{
		(void) executeQuery(query, context, true);
	}
	catch (...)
	{
		tryLogCurrentException(__PRETTY_FUNCTION__);
		flag = false;
	}

	writeBinary(flag, out);
	out.next();
}

bool Client::executeQuery(const InterserverIOEndpointLocation & location, const std::string & query)
{
	ReadBufferFromHTTP::Params params =
	{
		{"endpoint", getEndpointId(location.name)},
		{"compress", "false"},
		{"query", query}
	};

	ReadBufferFromHTTP in{location.host, location.port, params};

	bool flag;
	readBinary(flag, in);
	assertEOF(in);

	return flag;
}

}

}