added save path assertion.

KatanaGraph · Oct 11, 2021 · ad7631c · ad7631c
1 parent fc23430
commit ad7631c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tools/graph-stats/graph-memory-stats.cpp b/tools/graph-stats/graph-memory-stats.cpp
@@ -23,6 +23,7 @@
 #include <arrow/type_traits.h>
 
 #include "katana/ArrowVisitor.h"
+#include "katana/ErrorCode.h"
 #include "katana/Galois.h"
 #include "katana/JSON.h"
 #include "katana/Logging.h"
@@ -138,6 +139,13 @@ SaveToJson(
   }
 
   myfile.open(path_to_save);
+
+  if (!myfile) {
+    return KATANA_ERROR(
+        katana::ErrorCode::ArrowError, "Could not open file at {}",
+        path_to_save);
+  }
+
   myfile << serialized;
   myfile.close();
   return katana::ResultSuccess();