querqy.elasticsearch.infologging.LogMessage Maven / Gradle / Ivy

Go to download
package querqy.elasticsearch.infologging;

import org.apache.logging.log4j.core.util.JsonUtils;
import org.apache.logging.log4j.message.Message;
import org.apache.logging.log4j.util.StringBuilderFormattable;

import java.util.Collection;
import java.util.List;
import java.util.Map;

public abstract class LogMessage implements Message, StringBuilderFormattable {

    private final Map> messages;
    private final String id;

    public LogMessage(final String id, final Map> messages) {
        this.id = id;
        this.messages = messages;
    }


    @Override
    public String getFormattedMessage() {
        final StringBuilder builder = new StringBuilder();
        formatTo(builder);
        return builder.toString();
    }

    @Override
    public String getFormat() {
        return "";
    }

    @Override
    public Object[] getParameters() {
        return null;
    }

    @Override
    public Throwable getThrowable() {
        return null;
    }

    @Override
    public void formatTo(final StringBuilder buffer) {
        buffer.append('{');
        if (id != null) {
            buffer.append("\"id\":\"");
            JsonUtils.quoteAsString(id, buffer);
            buffer.append("\",");
        }
        buffer.append("\"msg\":");
        formatMessagesTo(messages, buffer);
        buffer.append('}');
    }

    public abstract void formatMessagesTo(Map> messages, final StringBuilder buffer);

    public static class DetailLogMessage extends LogMessage {

        public DetailLogMessage(final String id, final Map> messages) {
            super(id, messages);
        }

        @Override
        public void formatMessagesTo(Map> messages, final StringBuilder buffer) {
            appendValue(messages, buffer);
        }
    }

    public static class RewriterIdLogMessage extends LogMessage {

        public RewriterIdLogMessage(final String id, final Map> messages) {
            super(id, messages);
        }

        @Override
        public void formatMessagesTo(Map> messages, final StringBuilder buffer) {
            appendValue(messages.keySet(), buffer);
        }
    }

    public static void appendValue(final Object value, final StringBuilder buffer) {
        if (value == null) {
            buffer.append("null");
        } else if (value instanceof Map) {
            final Map map = (Map) value;
            buffer.append('{');
            boolean first = true;
            for (final Map.Entry entry : map.entrySet()) {
                if (first) {
                    first = false;
                } else {
                    buffer.append(',');
                }
                buffer.append('"');
                JsonUtils.quoteAsString(entry.getKey(), buffer);
                buffer.append("\":");
                appendValue(entry.getValue(), buffer);
            }
            buffer.append('}');
        } else if (value instanceof Collection) {
            final Collection coll = (Collection) value;
            buffer.append('[');
            boolean first = true;
            for (final Object element: coll) {
                if (first) {
                    first = false;
                } else {
                    buffer.append(',');
                }
                appendValue(element, buffer);
            }
            buffer.append(']');
        } else if (value instanceof String) {
            buffer.append('"');
            JsonUtils.quoteAsString((String) value, buffer);
            buffer.append('"');
        } else if (value instanceof Number || value instanceof Boolean) {
            buffer.append(value.toString());
        } else {
            // whatever it is
            buffer.append('"');
            JsonUtils.quoteAsString(value.toString(), buffer);
            buffer.append('"');
        }
    }
}