mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-09 18:10:48 +00:00
97f2a2213e
* Move some code outside dbms/src folder * Fix paths
58 lines
1.7 KiB
C++
58 lines
1.7 KiB
C++
#include "HierarchyFormatReader.h"
|
|
|
|
#include <IO/ReadHelpers.h>
|
|
#include <IO/WriteHelpers.h>
|
|
|
|
|
|
bool RegionsHierarchyFormatReader::readNext(RegionEntry & entry)
|
|
{
|
|
while (!input->eof())
|
|
{
|
|
/** Our internal geobase has negative numbers,
|
|
* that means "this is garbage, ignore this row".
|
|
*/
|
|
Int32 read_region_id = 0;
|
|
Int32 read_parent_id = 0;
|
|
Int8 read_type = 0;
|
|
|
|
DB::readIntText(read_region_id, *input);
|
|
DB::assertChar('\t', *input);
|
|
DB::readIntText(read_parent_id, *input);
|
|
DB::assertChar('\t', *input);
|
|
DB::readIntText(read_type, *input);
|
|
|
|
/** Then there can be a newline (old version)
|
|
* or tab, the region's population, line feed (new version).
|
|
*/
|
|
RegionPopulation population = 0;
|
|
if (!input->eof() && *input->position() == '\t')
|
|
{
|
|
++input->position();
|
|
UInt64 population_big = 0;
|
|
DB::readIntText(population_big, *input);
|
|
population = population_big > std::numeric_limits<RegionPopulation>::max() ? std::numeric_limits<RegionPopulation>::max()
|
|
: population_big;
|
|
}
|
|
DB::assertChar('\n', *input);
|
|
|
|
if (read_region_id <= 0 || read_type < 0)
|
|
continue;
|
|
|
|
RegionID region_id = read_region_id;
|
|
RegionID parent_id = 0;
|
|
|
|
if (read_parent_id >= 0)
|
|
parent_id = read_parent_id;
|
|
|
|
RegionType type = static_cast<RegionType>(read_type);
|
|
|
|
entry.id = region_id;
|
|
entry.parent_id = parent_id;
|
|
entry.type = type;
|
|
entry.population = population;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|