Record the tfrecord name

cheyang · Mar 19, 2020 · 5401562 · 5401562
1 parent 590d6ee
commit 5401562
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tensorflow/core/lib/io/random_inputstream.h b/tensorflow/core/lib/io/random_inputstream.h
@@ -49,9 +49,8 @@ class RandomAccessInputStream : public InputStreamInterface {
   }
 
   Status Reset() override { return Seek(0); }
-
- private:
   RandomAccessFile* file_;  // Not owned.
+ private:
   int64 pos_ = 0;           // Tracks where we are in the file.
   bool owns_file_ = false;
 };

diff --git a/tensorflow/core/lib/io/record_reader.cc b/tensorflow/core/lib/io/record_reader.cc
@@ -102,7 +102,11 @@ Status RecordReader::ReadChecksummed(uint64 offset, size_t n, string* result) {
 
   const uint32 masked_crc = core::DecodeFixed32(result->data() + n);
   if (crc32c::Unmask(masked_crc) != crc32c::Value(result->data(), n)) {
-    return errors::DataLoss("corrupted record at ", offset);
+    auto rai = dynamic_cast<RandomAccessInputStream *>(input_stream_.get());
+
+    tensorflow::StringPiece file_name;
+    rai->file_->Name(&file_name);
+    return errors::DataLoss("corrupted record at file:", file_name, " offset: ", offset);
   }
   result->resize(n);
   return Status::OK();