abstractdog commented on code in PR #341:
URL: https://github.com/apache/tez/pull/341#discussion_r1547310433
##########
tez-mapreduce/src/main/java/org/apache/tez/mapreduce/hadoop/MRInputHelpers.java:
##########
@@ -889,4 +895,29 @@ public static int getDagAttemptNumber(Configuration conf) {
return getIntProperty(conf, MRInput.TEZ_MAPREDUCE_DAG_ATTEMPT_NUMBER);
}
+ public static MRSplitProto getProto(InputDataInformationEvent initEvent,
JobConf jobConf) throws IOException {
+ return Strings.isNullOrEmpty(initEvent.getSerializedPath()) ?
readProtoFromPayload(initEvent)
+ : readProtoFromFs(initEvent, jobConf);
+ }
+
+ private static MRSplitProto readProtoFromFs(InputDataInformationEvent
initEvent, JobConf jobConf) throws IOException {
+ String serializedPath = initEvent.getSerializedPath();
+ Path filePath = new Path(serializedPath);
+ LOG.info("Reading InputDataInformationEvent from path: {}", filePath);
+
+ MRSplitProto splitProto = null;
+ FileSystem fs = FileSystem.get(filePath.toUri(), jobConf);
+
+ try (FSDataInputStream in = fs.open(filePath)) {
+ splitProto = MRSplitProto.parseFrom(in);
+ fs.delete(filePath, false);
+ }
+ return splitProto;
+ }
+
+ private static MRSplitProto readProtoFromPayload(InputDataInformationEvent
initEvent) throws IOException {
+ ByteBuffer payload = initEvent.getUserPayload();
+ LOG.info("Reading InputDataInformationEvent from payload, size: {}
bytes}", payload.limit());
Review Comment:
ack, will do this below:
```
LOG.info("Reading InputDataInformationEvent from payload: {}", payload);
```
--
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: issues-unsubscr...@tez.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
