mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 01:12:12 +00:00
48 lines
1.6 KiB
C++
48 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#include <DataStreams/IBlockInputStream.h>
|
|
#include <Common/SharedBlockRowRef.h>
|
|
#include <Core/SortDescription.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
|
|
/** Implements the LIMIT relational operation.
|
|
*/
|
|
class LimitBlockInputStream : public IBlockInputStream
|
|
{
|
|
public:
|
|
/** If always_read_till_end = false (by default), then after reading enough data,
|
|
* returns an empty block, and this causes the query to be canceled.
|
|
* If always_read_till_end = true - reads all the data to the end, but ignores them. This is necessary in rare cases:
|
|
* when otherwise, due to the cancellation of the request, we would not have received the data for GROUP BY WITH TOTALS from the remote server.
|
|
* If use_limit_as_total_rows_approx = true, then addTotalRowsApprox is called to use the limit in progress & stats
|
|
* with_ties = true, when query has WITH TIES modifier. If so, description should be provided
|
|
* description lets us know which row we should check for equality
|
|
*/
|
|
LimitBlockInputStream(
|
|
const BlockInputStreamPtr & input, UInt64 limit_, UInt64 offset_,
|
|
bool always_read_till_end_ = false, bool use_limit_as_total_rows_approx = false,
|
|
bool with_ties_ = false, const SortDescription & description_ = {});
|
|
|
|
String getName() const override { return "Limit"; }
|
|
|
|
Block getHeader() const override { return children.at(0)->getHeader(); }
|
|
|
|
protected:
|
|
Block readImpl() override;
|
|
|
|
private:
|
|
UInt64 limit;
|
|
UInt64 offset;
|
|
UInt64 pos = 0;
|
|
bool always_read_till_end;
|
|
bool with_ties;
|
|
const SortDescription description;
|
|
SharedBlockRowRef ties_row_ref;
|
|
};
|
|
|
|
}
|