mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-24 18:50:49 +00:00
82 lines
2.0 KiB
C++
82 lines
2.0 KiB
C++
#pragma once
|
|
|
|
#include <common/find_symbols.h>
|
|
#include "domain.h"
|
|
#include "tldLookup.h"
|
|
|
|
namespace DB
|
|
{
|
|
|
|
struct ExtractFirstSignificantSubdomain
|
|
{
|
|
static size_t getReserveLengthForElement() { return 10; }
|
|
|
|
static void execute(const Pos data, const size_t size, Pos & res_data, size_t & res_size, Pos * out_domain_end = nullptr)
|
|
{
|
|
res_data = data;
|
|
res_size = 0;
|
|
|
|
Pos tmp;
|
|
size_t domain_length;
|
|
ExtractDomain<true>::execute(data, size, tmp, domain_length);
|
|
|
|
if (domain_length == 0)
|
|
return;
|
|
|
|
if (out_domain_end)
|
|
*out_domain_end = tmp + domain_length;
|
|
|
|
/// cut useless dot
|
|
if (tmp[domain_length - 1] == '.')
|
|
--domain_length;
|
|
|
|
res_data = tmp;
|
|
res_size = domain_length;
|
|
|
|
auto begin = tmp;
|
|
auto end = begin + domain_length;
|
|
const char * last_3_periods[3]{};
|
|
|
|
auto pos = find_first_symbols<'.'>(begin, end);
|
|
while (pos < end)
|
|
{
|
|
last_3_periods[2] = last_3_periods[1];
|
|
last_3_periods[1] = last_3_periods[0];
|
|
last_3_periods[0] = pos;
|
|
pos = find_first_symbols<'.'>(pos + 1, end);
|
|
}
|
|
|
|
if (!last_3_periods[0])
|
|
return;
|
|
|
|
if (!last_3_periods[1])
|
|
{
|
|
res_size = last_3_periods[0] - begin;
|
|
return;
|
|
}
|
|
|
|
if (!last_3_periods[2])
|
|
last_3_periods[2] = begin - 1;
|
|
|
|
auto end_of_level_domain = find_first_symbols<'/'>(last_3_periods[0], end);
|
|
if (!end_of_level_domain)
|
|
{
|
|
end_of_level_domain = end;
|
|
}
|
|
|
|
if (tldLookup::isValid(last_3_periods[1] + 1, end_of_level_domain - last_3_periods[1] - 1) != nullptr)
|
|
{
|
|
res_data += last_3_periods[2] + 1 - begin;
|
|
res_size = last_3_periods[1] - last_3_periods[2] - 1;
|
|
}
|
|
else
|
|
{
|
|
res_data += last_3_periods[1] + 1 - begin;
|
|
res_size = last_3_periods[0] - last_3_periods[1] - 1;
|
|
}
|
|
}
|
|
};
|
|
|
|
}
|
|
|