mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-04 21:42:39 +00:00
99 lines
2.4 KiB
C++
99 lines
2.4 KiB
C++
#pragma once
|
|
|
|
#include <Common/config.h>
|
|
|
|
#if USE_AWS_S3
|
|
|
|
#include "Client/Connection.h"
|
|
#include <Interpreters/Cluster.h>
|
|
#include <IO/S3Common.h>
|
|
#include <Storages/StorageS3.h>
|
|
|
|
#include <memory>
|
|
#include <optional>
|
|
#include "ext/shared_ptr_helper.h"
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int LOGICAL_ERROR;
|
|
}
|
|
|
|
|
|
class Context;
|
|
|
|
struct ClientAuthentificationBuilder
|
|
{
|
|
String access_key_id;
|
|
String secret_access_key;
|
|
UInt64 max_connections;
|
|
};
|
|
|
|
class StorageS3Distributed : public ext::shared_ptr_helper<StorageS3Distributed>, public IStorage
|
|
{
|
|
friend struct ext::shared_ptr_helper<StorageS3Distributed>;
|
|
public:
|
|
std::string getName() const override { return "S3Distributed"; }
|
|
|
|
Pipe read(
|
|
const Names & /*column_names*/,
|
|
const StorageMetadataPtr & /*metadata_snapshot*/,
|
|
SelectQueryInfo & /*query_info*/,
|
|
const Context & /*context*/,
|
|
QueryProcessingStage::Enum /*processed_stage*/,
|
|
size_t /*max_block_size*/,
|
|
unsigned /*num_streams*/) override;
|
|
|
|
QueryProcessingStage::Enum getQueryProcessingStage(const Context & context, QueryProcessingStage::Enum /*to_stage*/, SelectQueryInfo &) const override;
|
|
|
|
NamesAndTypesList getVirtuals() const override;
|
|
|
|
protected:
|
|
StorageS3Distributed(
|
|
const String & filename_,
|
|
const String & access_key_id_,
|
|
const String & secret_access_key_,
|
|
const StorageID & table_id_,
|
|
String cluster_name_,
|
|
const String & format_name_,
|
|
UInt64 max_connections_,
|
|
const ColumnsDescription & columns_,
|
|
const ConstraintsDescription & constraints_,
|
|
const Context & context_,
|
|
const String & compression_method_);
|
|
|
|
private:
|
|
/// Connections from initiator to other nodes
|
|
std::vector<std::shared_ptr<Connection>> connections;
|
|
StorageS3::ClientAuthentificaiton client_auth;
|
|
|
|
String filename;
|
|
std::string cluster_name;
|
|
ClusterPtr cluster;
|
|
|
|
String format_name;
|
|
String compression_method;
|
|
|
|
|
|
struct DistributedFileIterator : public StorageS3Source::FileIterator
|
|
{
|
|
DistributedFileIterator(ReadTaskCallback callback_, String identifier_)
|
|
: callback(callback_), identifier(identifier_) {}
|
|
|
|
ReadTaskCallback callback;
|
|
String identifier;
|
|
|
|
std::optional<String> next() override
|
|
{
|
|
return callback(identifier);
|
|
}
|
|
};
|
|
};
|
|
|
|
|
|
}
|
|
|
|
#endif
|