com.arpnetworking.logback.serialization.MapSerialziationStrategy.java Source code

Introduction

Here is the source code for com.arpnetworking.logback.serialization.MapSerialziationStrategy.java

Source

/**
 * Copyright 2015 Groupon.com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.arpnetworking.logback.serialization;

import ch.qos.logback.classic.spi.ILoggingEvent;
import com.arpnetworking.logback.StenoEncoder;
import com.fasterxml.jackson.core.JsonFactory;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.io.IOException;
import java.io.StringWriter;
import java.util.Map;

/**
 * Serialization strategy for map based message specifications.
 *
 * @author Ville Koskela (vkoskela at groupon dot com)
 * @since 1.3.1
 */
public class MapSerialziationStrategy extends BaseSerializationStrategy {

    /**
     * Public constructor.
     *
     * @param encoder Instance of <code>StenoEncoder</code>.
     * @param jsonFactory Instance of <code>JsonFactory</code>.
     * @param objectMapper Instance of <code>ObjectMapper</code>.
     */
    public MapSerialziationStrategy(final StenoEncoder encoder, final JsonFactory jsonFactory,
            final ObjectMapper objectMapper) {
        super(encoder);
        _jsonFactory = jsonFactory;
        _objectMapper = objectMapper;
    }

    /**
     * Serialize an event.
     *
     * @param event The event.
     * @param eventName The event name.
     * @param map The message key to value pairs.
     * @return Serialization of message as a <code>String</code>.
     */
    public String serialize(final ILoggingEvent event, final String eventName,
            final Map<String, ? extends Object> map) {
        final StringWriter jsonWriter = new StringWriter();

        try {
            final JsonGenerator jsonGenerator = _jsonFactory.createGenerator(jsonWriter);
            // Start wrapper
            startStenoWrapper(event, eventName, jsonGenerator, _objectMapper);

            // Write event data
            jsonGenerator.writeObjectFieldStart("data");
            if (map != null) {
                for (final Map.Entry<String, ? extends Object> entry : map.entrySet()) {
                    if (isSimpleType(entry.getValue())) {
                        jsonGenerator.writeObjectField(entry.getKey(), entry.getValue());
                    } else {
                        jsonGenerator.writeFieldName(entry.getKey());
                        _objectMapper.writeValue(jsonGenerator, entry.getValue());
                    }
                }
            }
            jsonGenerator.writeEndObject(); // End 'data' field

            // Output throwable
            writeThrowable(event.getThrowableProxy(), jsonGenerator, _objectMapper);

            // End wrapper
            endStenoWrapper(event, eventName, jsonGenerator, _objectMapper);
        } catch (final IOException e) {
            return "Unknown exception: " + e.getMessage();
        }

        return jsonWriter.toString();
    }

    private final JsonFactory _jsonFactory;
    private final ObjectMapper _objectMapper;
}