mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 17:32:32 +00:00
51 lines
1.1 KiB
C++
51 lines
1.1 KiB
C++
#pragma once
|
|
#include <Processors/ISimpleTransform.h>
|
|
#include <DataStreams/SizeLimits.h>
|
|
#include <Poco/Timespan.h>
|
|
#include <Interpreters/ProcessList.h>
|
|
|
|
#include <DataStreams/StreamLocalLimits.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/// Information for profiling.
|
|
struct ProcessorProfileInfo
|
|
{
|
|
bool started = false;
|
|
Stopwatch total_stopwatch {CLOCK_MONOTONIC_COARSE}; /// Time with waiting time
|
|
|
|
size_t rows = 0;
|
|
size_t blocks = 0;
|
|
size_t bytes = 0;
|
|
|
|
void update(const Chunk & block);
|
|
};
|
|
|
|
class LimitsCheckingTransform : public ISimpleTransform
|
|
{
|
|
public:
|
|
|
|
LimitsCheckingTransform(const Block & header_, StreamLocalLimits limits_);
|
|
|
|
String getName() const override { return "LimitsCheckingTransform"; }
|
|
|
|
void setQuota(const std::shared_ptr<const EnabledQuota> & quota_) { quota = quota_; }
|
|
|
|
protected:
|
|
void transform(Chunk & chunk) override;
|
|
|
|
private:
|
|
StreamLocalLimits limits;
|
|
|
|
std::shared_ptr<const EnabledQuota> quota;
|
|
UInt64 prev_elapsed = 0;
|
|
|
|
ProcessorProfileInfo info;
|
|
|
|
bool checkTimeLimit();
|
|
void checkQuota(Chunk & chunk);
|
|
};
|
|
|
|
}
|