ClickHouse/libs/libcommon/src/DateLUT.cpp

#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 determine 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;
}