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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of querqy-elasticsearch Show documentation

Querqy library for query rewriting: Querqy for Elasticsearch

There is a newer version: 1.7.es8150.0

package querqy.elasticsearch.infologging;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.Marker;
import org.apache.logging.log4j.MarkerManager;
import querqy.elasticsearch.query.InfoLoggingSpec;
import querqy.infologging.Sink;
import querqy.rewrite.SearchEngineRequestAdapter;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

public class Log4jSink implements Sink {

    private static final String CONTEXT_KEY = Log4jSink.class.getName() + ".MESSAGES";
    private static final InfoLoggingSpec DEFAULT_SPEC = new InfoLoggingSpec();

    private final Logger logger = LogManager.getLogger(Log4jSink.class);
    private static final Marker MARKER_QUERQY_REWRITER = MarkerManager.getMarker("QUERQY");
    public static final Marker MARKER_QUERQY_REWRITER_ID = MarkerManager.getMarker("REWRITER_ID")
            .addParents(MARKER_QUERQY_REWRITER);
    public static final Marker MARKER_QUERQY_REWRITER_DETAIL = MarkerManager.getMarker("DETAIL")
            .addParents(MARKER_QUERQY_REWRITER);

    @Override
    public void log(final Object message, final String rewriterId,
                    final SearchEngineRequestAdapter searchEngineRequestAdapter) {

        final Map> messages = (Map>) searchEngineRequestAdapter
                .getContext().computeIfAbsent(CONTEXT_KEY, key -> new TreeMap<>());

        messages.computeIfAbsent(rewriterId, key -> new ArrayList<>()).add(message);

    }

    @Override
    public void endOfRequest(final SearchEngineRequestAdapter searchEngineRequestAdapter) {

        final Map> messages = (Map>) searchEngineRequestAdapter.getContext()
                .get(CONTEXT_KEY);

        if (messages != null && !messages.isEmpty()) {
            if (searchEngineRequestAdapter instanceof InfoLoggingSpecProvider) {
                final InfoLoggingSpec spec = ((InfoLoggingSpecProvider) searchEngineRequestAdapter).getInfoLoggingSpec()
                        .orElse(DEFAULT_SPEC);
                switch (spec.getPayloadType()) {
                        case REWRITER_ID:
                        logger.info(MARKER_QUERQY_REWRITER_ID,
                                new LogMessage.RewriterIdLogMessage(spec.getId().orElse(null), messages));
                        break;
                        case DETAIL:
                            logger.info(MARKER_QUERQY_REWRITER_DETAIL,
                                    new LogMessage.DetailLogMessage(spec.getId().orElse(null), messages));
                            break;

                    };
            }
        }
    }

}