diff --git a/src/main/java/at/compax/tools/sql/main/Main.java b/src/main/java/at/compax/tools/sql/main/Main.java index 22f787b..250151d 100644 --- a/src/main/java/at/compax/tools/sql/main/Main.java +++ b/src/main/java/at/compax/tools/sql/main/Main.java @@ -192,11 +192,13 @@ public class Main { ObjectMapper mapper = new ObjectMapper(); QueryParameter queryParameter = mapper.readValue(commandBuffer.toString(), QueryParameter.class); - String query = queryParameter.getSql(); + String query = queryParameter.getSql().trim(); String result; if (query.toLowerCase().startsWith("select")) { result = formatJsonObject(SqlExecutor.executeQuery(conn, query, queryParameter.getRowLimit())); - } else if (query.toLowerCase().startsWith("update") || query.toLowerCase().startsWith("insert")) { + } else if ( query.toLowerCase().startsWith("update") + || query.toLowerCase().startsWith("insert") + || query.toLowerCase().startsWith("delete")) { result = formatJsonObject(SqlExecutor.executeUpdate(conn, query)); } else { result = formatJsonObject(SqlExecutor.execute(conn, query));