From 7d9a6881248ad5f10eaee66b56f508ca1ae27fcc Mon Sep 17 00:00:00 2001
From: kg29 <kunalghogale@gmail.com>
Date: Sun, 9 Apr 2017 11:35:54 -0700
Subject: [PATCH] query timeout

---
 core/src/main/java/org/sql2o/Query.java | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/core/src/main/java/org/sql2o/Query.java b/core/src/main/java/org/sql2o/Query.java
index f60f62a1..c355c451 100644
--- a/core/src/main/java/org/sql2o/Query.java
+++ b/core/src/main/java/org/sql2o/Query.java
@@ -403,6 +403,16 @@ public void close() {
     // -------------------- Execute -------------------
     // ------------------------------------------------
 
+    PreparedStatement buildPreparedStatementWithTimeout(int seconds) {
+        try {
+            PreparedStatement preparedStatement = buildPreparedStatement(true);
+            preparedStatement.setQueryTimeout(seconds);
+            return preparedStatement;
+        } catch (SQLException ex) {
+            throw new Sql2oException(String.format("Error preparing statement - %s", ex.getMessage()), ex);
+        }
+    }
+
     // visible for testing
     PreparedStatement buildPreparedStatement() {
         return buildPreparedStatement(true);