From f3f8900fd457eb5c62398ed301af5531c5865b17 Mon Sep 17 00:00:00 2001 From: Rahul Rudragoudar Date: Sun, 7 Apr 2019 20:50:56 +0530 Subject: [PATCH] Minor fixes Signed-off-by: Rahul Rudragoudar --- src/main/scala/lc/Main.scala | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/main/scala/lc/Main.scala b/src/main/scala/lc/Main.scala index ae913a1..a92c510 100644 --- a/src/main/scala/lc/Main.scala +++ b/src/main/scala/lc/Main.scala @@ -92,16 +92,15 @@ class Captcha(throttle: Int) { def getChallenge(param: Parameters): Id = { val rs = stmt.executeQuery("SELECT token FROM challenge WHERE solved=FALSE LIMIT 1") - var id: String = null - if(rs.next()){ - id = rs.getString("token") + val id = if(rs.next()){ + rs.getString("token") } else { - id = generateChallenge(param) + generateChallenge(param) } Id(id) } - def getAnswer(answer: Answer): Boolean = { + def checkAnswer(answer: Answer): Boolean = { selectPstmt.setString(1, answer.id) val rs: ResultSet = selectPstmt.executeQuery() rs.next() @@ -169,7 +168,7 @@ class Server(port: Int){ val body = req.getJson() val json = parse(body) val answer = json.extract[Answer] - val result = captcha.getAnswer(answer) + val result = captcha.checkAnswer(answer) resp.getHeaders().add("Content-Type","application/json") val responseContent = if(result) """{"result":"True"}""" else """{"result":"False"}""" resp.send(200,responseContent)