thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-25 17:12:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

dbms: Server: Support for multiple time zones: take into account the existence of equivalent time zones [#METR-15618]

Browse Source
This commit is contained in:
Alexey Arno 2015-06-26 20:57:49 +03:00
parent 37f008afd0
commit 779a90d438
2 changed files with 40 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
libs/libcommon/src/DateLUT.cpp
View File

@ -1,8 +1,9 @@
#include <Yandex/DateLUT.h>
#include <Poco/Exception.h>
#include <Yandex/likely.h>
#include <unicode/timezone.h>
#include <unicode/unistr.h>
#include <memory>
std::string DateLUT::default_time_zone;
@ -27,13 +28,35 @@ DateLUT::DateLUT()
if (zone_id == nullptr)
throw Poco::Exception("No time zone available.");
std::vector<UnicodeString> time_zones;
while ((zone_id != nullptr) && (status == U_ZERO_ERROR))
{
std::string zone_id_str;
zone_id->toUTF8String(zone_id_str);
date_lut_impl_list.emplace(std::piecewise_construct, std::forward_as_tuple(zone_id_str), std::forward_as_tuple(nullptr));
time_zones.push_back(*zone_id);
zone_id = time_zone_ids->snext(status);
}
for (const auto & time_zone : time_zones)
{
auto count = TimeZone::countEquivalentIDs(time_zone);
const UnicodeString & u_group_id = TimeZone::getEquivalentID(time_zone, 0);
std::string group_id;
u_group_id.toUTF8String(group_id);
auto it = time_zone_to_group.find(group_id);
if (it == time_zone_to_group.end())
{
for (auto i = 1; i < count; ++i)
{
const UnicodeString & u_equivalent_id = TimeZone::getEquivalentID(time_zone, i);
std::string equivalent_id;
u_equivalent_id.toUTF8String(equivalent_id);
time_zone_to_group.insert(std::make_pair(equivalent_id, group_id));
}
}
date_lut_impl_list.emplace(std::piecewise_construct, std::forward_as_tuple(group_id), std::forward_as_tuple(nullptr));
}
}
DateLUTImpl & DateLUT::instance(const std::string & time_zone)
@ -44,16 +67,21 @@ DateLUTImpl & DateLUT::instance(const std::string & time_zone)
DateLUTImpl & DateLUT::get(const std::string & time_zone)
{
auto it = date_lut_impl_list.find(time_zone);
if (it == date_lut_impl_list.end())
auto it = time_zone_to_group.find(time_zone);
if (it == time_zone_to_group.end())
throw Poco::Exception("Invalid time zone " + time_zone);
const auto & group_id = it->second;
auto & wrapper = it->second;
auto it2 = date_lut_impl_list.find(group_id);
if (it2 == date_lut_impl_list.end())
throw Poco::Exception("Invalid group of equivalent time zones.");
auto & wrapper = it2->second;
DateLUTImpl * tmp = wrapper.load(std::memory_order_acquire);
if (tmp == nullptr)
{
std::lock_guard<std::mutex> guard(mux);
std::lock_guard<std::mutex> guard(mutex);
tmp = wrapper.load(std::memory_order_acquire);
if (tmp == nullptr)
{

7
libs/libcommon/src/DateLUTImpl.cpp
View File

@ -1,8 +1,9 @@
#include <cstring>
#include <Yandex/DateLUTImpl.h>
#include <Poco/Exception.h>
#include <glib.h>
#include <memory>
#include <cstring>
#include <glib.h>
namespace details { namespace {
@ -121,7 +122,7 @@ DateLUTImpl::DateLUTImpl(const std::string & time_zone)
while (start_of_day <= DATE_LUT_MAX);
/// Заполняем lookup таблицу для годов
memset(years_lut, 0, DATE_LUT_YEARS * sizeof(years_lut[0]));
::memset(years_lut, 0, DATE_LUT_YEARS * sizeof(years_lut[0]));
for (size_t day = 0; day < i && lut[day].year <= DATE_LUT_MAX_YEAR; ++day)
{
if (lut[day].month == 1 && lut[day].day_of_month == 1)