| 1518 | } |
| 1519 | |
| 1520 | Status SegmentBundleReader::Read(void* data, int64 size) { |
| 1521 | if (size > remain_size_) { |
| 1522 | return errors::InvalidArgument("SegmentBundleReader Read Exhuasted"); |
| 1523 | } |
| 1524 | size_t read_size; |
| 1525 | TF_RETURN_IF_ERROR(input_->ReadNBytes(size, (char*)data, &read_size)); |
| 1526 | remain_size_ -= size; |
| 1527 | return Status::OK(); |
| 1528 | } |
| 1529 | |
| 1530 | Status SegmentBundleReader::Skip(int64 size) { |
| 1531 | if (size > remain_size_) { |
no test coverage detected