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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
#include "clickhouse_config.h"
#if USE_MYSQL
#include <Storages/StorageMaterializedMySQL.h>
#include <Interpreters/ExpressionAnalyzer.h>
#include <Interpreters/TreeRewriter.h>
#include <Parsers/ASTTablesInSelectQuery.h>
#include <QueryPipeline/Pipe.h>
#include <Processors/Transforms/FilterTransform.h>
#include <Databases/MySQL/DatabaseMaterializedMySQL.h>
#include <Storages/ReadFinalForExternalReplicaStorage.h>
#include <Storages/SelectQueryInfo.h>
namespace DB
{
StorageMaterializedMySQL::StorageMaterializedMySQL(const StoragePtr & nested_storage_, const IDatabase * database_)
: StorageProxy(nested_storage_->getStorageID()), nested_storage(nested_storage_), database(database_)
{
StorageInMemoryMetadata in_memory_metadata;
in_memory_metadata = nested_storage->getInMemoryMetadata();
setInMemoryMetadata(in_memory_metadata);
}
bool StorageMaterializedMySQL::needRewriteQueryWithFinal(const Names & column_names) const
{
return needRewriteQueryWithFinalForStorage(column_names, nested_storage);
}
void StorageMaterializedMySQL::read(
QueryPlan & query_plan,
const Names & column_names,
const StorageSnapshotPtr & /*storage_snapshot*/,
SelectQueryInfo & query_info,
ContextPtr context,
QueryProcessingStage::Enum processed_stage,
size_t max_block_size,
size_t num_streams)
{
if (const auto * db = typeid_cast<const DatabaseMaterializedMySQL *>(database))
db->rethrowExceptionIfNeeded();
readFinalFromNestedStorage(query_plan, nested_storage, column_names,
query_info, context, processed_stage, max_block_size, num_streams);
}
NamesAndTypesList StorageMaterializedMySQL::getVirtuals() const
{
if (const auto * db = typeid_cast<const DatabaseMaterializedMySQL *>(database))
db->rethrowExceptionIfNeeded();
return nested_storage->getVirtuals();
}
IStorage::ColumnSizeByName StorageMaterializedMySQL::getColumnSizes() const
{
auto sizes = nested_storage->getColumnSizes();
auto nested_header = nested_storage->getInMemoryMetadataPtr()->getSampleBlock();
String sign_column_name = nested_header.getByPosition(nested_header.columns() - 2).name;
String version_column_name = nested_header.getByPosition(nested_header.columns() - 1).name;
sizes.erase(sign_column_name);
sizes.erase(version_column_name);
return sizes;
}
}
#endif
|
