Detect IO error in load

rimms · Feb 28, 2014 · 5c2c2c0 · kmaehashi · Mar 6, 2014 · rimms
1 parent da67929
commit 5c2c2c0
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/jubatus/server/framework/server_base.cpp b/jubatus/server/framework/server_base.cpp
@@ -58,8 +58,16 @@ void load_file_impl(server_base& server,
       << core::common::exception::error_errno(errno));
   }
 
-  framework::load_server(ifs, server, id);
-  ifs.close();
+  ifs.exceptions(std::ios_base::failbit | std::ios_base::badbit);
+  try {
+    framework::load_server(ifs, server, id);
+    ifs.close();
+  } catch (const std::ios_base::failure&) {
+    throw JUBATUS_EXCEPTION(
+      core::common::exception::runtime_error("cannot read output file")
+      << core::common::exception::error_file_name(path)
+      << core::common::exception::error_errno(errno));
+  }
 
   server.update_loaded_status(path);
   LOG(INFO) << "loaded from " << path;