com.github.pukkaone.gelf.protocol.GelfMessage Maven / Gradle / Ivy

Go to download
package com.github.pukkaone.gelf.protocol;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.util.HashMap;
import java.util.Map;

public class GelfMessage {

    public static final String FACILITY = "facility";

    private static final String VERSION_VALUE = "1.1";
    private static final String HOST = "host";
    private static final String SHORT_MESSAGE = "short_message";
    private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper();

    private long timestampMillis;
    private Map fieldNameToValueMap = new HashMap<>();

    public GelfMessage() {
        fieldNameToValueMap.put("version", VERSION_VALUE);
    }

    public long getTimestampMillis() {
        return timestampMillis;
    }

    public GelfMessage setTimestampMillis(long timestampMillis) {
        this.timestampMillis = timestampMillis;

        fieldNameToValueMap.put("timestamp", timestampMillis / 1000.0);
        return this;
    }

    public String getHost() {
        return (String) fieldNameToValueMap.get(HOST);
    }

    public GelfMessage setHost(String host) {
        fieldNameToValueMap.put(HOST, host);
        return this;
    }

    public GelfMessage setLevel(int level) {
        fieldNameToValueMap.put("level", level);
        return this;
    }

    private String prefix(String name) {
      return '_' + name;
    }

    public GelfMessage setFacility(String facility) {
        fieldNameToValueMap.put(prefix(FACILITY), facility);
        return this;
    }

    public GelfMessage setFile(String file) {
        fieldNameToValueMap.put("_file", file);
        return this;
    }

    public GelfMessage setLine(int line) {
        fieldNameToValueMap.put("_line", line);
        return this;
    }

    public GelfMessage setLogger(String logger) {
        fieldNameToValueMap.put("_logger", logger);
        return this;
    }

    public GelfMessage setMarker(String marker) {
        fieldNameToValueMap.put("_marker", marker);
        return this;
    }

    public GelfMessage setThread(String thread) {
        fieldNameToValueMap.put("_thread", thread);
        return this;
    }

    public GelfMessage setShortMessage(String shortMessage) {
        fieldNameToValueMap.put(SHORT_MESSAGE, shortMessage);
        return this;
    }

    public GelfMessage setFullMessage(String fullMessage) {
        fieldNameToValueMap.put("full_message", fullMessage);
        return this;
    }

    public Object getField(String name) {
        return fieldNameToValueMap.get(prefix(name));
    }

    public GelfMessage addField(String name, Object value) {
        // Disallow sending id additional field.
        if (!"id".equals(name)) {
            fieldNameToValueMap.put('_' + name, value);
        }
        return this;
    }

    private boolean isNotBlank(String fieldName) {
        Object value = fieldNameToValueMap.get(fieldName);
        if (value == null) {
            return false;
        }

        if (value instanceof String) {
            return !((String) value).trim().isEmpty();
        }

        return false;
    }

    public boolean isValid() {
        return isNotBlank(HOST) && isNotBlank(SHORT_MESSAGE);
    }

    public String toJson() {
        try {
            return OBJECT_MAPPER.writeValueAsString(fieldNameToValueMap);
        } catch (JsonProcessingException e) {
            throw new IllegalStateException("Cannot write JSON", e);
        }
    }
}