com.datastax.dse.graph.internal.serde.DseGraphModule Maven / Gradle / Ivy
/*
* Copyright (C) 2012-2016 DataStax Inc.
*
* This software can be used solely with DataStax Enterprise. Please consult the license at
* http://www.datastax.com/terms/datastax-dse-driver-license-terms
*/
package com.datastax.dse.graph.internal.serde;
import com.datastax.driver.dse.geometry.Distance;
import com.datastax.driver.dse.geometry.Geometry;
import com.datastax.driver.dse.geometry.LineString;
import com.datastax.driver.dse.geometry.Point;
import com.datastax.driver.dse.geometry.Polygon;
import com.datastax.dse.graph.api.predicates.Geo;
import com.datastax.dse.graph.api.predicates.Search;
import com.datastax.dse.graph.internal.GeoPredicate;
import com.datastax.dse.graph.internal.SearchPredicate;
import org.apache.tinkerpop.gremlin.process.traversal.P;
import org.apache.tinkerpop.gremlin.process.traversal.util.AndP;
import org.apache.tinkerpop.gremlin.process.traversal.util.ConnectiveP;
import org.apache.tinkerpop.gremlin.process.traversal.util.OrP;
import org.apache.tinkerpop.gremlin.structure.io.graphson.AbstractObjectDeserializer;
import org.apache.tinkerpop.gremlin.structure.io.graphson.GraphSONTokens;
import org.apache.tinkerpop.gremlin.structure.io.graphson.TinkerPopJacksonModule;
import org.apache.tinkerpop.shaded.jackson.core.JsonGenerator;
import org.apache.tinkerpop.shaded.jackson.core.JsonParser;
import org.apache.tinkerpop.shaded.jackson.databind.DeserializationContext;
import org.apache.tinkerpop.shaded.jackson.databind.SerializerProvider;
import org.apache.tinkerpop.shaded.jackson.databind.deser.std.StdDeserializer;
import org.apache.tinkerpop.shaded.jackson.databind.ser.std.StdScalarSerializer;
import java.io.IOException;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* A Jackson Module to use for TinkerPop serialization/deserialization. It extends
* {@link org.apache.tinkerpop.gremlin.structure.io.graphson.TinkerPopJacksonModule} because of the specific typing format used in GraphSON.
*/
public class DseGraphModule extends TinkerPopJacksonModule {
public DseGraphModule() {
super("dse-driver-2.0");
addSerializer(LineString.class, new LineStringGeometrySerializer());
addSerializer(Distance.class, new DistanceGeometrySerializer());
addSerializer(Point.class, new PointGeometrySerializer());
addSerializer(Polygon.class, new PolygonGeometrySerializer());
//override TinkerPop's P predicates because of DSE's Search and Geo predicates
addSerializer(P.class, new DsePJacksonSerializer());
addDeserializer(LineString.class, new LineStringGeometryDeserializer());
addDeserializer(Point.class, new PointGeometryDeserializer());
addDeserializer(Polygon.class, new PolygonGeometryDeserializer());
addDeserializer(Distance.class, new DistanceGeometryDeserializer());
//override TinkerPop's P predicates because of DSE's Search and Geo predicates
addDeserializer(P.class, new DsePJacksonDeserializer());
}
@Override
public Map getTypeDefinitions() {
Map definitions = new HashMap<>();
definitions.put(LineString.class, "LineString");
definitions.put(Point.class, "Point");
definitions.put(Polygon.class, "Polygon");
definitions.put(byte[].class, "Blob");
definitions.put(Distance.class, "Distance");
definitions.put(P.class, "P");
return definitions;
}
@Override
public String getTypeNamespace() {
return "dse";
}
abstract static class AbstractGeometryJacksonDeserializer extends StdDeserializer {
public AbstractGeometryJacksonDeserializer(final Class clazz) {
super(clazz);
}
public abstract T parse(final String val);
@Override
public T deserialize(final JsonParser jsonParser, final DeserializationContext deserializationContext) throws IOException {
return parse(jsonParser.getText());
}
}
static abstract class AbstractGeometryJacksonSerializer extends StdScalarSerializer {
public AbstractGeometryJacksonSerializer(final Class clazz) {
super(clazz);
}
@Override
public void serialize(final T value, final JsonGenerator gen,
final SerializerProvider serializerProvider) throws IOException {
gen.writeString(value.asWellKnownText());
}
}
public static class LineStringGeometrySerializer extends AbstractGeometryJacksonSerializer {
public LineStringGeometrySerializer() {
super(LineString.class);
}
}
public static class LineStringGeometryDeserializer extends AbstractGeometryJacksonDeserializer {
public LineStringGeometryDeserializer() {
super(LineString.class);
}
@Override
public LineString parse(final String val) {
return LineString.fromWellKnownText(val);
}
}
public static class PolygonGeometrySerializer extends AbstractGeometryJacksonSerializer {
public PolygonGeometrySerializer() {
super(Polygon.class);
}
}
public static class PolygonGeometryDeserializer extends AbstractGeometryJacksonDeserializer {
public PolygonGeometryDeserializer() {
super(Polygon.class);
}
@Override
public Polygon parse(final String val) {
return Polygon.fromWellKnownText(val);
}
}
public static class PointGeometrySerializer extends AbstractGeometryJacksonSerializer {
public PointGeometrySerializer() {
super(Point.class);
}
}
public static class PointGeometryDeserializer extends AbstractGeometryJacksonDeserializer {
public PointGeometryDeserializer() {
super(Point.class);
}
@Override
public Point parse(final String val) {
return Point.fromWellKnownText(val);
}
}
public static class DistanceGeometrySerializer extends AbstractGeometryJacksonSerializer {
public DistanceGeometrySerializer() {
super(Distance.class);
}
}
public static class DistanceGeometryDeserializer extends AbstractGeometryJacksonDeserializer {
public DistanceGeometryDeserializer() {
super(Distance.class);
}
@Override
public Distance parse(final String val) {
return Distance.fromWellKnownText(val);
}
}
final static class DsePJacksonSerializer extends StdScalarSerializer {
public DsePJacksonSerializer() {
super(P.class);
}
@Override
public void serialize(final P p, final JsonGenerator jsonGenerator, final SerializerProvider serializerProvider) throws IOException {
jsonGenerator.writeStartObject();
jsonGenerator.writeStringField("predicateType", getPredicateType(p));
jsonGenerator.writeStringField(GraphSONTokens.PREDICATE,
p instanceof ConnectiveP ?
p instanceof AndP ?
GraphSONTokens.AND :
GraphSONTokens.OR :
p.getBiPredicate().toString());
if (p instanceof ConnectiveP) {
jsonGenerator.writeArrayFieldStart(GraphSONTokens.VALUE);
for (final P predicate : ((ConnectiveP) p).getPredicates()) {
jsonGenerator.writeObject(predicate);
}
jsonGenerator.writeEndArray();
} else {
if (p.getValue() instanceof Collection) {
jsonGenerator.writeArrayFieldStart(GraphSONTokens.VALUE);
for (final Object object : (Collection) p.getValue()) {
jsonGenerator.writeObject(object);
}
jsonGenerator.writeEndArray();
} else {
jsonGenerator.writeObjectField(GraphSONTokens.VALUE, p.getValue());
}
}
jsonGenerator.writeEndObject();
}
private String getPredicateType(P p) {
if (p.getBiPredicate() instanceof SearchPredicate)
return Search.class.getSimpleName();
else if (p.getBiPredicate() instanceof GeoPredicate)
return Geo.class.getSimpleName();
else
return P.class.getSimpleName();
}
}
final static class DsePJacksonDeserializer extends AbstractObjectDeserializer
{
public DsePJacksonDeserializer() {
super(P.class);
}
@Override
public P createObject(final Map data) {
final String predicate = (String) data.get(GraphSONTokens.PREDICATE);
final String predicateType = (String) data.get("predicateType");
final Object value = data.get(GraphSONTokens.VALUE);
if (predicate.equals(GraphSONTokens.AND) || predicate.equals(GraphSONTokens.OR)) {
return predicate.equals(GraphSONTokens.AND) ? new AndP((List) value) : new OrP((List
) value);
} else {
try {
if (value instanceof Collection) {
if (predicate.equals("between"))
return P.between(((List) value).get(0), ((List) value).get(1));
else if (predicateType.equals(P.class.getSimpleName()) && predicate.equals("inside"))
return P.between(((List) value).get(0), ((List) value).get(1));
else if (predicate.equals("outside"))
return P.outside(((List) value).get(0), ((List) value).get(1));
else if (predicate.equals("within"))
return P.within((Collection) value);
else if (predicate.equals("without"))
return P.without((Collection) value);
else
return (P) P.class.getMethod(predicate, Collection.class).invoke(null, (Collection) value);
} else {
if (predicate.equals(SearchPredicate.prefix.name()))
return Search.prefix(value);
else if (predicate.equals(SearchPredicate.tokenPrefix.name()))
return Search.tokenPrefix(value);
else if (predicate.equals(SearchPredicate.regex.name()))
return Search.regex(value);
else if (predicate.equals(SearchPredicate.tokenRegex.name()))
return Search.tokenRegex(value);
else if (predicate.equals(SearchPredicate.token.name()))
return Search.token(value);
else if (predicateType.equals(Geo.class.getSimpleName()) && predicate.equals(GeoPredicate.inside.name()))
return Geo.inside(value);
else
return (P) P.class.getMethod(predicate, Object.class).invoke(null, value);
}
} catch (final Exception e) {
throw new IllegalStateException(e.getMessage(), e);
}
}
}
}
}