thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-24 18:50:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed errors

Browse Source
This commit is contained in:
Alexey Milovidov 2020-03-06 05:34:35 +03:00
parent ca24156be9
commit 1212339023
1 changed files with 31 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
dbms/src/Storages/StorageGenerateRandom.cpp
View File

@ -63,9 +63,23 @@ ColumnPtr fillColumnWithRandomData(
{
case TypeIndex::String:
{
Block block{ColumnWithTypeAndName{nullptr, type, "result"}};
FunctionFactory::instance().get("randomPrintableASCII", context)->build({})->execute(block, {}, 0, limit);
return block.getByPosition(0).column;
auto size_column = ColumnUInt32::create();
auto & sizes = size_column->getData();
sizes.resize(limit);
for (UInt64 i = 0; i < limit; ++i)
sizes[i] = static_cast<UInt32>(rng()) % max_string_length; /// Slow
ColumnWithTypeAndName argument{std::move(size_column), std::make_shared<DataTypeUInt32>(), "size"};
Block block
{
argument,
{nullptr, type, "result"}
};
FunctionFactory::instance().get("randomPrintableASCII", context)->build({argument})->execute(block, {0}, 1, limit);
return block.getByPosition(1).column;
}
case TypeIndex::Enum8:
@ -238,8 +252,7 @@ ColumnPtr fillColumnWithRandomData(
fillBufferWithRandomData(reinterpret_cast<char *>(column_concrete.getData().data()), limit * sizeof(Decimal32), rng);
return column;
}
case TypeIndex::Decimal64: [[fallthrough]];
case TypeIndex::DateTime64:
case TypeIndex::Decimal64:
{
auto column = type->createColumn();
auto & column_concrete = typeid_cast<ColumnDecimal<Decimal64> &>(*column);
@ -255,6 +268,19 @@ ColumnPtr fillColumnWithRandomData(
fillBufferWithRandomData(reinterpret_cast<char *>(column_concrete.getData().data()), limit * sizeof(Decimal128), rng);
return column;
}
case TypeIndex::DateTime64:
{
auto column = type->createColumn();
auto & column_concrete = typeid_cast<ColumnDecimal<Decimal64> &>(*column);
column_concrete.getData().resize(limit);
UInt64 range = (1ULL << 32) * intExp10(typeid_cast<const DataTypeDateTime64 &>(*type).getScale());
for (size_t i = 0; i < limit; ++i)
column_concrete.getData()[i] = rng() % range; /// Slow
return column;
}
default:
throw Exception("The 'GenerateRandom' is not implemented for type " + type->getName(), ErrorCodes::NOT_IMPLEMENTED);