Make LoggingHandler.logEntryFor extensible

googleapis · Jul 30, 2022 · 6b511fb · 6b511fb
1 parent 85b3ebd
commit 6b511fb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/google-cloud-logging/src/main/java/com/google/cloud/logging/LoggingHandler.java b/google-cloud-logging/src/main/java/com/google/cloud/logging/LoggingHandler.java
@@ -305,7 +305,7 @@ public void publish(LogRecord record) {
     }
     LogEntry logEntry;
     try {
-      logEntry = logEntryFor(record);
+      logEntry = logEntryFor(record).build();
     } catch (Exception ex) {
       getErrorManager().error(null, ex, ErrorManager.FORMAT_FAILURE);
       return;
@@ -343,7 +343,7 @@ private MonitoredResource getMonitoredResource() {
     return null;
   }
 
-  private LogEntry logEntryFor(LogRecord record) throws Exception {
+  protected LogEntry.Builder logEntryFor(LogRecord record) throws Exception {
     String payload = getFormatter().format(record);
     Level level = record.getLevel();
     LogEntry.Builder builder =
@@ -360,7 +360,7 @@ private LogEntry logEntryFor(LogRecord record) throws Exception {
     for (LoggingEnhancer enhancer : enhancers) {
       enhancer.enhanceLogEntry(builder);
     }
-    return builder.build();
+    return builder;
   }
 
   @Override