From 741637eba41f66b51bd1764900e75cc7d5bd9ce6 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Mon, 16 Jan 2017 16:14:29 +0800
Subject: [PATCH] Add printf method to Status.

---
 paddle/utils/Status.h              | 23 +++++++++++++++++++++++
 paddle/utils/tests/test_Status.cpp |  5 +++++
 2 files changed, 28 insertions(+)

diff --git a/paddle/utils/Status.h b/paddle/utils/Status.h
index 3456d7b686f..db1edfb7c73 100644
--- a/paddle/utils/Status.h
+++ b/paddle/utils/Status.h
@@ -14,6 +14,7 @@ limitations under the License. */
 
 #pragma once
 
+#include <stdio.h>
 #include <memory>
 #include <string>
 
@@ -45,6 +46,28 @@ public:
     errMsg_.reset(new std::string(msg));
   }
 
+  /**
+   * @brief set a error message for status. Use C style printf
+   * @param fmt
+   */
+  template <typename... ARGS>
+  inline void setByPrintf(const char* fmt, ARGS... args) noexcept {
+    constexpr size_t bufferSize = 4096;
+    char buffer[bufferSize];
+    snprintf(buffer, bufferSize, fmt, args...);
+    errMsg_.reset(new std::string(buffer));
+  }
+
+  /**
+   * create a error status by C style printf.
+   */
+  template <typename... ARGS>
+  inline static Status printf(const char* fmt, ARGS... args) noexcept {
+    Status s;
+    s.setByPrintf(fmt, args...);
+    return s;
+  }
+
   /**
    * @brief what will return the error message. If status is OK, return nullptr.
    */
diff --git a/paddle/utils/tests/test_Status.cpp b/paddle/utils/tests/test_Status.cpp
index e2c2ae537d8..04cef095792 100644
--- a/paddle/utils/tests/test_Status.cpp
+++ b/paddle/utils/tests/test_Status.cpp
@@ -26,4 +26,9 @@ TEST(Status, testAll) {
   paddle::Status status2("error2");
   ASSERT_FALSE(status2.isOK());
   ASSERT_STREQ("error2", status2.what());
+
+  int i = 3;
+  auto status3 = paddle::Status::printf("error%d", i);
+  ASSERT_FALSE(status3.isOK());
+  ASSERT_STREQ("error3", status3.what());
 }
-- 
GitLab