(p []byte, off int64)
| 138 | } |
| 139 | |
| 140 | func (wr *walReaderAt) ReadAt(p []byte, off int64) (int, error) { |
| 141 | if err := wr.ctx.Err(); err != nil { |
| 142 | return 0, err |
| 143 | } |
| 144 | |
| 145 | // parquet-go will call ReadAt when reading data from a parquet file. |
| 146 | n, err := wr.r.ReadAt(p, off) |
| 147 | // ReadAt can read less than len(p) bytes in some cases |
| 148 | wr.bytesRead.Add(uint64(n)) |
| 149 | |
| 150 | if err != nil { |
| 151 | return 0, err |
| 152 | } |
| 153 | |
| 154 | return n, nil |
| 155 | } |
| 156 | |
| 157 | func (wr *walReaderAt) BytesRead() uint64 { |
| 158 | return wr.bytesRead.Load() |