com.arangodb.internal.velocypack.VPackDeserializers Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver Show documentation
ArangoDB Java Driver
There is a newer version: 7.15.0
/*
 * DISCLAIMER
 *
 * Copyright 2016 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.internal.velocypack;

import com.arangodb.entity.*;
import com.arangodb.entity.arangosearch.*;
import com.arangodb.velocypack.VPackDeserializer;
import com.arangodb.velocypack.VPackSlice;
import com.arangodb.velocystream.Response;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

/**
 * @author Mark Vollmary
 *
 */
public class VPackDeserializers {

	private static final Logger LOGGER = LoggerFactory.getLogger(VPackDeserializers.class);
	private static final String DATE_TIME_FORMAT = "yyyy-MM-dd'T'HH:mm:ss.SSSZ";

	public static final VPackDeserializer RESPONSE = (parent, vpack, context) -> {
		final Response response = new Response();
		response.setVersion(vpack.get(0).getAsInt());
		response.setType(vpack.get(1).getAsInt());
		response.setResponseCode(vpack.get(2).getAsInt());
		if (vpack.size() > 3) {
			response.setMeta(context.deserialize(vpack.get(3), Map.class));
		}
		return response;
	};

	public static final VPackDeserializer COLLECTION_TYPE = (parent, vpack, context) -> CollectionType.fromType(vpack.getAsInt());

	public static final VPackDeserializer COLLECTION_STATUS = (parent, vpack, context) -> CollectionStatus.fromStatus(vpack.getAsInt());

	@SuppressWarnings("unchecked")
	public static final VPackDeserializer BASE_DOCUMENT = (parent, vpack, context) -> new BaseDocument((Map) context.deserialize(vpack, Map.class));

	@SuppressWarnings("unchecked")
	public static final VPackDeserializer BASE_EDGE_DOCUMENT = (parent, vpack, context) -> new BaseEdgeDocument((Map) context.deserialize(vpack, Map.class));

	public static final VPackDeserializer DATE_STRING = (parent, vpack, context) -> {
		try {
			return new SimpleDateFormat(DATE_TIME_FORMAT).parse(vpack.getAsString());
		} catch (final ParseException e) {
			if (LOGGER.isDebugEnabled()) {
				LOGGER.debug("got ParseException for date string: " + vpack.getAsString());
			}
		}
		return null;
	};

	public static final VPackDeserializer LOG_LEVEL = (parent, vpack, context) -> LogLevel.fromLevel(vpack.getAsInt());

	public static final VPackDeserializer LICENSE = (parent, vpack, context) -> License.valueOf(vpack.getAsString().toUpperCase());

	public static final VPackDeserializer PERMISSIONS = (parent, vpack, context) -> Permissions.valueOf(vpack.getAsString().toUpperCase());

	public static final VPackDeserializer QUERY_EXECUTION_STATE = (parent, vpack, context) -> QueryExecutionState.valueOf(vpack.getAsString().toUpperCase().replaceAll(" ", "_"));

	public static final VPackDeserializer REPLICATION_FACTOR = (parent, vpack, context) -> {
		final ReplicationFactor replicationFactor = new ReplicationFactor();
		if (vpack.isString() && vpack.getAsString().equals("satellite")) {
			replicationFactor.setSatellite(true);
		} else {
			replicationFactor.setReplicationFactor(vpack.getAsInt());
		}
		return replicationFactor;
	};

	public static final VPackDeserializer MIN_REPLICATION_FACTOR = (parent, vpack, context) -> {
		final MinReplicationFactor minReplicationFactor = new MinReplicationFactor();
		minReplicationFactor.setMinReplicationFactor(vpack.getAsInt());
		return minReplicationFactor;
	};

	public static final VPackDeserializer VIEW_TYPE = (parent, vpack, context) -> "arangosearch".equals(vpack.getAsString()) ? ViewType.ARANGO_SEARCH
			: ViewType.valueOf(vpack.getAsString().toUpperCase());

	public static final VPackDeserializer ARANGO_SEARCH_PROPERTIES = (parent, vpack, context) -> {
		final ArangoSearchProperties properties = new ArangoSearchProperties();
		final VPackSlice consolidationIntervalMsec = vpack.get("consolidationIntervalMsec");
		if (consolidationIntervalMsec.isInteger()) {
			properties.setConsolidationIntervalMsec(consolidationIntervalMsec.getAsLong());
		}

		final VPackSlice commitIntervalMsec = vpack.get("commitIntervalMsec");
		if (commitIntervalMsec.isInteger()) {
			properties.setCommitIntervalMsec(commitIntervalMsec.getAsLong());
		}

		final VPackSlice cleanupIntervalStep = vpack.get("cleanupIntervalStep");
		if (cleanupIntervalStep.isInteger()) {
			properties.setCleanupIntervalStep(cleanupIntervalStep.getAsLong());
		}

		final VPackSlice consolidationPolicy = vpack.get("consolidationPolicy");
		if (consolidationPolicy.isObject()) {
			properties.setConsolidationPolicy(
					context.deserialize(consolidationPolicy, ConsolidationPolicy.class));
		}

		final VPackSlice links = vpack.get("links");
		if (links.isObject()) {
			final Iterator> collectionIterator = links.objectIterator();
			for (; collectionIterator.hasNext(); ) {
				final Entry entry = collectionIterator.next();
				final VPackSlice value = entry.getValue();
				final CollectionLink link = CollectionLink.on(entry.getKey());
				final VPackSlice analyzers = value.get("analyzers");
				if (analyzers.isArray()) {
					final Iterator analyzerIterator = analyzers.arrayIterator();
					for (; analyzerIterator.hasNext(); ) {
						link.analyzers(analyzerIterator.next().getAsString());
					}
				}
				final VPackSlice includeAllFields = value.get("includeAllFields");
				if (includeAllFields.isBoolean()) {
					link.includeAllFields(includeAllFields.getAsBoolean());
				}
				final VPackSlice trackListPositions = value.get("trackListPositions");
				if (trackListPositions.isBoolean()) {
					link.trackListPositions(trackListPositions.getAsBoolean());
				}
				final VPackSlice storeValues = value.get("storeValues");
				if (storeValues.isString()) {
					link.storeValues(StoreValuesType.valueOf(storeValues.getAsString().toUpperCase()));
				}
				final VPackSlice fields = value.get("fields");
				if (fields.isObject()) {
					final Iterator> fieldsIterator = fields.objectIterator();
					for (; fieldsIterator.hasNext(); ) {
						link.fields(deserializeField(fieldsIterator.next()));
					}
				}
				properties.addLink(link);
			}
		}

		final VPackSlice primarySorts = vpack.get("primarySort");
		if (primarySorts.isArray()) {
			final Iterator primarySortsIterator = primarySorts.arrayIterator();
			for (; primarySortsIterator.hasNext(); ) {
				final VPackSlice entry = primarySortsIterator.next();
				if (entry.isObject()) {
					if (entry.get("field").isString() && entry.get("asc").isBoolean()) {
						final PrimarySort primarySort = PrimarySort.on(entry.get("field").getAsString());
						primarySort.ascending(entry.get("asc").getAsBoolean());
						properties.addPrimarySort(primarySort);
					}
				}
			}
		}

		return properties;
	};

	protected static FieldLink deserializeField(final Entry field) {
		final VPackSlice value = field.getValue();
		final FieldLink link = FieldLink.on(field.getKey());
		final VPackSlice analyzers = value.get("analyzers");
		if (analyzers.isArray()) {
			final Iterator analyzerIterator = analyzers.arrayIterator();
			for (; analyzerIterator.hasNext();) {
				link.analyzers(analyzerIterator.next().getAsString());
			}
		}
		final VPackSlice includeAllFields = value.get("includeAllFields");
		if (includeAllFields.isBoolean()) {
			link.includeAllFields(includeAllFields.getAsBoolean());
		}
		final VPackSlice trackListPositions = value.get("trackListPositions");
		if (trackListPositions.isBoolean()) {
			link.trackListPositions(trackListPositions.getAsBoolean());
		}
		final VPackSlice storeValues = value.get("storeValues");
		if (storeValues.isString()) {
			link.storeValues(StoreValuesType.valueOf(storeValues.getAsString().toUpperCase()));
		}
		final VPackSlice fields = value.get("fields");
		if (fields.isObject()) {
			final Iterator> fieldsIterator = fields.objectIterator();
			for (; fieldsIterator.hasNext();) {
				link.fields(deserializeField(fieldsIterator.next()));
			}
		}
		return link;
	}

	public static final VPackDeserializer ARANGO_SEARCH_PROPERTIES_ENTITY = (parent, vpack, context) -> {
		final ViewEntity entity = context.deserialize(vpack, ViewEntity.class);
		final ArangoSearchProperties properties = context.deserialize(vpack, ArangoSearchProperties.class);
		return new ArangoSearchPropertiesEntity(entity.getId(),
				entity.getName(), entity.getType(), properties);
	};

	public static final VPackDeserializer CONSOLIDATE = (parent, vpack, context) -> {
		final VPackSlice type = vpack.get("type");
		if (type.isString()) {
			final ConsolidationPolicy consolidate = ConsolidationPolicy
					.of(ConsolidationType.valueOf(type.getAsString().toUpperCase()));
			final VPackSlice threshold = vpack.get("threshold");
			if (threshold.isNumber()) {
				consolidate.threshold(threshold.getAsDouble());
			}
			final VPackSlice segmentThreshold = vpack.get("segmentThreshold");
			if (segmentThreshold.isInteger()) {
				consolidate.segmentThreshold(segmentThreshold.getAsLong());
			}
			return consolidate;
		}
		return null;
	};

}