2019-05-31 07:27:14 +00:00
|
|
|
#pragma once
|
|
|
|
|
2019-12-06 14:37:21 +00:00
|
|
|
#include <Common/config.h>
|
|
|
|
|
|
|
|
#if USE_AWS_S3
|
|
|
|
|
2019-11-04 19:57:03 +00:00
|
|
|
#include <TableFunctions/ITableFunction.h>
|
2019-05-31 07:27:14 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
2019-09-22 22:13:42 +00:00
|
|
|
|
|
|
|
class Context;
|
|
|
|
|
2019-11-04 19:57:03 +00:00
|
|
|
/* s3(source, [access_key_id, secret_access_key,] format, structure) - creates a temporary storage for a file in S3
|
2019-05-31 07:27:14 +00:00
|
|
|
*/
|
2019-11-04 19:57:03 +00:00
|
|
|
class TableFunctionS3 : public ITableFunction
|
2019-05-31 07:27:14 +00:00
|
|
|
{
|
|
|
|
public:
|
2020-07-17 03:33:29 +00:00
|
|
|
static constexpr auto name = "s3";
|
2019-05-31 07:27:14 +00:00
|
|
|
std::string getName() const override
|
|
|
|
{
|
|
|
|
return name;
|
|
|
|
}
|
2020-10-14 12:19:29 +00:00
|
|
|
bool hasStaticStructure() const override { return true; }
|
2019-05-31 07:27:14 +00:00
|
|
|
|
2020-07-17 03:33:29 +00:00
|
|
|
protected:
|
2019-11-04 19:57:03 +00:00
|
|
|
StoragePtr executeImpl(
|
|
|
|
const ASTPtr & ast_function,
|
|
|
|
const Context & context,
|
2020-09-02 15:07:53 +00:00
|
|
|
const std::string & table_name,
|
2020-10-14 12:19:29 +00:00
|
|
|
ColumnsDescription cached_columns) const override;
|
2019-11-04 19:57:03 +00:00
|
|
|
|
2020-07-17 03:33:29 +00:00
|
|
|
const char * getStorageTypeName() const override { return "S3"; }
|
2020-10-14 12:19:29 +00:00
|
|
|
|
|
|
|
ColumnsDescription getActualTableStructure(const Context & context) const override;
|
|
|
|
void parseArguments(const ASTPtr & ast_function, const Context & context) override;
|
|
|
|
|
|
|
|
String filename;
|
|
|
|
String format;
|
|
|
|
String structure;
|
|
|
|
String access_key_id;
|
|
|
|
String secret_access_key;
|
|
|
|
String compression_method = "auto";
|
2020-07-17 03:33:29 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
class TableFunctionCOS : public TableFunctionS3
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
static constexpr auto name = "cosn";
|
|
|
|
std::string getName() const override
|
|
|
|
{
|
|
|
|
return name;
|
|
|
|
}
|
|
|
|
private:
|
|
|
|
const char * getStorageTypeName() const override { return "COSN"; }
|
2019-05-31 07:27:14 +00:00
|
|
|
};
|
2019-09-22 22:13:42 +00:00
|
|
|
|
2019-05-31 07:27:14 +00:00
|
|
|
}
|
2019-12-06 14:37:21 +00:00
|
|
|
|
2019-12-09 12:36:06 +00:00
|
|
|
#endif
|