src/main/java/com/uber/rss/clients/RecordSocketReadClient.java [76:89]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    public TaskDataBlock readDataBlock() {
        try {
            DataBlock dataBlock = dataBlockSocketReadClient.readDataBlock();
            if (dataBlock == null) {
                return null;
            }
            shuffleReadBytes += DataBlockHeader.NUM_BYTES + dataBlock.getPayload().length;
            return new TaskDataBlock(dataBlock.getPayload(), dataBlock.getHeader().getTaskAttemptId());
        } catch (RuntimeException ex) {
            logger.warn(String.format("Failed to read shuffle data %s", this), ex);
            close();
            throw ex;
        }
    }
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/main/java/com/uber/rss/clients/ShuffleDataSocketReadClient.java [76:89]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  public TaskDataBlock readDataBlock() {
    try {
      DataBlock dataBlock = dataBlockSocketReadClient.readDataBlock();
      if (dataBlock == null) {
        return null;
      }
      shuffleReadBytes += DataBlockHeader.NUM_BYTES + dataBlock.getPayload().length;
      return new TaskDataBlock(dataBlock.getPayload(), dataBlock.getHeader().getTaskAttemptId());
    } catch (RuntimeException ex) {
      logger.warn(String.format("Failed to read shuffle data %s", this), ex);
      close();
      throw ex;
    }
  }
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -