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
136
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.log;
import java.io.UnsupportedEncodingException;
import java.util.logging.Level;
import java.util.logging.LogRecord;
import java.util.logging.StreamHandler;
/**
* @author Bjorn Borud
* @author arnej27959
*/
class VespaLogHandler extends StreamHandler {
private final LogTarget logTarget;
private final String serviceName;
private final String appPrefix;
private final LevelControllerRepo repo;
private final RejectFilter logRejectFilter;
/**
* Construct handler which logs to specified logTarget. The logTarget
* may be of the following formats:
*
* <DL>
* <DT> <code>fd:<number></code>
* <DD> Log to specified file descriptor number. Only "fd:2"
* is supported.
*
* <DT> <code>file:<filename></code>
* <DD> Log to specified file in append mode
* </DL>
*/
public VespaLogHandler(LogTarget logTarget,
LevelControllerRepo levelControllerRepo, String serviceName,
String applicationPrefix) {
this.logTarget = logTarget;
this.serviceName = serviceName;
this.appPrefix = applicationPrefix;
this.repo = levelControllerRepo;
this.logRejectFilter = RejectFilter.createDefaultRejectFilter();
initialize();
}
/**
* Publish a log record into the Vespa log target.
*/
public synchronized void publish (LogRecord record) {
Level level = record.getLevel();
String component = record.getLoggerName();
LevelController ctrl = getLevelControl(component);
if (!ctrl.shouldLog(level)) {
return;
}
if (logRejectFilter.shouldReject(record.getMessage())) {
return;
}
try {
// provokes rotation of target
setOutputStream(logTarget.open());
} catch (RuntimeException e) {
LogRecord r = new LogRecord(Level.SEVERE,
"Unable to open file target");
r.setThrown(e);
emergencyLog(r);
setOutputStream(System.err);
}
super.publish(record);
flush();
closeFileTarget();
}
public LevelController getLevelControl(String component) {
return repo.getLevelController(component);
}
/**
* Initalize the handler. The main invariant is that
* outputStream is always set to something valid when this method
* returns.
*/
private void initialize () {
try {
setFormatter(new VespaFormatter(serviceName, appPrefix));
setLevel(LogLevel.ALL);
setEncoding("UTF-8");
// System.err.println("initialize vespa logging, default level: "+defaultLogLevel);
setOutputStream(logTarget.open());
}
catch (UnsupportedEncodingException uee) {
LogRecord r = new LogRecord(Level.SEVERE, "Unable to set log encoding to UTF-8");
r.setThrown(uee);
emergencyLog(r);
}
catch (RuntimeException e) {
LogRecord r = new LogRecord(Level.SEVERE,
"Unable to open file target");
r.setThrown(e);
emergencyLog(r);
setOutputStream(System.err);
}
}
/** Closes the target log file, if there is one */
public void closeFileTarget() {
try {
logTarget.close();
}
catch (RuntimeException e) {
LogRecord r = new LogRecord(Level.WARNING, "Unable to close log");
r.setThrown(e);
emergencyLog(r);
}
}
/**
* If the logging system experiences problems we can't be expected
* to log it through normal channels, so we have an emergency log
* method which just uses STDERR for formatting the log messages.
* (Which might be right, and might be wrong).
*
* @param record The log record to be logged
*/
private void emergencyLog(LogRecord record) {
record.setLoggerName(VespaLogHandler.class.getName());
System.err.println(getFormatter().format(record));
}
public void cleanup() {
repo.close();
}
}
|
