-
Notifications
You must be signed in to change notification settings - Fork 2
/
ContextBuildingJsonLdSerializer.java
60 lines (53 loc) · 2.59 KB
/
ContextBuildingJsonLdSerializer.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package cz.cvut.kbss.jsonld.serialization;
import cz.cvut.kbss.jsonld.ConfigParam;
import cz.cvut.kbss.jsonld.Configuration;
import cz.cvut.kbss.jsonld.JsonLd;
import cz.cvut.kbss.jsonld.serialization.context.ContextMappingSerializationContextFactory;
import cz.cvut.kbss.jsonld.serialization.context.JsonLdContext;
import cz.cvut.kbss.jsonld.serialization.model.CollectionNode;
import cz.cvut.kbss.jsonld.serialization.model.JsonNode;
import cz.cvut.kbss.jsonld.serialization.model.ObjectNode;
import cz.cvut.kbss.jsonld.serialization.traversal.ObjectGraphTraverser;
import java.util.Collection;
/**
* JSON-LD serializer outputting compacted JSON-LD with context.
*/
public class ContextBuildingJsonLdSerializer extends JsonLdSerializer {
ContextBuildingJsonLdSerializer(JsonGenerator jsonGenerator) {
super(jsonGenerator);
}
ContextBuildingJsonLdSerializer(JsonGenerator jsonGenerator, Configuration configuration) {
super(jsonGenerator, configuration);
}
@Override
protected JsonNode buildJsonTree(Object root) {
final JsonLdContext context = new JsonLdContext();
final ObjectGraphTraverser traverser =
new ObjectGraphTraverser(new ContextMappingSerializationContextFactory(context));
traverser.setRequireId(configuration().is(ConfigParam.REQUIRE_ID));
final JsonLdTreeBuilder treeBuilder =
new JsonLdTreeBuilder(new ObjectGraphValueSerializers(serializers, traverser));
traverser.setVisitor(treeBuilder);
if (root instanceof Collection) {
return buildObjectWithContextAndGraph(traverser, context, (Collection<?>) root);
}
traverser.traverse(root);
treeBuilder.getTreeRoot().prependItem(context.getContextNode());
return treeBuilder.getTreeRoot();
}
private JsonNode buildObjectWithContextAndGraph(ObjectGraphTraverser traverser, JsonLdContext context,
Collection<?> items) {
final CollectionNode<?> graph = JsonNodeFactory.createCollectionNodeFromArray(JsonLd.GRAPH);
items.stream().map(item -> {
final JsonLdTreeBuilder treeBuilder =
new JsonLdTreeBuilder(new ObjectGraphValueSerializers(serializers, traverser));
traverser.setVisitor(treeBuilder);
traverser.traverse(item);
return treeBuilder.getTreeRoot();
}).forEach(graph::addItem);
final ObjectNode result = JsonNodeFactory.createObjectNode();
result.addItem(context.getContextNode());
result.addItem(graph);
return result;
}
}