mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-22 00:10:52 +00:00
116 lines
3.0 KiB
C++
116 lines
3.0 KiB
C++
#include <Yandex/DateLUT.h>
|
|
#include <Poco/Exception.h>
|
|
|
|
#include <unicode/timezone.h>
|
|
#include <unicode/unistr.h>
|
|
|
|
std::string DateLUT::default_time_zone;
|
|
|
|
DateLUT::DateLUT()
|
|
{
|
|
using namespace icu;
|
|
|
|
std::unique_ptr<TimeZone> tz(TimeZone::createDefault());
|
|
if (tz == nullptr)
|
|
throw Poco::Exception("Failed to query the host time zone.");
|
|
|
|
UnicodeString u_out;
|
|
tz->getID(u_out);
|
|
u_out.toUTF8String(default_time_zone);
|
|
|
|
std::unique_ptr<StringEnumeration> time_zone_ids(TimeZone::createEnumeration());
|
|
if (time_zone_ids == nullptr)
|
|
throw Poco::Exception("Failed to query the list of time zones.");
|
|
|
|
UErrorCode status = U_ZERO_ERROR;
|
|
const UnicodeString * zone_id = time_zone_ids->snext(status);
|
|
if (zone_id == nullptr)
|
|
throw Poco::Exception("No time zone available.");
|
|
|
|
std::vector<UnicodeString> time_zones;
|
|
while ((zone_id != nullptr) && (status == U_ZERO_ERROR))
|
|
{
|
|
time_zones.push_back(*zone_id);
|
|
zone_id = time_zone_ids->snext(status);
|
|
}
|
|
|
|
size_t group_id = 0;
|
|
|
|
for (const auto & time_zone : time_zones)
|
|
{
|
|
const UnicodeString & u_group_name = TimeZone::getEquivalentID(time_zone, 0);
|
|
std::string group_name;
|
|
|
|
if (u_group_name.isEmpty())
|
|
{
|
|
time_zone.toUTF8String(group_name);
|
|
|
|
auto res = time_zone_to_group.insert(std::make_pair(group_name, group_id));
|
|
if (!res.second)
|
|
throw Poco::Exception("Failed to initialize time zone information.");
|
|
++group_id;
|
|
}
|
|
else
|
|
{
|
|
u_group_name.toUTF8String(group_name);
|
|
|
|
auto it = time_zone_to_group.find(group_name);
|
|
if (it == time_zone_to_group.end())
|
|
{
|
|
auto count = TimeZone::countEquivalentIDs(time_zone);
|
|
if (count == 0)
|
|
throw Poco::Exception("Inconsistent time zone information.");
|
|
|
|
for (auto i = 0; i < count; ++i)
|
|
{
|
|
const UnicodeString & u_name = TimeZone::getEquivalentID(time_zone, i);
|
|
std::string name;
|
|
u_name.toUTF8String(name);
|
|
auto res = time_zone_to_group.insert(std::make_pair(name, group_id));
|
|
if (!res.second)
|
|
throw Poco::Exception("Failed to initialize time zone information.");
|
|
}
|
|
++group_id;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (group_id == 0)
|
|
throw Poco::Exception("Could not find any time zone information.");
|
|
|
|
date_lut_impl_list = std::make_unique<DateLUTImplList>(group_id);
|
|
}
|
|
|
|
DateLUTImpl & DateLUT::instance(const std::string & time_zone)
|
|
{
|
|
auto & date_lut = Singleton<DateLUT>::instance();
|
|
return date_lut.get(time_zone);
|
|
}
|
|
|
|
DateLUTImpl & DateLUT::get(const std::string & time_zone)
|
|
{
|
|
const std::string & actual_time_zone = time_zone.empty() ? default_time_zone : time_zone;
|
|
|
|
auto it = time_zone_to_group.find(actual_time_zone);
|
|
if (it == time_zone_to_group.end())
|
|
throw Poco::Exception("Invalid time zone " + actual_time_zone);
|
|
|
|
const auto & group_id = it->second;
|
|
auto & wrapper = (*date_lut_impl_list)[group_id];
|
|
|
|
DateLUTImpl * tmp = wrapper.load(std::memory_order_acquire);
|
|
if (tmp == nullptr)
|
|
{
|
|
std::lock_guard<std::mutex> guard(mutex);
|
|
tmp = wrapper.load(std::memory_order_acquire);
|
|
if (tmp == nullptr)
|
|
{
|
|
tmp = new DateLUTImpl(actual_time_zone);
|
|
wrapper.store(tmp, std::memory_order_release);
|
|
}
|
|
}
|
|
|
|
return *tmp;
|
|
}
|
|
|