add less expensive setContext instead of registerContext

Author: Neenu1995

java-bigquery/google-cloud-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryArrowResultSet.java

@@ -212,12 +212,10 @@ public void close() {
 
   @Override
   public boolean next() throws SQLException {
-    try (BigQueryJdbcMdc.MdcCloseable mdc =
-        BigQueryJdbcMdc.registerInstance(this.connection, this.connectionId)) {
-      LOG.finest("++enter++");
-      checkClosed();
-      return nextImpl();
-    }
+    BigQueryJdbcMdc.setContextPersistent(this.connectionId);
+    LOG.finest("++enter++");
+    checkClosed();
+    return nextImpl();
   }
 
   private boolean nextImpl() throws SQLException {
@@ -327,13 +325,11 @@ private Object getObjectInternal(int columnIndex) throws SQLException {
 
   @Override
   public Object getObject(int columnIndex) throws SQLException {
-    try (BigQueryJdbcMdc.MdcCloseable mdc =
-        BigQueryJdbcMdc.registerInstance(this.connection, this.connectionId)) {
-      // columnIndex is SQL index starting at 1
-      LOG.finest("++enter++");
-      checkClosed();
-      return getObjectImpl(columnIndex);
-    }
+    BigQueryJdbcMdc.setContextPersistent(this.connectionId);
+    // columnIndex is SQL index starting at 1
+    LOG.finest("++enter++");
+    checkClosed();
+    return getObjectImpl(columnIndex);
   }
 
   private Object getObjectImpl(int columnIndex) throws SQLException {

java-bigquery/google-cloud-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryConnection.java

@@ -745,8 +745,7 @@ Long getListenerPoolSize() {
 
   @Override
   public boolean isValid(int timeout) throws SQLException {
-    try (BigQueryJdbcMdc.MdcCloseable mdc =
-        BigQueryJdbcMdc.registerInstance(this, this.connectionId)) {
+    try (BigQueryJdbcMdc.MdcCloseable mdc = BigQueryJdbcMdc.setContext(this.connectionId)) {
       LOG.finest("++enter++");
       LOG.info("Validating connection");
       if (timeout < 0) {
@@ -771,8 +770,7 @@ public boolean isValid(int timeout) throws SQLException {
 
   @Override
   public void abort(Executor executor) throws SQLException {
-    try (BigQueryJdbcMdc.MdcCloseable mdc =
-        BigQueryJdbcMdc.registerInstance(this, this.connectionId)) {
+    try (BigQueryJdbcMdc.MdcCloseable mdc = BigQueryJdbcMdc.setContext(this.connectionId)) {
       LOG.finest("++enter++");
       close();
     }
@@ -822,8 +820,7 @@ public void clearWarnings() {
 
   @Override
   public boolean getAutoCommit() {
-    try (BigQueryJdbcMdc.MdcCloseable mdc =
-        BigQueryJdbcMdc.registerInstance(this, this.connectionId)) {
+    try (BigQueryJdbcMdc.MdcCloseable mdc = BigQueryJdbcMdc.setContext(this.connectionId)) {
       LOG.finest("++enter++");
       checkClosed();
       return this.autoCommit;
@@ -832,8 +829,7 @@ public boolean getAutoCommit() {
 
   @Override
   public void setAutoCommit(boolean autoCommit) throws SQLException {
-    try (BigQueryJdbcMdc.MdcCloseable mdc =
-        BigQueryJdbcMdc.registerInstance(this, this.connectionId)) {
+    try (BigQueryJdbcMdc.MdcCloseable mdc = BigQueryJdbcMdc.setContext(this.connectionId)) {
       LOG.finest("++enter++");
       checkClosed();
       checkIfEnabledSession("setAutoCommit");
@@ -932,8 +928,7 @@ public int getTransactionIsolation() {
 
   @Override
   public void setTransactionIsolation(int level) throws SQLException {
-    try (BigQueryJdbcMdc.MdcCloseable mdc =
-        BigQueryJdbcMdc.registerInstance(this, this.connectionId)) {
+    try (BigQueryJdbcMdc.MdcCloseable mdc = BigQueryJdbcMdc.setContext(this.connectionId)) {
       LOG.finest("++enter++");
       if (level != Connection.TRANSACTION_SERIALIZABLE) {
         throw new BigQueryJdbcSqlFeatureNotSupportedException(
@@ -951,8 +946,7 @@ public int getHoldability() {
 
   @Override
   public void setHoldability(int holdability) throws SQLException {
-    try (BigQueryJdbcMdc.MdcCloseable mdc =
-        BigQueryJdbcMdc.registerInstance(this, this.connectionId)) {
+    try (BigQueryJdbcMdc.MdcCloseable mdc = BigQueryJdbcMdc.setContext(this.connectionId)) {
       if (holdability != ResultSet.CLOSE_CURSORS_AT_COMMIT) {
         throw new BigQueryJdbcSqlFeatureNotSupportedException(
             "CLOSE_CURSORS_AT_COMMIT not supported");