track_header_stream.go

package moqtransport

import "github.com/mengelbart/moqtransport/internal/wire"

type trackHeaderStream struct {
	stream SendStream
}

func newTrackHeaderStream(stream SendStream, subscribeID, trackAlias uint64, publisherPriority uint8) (*trackHeaderStream, error) {
	shtm := &wire.StreamHeaderTrackMessage{
		SubscribeID:       subscribeID,
		TrackAlias:        trackAlias,
		PublisherPriority: publisherPriority,
	}
	buf := make([]byte, 0, 32)
	buf = shtm.Append(buf)
	_, err := stream.Write(buf)
	if err != nil {
		return nil, err
	}
	return &trackHeaderStream{
		stream: stream,
	}, nil
}

func (s *trackHeaderStream) writeObject(groupID, objectID uint64, payload []byte) (int, error) {
	shto := wire.StreamHeaderTrackObject{
		GroupID:       groupID,
		ObjectID:      objectID,
		ObjectPayload: payload,
	}
	buf := make([]byte, 0, 32+len(payload))
	buf = shto.Append(buf)
	return s.stream.Write(buf)
}

func (s *trackHeaderStream) Close() error {
	return s.stream.Close()
}