Intermediate changes

commit_hash:1169df9928696fd31869ba48a97f31688946588d

1 files changed, 19 insertions, 2 deletions

diff --git a/yt/yt/library/profiling/solomon/encoder.cpp b/yt/yt/library/profiling/solomon/encoder.cpp
index 91b7ca897e9..1266737dc41 100644
--- a/yt/yt/library/profiling/solomon/encoder.cpp
+++ b/yt/yt/library/profiling/solomon/encoder.cpp
@@ -39,10 +39,22 @@ void TSensorEncoder::Close()
 { }
 
 void TSensorEncoder::OnStreamBegin()
-{ }
+{
+    Cubes_.clear();
+    CommonTagIds_.clear();
+    TagRegistry_ = TTagRegistry();
+}
 
 void TSensorEncoder::OnStreamEnd()
-{ }
+{
+    for (auto& [_, cube] : Cubes_) {
+        std::visit(
+            [] (auto&& cube) {
+                cube.FinishIteration();
+            },
+            cube);
+    }
+}
 
 void TSensorEncoder::OnCommonTime(TInstant /*time*/)
 { }
@@ -96,6 +108,11 @@ void TSensorEncoder::OnLabelsEnd()
                 THROW_ERROR_EXCEPTION("Unsupported metric type %Qv", ToString(SensorContext_->Type));
         }
         cube = Cubes_.FindPtr(SensorContext_->Name);
+        std::visit(
+            [] (auto&& cube) {
+                cube.StartIteration();
+            },
+            *cube);
     }
 
     SensorContext_->Cube = cube;