intermediate changes

ref:fd45b675f102830e5f7311e0b3c6aaa6ee7d4eaa

1 files changed, 18 insertions, 0 deletions

diff --git a/library/cpp/lwtrace/control.h b/library/cpp/lwtrace/control.h
index 16b24eafd2..b7c47aaec6 100644
--- a/library/cpp/lwtrace/control.h
+++ b/library/cpp/lwtrace/control.h
@@ -200,6 +200,12 @@ namespace NLWTrace {
             CyclicDepot.ReadItems(r);
             DurationDepot.ReadItems(now, duration, r);
         }
+
+        template <class TReader>
+        void ExtractItemsFromCyclicDepot(TReader& r) const {
+            CyclicDepot.ExtractItems(r);
+        }
+
         void ToProtobuf(TLogPb& pb) const;
     };
 
@@ -310,6 +316,18 @@ namespace NLWTrace {
             }
         }
 
+        template <class TReader>
+        void ExtractItemsFromCyclicDepot(const TString& id, TReader& reader) {
+            TGuard<TMutex> g(Mtx);
+            TTraces::iterator it = Traces.find(id);
+            if (it == Traces.end()) {
+                ythrow yexception() << "trace id '" << id << "' is not used";
+            } else {
+                TSession* trace = it->second;
+                trace->ExtractItemsFromCyclicDepot(reader);
+            }
+        }
+
         bool GetDestructiveActionsAllowed() {
             return DestructiveActionsAllowed;
         }