ClickHouse/dbms/src/Parsers/ParserSampleRatio.cpp

#include <common/exp10.h>

#include <DB/Parsers/CommonParsers.h>
#include <DB/Parsers/ParserSampleRatio.h>
#include <DB/Parsers/ASTSampleRatio.h>
#include <DB/IO/ReadHelpers.h>


namespace DB
{


static bool parseDecimal(IParser::Pos & pos, IParser::Pos end, ASTSampleRatio::Rational & res, IParser::Pos & max_parsed_pos)
{
	ParserWhiteSpaceOrComments ws;
	ws.ignore(pos, end);

	UInt64 num_before = 0;
	UInt64 num_after = 0;
	Int64 exponent = 0;

	IParser::Pos pos_after_first_num = tryReadIntText(num_before, pos, end);

	bool has_num_before_point = pos_after_first_num > pos;
	pos = pos_after_first_num;
	bool has_point = pos < end && *pos == '.';

	if (has_point)
		++pos;

	if (!has_num_before_point && !has_point)
		return false;

	size_t number_of_digits_after_point = 0;

	if (has_point)
	{
		IParser::Pos pos_after_second_num = tryReadIntText(num_after, pos, end);
		number_of_digits_after_point = pos_after_second_num - pos;
		pos = pos_after_second_num;
	}

	bool has_exponent = pos < end && (*pos == 'e' || *pos == 'E');

	if (has_exponent)
	{
		++pos;
		IParser::Pos pos_after_exponent = tryReadIntText(exponent, pos, end);

		if (pos_after_exponent == pos)
			return false;

		pos = pos_after_exponent;
	}

	res.numerator = num_before * exp10(number_of_digits_after_point) + num_after;
	res.denominator = exp10(number_of_digits_after_point);

	if (exponent > 0)
		res.numerator *= exp10(exponent);
	if (exponent < 0)
		res.denominator *= exp10(-exponent);

	/// NOTE Удаление общих степеней десяти из числителя и знаменателя - не нужно.
	return true;
}


/** Возможные варианты:
  *
  * 12345
  * - целое число
  *
  * 0.12345
  * .12345
  * 0.
  * - дробь в обычной десятичной записи
  *
  * 1.23e-1
  * - дробь в инженерной десятичной записи
  *
  * 123 / 456
  * - дробь с произвольным знаменателем
  *
  * На всякий случай, в числителе и знаменателе дроби, поддерживаем предыдущие случаи.
  * Пример:
  * 123.0 / 456e0
  */
bool ParserSampleRatio::parseImpl(IParser::Pos & pos, IParser::Pos end, ASTPtr & node, IParser::Pos & max_parsed_pos, Expected & expected)
{
	auto begin = pos;

	ParserWhiteSpaceOrComments ws;

	ASTSampleRatio::Rational numerator;
	ASTSampleRatio::Rational denominator;
	ASTSampleRatio::Rational res;

	ws.ignore(pos, end);

	if (!parseDecimal(pos, end, numerator, max_parsed_pos))
		return false;

	ws.ignore(pos, end);

	bool has_slash = pos < end && *pos == '/';

	if (has_slash)
	{
		++pos;
		ws.ignore(pos, end);

		if (!parseDecimal(pos, end, denominator, max_parsed_pos))
			return false;

		res.numerator = numerator.numerator * denominator.denominator;
		res.denominator = numerator.denominator * denominator.numerator;
	}
	else
	{
		res = numerator;
	}

	ws.ignore(pos, end);

	node = std::make_shared<ASTSampleRatio>(StringRange(begin, pos), res);
	return true;
}

}