Movec cache utility types to zenutil and fixed unit tests.

1 files changed, 161 insertions, 0 deletions

diff --git a/zenutil/cache/cachepolicy.cpp b/zenutil/cache/cachepolicy.cpp
new file mode 100644
index 000000000..968f6aa53
--- /dev/null
+++ b/zenutil/cache/cachepolicy.cpp
@@ -0,0 +1,161 @@
+// Copyright Epic Games, Inc. All Rights Reserved.
+
+#include <zenutil/cache/cachepolicy.h>
+
+#include <zencore/compactbinary.h>
+#include <zencore/compactbinarybuilder.h>
+#include <zencore/string.h>
+
+namespace zen {
+
+using namespace std::literals;
+
+namespace detail { namespace cacheopt {
+	constexpr std::string_view Local	   = "local"sv;
+	constexpr std::string_view Remote	   = "remote"sv;
+	constexpr std::string_view Data		   = "data"sv;
+	constexpr std::string_view Meta		   = "meta"sv;
+	constexpr std::string_view Value	   = "value"sv;
+	constexpr std::string_view Attachments = "attachments"sv;
+}}	// namespace detail::cacheopt
+
+CachePolicy
+ParseQueryCachePolicy(std::string_view QueryPolicy, CachePolicy Default)
+{
+	if (QueryPolicy.empty())
+	{
+		return Default;
+	}
+
+	CachePolicy Result = CachePolicy::None;
+
+	ForEachStrTok(QueryPolicy, ',', [&Result](const std::string_view& Token) {
+		if (Token == detail::cacheopt::Local)
+		{
+			Result |= CachePolicy::QueryLocal;
+		}
+		if (Token == detail::cacheopt::Remote)
+		{
+			Result |= CachePolicy::QueryRemote;
+		}
+		return true;
+	});
+
+	return Result;
+}
+
+CachePolicy
+ParseStoreCachePolicy(std::string_view StorePolicy, CachePolicy Default)
+{
+	if (StorePolicy.empty())
+	{
+		return Default;
+	}
+
+	CachePolicy Result = CachePolicy::None;
+
+	ForEachStrTok(StorePolicy, ',', [&Result](const std::string_view& Token) {
+		if (Token == detail::cacheopt::Local)
+		{
+			Result |= CachePolicy::StoreLocal;
+		}
+		if (Token == detail::cacheopt::Remote)
+		{
+			Result |= CachePolicy::StoreRemote;
+		}
+		return true;
+	});
+
+	return Result;
+}
+
+CachePolicy
+ParseSkipCachePolicy(std::string_view SkipPolicy, CachePolicy Default)
+{
+	if (SkipPolicy.empty())
+	{
+		return Default;
+	}
+
+	CachePolicy Result = CachePolicy::None;
+
+	ForEachStrTok(SkipPolicy, ',', [&Result](const std::string_view& Token) {
+		if (Token == detail::cacheopt::Meta)
+		{
+			Result |= CachePolicy::SkipMeta;
+		}
+		if (Token == detail::cacheopt::Value)
+		{
+			Result |= CachePolicy::SkipValue;
+		}
+		if (Token == detail::cacheopt::Attachments)
+		{
+			Result |= CachePolicy::SkipAttachments;
+		}
+		if (Token == detail::cacheopt::Data)
+		{
+			Result |= CachePolicy::SkipData;
+		}
+		return true;
+	});
+
+	return Result;
+}
+
+CachePolicy
+CacheRecordPolicy::GetPayloadPolicy(const Oid& PayloadId) const
+{
+	if (const auto It = m_PayloadPolicies.find(PayloadId); It != m_PayloadPolicies.end())
+	{
+		return It->second;
+	}
+
+	return m_DefaultPayloadPolicy;
+}
+
+bool
+CacheRecordPolicy::Load(CbObjectView RecordPolicyObject, CacheRecordPolicy& OutRecordPolicy)
+{
+	using namespace std::literals;
+
+	const uint32_t RecordPolicy = RecordPolicyObject["RecordPolicy"sv].AsUInt32(static_cast<uint32_t>(CachePolicy::Default));
+	const uint32_t DefaultPayloadPolicy =
+		RecordPolicyObject["DefaultPayloadPolicy"sv].AsUInt32(static_cast<uint32_t>(CachePolicy::Default));
+
+	OutRecordPolicy.m_RecordPolicy		   = static_cast<CachePolicy>(RecordPolicy);
+	OutRecordPolicy.m_DefaultPayloadPolicy = static_cast<CachePolicy>(DefaultPayloadPolicy);
+
+	for (CbFieldView PayloadPolicyView : RecordPolicyObject["PayloadPolicies"sv])
+	{
+		CbObjectView   PayloadPolicyObject = PayloadPolicyView.AsObjectView();
+		const Oid	   PayloadId		   = PayloadPolicyObject["Id"sv].AsObjectId();
+		const uint32_t PayloadPolicy	   = PayloadPolicyObject["Policy"sv].AsUInt32();
+
+		if (PayloadId != Oid::Zero && PayloadPolicy != 0)
+		{
+			OutRecordPolicy.m_PayloadPolicies.emplace(PayloadId, static_cast<CachePolicy>(PayloadPolicy));
+		}
+	}
+
+	return true;
+}
+
+void
+CacheRecordPolicy::Save(const CacheRecordPolicy& Policy, CbWriter& Writer)
+{
+	Writer << "RecordPolicy"sv << static_cast<uint32_t>(Policy.GetRecordPolicy());
+	Writer << "DefaultPayloadPolicy"sv << static_cast<uint32_t>(Policy.GetDefaultPayloadPolicy());
+
+	if (!Policy.m_PayloadPolicies.empty())
+	{
+		Writer.BeginArray("PayloadPolicies"sv);
+		for (const auto& Kv : Policy.m_PayloadPolicies)
+		{
+			Writer.AddObjectId("Id"sv, Kv.first);
+			Writer << "Policy"sv << static_cast<uint32_t>(Kv.second);
+		}
+		Writer.EndArray();
+	}
+}
+
+}  // namespace zen