mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-13 01:41:59 +00:00
54 lines
1.6 KiB
C++
54 lines
1.6 KiB
C++
#include <TableFunctions/TableFunctionFile.h>
|
|
#include <TableFunctions/parseColumnsListForTableFunction.h>
|
|
|
|
#include "registerTableFunctions.h"
|
|
#include <Access/Common/AccessFlags.h>
|
|
#include <Interpreters/Context.h>
|
|
#include <Storages/ColumnsDescription.h>
|
|
#include <Storages/StorageFile.h>
|
|
#include <TableFunctions/TableFunctionFactory.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
StoragePtr TableFunctionFile::getStorage(const String & source,
|
|
const String & format_, const ColumnsDescription & columns,
|
|
ContextPtr global_context, const std::string & table_name,
|
|
const std::string & compression_method_) const
|
|
{
|
|
// For `file` table function, we are going to use format settings from the
|
|
// query context.
|
|
StorageFile::CommonArguments args{
|
|
WithContext(global_context),
|
|
StorageID(getDatabaseName(), table_name),
|
|
format_,
|
|
std::nullopt /*format settings*/,
|
|
compression_method_,
|
|
columns,
|
|
ConstraintsDescription{},
|
|
String{},
|
|
};
|
|
|
|
return StorageFile::create(source, global_context->getUserFilesPath(), args);
|
|
}
|
|
|
|
ColumnsDescription TableFunctionFile::getActualTableStructure(ContextPtr context) const
|
|
{
|
|
if (structure == "auto")
|
|
{
|
|
size_t total_bytes_to_read = 0;
|
|
Strings paths = StorageFile::getPathsList(filename, context->getUserFilesPath(), context, total_bytes_to_read);
|
|
return StorageFile::getTableStructureFromFile(format, paths, compression_method, std::nullopt, context);
|
|
}
|
|
|
|
|
|
return parseColumnsListFromString(structure, context);
|
|
}
|
|
|
|
void registerTableFunctionFile(TableFunctionFactory & factory)
|
|
{
|
|
factory.registerFunction<TableFunctionFile>();
|
|
}
|
|
|
|
}
|