2016-06-07 08:23:15 +00:00
|
|
|
#include <iostream>
|
|
|
|
#include <iomanip>
|
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <DataTypes/DataTypesNumber.h>
|
|
|
|
#include <DataTypes/DataTypeString.h>
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Columns/ColumnsNumber.h>
|
|
|
|
#include <Columns/ColumnString.h>
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <DataStreams/OneBlockInputStream.h>
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <Interpreters/Aggregator.h>
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 09:19:00 +00:00
|
|
|
#include <AggregateFunctions/AggregateFunctionFactory.h>
|
2016-06-07 08:23:15 +00:00
|
|
|
|
|
|
|
|
|
|
|
int main(int argc, char ** argv)
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
using namespace DB;
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
try
|
|
|
|
{
|
|
|
|
size_t n = argc == 2 ? atoi(argv[1]) : 10;
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
Block block;
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
{
|
|
|
|
ColumnWithTypeAndName column;
|
|
|
|
column.name = "x";
|
|
|
|
column.type = std::make_shared<DataTypeInt16>();
|
|
|
|
auto col = ColumnInt16::create();
|
|
|
|
auto & vec_x = col->getData();
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
vec_x.resize(n);
|
|
|
|
for (size_t i = 0; i < n; ++i)
|
|
|
|
vec_x[i] = i % 9;
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
column.column = std::move(col);
|
|
|
|
block.insert(column);
|
|
|
|
}
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
const char * strings[] = {"abc", "def", "abcd", "defg", "ac"};
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
{
|
|
|
|
ColumnWithTypeAndName column;
|
|
|
|
column.name = "s1";
|
|
|
|
column.type = std::make_shared<DataTypeString>();
|
|
|
|
auto col = ColumnString::create();
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
for (size_t i = 0; i < n; ++i)
|
|
|
|
col->insert(std::string(strings[i % 5]));
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
column.column = std::move(col);
|
|
|
|
block.insert(column);
|
|
|
|
}
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
{
|
|
|
|
ColumnWithTypeAndName column;
|
|
|
|
column.name = "s2";
|
|
|
|
column.type = std::make_shared<DataTypeString>();
|
|
|
|
auto col = ColumnString::create();
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
for (size_t i = 0; i < n; ++i)
|
|
|
|
col->insert(std::string(strings[i % 3]));
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-12-16 06:19:21 +00:00
|
|
|
column.column = std::move(col);
|
|
|
|
block.insert(column);
|
|
|
|
}
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
BlockInputStreamPtr stream = std::make_shared<OneBlockInputStream>(block);
|
|
|
|
AggregatedDataVariants aggregated_data_variants;
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
AggregateFunctionFactory factory;
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
AggregateDescriptions aggregate_descriptions(1);
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
DataTypes empty_list_of_types;
|
|
|
|
aggregate_descriptions[0].function = factory.get("count", empty_list_of_types);
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2018-01-06 18:10:44 +00:00
|
|
|
Aggregator::Params params(
|
|
|
|
stream->getHeader(), {0, 1}, aggregate_descriptions,
|
2019-08-27 00:23:07 +00:00
|
|
|
false, 0, OverflowMode::THROW, 0, 0, 0, false, "", 1, 0);
|
2018-01-06 18:10:44 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
Aggregator aggregator(params);
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
{
|
|
|
|
Stopwatch stopwatch;
|
|
|
|
stopwatch.start();
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
aggregator.execute(stream, aggregated_data_variants);
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
stopwatch.stop();
|
|
|
|
std::cout << std::fixed << std::setprecision(2)
|
|
|
|
<< "Elapsed " << stopwatch.elapsedSeconds() << " sec."
|
|
|
|
<< ", " << n / stopwatch.elapsedSeconds() << " rows/sec."
|
|
|
|
<< std::endl;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (const Exception & e)
|
|
|
|
{
|
|
|
|
std::cerr << e.displayText() << std::endl;
|
|
|
|
}
|
2016-06-07 08:23:15 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
return 0;
|
2016-06-07 08:23:15 +00:00
|
|
|
}
|