wal_file_page_reader.go

package walparser

import (
	"github.com/pkg/errors"
	"io"
)

type WalPageReader struct {
	walFileReader io.Reader
}

func NewWalPageReader(walFileReader io.Reader) *WalPageReader {
	return &WalPageReader{walFileReader}
}

// ReadPageData reads data corresponding to one page
func (reader *WalPageReader) ReadPageData() ([]byte, error) {
	page := make([]byte, WalPageSize)
	_, err := io.ReadFull(reader.walFileReader, page)
	if err != nil && err != io.EOF {
		return nil, errors.WithStack(err)
	}
	return page, err
}