|
16 | 16 |
|
17 | 17 | package com.google.cloud.bigquery.jdbc; |
18 | 18 |
|
| 19 | +import com.google.cloud.logging.LogEntry; |
19 | 20 | import com.google.cloud.logging.Logging; |
| 21 | +import com.google.cloud.logging.Payload; |
20 | 22 | import io.opentelemetry.api.OpenTelemetry; |
21 | 23 | import io.opentelemetry.api.baggage.Baggage; |
22 | 24 | import io.opentelemetry.api.common.AttributeKey; |
|
26 | 28 | import io.opentelemetry.api.trace.SpanContext; |
27 | 29 | import io.opentelemetry.context.Context; |
28 | 30 | import java.time.Instant; |
| 31 | +import java.util.Collections; |
29 | 32 | import java.util.logging.Handler; |
30 | 33 | import java.util.logging.Level; |
31 | 34 | import java.util.logging.LogRecord; |
@@ -84,7 +87,35 @@ public void publish(LogRecord record) { |
84 | 87 | } |
85 | 88 |
|
86 | 89 | private void publishToGcp(LogRecord record, String traceId, String spanId, String connectionId) { |
87 | | - // TODO(b/491238299) |
| 90 | + if (loggingClient == null) { |
| 91 | + return; |
| 92 | + } |
| 93 | + |
| 94 | + LogEntry.Builder builder = |
| 95 | + LogEntry.newBuilder(Payload.StringPayload.of(record.getMessage())) |
| 96 | + .setSeverity(mapGcpSeverity(record.getLevel())) |
| 97 | + .setTimestamp(record.getMillis()); |
| 98 | + |
| 99 | + if (traceId != null) { |
| 100 | + builder.setTrace(traceId); |
| 101 | + } |
| 102 | + if (spanId != null) { |
| 103 | + builder.setSpanId(spanId); |
| 104 | + } |
| 105 | + if (connectionId != null) { |
| 106 | + builder.addLabel("jdbc.connection_id", connectionId); |
| 107 | + } |
| 108 | + |
| 109 | + loggingClient.write(Collections.singleton(builder.build())); |
| 110 | + } |
| 111 | + |
| 112 | + private com.google.cloud.logging.Severity mapGcpSeverity(Level level) { |
| 113 | + if (level == Level.SEVERE) return com.google.cloud.logging.Severity.ERROR; |
| 114 | + if (level == Level.WARNING) return com.google.cloud.logging.Severity.WARNING; |
| 115 | + if (level == Level.INFO) return com.google.cloud.logging.Severity.INFO; |
| 116 | + if (level == Level.CONFIG) return com.google.cloud.logging.Severity.INFO; |
| 117 | + if (level == Level.FINE) return com.google.cloud.logging.Severity.DEBUG; |
| 118 | + return com.google.cloud.logging.Severity.DEBUG; |
88 | 119 | } |
89 | 120 |
|
90 | 121 | private void publishToOTel(LogRecord record, String traceId, String spanId, String connectionId) { |
|
0 commit comments