mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-25 00:52:02 +00:00
121 lines
3.6 KiB
C++
121 lines
3.6 KiB
C++
#pragma once
|
||
|
||
#include <string>
|
||
#include <vector>
|
||
#include <Poco/Exception.h>
|
||
#include <common/Common.h>
|
||
#include <DB/Core/StringRef.h>
|
||
|
||
|
||
/** Класс, позволяющий узнавать по id региона его текстовое название на одном из поддерживаемых языков: ru, en, ua, by, kz, tr.
|
||
*
|
||
* Информацию об именах регионов загружает из текстовых файлов с названиями следующего формата:
|
||
* regions_names_xx.txt,
|
||
* где xx - одно из двух буквенных обозначений следующих поддерживаемых языков:
|
||
* ru, en, ua, by, kz, tr.
|
||
*
|
||
* Умеет, по запросу, обновлять данные.
|
||
*/
|
||
class RegionsNames
|
||
{
|
||
public:
|
||
enum class Language
|
||
{
|
||
RU = 0,
|
||
EN,
|
||
UA,
|
||
BY,
|
||
KZ,
|
||
TR,
|
||
};
|
||
|
||
private:
|
||
static const size_t ROOT_LANGUAGE = 0;
|
||
static const size_t SUPPORTED_LANGUAGES_COUNT = 6;
|
||
static const size_t LANGUAGE_ALIASES_COUNT = 7;
|
||
|
||
static const char ** getSupportedLanguages()
|
||
{
|
||
static const char * res[] { "ru", "en", "ua", "by", "kz", "tr" };
|
||
return res;
|
||
}
|
||
|
||
struct language_alias { const char * const name; const Language lang; };
|
||
static const language_alias * getLanguageAliases()
|
||
{
|
||
static constexpr const language_alias language_aliases[] {
|
||
{ "ru", Language::RU },
|
||
{ "en", Language::EN },
|
||
{ "ua", Language::UA },
|
||
{ "uk", Language::UA },
|
||
{ "by", Language::BY },
|
||
{ "kz", Language::KZ },
|
||
{ "tr", Language::TR }
|
||
};
|
||
|
||
return language_aliases;
|
||
}
|
||
|
||
using RegionID = UInt32;
|
||
|
||
using Chars = std::vector<char>;
|
||
using CharsForLanguageID = std::vector<Chars>;
|
||
using ModificationTimes = std::vector<time_t>;
|
||
using StringRefs = std::vector<StringRef>; /// Lookup table RegionID -> StringRef
|
||
using StringRefsForLanguageID = std::vector<StringRefs>;
|
||
|
||
public:
|
||
/** Перезагружает, при необходимости, имена регионов.
|
||
*/
|
||
void reload();
|
||
void reload(const std::string & directory);
|
||
|
||
/// Has corresponding section in configuration file.
|
||
static bool isConfigured();
|
||
|
||
|
||
StringRef getRegionName(RegionID region_id, Language language = Language::RU) const
|
||
{
|
||
size_t language_id = static_cast<size_t>(language);
|
||
|
||
if (region_id > names_refs[language_id].size())
|
||
return StringRef("", 0);
|
||
|
||
StringRef ref = names_refs[language_id][region_id];
|
||
|
||
while (ref.size == 0 && language_id != ROOT_LANGUAGE)
|
||
{
|
||
static const size_t FALLBACK[] = { 0, 0, 0, 0, 0, 1 };
|
||
language_id = FALLBACK[language_id];
|
||
ref = names_refs[language_id][region_id];
|
||
}
|
||
|
||
return ref;
|
||
}
|
||
|
||
static Language getLanguageEnum(const std::string & language)
|
||
{
|
||
if (language.size() == 2)
|
||
{
|
||
for (size_t i = 0; i < LANGUAGE_ALIASES_COUNT; ++i)
|
||
{
|
||
const auto & alias = getLanguageAliases()[i];
|
||
if (language[0] == alias.name[0] && language[1] == alias.name[1])
|
||
return alias.lang;
|
||
}
|
||
}
|
||
throw Poco::Exception("Unsupported language for region name. Supported languages are: " + dumpSupportedLanguagesNames() + ".");
|
||
}
|
||
|
||
private:
|
||
static std::string dumpSupportedLanguagesNames();
|
||
|
||
ModificationTimes file_modification_times = ModificationTimes(SUPPORTED_LANGUAGES_COUNT);
|
||
|
||
/// Байты имен для каждого языка, уложенные подряд, разделенные нулями
|
||
CharsForLanguageID chars = CharsForLanguageID(SUPPORTED_LANGUAGES_COUNT);
|
||
|
||
/// Отображение для каждого языка из id региона в указатель на диапазон байт имени
|
||
StringRefsForLanguageID names_refs = StringRefsForLanguageID(SUPPORTED_LANGUAGES_COUNT);
|
||
};
|