1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
package eu.mulk.quarkus.googlecloud.jsonlogging;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.logging.Level;
import org.jboss.logmanager.ExtFormatter;
import org.jboss.logmanager.ExtLogRecord;
/**
* Formats log records as JSON for consumption by Google Cloud Logging.
*
* <p>Meant to be used in containers running on Google Kubernetes Engine (GKE).
*
* @see LogEntry
*/
public class Formatter extends ExtFormatter {
private static final String TRACE_LEVEL = "TRACE";
private static final String DEBUG_LEVEL = "DEBUG";
private static final String INFO_LEVEL = "INFO";
private static final String WARNING_LEVEL = "WARNING";
private static final String ERROR_LEVEL = "ERROR";
private static final String ERROR_EVENT_TYPE =
"type.googleapis.com/google.devtools.clouderrorreporting.v1beta1.ReportedErrorEvent";
private final List<StructuredParameterProvider> parameterProviders;
private final List<LabelProvider> labelProviders;
public Formatter(
Collection<StructuredParameterProvider> parameterProviders,
Collection<LabelProvider> labelProviders) {
this.parameterProviders = List.copyOf(parameterProviders);
this.labelProviders = List.copyOf(labelProviders);
}
@Override
public String format(ExtLogRecord logRecord) {
var message = formatMessageWithStackTrace(logRecord);
List<StructuredParameter> parameters = new ArrayList<>();
Map<String, String> labels = new HashMap<>();
for (var parameterProvider : parameterProviders) {
var parameter = parameterProvider.getParameter();
if (parameter != null) {
parameters.add(parameter);
}
}
for (var labelProvider : labelProviders) {
var providedLabels = labelProvider.getLabels();
if (providedLabels != null) {
for (var label : providedLabels) {
labels.put(label.key(), label.value());
}
}
}
if (logRecord.getParameters() != null) {
for (var parameter : logRecord.getParameters()) {
if (parameter instanceof StructuredParameter) {
parameters.add((StructuredParameter) parameter);
} else if (parameter instanceof Label) {
var label = (Label) parameter;
labels.put(label.key(), label.value());
}
}
}
var mdc = logRecord.getMdcCopy();
var ndc = logRecord.getNdc();
var sourceLocation =
new LogEntry.SourceLocation(
logRecord.getSourceFileName(),
String.valueOf(logRecord.getSourceLineNumber()),
String.format(
"%s.%s", logRecord.getSourceClassName(), logRecord.getSourceMethodName()));
var entry =
new LogEntry(
message,
severityOf(logRecord.getLevel()),
new LogEntry.Timestamp(logRecord.getInstant()),
null,
null,
sourceLocation,
labels,
parameters,
mdc,
ndc,
logRecord.getLevel().intValue() >= 1000 ? ERROR_EVENT_TYPE : null);
return entry.json().build().toString() + "\n";
}
/**
* Formats the log message corresponding to {@code logRecord} including a stack trace of the
* {@link ExtLogRecord#getThrown()} exception if any.
*/
private String formatMessageWithStackTrace(ExtLogRecord logRecord) {
var messageStringWriter = new StringWriter();
var messagePrintWriter = new PrintWriter(messageStringWriter);
messagePrintWriter.append(this.formatMessage(logRecord));
if (logRecord.getThrown() != null) {
messagePrintWriter.println();
logRecord.getThrown().printStackTrace(messagePrintWriter);
}
messagePrintWriter.close();
return messageStringWriter.toString();
}
/** Computes the Google Cloud Logging severity corresponding to a given {@link Level}. */
private static String severityOf(Level level) {
if (level.intValue() < 500) {
return TRACE_LEVEL;
} else if (level.intValue() < 700) {
return DEBUG_LEVEL;
} else if (level.intValue() < 900) {
return INFO_LEVEL;
} else if (level.intValue() < 1000) {
return WARNING_LEVEL;
} else {
return ERROR_LEVEL;
}
}
}
|
