com.cedarsoft.couchdb.io.RawCouchDocSerializer.java Source code

Introduction

Here is the source code for com.cedarsoft.couchdb.io.RawCouchDocSerializer.java

Source

/**
 * Copyright (C) cedarsoft GmbH.
 *
 * Licensed under the GNU General Public License version 3 (the "License")
 * with Classpath Exception; you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 *
 *         http://www.cedarsoft.org/gpl3ce
 *         (GPL 3 with Classpath Exception)
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 3 only, as
 * published by the Free Software Foundation. cedarsoft GmbH designates this
 * particular file as subject to the "Classpath" exception as provided
 * by cedarsoft GmbH in the LICENSE file that accompanied this code.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 3 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 3 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact cedarsoft GmbH, 72810 Gomaringen, Germany,
 * or visit www.cedarsoft.com if you need additional information or
 * have any questions.
 */

package com.cedarsoft.couchdb.io;

import com.cedarsoft.couchdb.core.DocId;
import com.cedarsoft.couchdb.core.RawCouchDoc;
import com.cedarsoft.couchdb.core.Revision;
import com.cedarsoft.serialization.jackson.JacksonParserWrapper;
import com.cedarsoft.serialization.jackson.JacksonSupport;
import com.fasterxml.jackson.core.JsonEncoding;
import com.fasterxml.jackson.core.JsonFactory;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonToken;

import javax.annotation.Nonnull;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

/**
 * @author Johannes Schneider (<a href="mailto:js@cedarsoft.com">js@cedarsoft.com</a>)
 */
public class RawCouchDocSerializer {

    public static final String PROPERTY_ID = "_id";

    public static final String PROPERTY_REV = "_rev";

    @Nonnull
    public byte[] serialize(@Nonnull RawCouchDoc info) throws IOException {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        serialize(info, out);
        return out.toByteArray();
    }

    public void serialize(@Nonnull RawCouchDoc doc, @Nonnull OutputStream out) throws IOException {
        try (JsonGenerator generator = createJsonGenerator(out)) {
            serialize(doc, generator);
        }
    }

    public void serialize(@Nonnull RawCouchDoc doc, @Nonnull JsonGenerator generator) throws IOException {
        generator.writeStartObject();
        serializeIdAndRev(generator, doc);
        generator.writeEndObject();
    }

    @Nonnull
    public RawCouchDoc deserialize(@Nonnull InputStream in) throws IOException {
        JsonParser parser = createJsonParser(in);
        RawCouchDoc doc = deserialize(parser);

        JacksonParserWrapper parserWrapper = new JacksonParserWrapper(parser);
        parserWrapper.ensureParserClosed();
        return doc;
    }

    @Nonnull
    public RawCouchDoc deserialize(@Nonnull JsonParser parser) throws IOException {
        JacksonParserWrapper parserWrapper = new JacksonParserWrapper(parser);
        parserWrapper.nextToken(JsonToken.START_OBJECT);

        parserWrapper.nextFieldValue(PROPERTY_ID);
        String id = parser.getText();

        parserWrapper.nextFieldValue(PROPERTY_REV);
        String rev = parser.getText();

        parser.nextToken();

        parserWrapper.ensureObjectClosed();
        return new RawCouchDoc(new DocId(id), new Revision(rev));
    }

    public static void serializeIdAndRev(@Nonnull JsonGenerator serializeTo, @Nonnull RawCouchDoc doc)
            throws IOException {
        serializeTo.writeStringField(PROPERTY_ID, doc.getId().asString());

        Revision rev = doc.getRev();
        if (rev != null) {
            serializeTo.writeStringField(PROPERTY_REV, rev.asString());
        }
    }

    @Nonnull
    protected static JsonParser createJsonParser(@Nonnull InputStream in) throws IOException {
        JsonFactory jsonFactory = JacksonSupport.getJsonFactory();
        return jsonFactory.createJsonParser(in);
    }

    @Nonnull
    protected static JsonGenerator createJsonGenerator(@Nonnull OutputStream out) throws IOException {
        JsonFactory jsonFactory = JacksonSupport.getJsonFactory();
        return jsonFactory.createJsonGenerator(out, JsonEncoding.UTF8);
    }
}