ClickHouse/dbms/src/Interpreters/tests/expression.cpp

#include <iostream>
#include <iomanip>

#include <Poco/Stopwatch.h>

#include <DB/IO/WriteBufferFromOStream.h>

#include <DB/Columns/ColumnString.h>

#include <DB/DataTypes/DataTypesNumberFixed.h>
#include <DB/DataTypes/DataTypeString.h>

#include <DB/Parsers/ASTSelectQuery.h>
#include <DB/Parsers/ParserSelectQuery.h>
#include <DB/Parsers/formatAST.h>

#include <DB/DataStreams/TabSeparatedRowOutputStream.h>
#include <DB/DataStreams/LimitBlockInputStream.h>
#include <DB/DataStreams/OneBlockInputStream.h>
#include <DB/DataStreams/BlockOutputStreamFromRowOutputStream.h>
#include <DB/DataStreams/copyData.h>

#include <DB/Interpreters/ExpressionAnalyzer.h>


int main(int argc, char ** argv)
{
	using namespace DB;

	try
	{
		ParserSelectQuery parser;
		ASTPtr ast;
		std::string input = "SELECT x, s1, s2, "
			"/*"
			"2 + x * 2, x * 2, x % 3 == 1, "
			"s1 == 'abc', s1 == s2, s1 != 'abc', s1 != s2, "
			"s1 <  'abc', s1 <  s2, s1 >  'abc', s1 >  s2, "
			"s1 <= 'abc', s1 <= s2, s1 >= 'abc', s1 >= s2, "
			"*/"
			"s1 < s2 AND x % 3 < x % 5";
		Expected expected = "";

		const char * begin = input.data();
		const char * end = begin + input.size();
		const char * pos = begin;

		if (parser.parse(pos, end, ast, expected))
		{
			std::cout << "Success." << std::endl;
			formatAST(*ast, std::cout);
			std::cout << std::endl << ast->getTreeID() << std::endl;
		}
		else
		{
			std::cout << "Failed at position " << (pos - begin) << ": "
				<< mysqlxx::quote << input.substr(pos - begin, 10)
				<< ", expected " << expected << "." << std::endl;
		}

		Context context;
		NamesAndTypesList columns;
		columns.emplace_back("x", new DataTypeInt16);
		columns.emplace_back("s1", new DataTypeString);
		columns.emplace_back("s2", new DataTypeString);
		context.setColumns(columns);

		ExpressionAnalyzer analyzer(ast, context, context.getColumns());
		ExpressionActionsChain chain;
		analyzer.appendSelect(chain, false);
		analyzer.appendProjectResult(chain, false);
		chain.finalize();
		ExpressionActionsPtr expression = chain.getLastActions();

		size_t n = argc == 2 ? atoi(argv[1]) : 10;

		Block block;

		ColumnWithNameAndType column_x;
		column_x.name = "x";
		column_x.type = new DataTypeInt16;
		ColumnInt16 * x = new ColumnInt16;
		column_x.column = x;
		auto & vec_x = x->getData();

		vec_x.resize(n);
		for (size_t i = 0; i < n; ++i)
			vec_x[i] = i;

		block.insert(column_x);

		const char * strings[] = {"abc", "def", "abcd", "defg", "ac"};

		ColumnWithNameAndType column_s1;
		column_s1.name = "s1";
		column_s1.type = new DataTypeString;
		column_s1.column = new ColumnString;

		for (size_t i = 0; i < n; ++i)
			column_s1.column->insert(String(strings[i % 5]));

		block.insert(column_s1);

		ColumnWithNameAndType column_s2;
		column_s2.name = "s2";
		column_s2.type = new DataTypeString;
		column_s2.column = new ColumnString;

		for (size_t i = 0; i < n; ++i)
			column_s2.column->insert(String(strings[i % 3]));

		block.insert(column_s2);

		{
			Poco::Stopwatch stopwatch;
			stopwatch.start();

			expression->execute(block);

			stopwatch.stop();
			std::cout << std::fixed << std::setprecision(2)
				<< "Elapsed " << stopwatch.elapsed() / 1000000.0 << " sec."
				<< ", " << n * 1000000 / stopwatch.elapsed() << " rows/sec."
				<< std::endl;
		}

		OneBlockInputStream * is = new OneBlockInputStream(block);
		LimitBlockInputStream lis(is, 20, std::max(0, static_cast<int>(n) - 20));
		WriteBufferFromOStream out_buf(std::cout);
		RowOutputStreamPtr os_ = new TabSeparatedRowOutputStream(out_buf, block);
		BlockOutputStreamFromRowOutputStream os(os_);

		copyData(lis, os);
	}
	catch (const Exception & e)
	{
		std::cerr << e.displayText() << std::endl;
	}

	return 0;
}