Blackrock: Added QUOTED_IDENTIFIER and CONCAT_NULL_YIELDS_NULL flag in connection

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

@@ -1451,8 +1451,8 @@ public interface ISQLServerDataSource extends javax.sql.CommonDataSource {
      * @return cacheBulkCopyMetadata boolean value
      */
     boolean getcacheBulkCopyMetadata();
-  
-   /**
+
+    /**
      * Returns value of 'retryExec' from Connection String.
      *
      * @param retryExec
@@ -1506,4 +1506,38 @@ public interface ISQLServerDataSource extends javax.sql.CommonDataSource {
      * @return useFlexibleCallableStatements
      */
     boolean getUseFlexibleCallableStatements();
+
+    /**
+     * Returns value of 'quotedIdentifier' from Connection String.
+     *
+     * @return true
+     *         if quotedIdentifier is set to true, false otherwise
+     */
+    String getQuotedIdentifier();
+
+    /**
+     * Sets the value for 'quotedIdentifier' property
+     *
+     * @param quotedIdentifier
+     *        boolean value
+     * 
+     */
+    void setQuotedIdentifier(String quotedIdentifier);
+
+    /**
+     * Returns value of 'concatNullYieldsNull' from Connection String.
+     *
+     * @return true
+     *         if concatNullYieldsNull is set to true, false otherwise
+     */
+    String getConcatNullYieldsNull();
+
+    /**
+     * Sets the value for 'concatNullYieldsNull' property
+     *
+     * @param concatNullYieldsNull
+     *        boolean value
+     * 
+     */
+    void setConcatNullYieldsNull(String concatNullYieldsNull);
 }

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

@@ -2038,7 +2038,7 @@
             return this;
     }
 
-    final void resetPooledConnection() {
+    final void resetPooledConnection() throws SQLServerException {
         tdsChannel.resetPooledConnection();
         initResettableValues();
 
@@ -2051,6 +2051,8 @@
         if (null != bulkCopyOperationCache) {
             bulkCopyOperationCache.clear();
         }
+
+        setSessionProperties();        
     }
 
     /**
@@ -3530,7 +3532,8 @@
             }
 
             state = State.OPENED;
-
+            setSessionProperties();
+
             // Socket timeout is bounded by loginTimeout during the login phase.
             // Reset socket timeout back to the original value.
             tdsChannel.resetTcpSocketTimeout();
@@ -3552,6 +3555,40 @@
         return this;
     }
 
+    private void setSessionProperties() throws SQLServerException {
+        // check QUOTED_IDENTIFIER property
+        String quotedIdentifierProperty = SQLServerDriverStringProperty.QUOTED_IDENTIFIER.toString();
+        String quotedIdentifierValue = activeConnectionProperties.getProperty(quotedIdentifierProperty);
+        if (null != quotedIdentifierValue) {
+            OnOffOption quotedIdentifierOption = OnOffOption.valueOfString(quotedIdentifierValue);
+            activeConnectionProperties.setProperty(quotedIdentifierProperty, quotedIdentifierValue);
+            switch (quotedIdentifierOption) {
+                case ON:
+                    connectionCommand("SET QUOTED_IDENTIFIER ON", "quotedIdentifier");
+                    break;
+                case OFF:
+                    connectionCommand("SET QUOTED_IDENTIFIER OFF", "quotedIdentifier");
+                    break;
+            }
+        }
+
+        // check CONCAT_NULL_YIELDS_NULL property
+        String concatNullYieldsNullProperty = SQLServerDriverStringProperty.CONCAT_NULL_YIELDS_NULL.toString();
+        String concatNullYieldsNullValue = activeConnectionProperties.getProperty(concatNullYieldsNullProperty);
+        if (null != concatNullYieldsNullValue) {
+            OnOffOption concatNullYieldsOption = OnOffOption.valueOfString(concatNullYieldsNullValue);
+            activeConnectionProperties.setProperty(concatNullYieldsNullProperty, concatNullYieldsNullValue);
+            switch (concatNullYieldsOption) {
+                case ON:
+                    connectionCommand("SET CONCAT_NULL_YIELDS_NULL ON", "concatNullYields");        
+                    break;
+                case OFF:
+                    connectionCommand("SET CONCAT_NULL_YIELDS_NULL OFF", "concatNullYields");
+                    break;
+            }
+        }
+    }
+
     // log open connection failures
     private void logConnectFailure(int attemptNumber, SQLServerException e, SQLServerError sqlServerError) {
         loggerResiliency.finer(toString() + " Connection open - connection failed on attempt: " + attemptNumber + ".");
@@ -4786,6 +4823,7 @@
              */
             boolean commandComplete = false;
             try {
+                newCommand.createCounter(null, activeConnectionProperties);
                 commandComplete = newCommand.execute(tdsChannel.getWriter(), tdsChannel.getReader(newCommand));
             } finally {
                 /*

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

@@ -948,17 +948,15 @@
 
     @Override
     public void setcacheBulkCopyMetadata(boolean cacheBulkCopyMetadata) {
-        setBooleanProperty(connectionProps,
-                SQLServerDriverBooleanProperty.ENABLE_BULK_COPY_CACHE.toString(),
+        setBooleanProperty(connectionProps, SQLServerDriverBooleanProperty.ENABLE_BULK_COPY_CACHE.toString(),
                 cacheBulkCopyMetadata);
     }
 
     @Override
     public boolean getcacheBulkCopyMetadata() {
-        boolean defaultValue = SQLServerDriverBooleanProperty.ENABLE_BULK_COPY_CACHE
-                .getDefaultValue();
-        return getBooleanProperty(connectionProps,
-                SQLServerDriverBooleanProperty.ENABLE_BULK_COPY_CACHE.toString(), defaultValue);
+        boolean defaultValue = SQLServerDriverBooleanProperty.ENABLE_BULK_COPY_CACHE.getDefaultValue();
+        return getBooleanProperty(connectionProps, SQLServerDriverBooleanProperty.ENABLE_BULK_COPY_CACHE.toString(),
+                defaultValue);
     }
 
     @Override
@@ -1392,7 +1390,6 @@
     @Override
     public void setUseFlexibleCallableStatements(boolean enable) {}
 
-
     /**
      * useFlexibleCallableStatements is temporarily removed.
      * This method is a no-op for backwards compatibility only.
@@ -1494,6 +1491,29 @@
         return getStringProperty(connectionProps, SQLServerDriverStringProperty.RETRY_EXEC.toString(), null);
     }
 
+    @Override
+    public String getQuotedIdentifier() {
+        return getStringProperty(connectionProps, SQLServerDriverStringProperty.QUOTED_IDENTIFIER.toString(),
+                SQLServerDriverStringProperty.QUOTED_IDENTIFIER.getDefaultValue());
+    }
+
+    @Override
+    public void setQuotedIdentifier(String quotedIdentifier) {
+        setStringProperty(connectionProps, SQLServerDriverStringProperty.QUOTED_IDENTIFIER.toString(), quotedIdentifier);
+    }
+
+    @Override
+    public String getConcatNullYieldsNull() {
+        return getStringProperty(connectionProps, SQLServerDriverStringProperty.CONCAT_NULL_YIELDS_NULL.toString(),
+                SQLServerDriverStringProperty.CONCAT_NULL_YIELDS_NULL.getDefaultValue());
+    }
+
+    @Override
+    public void setConcatNullYieldsNull(String concatNullYieldNull) {
+        setStringProperty(connectionProps, SQLServerDriverStringProperty.CONCAT_NULL_YIELDS_NULL.toString(),
+                concatNullYieldNull);
+    }
+
     /**
      * Sets the 'retryConn' setting.
      *

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

@@ -612,7 +612,9 @@
     DATETIME_DATATYPE("datetimeParameterType", DatetimeType.DATETIME2.toString()),
     ACCESS_TOKEN_CALLBACK_CLASS("accessTokenCallbackClass", ""),
     RETRY_EXEC("retryExec", ""),
-    RETRY_CONN("retryConn", "");
+    RETRY_CONN("retryConn", ""),
+    QUOTED_IDENTIFIER("quotedIdentifier", OnOffOption.ON.toString()),
+    CONCAT_NULL_YIELDS_NULL("concatNullYieldsNull", OnOffOption.ON.toString());
 
     private final String name;
     private final String defaultValue;
@@ -728,6 +730,47 @@
 }
 
 
+enum OnOffOption {
+    ON("ON"),
+    OFF("OFF");
+
+    private final String option;
+
+    private OnOffOption(String option) {
+        this.option = option;
+    }
+
+    @Override
+    public String toString() {
+        return option;
+    }
+
+    static OnOffOption valueOfString(String value) throws SQLServerException {
+        OnOffOption option = null;
+
+        if (value.toLowerCase(Locale.US).equalsIgnoreCase(OnOffOption.ON.toString())) {
+            option = OnOffOption.ON;
+        } else if (value.toLowerCase(Locale.US).equalsIgnoreCase(OnOffOption.OFF.toString())) {
+            option = OnOffOption.OFF;
+        } else {
+            MessageFormat form = new MessageFormat(SQLServerException.getErrString("R_InvalidConnectionSetting"));
+            Object[] msgArgs = {"OnOffOption", value};
+            throw new SQLServerException(form.format(msgArgs), null);
+        }
+        return option;
+    }
+
+    static boolean isValidOnOffOption(String option) {
+        for (OnOffOption t : OnOffOption.values()) {
+            if (option.equalsIgnoreCase(t.toString())) {
+                return true;
+            }
+        }
+        return false;
+    }
+}
+
+
 /**
  * Provides methods to connect to a SQL Server database and to obtain information about the JDBC driver.
  */
@@ -807,8 +850,8 @@
                     Boolean.toString(SQLServerDriverBooleanProperty.INTEGRATED_SECURITY.getDefaultValue()), false,
                     TRUE_FALSE),
             new SQLServerDriverPropertyInfo(SQLServerDriverBooleanProperty.USE_DEFAULT_GSS_CREDENTIAL.toString(),
-                    Boolean.toString(SQLServerDriverBooleanProperty.USE_DEFAULT_GSS_CREDENTIAL.getDefaultValue()), false,
-                    TRUE_FALSE),
+                    Boolean.toString(SQLServerDriverBooleanProperty.USE_DEFAULT_GSS_CREDENTIAL.getDefaultValue()),
+                    false, TRUE_FALSE),
             new SQLServerDriverPropertyInfo(SQLServerDriverStringProperty.KEY_STORE_AUTHENTICATION.toString(),
                     SQLServerDriverStringProperty.KEY_STORE_AUTHENTICATION.getDefaultValue(), false,
                     new String[] {KeyStoreAuthentication.JAVA_KEYSTORE_PASSWORD.toString()}),
@@ -1002,8 +1045,13 @@
             new SQLServerDriverPropertyInfo(SQLServerDriverIntProperty.CONNECT_RETRY_COUNT.toString(),
                     Integer.toString(SQLServerDriverIntProperty.CONNECT_RETRY_COUNT.getDefaultValue()), false, null),
             new SQLServerDriverPropertyInfo(SQLServerDriverIntProperty.CONNECT_RETRY_INTERVAL.toString(),
-                    Integer.toString(SQLServerDriverIntProperty.CONNECT_RETRY_INTERVAL.getDefaultValue()), false,
-                    null),};
+                    Integer.toString(SQLServerDriverIntProperty.CONNECT_RETRY_INTERVAL.getDefaultValue()), false, null),
+            new SQLServerDriverPropertyInfo(SQLServerDriverStringProperty.QUOTED_IDENTIFIER.toString(),
+                    SQLServerDriverStringProperty.QUOTED_IDENTIFIER.getDefaultValue(), false,
+                    new String[] {OnOffOption.OFF.toString(), OnOffOption.OFF.toString()}),
+            new SQLServerDriverPropertyInfo(SQLServerDriverStringProperty.CONCAT_NULL_YIELDS_NULL.toString(),
+                    SQLServerDriverStringProperty.CONCAT_NULL_YIELDS_NULL.getDefaultValue(), false,
+                    new String[] {OnOffOption.OFF.toString(), OnOffOption.OFF.toString()}),};
 
     /**
      * Properties that can only be set by using Properties. Cannot set in connection string

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

@@ -244,6 +244,8 @@ protected Object[][] getContents() {
         {"R_AADSecurePrincipalSecretPropertyDescription", "A Secret defined for a registered application which has been granted permission to the database connected."},
         {"R_accessTokenCallbackClassPropertyDescription", "The class to instantiate as the SQLServerAccessTokenCallback for acquiring tokens."},
         {"R_accessTokenCallbackPropertyDescription", "A SQLServerAccessTokenCallback object which is used to call a callback method to return an access token."},
+        {"R_quotedIdentifierPropertyDescription", "Indicates whether quotedIdentifier property is set."},
+        {"R_concatNullYieldsNullPropertyDescription", "Indicates whether concatNullYieldsNull property is set."},
         {"R_noParserSupport", "An error occurred while instantiating the required parser. Error: \"{0}\""},
         {"R_writeOnlyXML", "Cannot read from this SQLXML instance. This instance is for writing data only."},
         {"R_dataHasBeenReadXML", "Cannot read from this SQLXML instance. The data has already been read."},