CHECK to Status migration for Trainer.

1 files changed, 14 insertions, 6 deletions

diff --git a/src/util.cc b/src/util.cc
index d06d953..915f01f 100644
--- a/src/util.cc
+++ b/src/util.cc
@@ -220,7 +220,9 @@ namespace io {
 InputBuffer::InputBuffer(StringPiece filename)
     : is_(filename.empty() ? &std::cin
                            : new std::ifstream(WPATH(filename.data()))) {
-  CHECK_IFS(*is_, filename.data());
+  if (!*is_)
+    status_ = util::StatusBuilder(util::error::NOT_FOUND)
+              << "\"" << filename.data() << "\": " << std::strerror(errno);
 }
 
 InputBuffer::~InputBuffer() {
@@ -229,6 +231,8 @@ InputBuffer::~InputBuffer() {
   }
 }
 
+util::Status InputBuffer::status() const { return status_; }
+
 bool InputBuffer::ReadLine(std::string *line) {
   return static_cast<bool>(std::getline(*is_, *line));
 }
@@ -237,7 +241,9 @@ OutputBuffer::OutputBuffer(StringPiece filename)
     : os_(filename.empty()
               ? &std::cout
               : new std::ofstream(WPATH(filename.data()), OUTPUT_MODE)) {
-  CHECK_OFS(*os_, filename.data());
+  if (!*os_)
+    status_ = util::StatusBuilder(util::error::PERMISSION_DENIED)
+              << "\"" << filename.data() << "\": " << std::strerror(errno);
 }
 
 OutputBuffer::~OutputBuffer() {
@@ -246,13 +252,15 @@ OutputBuffer::~OutputBuffer() {
   }
 }
 
-void OutputBuffer::Write(StringPiece text) {
+util::Status OutputBuffer::status() const { return status_; }
+
+bool OutputBuffer::Write(StringPiece text) {
   os_->write(text.data(), text.size());
+  return os_->good();
 }
 
-void OutputBuffer::WriteLine(StringPiece text) {
-  Write(text);
-  Write("\n");
+bool OutputBuffer::WriteLine(StringPiece text) {
+  return Write(text) && Write("\n");
 }
 }  // namespace io
 }  // namespace sentencepiece