Fix error when job is not found

TheHive-Project · Jun 29, 2018 · 81d2a78 · 81d2a78
1 parent d7bc48d
commit 81d2a78
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/org/thp/cortex/services/JobSrv.scala b/app/org/thp/cortex/services/JobSrv.scala
@@ -150,7 +150,11 @@ class JobSrv(
       findForOrganization(organizationId, withId(jobId), Some("0-1"), Nil)
     }
       ._1
-      .runWith(Sink.head)
+      .runWith(Sink.headOption)
+      .flatMap {
+        case Some(j) ⇒ Future.successful(j)
+        case None    ⇒ Future.failed(NotFoundError(s"job $jobId not found"))
+      }
   }
 
   def delete(job: Job)(implicit authContext: AuthContext): Future[Job] = deleteSrv(job)