Added copy/consume constructors.

joshmg · joshmg · commit cc026485d8c6 · 2019-07-30T16:59:35.000-04:00
diff --git a/src/main/java/com/softwareverde/database/query/BatchedInsertQuery.java b/src/main/java/com/softwareverde/database/query/BatchedInsertQuery.java
@@ -1,10 +1,18 @@
 package com.softwareverde.database.query;
 
 public class BatchedInsertQuery extends Query {
+    protected BatchedInsertQuery(final Query query, final Boolean shouldConsumeQuery) {
+        super(query, shouldConsumeQuery);
+    }
+
     public BatchedInsertQuery(final String query) {
         super(query);
     }
 
+    public BatchedInsertQuery(final Query query) {
+        super(query);
+    }
+
     @Override
     public String getQueryString() {
         final Integer parameterCount = _parameters.size();
diff --git a/src/main/java/com/softwareverde/database/query/BatchedUpdateQuery.java b/src/main/java/com/softwareverde/database/query/BatchedUpdateQuery.java
@@ -4,10 +4,18 @@
 import java.util.regex.Pattern;
 
 public class BatchedUpdateQuery extends Query {
+    protected BatchedUpdateQuery(final Query query, final Boolean shouldConsumeQuery) {
+        super(query, shouldConsumeQuery);
+    }
+
     public BatchedUpdateQuery(final String query) {
         super(query);
     }
 
+    public BatchedUpdateQuery(final Query query) {
+        super(query);
+    }
+
     @Override
     public String getQueryString() {
         // UPDATE t SET v = x WHERE z IN (?)
diff --git a/src/main/java/com/softwareverde/database/query/Query.java b/src/main/java/com/softwareverde/database/query/Query.java
@@ -9,10 +9,26 @@
 
 public class Query {
     protected final String _query;
-    protected final List<TypedParameter> _parameters = new ArrayList<TypedParameter>();
+    protected final List<TypedParameter> _parameters;
+
+    protected Query(final Query query, final Boolean shouldConsumeQuery) {
+        if (shouldConsumeQuery) {
+            _query = query._query;
+            _parameters = query._parameters;
+        }
+        else {
+            _query = query._query;
+            _parameters = new ArrayList<TypedParameter>(query._parameters);
+        }
+    }
+
+    public Query(final Query query) {
+        this(query, false);
+    }
 
     public Query(final String query) {
         _query = query;
+        _parameters = new ArrayList<TypedParameter>();
     }
 
     public Query setParameter(final Boolean value) {
