package org.graylog2.logging;
import org.graylog2.*;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.net.InetAddress;
import java.net.SocketException;
import java.net.URISyntaxException;
import java.net.UnknownHostException;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.text.MessageFormat;
import java.util.HashMap;
import java.util.IllegalFormatConversionException;
import java.util.Map;
import java.util.logging.*;
public class GelfHandler
extends Handler {
private static final int MAX_SHORT_MESSAGE_LENGTH = 250;
private String graylogHost;
private String amqpURI;
private String amqpExchangeName;
private String amqpRoutingKey;
private int amqpMaxRetries;
private String originHost;
private int graylogPort;
private String facility;
private GelfSender gelfSender;
private boolean extractStacktrace;
private Map<String, String> fields;
public GelfHandler() {
final LogManager manager = LogManager.getLogManager();
final String prefix = getClass().getName();
graylogHost = manager.getProperty(prefix + ".graylogHost");
final String port = manager.getProperty(prefix + ".graylogPort");
graylogPort = null == port ? 12201 : Integer.parseInt(port);
originHost = manager.getProperty(prefix + ".originHost");
extractStacktrace = "true".equalsIgnoreCase(manager.getProperty(prefix + ".extractStacktrace"));
int fieldNumber = 0;
fields = new HashMap<String, String>();
while (true) {
final String property = manager.getProperty(prefix + ".additionalField." + fieldNumber);
if (null == property) {
break;
}
final int index = property.indexOf('=');
if (-1 != index) {
fields.put(property.substring(0, index), property.substring(index + 1));
}
fieldNumber++;
}
facility = manager.getProperty(prefix + ".facility");
amqpURI = manager.getProperty(prefix + ".amqpURI");
amqpExchangeName = manager.getProperty(prefix + ".amqpExchangeName");
amqpRoutingKey = manager.getProperty(prefix + ".amqpRoutingKey");
String maxRetries = manager.getProperty(prefix + ".amqpMaxRetries");
amqpMaxRetries = maxRetries == null ? 0 : Integer.valueOf(maxRetries);
final String level = manager.getProperty(prefix + ".level");
if (null != level) {
setLevel(Level.parse(level.trim()));
} else {
setLevel(Level.INFO);
}
final String filter = manager.getProperty(prefix + ".filter");
try {
if (null != filter) {
final Class clazz = ClassLoader.getSystemClassLoader().loadClass(filter);
setFilter((Filter) clazz.newInstance());
}
} catch (final Exception e) {
//ignore
}
//This only used for testing
final String testSender = manager.getProperty(prefix + ".graylogTestSenderClass");
try {
if (null != testSender) {
final Class clazz = ClassLoader.getSystemClassLoader().loadClass(testSender);
gelfSender = (GelfSender) clazz.newInstance();
}
} catch (final Exception e) {
//ignore
}
}
@Override
public synchronized void flush() {
}
private String getOriginHost() {
if (null == originHost) {
originHost = getLocalHostName();
}
return originHost;
}
private String getLocalHostName() {
try {
return InetAddress.getLocalHost().getHostName();
} catch (final UnknownHostException uhe) {
reportError("Unknown local hostname", uhe, ErrorManager.GENERIC_FAILURE);
}
return null;
}
@Override
public synchronized void publish(final LogRecord record) {
if (!isLoggable(record)) {
return;
}
if (null == gelfSender) {
if (graylogHost == null && amqpURI == null) {
reportError("Graylog2 hostname and amqp uri are empty!", null, ErrorManager.WRITE_FAILURE);
} else if (graylogHost != null && amqpURI != null) {
reportError("Graylog2 hostname and amqp uri are both informed!", null, ErrorManager.WRITE_FAILURE);
} else {
try {
if (graylogHost.startsWith("tcp:")) {
String tcpGraylogHost = graylogHost.substring(4, graylogHost.length());
gelfSender = new GelfTCPSender(tcpGraylogHost, graylogPort);
} else if (graylogHost.startsWith("udp:")) {
String udpGraylogHost = graylogHost.substring(4, graylogHost.length());
gelfSender = new GelfUDPSender(udpGraylogHost, graylogPort);
} else if (amqpURI != null) {
gelfSender = new GelfAMQPSender(amqpURI, amqpExchangeName, amqpRoutingKey, amqpMaxRetries);
} else {
gelfSender = new GelfUDPSender(graylogHost, graylogPort);
}
} catch (UnknownHostException e) {
reportError("Unknown Graylog2 hostname:" + graylogHost, e, ErrorManager.WRITE_FAILURE);
} catch (SocketException e) {
reportError("Socket exception", e, ErrorManager.WRITE_FAILURE);
} catch (IOException e) {
reportError("IO exception", e, ErrorManager.WRITE_FAILURE);
} catch (URISyntaxException e) {
reportError("AMQP uri exception", e, ErrorManager.WRITE_FAILURE);
} catch (NoSuchAlgorithmException e) {
reportError("AMQP algorithm exception", e, ErrorManager.WRITE_FAILURE);
} catch (KeyManagementException e) {
reportError("AMQP key exception", e, ErrorManager.WRITE_FAILURE);
}
}
}
if (null == gelfSender) {
reportError("Could not send GELF message", null, ErrorManager.WRITE_FAILURE);
} else {
GelfSenderResult gelfSenderResult = gelfSender.sendMessage(makeMessage(record));
if (!GelfSenderResult.OK.equals(gelfSenderResult)) {
reportError("Error during sending GELF message. Error code: " + gelfSenderResult.getCode() + ".", gelfSenderResult.getException(), ErrorManager.WRITE_FAILURE);
}
}
}
@Override
public void close() {
if (null != gelfSender) {
gelfSender.close();
gelfSender = null;
}
}
private GelfMessage makeMessage(final LogRecord record) {
String message = record.getMessage();
Object[] parameters = record.getParameters();
if (message == null) message = "";
if (parameters != null && parameters.length > 0) {
//by default, using {0}, {1}, etc. -> MessageFormat
message = MessageFormat.format(message, parameters);
if (message.equals(record.getMessage())) {
//if the text is the same, assuming this is String.format type log (%s, %d, etc.)
try {
message = String.format(message, parameters);
} catch (IllegalFormatConversionException e) {
//leaving message as it is to avoid compatibility problems
message = record.getMessage();
} catch (NullPointerException e) {
//ignore
}
}
}
final String shortMessage;
if (message.length() > MAX_SHORT_MESSAGE_LENGTH) {
shortMessage = message.substring(0, MAX_SHORT_MESSAGE_LENGTH - 1);
} else {
shortMessage = message;
}
if (extractStacktrace) {
final Throwable thrown = record.getThrown();
if (null != thrown) {
final StringWriter sw = new StringWriter();
thrown.printStackTrace(new PrintWriter(sw));
message += "\n\r" + sw.toString();
}
}
final GelfMessage gelfMessage =
new GelfMessage(shortMessage,
message,
record.getMillis(),
String.valueOf(levelToSyslogLevel(record.getLevel())));
gelfMessage.addField("SourceClassName", record.getSourceClassName());
gelfMessage.addField("SourceMethodName", record.getSourceMethodName());
if (null != getOriginHost()) {
gelfMessage.setHost(getOriginHost());
}
if (null != facility) {
gelfMessage.setFacility(facility);
}
if (null != fields) {
for (final Map.Entry<String, String> entry : fields.entrySet()) {
gelfMessage.addField(entry.getKey(), entry.getValue());
}
}
return gelfMessage;
}
private int levelToSyslogLevel(final Level level) {
final int syslogLevel;
if (level.intValue() == Level.SEVERE.intValue()) {
syslogLevel = 3;
} else if (level.intValue() == Level.WARNING.intValue()) {
syslogLevel = 4;
} else if (level.intValue() == Level.INFO.intValue()) {
syslogLevel = 6;
} else {
syslogLevel = 7;
}
return syslogLevel;
}
public void setExtractStacktrace(boolean extractStacktrace) {
this.extractStacktrace = extractStacktrace;
}
public void setGraylogPort(int graylogPort) {
this.graylogPort = graylogPort;
}
public void setOriginHost(String originHost) {
this.originHost = originHost;
}
public void setGraylogHost(String graylogHost) {
this.graylogHost = graylogHost;
}
public void setFacility(String facility) {
this.facility = facility;
}
public void setAdditionalField(String entry) {
if (entry == null) return;
final int index = entry.indexOf('=');
if (-1 != index) {
String key = entry.substring(0, index);
String val = entry.substring(index + 1);
if (key.equals("")) return;
fields.put(key, val);
}
}
public Map<String, String> getFields() {
return fields;
}
}