apps/donut
1
0
Fork 0
mirror of https://github.com/flavioribeiro/donut.git synced 2025-10-17 12:30:47 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

extract sub methods on stream

Browse Source
This commit is contained in:
Leandro Moreira
2024-01-28 00:01:00 -03:00
parent 3cd3bb836a
commit 852ad6c1ad
2 changed files with 33 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
internal/controller/streaming/controller.go
View File

@@ -28,7 +28,7 @@ func NewStreamingController(c *entity.Config, l *zap.Logger) *StreamingControlle
}
}
func (c *StreamingController) Stream(_ context.Context, srtConnection *astisrt.Connection, videoTrack *webrtc.TrackLocalStaticSample, metadataTrack *webrtc.DataChannel) {
func (c *StreamingController) Stream(srtConnection *astisrt.Connection, videoTrack *webrtc.TrackLocalStaticSample, metadataTrack *webrtc.DataChannel) {
r, w := io.Pipe()
defer r.Close()
defer w.Close()
@@ -54,27 +54,8 @@ func (c *StreamingController) Stream(_ context.Context, srtConnection *astisrt.C
}
if d.PMT != nil {
for _, es := range d.PMT.ElementaryStreams {
msg, _ := json.Marshal(entity.Message{
Type: entity.MessageTypeMetadata,
Message: es.StreamType.String(),
})
metadataTrack.SendText(string(msg))
if es.StreamType == astits.StreamTypeH264Video {
h264PID = es.ElementaryPID
}
}
for _, d := range d.PMT.ProgramDescriptors {
if d.MaximumBitrate != nil {
bitrateInMbitsPerSecond := float32(d.MaximumBitrate.Bitrate) / float32(125000)
msg, _ := json.Marshal(entity.Message{
Type: entity.MessageTypeMetadata,
Message: fmt.Sprintf("Bitrate %.2fMbps", bitrateInMbitsPerSecond),
})
metadataTrack.SendText(string(msg))
}
}
h264PID = c.captureMediaInfoAndSendToWebRTC(d, metadataTrack, h264PID)
c.captureBitrateAndSendToWebRTC(d, metadataTrack)
}
if d.PID == h264PID && d.PES != nil {
@@ -105,6 +86,35 @@ func (c *StreamingController) Stream(_ context.Context, srtConnection *astisrt.C
}
}
func (*StreamingController) captureBitrateAndSendToWebRTC(d *astits.DemuxerData, metadataTrack *webrtc.DataChannel) {
for _, d := range d.PMT.ProgramDescriptors {
if d.MaximumBitrate != nil {
bitrateInMbitsPerSecond := float32(d.MaximumBitrate.Bitrate) / float32(125000)
msg, _ := json.Marshal(entity.Message{
Type: entity.MessageTypeMetadata,
Message: fmt.Sprintf("Bitrate %.2fMbps", bitrateInMbitsPerSecond),
})
metadataTrack.SendText(string(msg))
}
}
}
func (*StreamingController) captureMediaInfoAndSendToWebRTC(d *astits.DemuxerData, metadataTrack *webrtc.DataChannel, h264PID uint16) uint16 {
for _, es := range d.PMT.ElementaryStreams {
msg, _ := json.Marshal(entity.Message{
Type: entity.MessageTypeMetadata,
Message: es.StreamType.String(),
})
metadataTrack.SendText(string(msg))
if es.StreamType == astits.StreamTypeH264Video {
h264PID = es.ElementaryPID
}
}
return h264PID
}
func (c *StreamingController) readFromSRTIntoWriterPipe(srtConnection *astisrt.Connection, w *io.PipeWriter) {
defer srtConnection.Close()

2
internal/web/signaling.go
View File

@@ -97,7 +97,7 @@ func (h *SignalingHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
return
}
go h.streamingController.Stream(r.Context(), srtConnection, videoTrack, metadataSender)
go h.streamingController.Stream(srtConnection, videoTrack, metadataSender)
if _, err := w.Write(localOfferDescription); err != nil {
ErrorToHTTP(w, err)