object_stream.go

package moqtransport

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

type objectStream struct {
	stream SendStream
}

func newObjectStream(stream SendStream, subscribeID, trackAlias, groupID, objectID uint64, publisherPriority uint8) (*objectStream, error) {
	osm := &wire.ObjectMessage{
		Type:              wire.ObjectStreamMessageType,
		SubscribeID:       subscribeID,
		TrackAlias:        trackAlias,
		GroupID:           groupID,
		ObjectID:          objectID,
		PublisherPriority: publisherPriority,
		ObjectStatus:      0,
		ObjectPayload:     nil,
	}
	buf := make([]byte, 0, 48)
	buf = osm.Append(buf)
	_, err := stream.Write(buf)
	if err != nil {
		return nil, err
	}
	return &objectStream{
		stream: stream,
	}, nil
}

func (s *objectStream) Write(payload []byte) (int, error) {
	return s.stream.Write(payload)
}

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