simplify FixedLengthBinRecReader via readFully call

thunder-project · Nov 4, 2014 · 7ea31c8 · 7ea31c8
1 parent bd268aa
commit 7ea31c8
Showing 1 changed file with 3 additions and 13 deletions.
diff --git a/scala/src/main/scala/thunder/util/io/hadoop/FixedLengthBinaryRecordReader.scala b/scala/src/main/scala/thunder/util/io/hadoop/FixedLengthBinaryRecordReader.scala
@@ -105,20 +105,10 @@ class FixedLengthBinaryRecordReader extends RecordReader[LongWritable, BytesWrit
 
       // setup a buffer to store the record
       val buffer = recordValue.getBytes
+      fileInputStream.readFully(buffer)
 
-      breakable {
-        var bytestoread: Int = recordLength
-        while (bytestoread > 0) {
-          val bytesread = fileInputStream.read(buffer, recordLength - bytestoread, bytestoread)
-          if (bytesread == -1) {
-            // EOF
-            break() 
-          }
-          // update our current position
-          currentPosition += bytesread
-          bytestoread -= bytesread
-        }
-      }
+      // update our current position
+      currentPosition += recordLength
 
       // return true
       return true