From cddb143cf7b300eed4f99623875c761cce87ae78 Mon Sep 17 00:00:00 2001 From: Rahul Rudragoudar Date: Thu, 7 Mar 2019 13:02:41 +0530 Subject: [PATCH] Added contentType:varchar field in db --- src/main/scala/lc/Main.scala | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/main/scala/lc/Main.scala b/src/main/scala/lc/Main.scala index faafcbb..4755a52 100644 --- a/src/main/scala/lc/Main.scala +++ b/src/main/scala/lc/Main.scala @@ -18,8 +18,8 @@ import scala.Array class Captcha { val con: Connection = DriverManager.getConnection("jdbc:h2:./captcha", "sa", "") val stmt: Statement = con.createStatement() - stmt.execute("CREATE TABLE IF NOT EXISTS challenge(token varchar, id varchar, secret varchar, provider varchar, image blob)") - val insertPstmt: PreparedStatement = con.prepareStatement("INSERT INTO challenge(token, id, secret, provider, image) VALUES (?, ?, ?, ?, ?)") + stmt.execute("CREATE TABLE IF NOT EXISTS challenge(token varchar, id varchar, secret varchar, provider varchar, contentType varchar, image blob)") + val insertPstmt: PreparedStatement = con.prepareStatement("INSERT INTO challenge(token, id, secret, provider, contentType, image) VALUES (?, ?, ?, ?, ?, ?)") val selectPstmt: PreparedStatement = con.prepareStatement("SELECT secret, provider FROM challenge WHERE token = ?") val imagePstmt: PreparedStatement = con.prepareStatement("SELECT image FROM challenge WHERE token = ?") @@ -48,7 +48,8 @@ class Captcha { insertPstmt.setString(2, provider.getId) insertPstmt.setString(3, challenge.secret) insertPstmt.setString(4, providerMap) - insertPstmt.setBlob(5, blob) + insertPstmt.setString(5, challenge.contentType) + insertPstmt.setBlob(6, blob) insertPstmt.executeUpdate() id } @@ -65,7 +66,8 @@ class Captcha { insertPstmt.setString(2, provider.getId) insertPstmt.setString(3, challenge.secret) insertPstmt.setString(4, providerMap) - insertPstmt.setBlob(5, blob) + insertPstmt.setString(5, challenge.contentType) + insertPstmt.setBlob(6, blob) insertPstmt.executeUpdate() } }