Added 404 code for single news getting

N1ghtF1re · May 12, 2019 · 25d3984 · 25d3984
1 parent 1507f81
commit 25d3984
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/kotlin/men/brakh/bsuirapi/controller/NewsServlet.kt b/src/main/kotlin/men/brakh/bsuirapi/controller/NewsServlet.kt
@@ -60,7 +60,8 @@ class NewsServlet : HttpServlet() {
         val multiParams: Map<String, Array<String>> = req.parameterMap
 
         if("id" in params) {
-            resp.writer.write(newsRepo.findById(params["id"]!!.toLong())?.toJson())
+            val news = newsRepo.findById(params["id"]!!.toLong())
+            news?.let { resp.writeJson(it) } ?: resp.writeError("News not found", 404)
             return
         }