Return the number of changes on JDBC update.

rapidoid · Dec 3, 2016 · b631fc9 · b631fc9
1 parent 74df325
commit b631fc9
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/rapidoid-sql/src/main/java/org/rapidoid/jdbc/JdbcClient.java b/rapidoid-sql/src/main/java/org/rapidoid/jdbc/JdbcClient.java
@@ -157,7 +157,7 @@ private static void close(ResultSet rs) {
 		}
 	}
 
-	public void execute(String sql, Object... args) {
+	public int execute(String sql, Object... args) {
 		ensureIsInitialized();
 
 		Log.info("SQL", "sql", sql, "args", args);
@@ -167,7 +167,13 @@ public void execute(String sql, Object... args) {
 
 		try {
 			stmt = JDBC.prepare(conn, sql, args);
-			stmt.execute();
+
+			if (sql.trim().toUpperCase().startsWith("UPDATE ")) {
+				return stmt.executeUpdate();
+
+			} else {
+				return stmt.execute() ? 1 : 0;
+			}
 
 		} catch (SQLException e) {
 			throw U.rte(e);