pomerium/internal/zero/reconciler/bundles_format.go

package reconciler

import (
	"bufio"
	"errors"
	"fmt"
	"io"

	"google.golang.org/protobuf/encoding/protodelim"

	"github.com/pomerium/pomerium/pkg/grpc/databroker"
)

var unmarshalOpts = protodelim.UnmarshalOptions{}

// ReadBundleRecords reads records in a protobuf wire format from src.
// Each record is expected to be a databroker.Record.
func ReadBundleRecords(src io.Reader) (databroker.RecordSetBundle, error) {
	r := bufio.NewReader(src)
	rsb := make(databroker.RecordSetBundle)
	for {
		record := new(databroker.Record)
		err := unmarshalOpts.UnmarshalFrom(r, record)
		if errors.Is(err, io.EOF) {
			break
		}
		if err != nil {
			return nil, fmt.Errorf("error reading protobuf record: %w", err)
		}

		rsb.Add(record)
	}

	return rsb, nil
}