spmallette commented on code in PR #1835:
URL: https://github.com/apache/tinkerpop/pull/1835#discussion_r1002767340
##########
hadoop-gremlin/src/main/java/org/apache/tinkerpop/gremlin/hadoop/structure/io/graphson/GraphSONRecordReader.java:
##########
@@ -56,11 +59,33 @@ public GraphSONRecordReader() {
public void initialize(final InputSplit genericSplit, final
TaskAttemptContext context) throws IOException {
this.lineRecordReader.initialize(genericSplit, context);
this.hasEdges =
context.getConfiguration().getBoolean(Constants.GREMLIN_HADOOP_GRAPH_READER_HAS_EDGES,
true);
- this.graphsonReader = GraphSONReader.build().mapper(
- GraphSONMapper.build().
-
version(GraphSONVersion.valueOf(context.getConfiguration().get(Constants.GREMLIN_HADOOP_GRAPHSON_VERSION,
"V3_0"))).
- typeInfo(TypeInfo.PARTIAL_TYPES).
-
addRegistries(IoRegistryHelper.createRegistries(ConfUtil.makeApacheConfiguration(context.getConfiguration()))).create()).create();
+ GraphSONVersion graphSONVersion =
+
GraphSONVersion.valueOf(context.getConfiguration().get(Constants.GREMLIN_HADOOP_GRAPHSON_VERSION,
"V3_0"));
+ Mapper mapper = null;
+
+ if (graphSONVersion == GraphSONVersion.V2_0) {
Review Comment:
might be nice if you make this code a bit more compact - like, create a
builder instance with the common things and then just `addCustomModule(...)` in
the if-then parts. i can do it after it's merged if you don't have time.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
