Incident#567732673 - Add provision to set SQLServerBulkCopy options in PreparedStatement

src/main/java/com/microsoft/sqlserver/jdbc/ISQLServerConnection.java

@@ -547,4 +547,139 @@ CallableStatement prepareCall(String sql, int nType, int nConcur, int nHold,
      * @return cacheBulkCopyMetadata boolean value
      */
     boolean getcacheBulkCopyMetadata();
+
+    /**
+     * Specifies the default batch size for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertBatchSize
+     *        integer value for bulkCopyForBatchInsertBatchSize.
+     */
+    void setBulkCopyForBatchInsertBatchSize(int bulkCopyForBatchInsertBatchSize);
+
+    /**
+     * Returns the default batch size for bulk copy operations created from batch insert operations.
+     * 
+     * @return integer value for bulkCopyForBatchInsertBatchSize.
+     */
+    int getBulkCopyForBatchInsertBatchSize();
+
+    /**
+     * Specifies the default timeout for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertTimeout
+     *        integer value for bulkCopyForBatchInsertTimeout.
+     */
+    void setBulkCopyForBatchInsertTimeout(int bulkCopyForBatchInsertTimeout);
+
+    /**
+     * Returns the default timeout for bulk copy operations created from batch insert operations.
+     * 
+     * @return integer value for bulkCopyForBatchInsertTimeout.
+     */
+    int getBulkCopyForBatchInsertTimeout();
+
+    /**
+     * Specifies the default check constraints for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertCheckConstraints
+     *        boolean value for bulkCopyForBatchInsertCheckConstraints.
+     */
+    void setBulkCopyForBatchInsertCheckConstraints(boolean bulkCopyForBatchInsertCheckConstraints);
+
+    /**
+     * Returns the default check constraints for bulk copy operations created from batch insert operations.
+     * 
+     * @return boolean value for bulkCopyForBatchInsertCheckConstraints.
+     */
+    boolean getBulkCopyForBatchInsertCheckConstraints();
+
+    /**
+     * Specifies the default fire triggers for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertFireTriggers
+     *        boolean value for bulkCopyForBatchInsertFireTriggers.
+     */
+    void setBulkCopyForBatchInsertFireTriggers(boolean bulkCopyForBatchInsertFireTriggers);
+
+    /**
+     * Returns the default fire triggers for bulk copy operations created from batch insert operations.
+     * 
+     * @return boolean value for bulkCopyForBatchInsertFireTriggers.
+     */
+    boolean getBulkCopyForBatchInsertFireTriggers();
+
+    /**
+     * Specifies the default keep identity for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertKeepIdentity
+     *        boolean value for bulkCopyForBatchInsertKeepIdentity.
+     */
+    void setBulkCopyForBatchInsertKeepIdentity(boolean bulkCopyForBatchInsertKeepIdentity);
+
+    /**
+     * Returns the default keep identity for bulk copy operations created from batch insert operations.
+     * 
+     * @return boolean value for bulkCopyForBatchInsertKeepIdentity.
+     */
+    boolean getBulkCopyForBatchInsertKeepIdentity();
+
+    /**
+     * Specifies the default keep nulls for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertKeepNulls
+     *        boolean value for bulkCopyForBatchInsertKeepNulls.
+     */
+    void setBulkCopyForBatchInsertKeepNulls(boolean bulkCopyForBatchInsertKeepNulls);
+
+    /**
+     * Returns the default keep nulls for bulk copy operations created from batch insert operations.
+     * 
+     * @return boolean value for bulkCopyForBatchInsertKeepNulls.
+     */
+    boolean getBulkCopyForBatchInsertKeepNulls();
+
+    /**
+     * Specifies the default table lock for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertTableLock
+     *        boolean value for bulkCopyForBatchInsertTableLock.
+     */
+    void setBulkCopyForBatchInsertTableLock(boolean bulkCopyForBatchInsertTableLock);
+
+    /**
+     * Returns the default table lock for bulk copy operations created from batch insert operations.
+     * 
+     * @return boolean value for bulkCopyForBatchInsertTableLock.
+     */
+    boolean getBulkCopyForBatchInsertTableLock();
+
+    /**
+     * Specifies the default use internal transaction for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertUseInternalTransaction
+     *        boolean value for bulkCopyForBatchInsertUseInternalTransaction.
+     */
+    void setBulkCopyForBatchInsertUseInternalTransaction(boolean bulkCopyForBatchInsertUseInternalTransaction);
+
+    /**
+     * Returns the default use internal transaction for bulk copy operations created from batch insert operations.
+     * 
+     * @return boolean value for bulkCopyForBatchInsertUseInternalTransaction.
+     */
+    boolean getBulkCopyForBatchInsertUseInternalTransaction();
+
+    /**
+     * Specifies the default allow encrypted value modifications for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertAllowEncryptedValueModifications
+     *        boolean value for bulkCopyForBatchInsertAllowEncryptedValueModifications.
+     */
+    void setBulkCopyForBatchInsertAllowEncryptedValueModifications(boolean bulkCopyForBatchInsertAllowEncryptedValueModifications);
+
+    /**
+     * Returns the default allow encrypted value modifications for bulk copy operations created from batch insert operations.
+     * 
+     * @return boolean value for bulkCopyForBatchInsertAllowEncryptedValueModifications.
+     */
+    boolean getBulkCopyForBatchInsertAllowEncryptedValueModifications();
 }

src/main/java/com/microsoft/sqlserver/jdbc/ISQLServerDataSource.java

@@ -980,6 +980,141 @@ public interface ISQLServerDataSource extends javax.sql.CommonDataSource {
      */
     void setUseBulkCopyForBatchInsert(boolean useBulkCopyForBatchInsert);
 
+    /**
+     * Sets the default batch size for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertBatchSize
+     *        the default batch size for bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertBatchSize(int bulkCopyForBatchInsertBatchSize);
+
+    /**
+     * Returns the default batch size for bulk copy operations created from batch insert operations.
+     * 
+     * @return the default batch size for bulk copy operations created from batch insert operations.
+     */
+    int getBulkCopyForBatchInsertBatchSize();
+
+    /**
+     * Sets the default timeout for bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertTimeout
+     *        the default timeout for bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertTimeout(int bulkCopyForBatchInsertTimeout);
+
+    /**
+     * Returns the default timeout for bulk copy operations created from batch insert operations.
+     * 
+     * @return the default timeout for bulk copy operations created from batch insert operations.
+     */
+    int getBulkCopyForBatchInsertTimeout();
+
+    /**
+     * Sets whether to check constraints during bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertCheckConstraints
+     *        indicates whether to check constraints during bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertCheckConstraints(boolean bulkCopyForBatchInsertCheckConstraints);
+
+    /**
+     * Returns whether to check constraints during bulk copy operations created from batch insert operations.
+     * 
+     * @return whether to check constraints during bulk copy operations created from batch insert operations.
+     */
+    boolean getBulkCopyForBatchInsertCheckConstraints();
+
+    /**
+     * Sets whether to fire triggers during bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertFireTriggers
+     *        indicates whether to fire triggers during bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertFireTriggers(boolean bulkCopyForBatchInsertFireTriggers);
+
+    /**
+     * Returns whether to fire triggers during bulk copy operations created from batch insert operations.
+     * 
+     * @return whether to fire triggers during bulk copy operations created from batch insert operations.
+     */
+    boolean getBulkCopyForBatchInsertFireTriggers();
+
+    /**
+     * Sets whether to keep identity values during bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertKeepIdentity
+     *        indicates whether to keep identity values during bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertKeepIdentity(boolean bulkCopyForBatchInsertKeepIdentity);
+
+    /**
+     * Returns whether to keep identity values during bulk copy operations created from batch insert operations.
+     * 
+     * @return whether to keep identity values during bulk copy operations created from batch insert operations.
+     */
+    boolean getBulkCopyForBatchInsertKeepIdentity();
+
+    /**
+     * Sets whether to keep null values during bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertKeepNulls
+     *        indicates whether to keep null values during bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertKeepNulls(boolean bulkCopyForBatchInsertKeepNulls);
+
+    /**
+     * Returns whether to keep null values during bulk copy operations created from batch insert operations.
+     * 
+     * @return whether to keep null values during bulk copy operations created from batch insert operations.
+     */
+    boolean getBulkCopyForBatchInsertKeepNulls();
+
+    /**
+     * Sets whether to use table lock during bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertTableLock
+     *        indicates whether to use table lock during bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertTableLock(boolean bulkCopyForBatchInsertTableLock);
+
+    /**
+     * Returns whether to use table lock during bulk copy operations created from batch insert operations.
+     * 
+     * @return whether to use table lock during bulk copy operations created from batch insert operations.
+     */
+    boolean getBulkCopyForBatchInsertTableLock();
+
+    /**
+     * Sets whether to use internal transaction during bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertUseInternalTransaction
+     *        indicates whether to use internal transaction during bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertUseInternalTransaction(boolean bulkCopyForBatchInsertUseInternalTransaction);
+
+    /**
+     * Returns whether to use internal transaction during bulk copy operations created from batch insert operations.
+     * 
+     * @return whether to use internal transaction during bulk copy operations created from batch insert operations.
+     */
+    boolean getBulkCopyForBatchInsertUseInternalTransaction();
+
+    /**
+     * Sets whether to allow encrypted value modifications during bulk copy operations created from batch insert operations.
+     * 
+     * @param bulkCopyForBatchInsertAllowEncryptedValueModifications
+     *        indicates whether to allow encrypted value modifications during bulk copy operations created from batch insert operations.
+     */
+    void setBulkCopyForBatchInsertAllowEncryptedValueModifications(boolean bulkCopyForBatchInsertAllowEncryptedValueModifications);
+
+    /**
+     * Returns whether to allow encrypted value modifications during bulk copy operations created from batch insert operations.
+     * 
+     * @return whether to allow encrypted value modifications during bulk copy operations created from batch insert operations.
+     */
+    boolean getBulkCopyForBatchInsertAllowEncryptedValueModifications();
+
     /**
      * Sets the client id to be used to retrieve the access token for a user-assigned Managed Identity.
      * 

src/main/java/com/microsoft/sqlserver/jdbc/SQLServerBulkCopyOptions.java

@@ -103,6 +103,22 @@ public SQLServerBulkCopyOptions() {
         useInternalTransaction = false;
         allowEncryptedValueModifications = false;
     }
+
+    /**
+     * Constructs a SQLServerBulkCopySettings class using defaults from given connection
+     */
+    SQLServerBulkCopyOptions(SQLServerConnection conn) {
+        batchSize = conn.getBulkCopyForBatchInsertBatchSize();
+        bulkCopyTimeout = conn.getBulkCopyForBatchInsertTimeout();
+        checkConstraints = conn.getBulkCopyForBatchInsertCheckConstraints();
+        fireTriggers = conn.getBulkCopyForBatchInsertFireTriggers();
+        keepIdentity = conn.getBulkCopyForBatchInsertKeepIdentity();
+        keepNulls = conn.getBulkCopyForBatchInsertKeepNulls();
+        tableLock = conn.getBulkCopyForBatchInsertTableLock();
+        useInternalTransaction = conn.getBulkCopyForBatchInsertUseInternalTransaction();
+        allowEncryptedValueModifications = conn.getBulkCopyForBatchInsertAllowEncryptedValueModifications();
+    }
+
 
     /**
      * Returns the number of rows in each batch. At the end of each batch, the rows in the batch are sent to the server.