com.arangodb.VelocyJack Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jackson-dataformat-velocypack Show documentation
Jackson Dataformat for Velocypack
There is a newer version: 4.4.0
Show newest version
/*
 * DISCLAIMER
 *
 * Copyright 2017 ArangoDB GmbH, Cologne, Germany
 *
 * 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.
 *
 * Copyright holder is ArangoDB GmbH, Cologne, Germany
 */

package com.arangodb;

import java.io.IOException;
import java.lang.reflect.Type;
import java.util.Arrays;
import java.util.Iterator;

import com.arangodb.jackson.dataformat.velocypack.VPackMapper;
import com.arangodb.util.ArangoSerialization;
import com.arangodb.util.ArangoSerializer;
import com.arangodb.velocypack.VPackSlice;
import com.fasterxml.jackson.annotation.JsonInclude.Include;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

/**
 * @author Mark Vollmary
 *
 */
public class VelocyJack implements ArangoSerialization {

	public interface ConfigureFunction {
		void configure(ObjectMapper mapper);
	}

	private final ObjectMapper vpackMapper;
	private final ObjectMapper vpackMapperNull;
	private final ObjectMapper jsonMapper;
	private final ObjectMapper jsonMapperNull;

	public VelocyJack() {
		super();
		vpackMapper = new VPackMapper().setSerializationInclusion(Include.NON_NULL);
		vpackMapperNull = new VPackMapper().setSerializationInclusion(Include.ALWAYS);
		jsonMapper = new ObjectMapper().setSerializationInclusion(Include.NON_NULL);
		jsonMapperNull = new ObjectMapper().setSerializationInclusion(Include.ALWAYS);
	}

	public void configure(final ConfigureFunction f) {
		f.configure(vpackMapper);
		f.configure(vpackMapperNull);
		f.configure(jsonMapper);
		f.configure(jsonMapperNull);
	}

	@Override
	public VPackSlice serialize(final Object entity) throws ArangoDBException {
		return serialize(entity, new ArangoSerializer.Options());
	}

	@SuppressWarnings("unchecked")
	@Override
	public VPackSlice serialize(final Object entity, final Options options) throws ArangoDBException {
		if (options.getType() == null) {
			options.type(entity.getClass());
		}
		try {
			final VPackSlice vpack;
			final Class type = entity.getClass();
			final boolean serializeNullValues = options.isSerializeNullValues();
			if (String.class.isAssignableFrom(type)) {
				final ObjectMapper p = serializeNullValues ? jsonMapperNull : jsonMapper;
				vpack = new VPackSlice(p.writeValueAsBytes((String) entity));
			} else if (options.isStringAsJson() && Iterable.class.isAssignableFrom(type)) {
				final Iterator iterator = Iterable.class.cast(entity).iterator();
				if (iterator.hasNext() && String.class.isAssignableFrom(iterator.next().getClass())) {
					final ObjectMapper p = serializeNullValues ? jsonMapperNull : jsonMapper;
					vpack = new VPackSlice(p.writeValueAsBytes((Iterable) entity));
				} else {
					final ObjectMapper vp = serializeNullValues ? vpackMapperNull : vpackMapper;
					vpack = new VPackSlice(vp.writeValueAsBytes(entity));
				}
			} else {
				final ObjectMapper vp = serializeNullValues ? vpackMapperNull : vpackMapper;
				vpack = new VPackSlice(vp.writeValueAsBytes(entity));
			}
			return vpack;
		} catch (final JsonProcessingException e) {
			throw new ArangoDBException(e);
		}
	}

	@SuppressWarnings("unchecked")
	@Override
	public  T deserialize(final VPackSlice vpack, final Type type) throws ArangoDBException {
		try {
			final T doc;
			if (type == String.class && !vpack.isString()) {
				final JsonNode node = vpackMapper.readTree(
					Arrays.copyOfRange(vpack.getBuffer(), vpack.getStart(), vpack.getStart() + vpack.getByteSize()));
				doc = (T) jsonMapper.writeValueAsString(node);
			} else {
				doc = vpackMapper.readValue(vpack.getBuffer(), vpack.getStart(), vpack.getStart() + vpack.getByteSize(),
					(Class) type);
			}
			return doc;
		} catch (final IOException e) {
			throw new ArangoDBException(e);
		}
	}

}