methods for print()

Author: youge325

src/file_manager.cpp

@@ -53,4 +53,37 @@ void FileManerger::setFileName(const std::string& value) {
   std::unique_lock<std::shared_mutex> lock(mutex_);
   file_name_ = value;
 }
+
+void FileManerger::captureStdout(std::function<void()> func) {
+  std::unique_lock<std::shared_mutex> lock(mutex_);
+
+  if (!file_stream_.is_open()) {
+    throw std::runtime_error(
+        "File stream is not open. Call createFile() first.");
+  }
+
+  // 保存原来的 cout buffer
+  std::streambuf* old_cout_buf = std::cout.rdbuf();
+
+  // 创建一个 stringstream 来捕获输出
+  std::stringstream captured_output;
+
+  // 重定向 cout 到 stringstream
+  std::cout.rdbuf(captured_output.rdbuf());
+
+  try {
+    // 执行函数
+    func();
+
+    // 恢复 cout
+    std::cout.rdbuf(old_cout_buf);
+
+    // 将捕获的输出写入文件
+    file_stream_ << captured_output.str();
+  } catch (...) {
+    // 确保恢复 cout
+    std::cout.rdbuf(old_cout_buf);
+    throw;
+  }
+}
 }  // namespace paddle_api_test

src/file_manager.h

@@ -1,5 +1,6 @@
 #pragma once
 #include <fstream>
+#include <functional>
 #include <mutex>
 #include <shared_mutex>
 #include <string>
@@ -16,6 +17,9 @@ class FileManerger {
   FileManerger& operator<<(const std::string& str);
   void saveFile();
 
+  // 捕获标准输出到文件
+  void captureStdout(std::function<void()> func);
+
  private:
   mutable std::shared_mutex mutex_;
   std::string basic_path_ = "/tmp/paddle_cpp_api_test/";