log exception messages

Author: Neenu1995

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

@@ -16,6 +16,8 @@
 
 package com.google.cloud.bigquery.jdbc;
 
+import java.io.PrintWriter;
+import java.io.StringWriter;
 import java.util.function.Supplier;
 import java.util.logging.Level;
 import java.util.logging.Logger;
@@ -53,6 +55,12 @@ private void logWithCaller(Level level, Supplier<String> msgSupplier) {
     logp(level, sourceClass, sourceMethod, msgSupplier);
   }
 
+  private String formatStackTrace(Throwable thrown) {
+    StringWriter sw = new StringWriter();
+    thrown.printStackTrace(new PrintWriter(sw));
+    return sw.toString();
+  }
+
   void finest(String format, Object... args) {
     logWithCaller(Level.FINEST, () -> String.format(format, args));
   }
@@ -77,8 +85,26 @@ void warning(String format, Object... args) {
     logWithCaller(Level.WARNING, () -> String.format(format, args));
   }
 
+  void warning(Throwable thrown, String msg) {
+    logWithCaller(Level.WARNING, () -> msg + System.lineSeparator() + formatStackTrace(thrown));
+  }
+
+  void warning(Throwable thrown, String format, Object... args) {
+    logWithCaller(Level.WARNING, () -> String.format(format, args) + System.lineSeparator() + formatStackTrace(thrown));
+  }
+
   void severe(String format, Object... args) {
     logWithCaller(Level.SEVERE, () -> String.format(format, args));
   }
+
+  void severe(Throwable thrown, String msg) {
+    logWithCaller(Level.SEVERE, () -> msg + System.lineSeparator() + formatStackTrace(thrown));
+  }
+
+  void severe(Throwable thrown, String format, Object... args) {
+    logWithCaller(Level.SEVERE, () -> String.format(format, args) + System.lineSeparator() + formatStackTrace(thrown));
+  }
 }
 
+
+

java-bigquery/google-cloud-bigquery-jdbc/src/test/java/com/google/cloud/bigquery/jdbc/BigQueryJdbcCustomLoggerTest.java

@@ -17,6 +17,7 @@
 package com.google.cloud.bigquery.jdbc;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -77,4 +78,20 @@ public void testLogWithCallerInference() {
     assertEquals("testLogWithCallerInference", record.getSourceMethodName());
     assertEquals(BigQueryJdbcCustomLoggerTest.class.getName(), record.getSourceClassName());
   }
+
+  @Test
+  public void testLogWithException() {
+    Exception ex = new Exception("Test exception");
+    logger.severe(ex, "Error occurred: %s", "detail");
+    
+    List<LogRecord> records = testHandler.getRecords();
+    assertEquals(1, records.size());
+    LogRecord record = records.get(0);
+    
+    assertEquals("testLogWithException", record.getSourceMethodName());
+    assertEquals(BigQueryJdbcCustomLoggerTest.class.getName(), record.getSourceClassName());
+    assertTrue(record.getMessage().contains("Error occurred: detail"));
+    assertTrue(record.getMessage().contains("java.lang.Exception: Test exception"));
+  }
 }
+