Fix no generated keys problem

1 files changed, 7 insertions, 3 deletions

diff --git a/src/org/traccar/database/QueryBuilder.java b/src/org/traccar/database/QueryBuilder.java
index 5e5c163a3..04e29d7a2 100644
--- a/src/org/traccar/database/QueryBuilder.java
+++ b/src/org/traccar/database/QueryBuilder.java
@@ -40,11 +40,13 @@ public class QueryBuilder {
     private final Connection connection;
     private PreparedStatement statement;
     private final String query;
+    private final boolean returnGeneratedKeys;
     
     private QueryBuilder(DataSource dataSource, String query, boolean returnGeneratedKeys) throws SQLException {
         indexMap = new HashMap<String, List<Integer>>();
         connection = dataSource.getConnection();
         this.query = query;
+        this.returnGeneratedKeys = returnGeneratedKeys;
         if (query != null) {
             String parsedQuery = parse(query.trim(), indexMap);
             try {
@@ -417,9 +419,11 @@ public class QueryBuilder {
         if (query != null) {
             try {
                 statement.executeUpdate();
-                ResultSet resultSet = statement.getGeneratedKeys();
-                if (resultSet.next()) {
-                    return resultSet.getLong(1);
+                if (returnGeneratedKeys) {
+                    ResultSet resultSet = statement.getGeneratedKeys();
+                    if (resultSet.next()) {
+                        return resultSet.getLong(1);
+                    }
                 }
             } finally {
                 statement.close();