From 9313768464789551f287304eadb2fb3f139cdf18 Mon Sep 17 00:00:00 2001
From: Per Larsson <p3rl@protonmail.com>
Date: Wed, 15 Dec 2021 20:12:41 +0100
Subject: Added trace scopes.

---
 zenserver/upstream/upstreamcache.cpp | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'zenserver/upstream/upstreamcache.cpp')

diff --git a/zenserver/upstream/upstreamcache.cpp b/zenserver/upstream/upstreamcache.cpp
index 6283457e3..38179f490 100644
--- a/zenserver/upstream/upstreamcache.cpp
+++ b/zenserver/upstream/upstreamcache.cpp
@@ -13,6 +13,7 @@
 #include <zencore/stats.h>
 #include <zencore/stream.h>
 #include <zencore/timer.h>
+#include <zencore/trace.h>
 
 #include <zenstore/cas.h>
 #include <zenstore/cidstore.h>
@@ -72,6 +73,8 @@ namespace detail {
 
 		virtual GetUpstreamCacheResult GetCacheRecord(CacheKey CacheKey, ZenContentType Type) override
 		{
+			ZEN_TRACE_CPU("Upstream::Horde::GetSingleCacheRecord");
+
 			try
 			{
 				CloudCacheSession Session(m_Client);
@@ -149,6 +152,8 @@ namespace detail {
 													   const CacheRecordPolicy&	  Policy,
 													   OnCacheRecordGetComplete&& OnComplete) override
 		{
+			ZEN_TRACE_CPU("Upstream::Horde::GetCacheRecords");
+
 			ZEN_UNUSED(Policy);
 
 			CloudCacheSession	   Session(m_Client);
@@ -203,6 +208,8 @@ namespace detail {
 
 		virtual GetUpstreamCacheResult GetCachePayload(const CacheKey&, const IoHash& PayloadId) override
 		{
+			ZEN_TRACE_CPU("Upstream::Horde::GetSingleCachePayload");
+
 			try
 			{
 				CloudCacheSession	   Session(m_Client);
@@ -232,6 +239,8 @@ namespace detail {
 														std::span<size_t>			 RequestIndex,
 														OnCachePayloadGetComplete&&	 OnComplete) override final
 		{
+			ZEN_TRACE_CPU("Upstream::Horde::GetCachePayloads");
+
 			CloudCacheSession	   Session(m_Client);
 			GetUpstreamCacheResult Result;
 
@@ -259,6 +268,8 @@ namespace detail {
 													  IoBuffer					 RecordValue,
 													  std::span<IoBuffer const>	 Payloads) override
 		{
+			ZEN_TRACE_CPU("Upstream::Horde::PutCacheRecord");
+
 			using namespace fmt::literals;
 
 			ZEN_ASSERT(CacheRecord.PayloadIds.size() == Payloads.size());
@@ -529,6 +540,8 @@ namespace detail {
 
 		virtual GetUpstreamCacheResult GetCacheRecord(CacheKey CacheKey, ZenContentType Type) override
 		{
+			ZEN_TRACE_CPU("Upstream::Zen::GetSingleCacheRecord");
+
 			try
 			{
 				ZenStructuredCacheSession Session(*m_Client);
@@ -559,6 +572,8 @@ namespace detail {
 													   const CacheRecordPolicy&	  Policy,
 													   OnCacheRecordGetComplete&& OnComplete) override
 		{
+			ZEN_TRACE_CPU("Upstream::Zen::GetCacheRecords");
+
 			std::vector<size_t> IndexMap;
 			IndexMap.reserve(KeyIndex.size());
 
@@ -623,6 +638,8 @@ namespace detail {
 
 		virtual GetUpstreamCacheResult GetCachePayload(const CacheKey& CacheKey, const IoHash& PayloadId) override
 		{
+			ZEN_TRACE_CPU("Upstream::Zen::GetSingleCachePayload");
+
 			try
 			{
 				ZenStructuredCacheSession Session(*m_Client);
@@ -652,6 +669,8 @@ namespace detail {
 														std::span<size_t>			 RequestIndex,
 														OnCachePayloadGetComplete&&	 OnComplete) override final
 		{
+			ZEN_TRACE_CPU("Upstream::Zen::GetCachePayloads");
+
 			std::vector<size_t> IndexMap;
 			IndexMap.reserve(RequestIndex.size());
 
@@ -736,6 +755,8 @@ namespace detail {
 													  IoBuffer					 RecordValue,
 													  std::span<IoBuffer const>	 Payloads) override
 		{
+			ZEN_TRACE_CPU("Upstream::Zen::PutCacheRecord");
+
 			ZEN_ASSERT(CacheRecord.PayloadIds.size() == Payloads.size());
 			const int32_t MaxAttempts = 3;
 
@@ -1020,6 +1041,8 @@ public:
 
 	virtual GetUpstreamCacheResult GetCacheRecord(CacheKey CacheKey, ZenContentType Type) override
 	{
+		ZEN_TRACE_CPU("Upstream::GetCacheRecord");
+
 		if (m_Options.ReadUpstream)
 		{
 			for (auto& Endpoint : m_Endpoints)
@@ -1053,6 +1076,8 @@ public:
 								 const CacheRecordPolicy&	Policy,
 								 OnCacheRecordGetComplete&& OnComplete) override final
 	{
+		ZEN_TRACE_CPU("Upstream::GetCacheRecords");
+
 		std::vector<size_t> MissingKeys(KeyIndex.begin(), KeyIndex.end());
 
 		if (m_Options.ReadUpstream)
@@ -1098,6 +1123,8 @@ public:
 								  std::span<size_t>			   RequestIndex,
 								  OnCachePayloadGetComplete&&  OnComplete) override final
 	{
+		ZEN_TRACE_CPU("Upstream::GetCachePayloads");
+
 		std::vector<size_t> MissingPayloads(RequestIndex.begin(), RequestIndex.end());
 
 		if (m_Options.ReadUpstream)
@@ -1142,6 +1169,8 @@ public:
 
 	virtual GetUpstreamCacheResult GetCachePayload(const CacheKey& CacheKey, const IoHash& PayloadId) override
 	{
+		ZEN_TRACE_CPU("Upstream::GetCachePayload");
+
 		if (m_Options.ReadUpstream)
 		{
 			for (auto& Endpoint : m_Endpoints)
@@ -1224,6 +1253,8 @@ public:
 private:
 	void ProcessCacheRecord(UpstreamCacheRecord CacheRecord)
 	{
+		ZEN_TRACE_CPU("Upstream::ProcessCacheRecord");
+
 		ZenCacheValue		  CacheValue;
 		std::vector<IoBuffer> Payloads;
 
-- 
cgit v1.2.3