ClickHouse/src/Storages/StorageS3Cluster.h

64 lines
1.6 KiB
C++
Raw Normal View History

2021-03-19 21:49:18 +00:00
#pragma once
#include "config.h"
2021-03-24 21:02:21 +00:00
#if USE_AWS_S3
2021-06-15 19:55:21 +00:00
#include <memory>
#include <optional>
2021-03-19 21:49:18 +00:00
#include "Client/Connection.h"
2021-04-08 00:09:15 +00:00
#include <Interpreters/Cluster.h>
#include <IO/S3Common.h>
#include <Storages/IStorageCluster.h>
2021-04-08 00:09:15 +00:00
#include <Storages/StorageS3.h>
2021-03-19 21:49:18 +00:00
namespace DB
{
class Context;
class StorageS3Cluster : public IStorageCluster
2021-03-19 21:49:18 +00:00
{
public:
struct Configuration : public StorageS3::Configuration
{
std::string cluster_name;
};
2021-04-12 21:42:52 +00:00
StorageS3Cluster(
const Configuration & configuration_,
2021-04-23 12:18:23 +00:00
const StorageID & table_id_,
const ColumnsDescription & columns_,
const ConstraintsDescription & constraints_,
ContextPtr context_,
bool structure_argument_was_provided_);
2021-03-19 21:49:18 +00:00
std::string getName() const override { return "S3Cluster"; }
Pipe read(const Names &, const StorageSnapshotPtr &, SelectQueryInfo &,
ContextPtr, QueryProcessingStage::Enum, size_t /*max_block_size*/, size_t /*num_streams*/) override;
QueryProcessingStage::Enum
getQueryProcessingStage(ContextPtr, QueryProcessingStage::Enum, const StorageSnapshotPtr &, SelectQueryInfo &) const override;
NamesAndTypesList getVirtuals() const override;
RemoteQueryExecutor::Extension getTaskIteratorExtension(ASTPtr query, ContextPtr context) const override;
ClusterPtr getCluster(ContextPtr context) const override;
2021-03-19 21:49:18 +00:00
private:
StorageS3::Configuration s3_configuration;
2021-04-12 19:35:26 +00:00
String cluster_name;
2021-03-22 17:12:31 +00:00
String format_name;
String compression_method;
2022-05-19 11:18:58 +00:00
NamesAndTypesList virtual_columns;
Block virtual_block;
bool structure_argument_was_provided;
2021-03-19 21:49:18 +00:00
};
}
2021-03-24 21:02:21 +00:00
#endif