io.logspace.jvm.agent.api.json.EventJsonDeserializer Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of logspace-jvm-agent Show documentation
The newest version!
/**
 * Logspace
 * Copyright (c) 2015 Indoqa Software Design und Beratung GmbH. All rights reserved.
 * This program and the accompanying materials are made available under the terms of
 * the Eclipse Public License Version 1.0, which accompanies this distribution and
 * is available at http://www.eclipse.org/legal/epl-v10.html.
 */
package io.logspace.jvm.agent.api.json;

import static io.logspace.jvm.agent.shaded.jackson.core.JsonToken.*;
import static io.logspace.jvm.agent.api.event.Event.*;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;
import java.util.List;

import io.logspace.jvm.agent.shaded.jackson.core.JsonParser;

import io.logspace.jvm.agent.api.event.*;

public final class EventJsonDeserializer extends AbstractJsonDeserializer {

    private EventJsonDeserializer(byte[] data) throws IOException {
        super();

        this.setData(data);
    }

    private EventJsonDeserializer(InputStream inputStream) throws IOException {
        super();

        this.setInputStream(inputStream);
    }

    private EventJsonDeserializer(JsonParser parser) {
        super();

        this.setJsonParser(parser);
    }

    public static Event eventFromJson(byte[] data) throws IOException {
        return new EventJsonDeserializer(data).deserializeSingleEvent();
    }

    public static List fromJson(byte[] data) throws IOException {
        return new EventJsonDeserializer(data).deserialize(true);
    }

    public static List fromJson(InputStream inputStream) throws IOException {
        return new EventJsonDeserializer(inputStream).deserialize(true);
    }

    public static List fromJson(JsonParser parser) throws IOException {
        return new EventJsonDeserializer(parser).deserialize(false);
    }

    private List deserialize(boolean validateEnd) throws IOException {
        List result = new ArrayList();

        this.prepareToken();
        this.validateTokenType(START_ARRAY);
        this.consumeToken();

        while (true) {
            this.prepareToken();

            if (this.hasToken(END_ARRAY)) {
                this.consumeToken();
                break;
            }

            result.add(this.deserializeSingleEvent());
        }

        this.prepareToken();

        if (validateEnd) {
            this.validateEnd();
        }

        return result;
    }

    private Event deserializeSingleEvent() throws IOException {
        this.prepareToken();
        this.validateTokenType(START_OBJECT);
        this.consumeToken();

        Event result = this.readEvent();

        this.prepareToken();
        this.validateTokenType(END_OBJECT);
        this.consumeToken();

        return result;
    }

    private Event readEvent() throws IOException {
        DeserializedEvent event = new DeserializedEvent();

        event.setId(this.readMandatoryField(FIELD_ID));
        event.setType(this.readOptionalField(FIELD_TYPE));
        event.setSystem(this.readMandatoryField(FIELD_SYSTEM));
        event.setAgentId(this.readMandatoryField(FIELD_AGENT_ID));
        event.setTimestamp(this.readMandatoryDateField(FIELD_TIMESTAMP));
        event.setParentEventId(this.readOptionalField(FIELD_PARENT_EVENT_ID));
        event.setGlobalEventId(this.readOptionalField(FIELD_GLOBAL_EVENT_ID));
        event.setMarker(this.readOptionalField(FIELD_MARKER));
        event.setProperties(this.readProperties());

        return event;
    }

    private EventProperties readProperties() throws IOException {
        EventProperties result = new EventProperties();

        while (true) {
            this.prepareToken();
            if (!this.hasToken(FIELD_NAME)) {
                return result;
            }

            String fieldName = this.getCurrentName();
            EventPropertyJsonHandler eventPropertyJsonHandler = EventPropertyJsonHandlers.getHandler(fieldName);

            this.validateFieldName(FIELD_BOOLEAN_PROPERTIES, FIELD_DATE_PROPERTIES, FIELD_DOUBLE_PROPERTIES, FIELD_FLOAT_PROPERTIES,
                FIELD_INTEGER_PROPERTIES, FIELD_LONG_PROPERTIES, FIELD_STRING_PROPERTIES);

            this.prepareToken();
            this.validateTokenType(START_OBJECT);
            this.consumeToken();

            while (true) {
                this.prepareToken();
                this.validateTokenType(FIELD_NAME, END_OBJECT);

                if (this.hasToken(END_OBJECT)) {
                    this.consumeToken();
                    break;
                }

                this.validateTokenType(FIELD_NAME);
                String propertyName = this.getCurrentName();
                this.getJsonParser().nextToken();

                this.validateTokenType(START_ARRAY);
                this.consumeToken();

                eventPropertyJsonHandler.readEventProperties(result, propertyName, this.getJsonParser());

                this.prepareToken();
                this.validateTokenType(END_ARRAY);
                this.consumeToken();
            }
        }
    }

    public static class DeserializedEvent implements Event {

        private String id;
        private String system;
        private String agentId;
        private Date timestamp;
        private String parentEventId;
        private String globalEventId;
        private String type;
        private String marker;

        private EventProperties properties = new EventProperties();

        @Override
        public String getAgentId() {
            return this.agentId;
        }

        @Override
        public Collection getBooleanProperties() {
            return this.properties.getBooleanProperties();
        }

        @Override
        public Collection getDateProperties() {
            return this.properties.getDateProperties();
        }

        @Override
        public Collection getDoubleProperties() {
            return this.properties.getDoubleProperties();
        }

        @Override
        public Collection getFloatProperties() {
            return this.properties.getFloatProperties();
        }

        @Override
        public String getGlobalEventId() {
            return this.globalEventId;
        }

        @Override
        public String getId() {
            return this.id;
        }

        @Override
        public Collection getIntegerProperties() {
            return this.properties.getIntegerProperties();
        }

        @Override
        public Collection getLongProperties() {
            return this.properties.getLongProperties();
        }

        @Override
        public String getMarker() {
            return this.marker;
        }

        @Override
        public String getParentEventId() {
            return this.parentEventId;
        }

        @Override
        public Collection getStringProperties() {
            return this.properties.getStringProperties();
        }

        @Override
        public String getSystem() {
            return this.system;
        }

        @Override
        public Date getTimestamp() {
            return this.timestamp;
        }

        @Override
        public String getType() {
            return this.type;
        }

        @Override
        public boolean hasProperties() {
            return this.properties != null && !this.properties.isEmpty();
        }

        public void setAgentId(String agentId) {
            this.agentId = agentId;
        }

        public void setGlobalEventId(String globalEventId) {
            this.globalEventId = globalEventId;
        }

        public void setId(String id) {
            this.id = id;
        }

        public void setMarker(String marker) {
            this.marker = marker;
        }

        public void setParentEventId(String parentEventId) {
            this.parentEventId = parentEventId;
        }

        public void setProperties(EventProperties properties) {
            this.properties = properties;
        }

        public void setSystem(String system) {
            this.system = system;
        }

        public void setTimestamp(Date timestamp) {
            this.timestamp = timestamp;
        }

        public void setType(String type) {
            this.type = type;
        }
    }
}