blob: 09833ff396be7fdceb40f342a7080af34956c330 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#pragma once
#include <Processors/Transforms/ExceptionKeepingTransform.h>
#include <Storages/ConstraintsDescription.h>
#include <Interpreters/StorageID.h>
namespace DB
{
/** Check for constraints violation. If anything is found - throw an exception with detailed error message.
* Otherwise just pass block to output unchanged.
*/
class CheckConstraintsTransform final : public ExceptionKeepingTransform
{
public:
CheckConstraintsTransform(
const StorageID & table_,
const Block & header,
const ConstraintsDescription & constraints_,
ContextPtr context_);
String getName() const override { return "CheckConstraintsTransform"; }
void onConsume(Chunk chunk) override;
GenerateResult onGenerate() override
{
GenerateResult res;
res.chunk = std::move(cur_chunk);
return res;
}
private:
StorageID table_id;
const ASTs constraints_to_check;
const ConstraintsExpressions expressions;
size_t rows_written = 0;
Chunk cur_chunk;
};
}
|
