io.floodplain.replication.impl.json.JSONReplicationMessageParserImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of replication-json Show documentation
Transforms CDC streams in Kotlin
There is a newer version: 1.13.0
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 io.floodplain.replication.impl.json;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectWriter;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import io.floodplain.replication.api.ReplicationMessage;
import io.floodplain.replication.api.ReplicationMessageParser;
import io.floodplain.replication.factory.ReplicationFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.enterprise.context.ApplicationScoped;
import javax.inject.Named;
import java.io.IOException;
import java.io.InputStream;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Optional;

@Named("json")
@ApplicationScoped
public class JSONReplicationMessageParserImpl implements ReplicationMessageParser {

    private final static Logger logger = LoggerFactory.getLogger(JSONReplicationMessageParserImpl.class);

    private final boolean includeNullValues = true;

    public ReplicationMessage parseJson(Optional source, ObjectNode on) {
        return ReplicationJSON.parseJSON(source, on);
    }

    @Override
    public ReplicationMessage parseBytes(Optional source, byte[] data) {
        if (data == null) {
            return null;
        }
        try {
            return ReplicationJSON.parseReplicationMessage(data, source);
        } catch (JsonProcessingException e) {
            return ReplicationFactory.createErrorReplicationMessage(e);
        } catch (Throwable e) {
            return ReplicationFactory.createErrorReplicationMessage(e);
        }
    }


    @Override
    public ReplicationMessage parseStream(InputStream data) {
        return parseStream(Optional.empty(), data);
    }

    @Override
    public ReplicationMessage parseStream(Optional source, InputStream data) {
        JsonNode node;
        try {
            node = ReplicationJSON.objectMapper.readTree(data);
            return ReplicationJSON.parseJSON(source, (ObjectNode) node);
        } catch (IOException e) {
            return ReplicationFactory.createErrorReplicationMessage(e);
        }
    }

    @Override
    public List parseMessageList(Optional source, byte[] data) {
        try {
            JsonNode node = ReplicationJSON.objectMapper.readTree(data);
            return parseJSONNode(source, node);
        } catch (JsonProcessingException e) {
            List result = new LinkedList<>();
            result.add(ReplicationFactory.createErrorReplicationMessage(e));
            return Collections.unmodifiableList(result);
        } catch (Throwable e) {
            List result = new LinkedList<>();
            result.add(ReplicationFactory.createErrorReplicationMessage(e));
            return Collections.unmodifiableList(result);
        }
    }

    private List parseJSONNode(Optional source, JsonNode node) {
        List result = new LinkedList<>();
        if (!(node instanceof ArrayNode)) {
            logger.warn("Node is not an array, so can't parse to list, will create list of one!");
            ObjectNode on = (ObjectNode) node;
            ReplicationMessage single = ReplicationJSON.parseJSON(source, on);
            result.add(single);
            return Collections.unmodifiableList(result);
        }
        ArrayNode elements = (ArrayNode) node;
        for (JsonNode jsonNode : elements) {
            ObjectNode on = (ObjectNode) jsonNode;
            result.add(ReplicationJSON.parseJSON(source, on));
        }
        return Collections.unmodifiableList(result);
    }


    @Override
    public byte[] serializeMessageList(List data) {
        ArrayNode list = ReplicationJSON.objectMapper.createArrayNode();
        data.stream().map(msg -> ReplicationJSON.toJSON(msg, includeNullValues)).forEach(list::add);
        try {
            ObjectWriter w = ReplicationMessage.usePrettyPrint() ? ReplicationJSON.objectMapper.writerWithDefaultPrettyPrinter() : ReplicationJSON.objectMapper.writer();
            return w.writeValueAsBytes(list);
        } catch (JsonProcessingException e) {
            logger.error("Error: ", e);
            // TODO, what is wise to return in this case?
            return null;
        }
    }

    @Override
    public byte[] serialize(ReplicationMessage msg) {
        return ReplicationJSON.jsonSerializer(msg, includeNullValues);
    }

    @Override
    public String describe(ReplicationMessage msg) {
        return ReplicationJSON.jsonDescriber(msg);
    }

    @Override
    public List parseMessageList(Optional source, InputStream data) {
        try {
            JsonNode node = ReplicationJSON.objectMapper.readTree(data);
            return parseJSONNode(source, node);
        } catch (Throwable e) {
            List result = new LinkedList<>();
            result.add(ReplicationFactory.createErrorReplicationMessage(e));
            return Collections.unmodifiableList(result);
        }
    }

    @Override
    public List parseMessageList(byte[] data) {
        return parseMessageList(Optional.empty(), data);
    }


}