com.arangodb.internal.serde.InternalDeserializers Maven / Gradle / Ivy
package com.arangodb.internal.serde;
import com.arangodb.entity.CollectionStatus;
import com.arangodb.entity.CollectionType;
import com.arangodb.entity.InvertedIndexPrimarySort;
import com.arangodb.entity.ReplicationFactor;
import com.arangodb.entity.arangosearch.CollectionLink;
import com.arangodb.entity.arangosearch.FieldLink;
import com.arangodb.util.RawBytes;
import com.arangodb.util.RawJson;
import com.arangodb.internal.InternalResponse;
import com.arangodb.shaded.fasterxml.jackson.core.JsonGenerator;
import com.arangodb.shaded.fasterxml.jackson.core.JsonParser;
import com.arangodb.shaded.fasterxml.jackson.core.TreeNode;
import com.arangodb.shaded.fasterxml.jackson.databind.DeserializationContext;
import com.arangodb.shaded.fasterxml.jackson.databind.JsonDeserializer;
import com.arangodb.shaded.fasterxml.jackson.databind.JsonNode;
import com.arangodb.shaded.fasterxml.jackson.databind.node.*;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.Map;
public final class InternalDeserializers {
static final JsonDeserializer RAW_JSON_DESERIALIZER = new JsonDeserializer() {
@Override
public RawJson deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
// TODO: find a way to access raw bytes directly
return RawJson.of(SerdeUtils.INSTANCE.writeJson(p.readValueAsTree()));
}
};
static final JsonDeserializer RAW_BYTES_DESERIALIZER = new JsonDeserializer() {
@Override
public RawBytes deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
// TODO: find a way to access raw bytes directly
ByteArrayOutputStream os = new ByteArrayOutputStream();
try (JsonGenerator g = p.getCodec().getFactory().createGenerator(os)) {
g.writeTree(p.readValueAsTree());
}
return RawBytes.of(os.toByteArray());
}
};
static final JsonDeserializer COLLECTION_STATUS = new JsonDeserializer() {
@Override
public CollectionStatus deserialize(final JsonParser p, final DeserializationContext ctxt) throws IOException {
return CollectionStatus.fromStatus(p.getIntValue());
}
};
static final JsonDeserializer COLLECTION_TYPE = new JsonDeserializer() {
@Override
public CollectionType deserialize(final JsonParser p, final DeserializationContext ctxt) throws IOException {
return CollectionType.fromType(p.getIntValue());
}
};
static final JsonDeserializer REPLICATION_FACTOR = new JsonDeserializer() {
@Override
public ReplicationFactor deserialize(final JsonParser p, final DeserializationContext ctxt) throws IOException {
TreeNode node = p.readValueAsTree();
if (node instanceof NumericNode) {
return ReplicationFactor.of(((NumericNode) node).intValue());
} else if (node instanceof TextNode && "satellite".equals(((TextNode) node).textValue())) {
return ReplicationFactor.ofSatellite();
} else throw new IllegalArgumentException();
}
};
@SuppressWarnings("unchecked")
static final JsonDeserializer RESPONSE = new JsonDeserializer() {
@Override
public InternalResponse deserialize(final JsonParser p, final DeserializationContext ctxt) throws IOException {
final InternalResponse response = new InternalResponse();
Iterator it = ((ArrayNode) p.readValueAsTree()).iterator();
response.setVersion(it.next().intValue());
response.setType(it.next().intValue());
response.setResponseCode(it.next().intValue());
if (it.hasNext()) {
response.putMetas(readTreeAsValue(p, ctxt, it.next(), Map.class));
}
return response;
}
};
static final JsonDeserializer INVERTED_INDEX_PRIMARY_SORT_FIELD = new JsonDeserializer() {
@Override
public InvertedIndexPrimarySort.Field deserialize(final JsonParser p, final DeserializationContext ctxt) throws IOException {
ObjectNode tree = p.readValueAsTree();
String field = tree.get("field").textValue();
InvertedIndexPrimarySort.Field.Direction direction = tree.get("asc").booleanValue() ?
InvertedIndexPrimarySort.Field.Direction.asc : InvertedIndexPrimarySort.Field.Direction.desc;
return new InvertedIndexPrimarySort.Field(field, direction);
}
};
private InternalDeserializers() {
}
private static T readTreeAsValue(JsonParser p, DeserializationContext ctxt, JsonNode n, Class targetType) throws IOException {
try (TreeTraversingParser t = new TreeTraversingParser(n, p.getCodec())) {
t.nextToken();
return ctxt.readValue(t, targetType);
}
}
public static class CollectionLinksDeserializer extends JsonDeserializer> {
@Override
public Collection deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
Collection out = new ArrayList<>();
ObjectNode tree = p.readValueAsTree();
Iterator> it = tree.fields();
while (it.hasNext()) {
Map.Entry e = it.next();
ObjectNode v = (ObjectNode) e.getValue();
v.put("name", e.getKey());
out.add(readTreeAsValue(p, ctxt, v, CollectionLink.class));
}
return out;
}
}
public static class FieldLinksDeserializer extends JsonDeserializer {
@Override
public FieldLink[] deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
Collection out = new ArrayList<>();
ObjectNode tree = p.readValueAsTree();
Iterator> it = tree.fields();
while (it.hasNext()) {
Map.Entry e = it.next();
ObjectNode v = (ObjectNode) e.getValue();
v.put("name", e.getKey());
out.add(readTreeAsValue(p, ctxt, v, FieldLink.class));
}
return out.toArray(new FieldLink[0]);
}
}
public static class CollectionSchemaRuleDeserializer extends JsonDeserializer {
@Override
public String deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
return SerdeUtils.INSTANCE.writeJson(p.readValueAsTree());
}
}
}