2020-01-30 20:12:00 +00:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <TableFunctions/ITableFunction.h>
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
2020-03-06 02:12:18 +00:00
|
|
|
|
2023-01-12 22:29:23 +00:00
|
|
|
/* generateRandom([structure, max_array_length, max_string_length, random_seed])
|
2020-03-06 02:12:18 +00:00
|
|
|
* - creates a temporary storage that generates columns with random data
|
2020-01-30 20:12:00 +00:00
|
|
|
*/
|
2020-03-06 02:12:18 +00:00
|
|
|
class TableFunctionGenerateRandom : public ITableFunction
|
2020-01-30 20:12:00 +00:00
|
|
|
{
|
|
|
|
public:
|
2020-03-06 02:12:18 +00:00
|
|
|
static constexpr auto name = "generateRandom";
|
2020-01-30 20:12:00 +00:00
|
|
|
std::string getName() const override { return name; }
|
2023-01-12 18:33:47 +00:00
|
|
|
bool hasStaticStructure() const override { return structure != "auto"; }
|
|
|
|
|
|
|
|
bool needStructureHint() const override { return structure == "auto"; }
|
|
|
|
void setStructureHint(const ColumnsDescription & structure_hint_) override { structure_hint = structure_hint_; }
|
|
|
|
|
2020-01-30 20:12:00 +00:00
|
|
|
private:
|
2021-04-10 23:33:54 +00:00
|
|
|
StoragePtr executeImpl(const ASTPtr & ast_function, ContextPtr context, const std::string & table_name, ColumnsDescription cached_columns) const override;
|
2020-04-06 05:19:40 +00:00
|
|
|
const char * getStorageTypeName() const override { return "GenerateRandom"; }
|
2020-10-14 12:19:29 +00:00
|
|
|
|
2021-04-10 23:33:54 +00:00
|
|
|
ColumnsDescription getActualTableStructure(ContextPtr context) const override;
|
|
|
|
void parseArguments(const ASTPtr & ast_function, ContextPtr context) override;
|
2020-10-14 12:19:29 +00:00
|
|
|
|
2023-01-12 18:33:47 +00:00
|
|
|
String structure = "auto";
|
2020-10-14 12:19:29 +00:00
|
|
|
UInt64 max_string_length = 10;
|
|
|
|
UInt64 max_array_length = 10;
|
|
|
|
std::optional<UInt64> random_seed;
|
2023-01-12 18:33:47 +00:00
|
|
|
ColumnsDescription structure_hint;
|
2020-01-30 20:12:00 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
}
|