2009-05-28 18:19:51 +00:00
|
|
|
#include <Yandex/DateLUT.h>
|
2014-07-08 23:52:53 +00:00
|
|
|
#include <Poco/Exception.h>
|
2015-06-26 17:57:49 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
#include <unicode/timezone.h>
|
|
|
|
#include <unicode/unistr.h>
|
2009-05-28 18:19:51 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
std::string DateLUT::default_time_zone;
|
2009-05-28 18:19:51 +00:00
|
|
|
|
2013-08-11 03:40:14 +00:00
|
|
|
DateLUT::DateLUT()
|
2009-05-28 18:19:51 +00:00
|
|
|
{
|
2015-06-26 15:11:31 +00:00
|
|
|
using namespace icu;
|
2014-11-06 02:39:48 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
std::unique_ptr<TimeZone> tz(TimeZone::createDefault());
|
|
|
|
if (tz == nullptr)
|
|
|
|
throw Poco::Exception("Failed to query the host time zone.");
|
2014-11-06 02:39:48 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
UnicodeString u_out;
|
|
|
|
tz->getID(u_out);
|
|
|
|
u_out.toUTF8String(default_time_zone);
|
2014-11-06 02:39:48 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
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.");
|
2013-08-11 03:40:14 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
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.");
|
2013-08-11 03:40:14 +00:00
|
|
|
|
2015-06-26 17:57:49 +00:00
|
|
|
std::vector<UnicodeString> time_zones;
|
2015-06-26 15:11:31 +00:00
|
|
|
while ((zone_id != nullptr) && (status == U_ZERO_ERROR))
|
|
|
|
{
|
2015-06-26 17:57:49 +00:00
|
|
|
time_zones.push_back(*zone_id);
|
2015-06-26 15:11:31 +00:00
|
|
|
zone_id = time_zone_ids->snext(status);
|
|
|
|
}
|
2015-06-26 17:57:49 +00:00
|
|
|
|
2015-06-29 13:54:08 +00:00
|
|
|
size_t group_id = 0;
|
|
|
|
|
2015-06-26 17:57:49 +00:00
|
|
|
for (const auto & time_zone : time_zones)
|
|
|
|
{
|
2015-06-29 13:54:08 +00:00
|
|
|
const UnicodeString & u_group_name = TimeZone::getEquivalentID(time_zone, 0);
|
|
|
|
std::string group_name;
|
2015-06-26 17:57:49 +00:00
|
|
|
|
2015-06-29 13:54:08 +00:00
|
|
|
if (u_group_name.isEmpty())
|
2015-06-29 11:54:57 +00:00
|
|
|
{
|
2015-06-29 13:54:08 +00:00
|
|
|
time_zone.toUTF8String(group_name);
|
2015-06-29 11:54:57 +00:00
|
|
|
|
2015-06-29 13:54:08 +00:00
|
|
|
auto res = time_zone_to_group.insert(std::make_pair(group_name, group_id));
|
2015-06-29 11:54:57 +00:00
|
|
|
if (!res.second)
|
|
|
|
throw Poco::Exception("Failed to initialize time zone information.");
|
2015-06-29 13:54:08 +00:00
|
|
|
++group_id;
|
2015-06-29 11:54:57 +00:00
|
|
|
}
|
|
|
|
else
|
2015-06-26 17:57:49 +00:00
|
|
|
{
|
2015-06-29 13:54:08 +00:00
|
|
|
u_group_name.toUTF8String(group_name);
|
2015-06-29 11:54:57 +00:00
|
|
|
|
2015-06-29 13:54:08 +00:00
|
|
|
auto it = time_zone_to_group.find(group_name);
|
2015-06-29 11:54:57 +00:00
|
|
|
if (it == time_zone_to_group.end())
|
2015-06-26 17:57:49 +00:00
|
|
|
{
|
2015-06-29 11:54:57 +00:00
|
|
|
auto count = TimeZone::countEquivalentIDs(time_zone);
|
2015-06-29 13:54:08 +00:00
|
|
|
if (count == 0)
|
|
|
|
throw Poco::Exception("Inconsistent time zone information.");
|
|
|
|
|
2015-06-29 11:54:57 +00:00
|
|
|
for (auto i = 0; i < count; ++i)
|
|
|
|
{
|
2015-06-29 13:54:08 +00:00
|
|
|
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));
|
2015-06-29 11:54:57 +00:00
|
|
|
if (!res.second)
|
|
|
|
throw Poco::Exception("Failed to initialize time zone information.");
|
|
|
|
}
|
2015-06-29 13:54:08 +00:00
|
|
|
++group_id;
|
2015-06-26 17:57:49 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-06-29 13:54:08 +00:00
|
|
|
|
|
|
|
if (group_id == 0)
|
|
|
|
throw Poco::Exception("Could not find any time zone information.");
|
|
|
|
|
|
|
|
date_lut_impl_list = std::make_unique<DateLUTImplList>(group_id);
|
2015-06-26 15:11:31 +00:00
|
|
|
}
|
2013-08-11 03:40:14 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
DateLUTImpl & DateLUT::instance(const std::string & time_zone)
|
|
|
|
{
|
|
|
|
auto & date_lut = Singleton<DateLUT>::instance();
|
|
|
|
return date_lut.get(time_zone);
|
|
|
|
}
|
2013-08-11 03:40:14 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
DateLUTImpl & DateLUT::get(const std::string & time_zone)
|
|
|
|
{
|
2015-06-26 17:57:49 +00:00
|
|
|
auto it = time_zone_to_group.find(time_zone);
|
|
|
|
if (it == time_zone_to_group.end())
|
2015-06-26 15:11:31 +00:00
|
|
|
throw Poco::Exception("Invalid time zone " + time_zone);
|
2015-06-26 17:57:49 +00:00
|
|
|
|
2015-06-29 13:54:08 +00:00
|
|
|
const auto & group_id = it->second;
|
|
|
|
auto & wrapper = (*date_lut_impl_list)[group_id];
|
2013-08-11 03:40:14 +00:00
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
DateLUTImpl * tmp = wrapper.load(std::memory_order_acquire);
|
|
|
|
if (tmp == nullptr)
|
2013-08-11 03:40:14 +00:00
|
|
|
{
|
2015-06-26 17:57:49 +00:00
|
|
|
std::lock_guard<std::mutex> guard(mutex);
|
2015-06-26 15:11:31 +00:00
|
|
|
tmp = wrapper.load(std::memory_order_acquire);
|
|
|
|
if (tmp == nullptr)
|
|
|
|
{
|
2015-06-29 13:54:08 +00:00
|
|
|
tmp = new DateLUTImpl(time_zone);
|
2015-06-26 15:11:31 +00:00
|
|
|
wrapper.store(tmp, std::memory_order_release);
|
|
|
|
}
|
2013-08-11 03:40:14 +00:00
|
|
|
}
|
|
|
|
|
2015-06-26 15:11:31 +00:00
|
|
|
return *tmp;
|
2009-05-28 18:19:51 +00:00
|
|
|
}
|
2015-06-26 15:11:31 +00:00
|
|
|
|