before attempting to write into RESULT_FILE make sure its parent fold…

…er exist
neowit · Jan 17, 2017 · 5f8f486 · 5f8f486
1 parent 827bcb0
commit 5f8f486
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/main/scala/com/neowit/response/protocols/vim/LoginOauth.scala b/src/main/scala/com/neowit/response/protocols/vim/LoginOauth.scala
@@ -25,6 +25,11 @@ import com.neowit.response.LoginOauthResult
 
 class LoginOauth(writer: ResponseWriterVim) extends VimProtocol[LoginOauthResult] {
     def send(result: LoginOauthResult): Unit = {
-        result.resultFileOpt.foreach(f => writer.send("RESULT_FILE=" + f.getAbsolutePath))
+
+        result.resultFileOpt.foreach{f =>
+            // make sure all path folders exist
+            f.getParentFile.mkdirs()
+            writer.send("RESULT_FILE=" + f.getAbsolutePath)
+        }
     }
 }
diff --git a/src/main/scala/com/neowit/response/protocols/vim/SoqlQuery.scala b/src/main/scala/com/neowit/response/protocols/vim/SoqlQuery.scala
@@ -30,7 +30,11 @@ class SoqlQuery(writer: ResponseWriterVim) extends VimProtocol[SoqlQueryResult]
             writer.send("RESULT_SIZE=" + size)
             writer.send(InfoMessage("Record Count: " + size))
         }
-        report.resultFileOpt.foreach(file => writer.send("RESULT_FILE=" + file.getAbsolutePath))
+        report.resultFileOpt.foreach{file =>
+            // make sure all path folders exist
+            file.getParentFile.mkdirs()
+            writer.send("RESULT_FILE=" + file.getAbsolutePath)
+        }
         report.errors.foreach(writer.send(_))
     }
 }