diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/backend/storage/aio/read_stream.c | 14 | ||||
| -rw-r--r-- | src/include/storage/read_stream.h | 2 |
2 files changed, 16 insertions, 0 deletions
diff --git a/src/backend/storage/aio/read_stream.c b/src/backend/storage/aio/read_stream.c index 7f0e07d9586..b8ea1180e25 100644 --- a/src/backend/storage/aio/read_stream.c +++ b/src/backend/storage/aio/read_stream.c @@ -798,6 +798,20 @@ read_stream_next_buffer(ReadStream *stream, void **per_buffer_data) } /* + * Transitional support for code that would like to perform or skip reads + * itself, without using the stream. Returns, and consumes, the next block + * number that would be read by the stream's look-ahead algorithm, or + * InvalidBlockNumber if the end of the stream is reached. Also reports the + * strategy that would be used to read it. + */ +BlockNumber +read_stream_next_block(ReadStream *stream, BufferAccessStrategy *strategy) +{ + *strategy = stream->ios[0].op.strategy; + return read_stream_get_block(stream, NULL); +} + +/* * Reset a read stream by releasing any queued up buffers, allowing the stream * to be used again for different blocks. This can be used to clear an * end-of-stream condition and start again, or to throw away blocks that were diff --git a/src/include/storage/read_stream.h b/src/include/storage/read_stream.h index 2f94ee744b9..2f98b600f7a 100644 --- a/src/include/storage/read_stream.h +++ b/src/include/storage/read_stream.h @@ -68,6 +68,8 @@ extern ReadStream *read_stream_begin_relation(int flags, void *callback_private_data, size_t per_buffer_data_size); extern Buffer read_stream_next_buffer(ReadStream *stream, void **per_buffer_data); +extern BlockNumber read_stream_next_block(ReadStream *stream, + BufferAccessStrategy *strategy); extern ReadStream *read_stream_begin_smgr_relation(int flags, BufferAccessStrategy strategy, SMgrRelation smgr, |