Add io stat for StarletOutputStream

Signed-off-by: PengFei Li <[email protected]>
StarRocks · Dec 26, 2024 · f345b5e · f345b5e
1 parent 7e6110f
commit f345b5e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/be/src/fs/fs_starlet.cpp b/be/src/fs/fs_starlet.cpp
@@ -39,6 +39,7 @@
 #include "io/throttled_seekable_input_stream.h"
 #include "service/staros_worker.h"
 #include "storage/olap_common.h"
+#include "util/defer_op.h"
 #include "util/stopwatch.hpp"
 #include "util/string_parser.hpp"
 
@@ -249,6 +250,9 @@ class StarletOutputStream : public starrocks::io::OutputStream {
     }
 
     Status close() override {
+        MonotonicStopWatch watch;
+        watch.start();
+        DeferOp defer([&]() { IOProfiler::add_sync(watch.elapsed_time()); });
         auto stream_st = _file_ptr->stream();
         if (!stream_st.ok()) {
             return to_status(stream_st.status());