|
|
// Copyright 2021, Chef. All rights reserved.
|
|
|
// https://github.com/q191201771/lal
|
|
|
//
|
|
|
// Use of this source code is governed by a MIT-style license
|
|
|
// that can be found in the License file.
|
|
|
//
|
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
|
|
package hls
|
|
|
|
|
|
import (
|
|
|
"github.com/q191201771/lal/pkg/base"
|
|
|
"github.com/q191201771/lal/pkg/mpegts"
|
|
|
)
|
|
|
|
|
|
// Queue
|
|
|
//
|
|
|
// 缓存流起始的一些数据，判断流中是否存在音频、视频，以及编码格式
|
|
|
// 一旦判断结束，该队列变成直进直出，不再有实际缓存
|
|
|
//
|
|
|
type Queue struct {
|
|
|
maxMsgSize int
|
|
|
data []base.RtmpMsg
|
|
|
observer IQueueObserver
|
|
|
|
|
|
audioCodecId int
|
|
|
videoCodecId int
|
|
|
done bool
|
|
|
}
|
|
|
|
|
|
type IQueueObserver interface {
|
|
|
// OnPatPmt
|
|
|
//
|
|
|
// 该回调一定发生在数据回调之前
|
|
|
// 只会返回两种格式，h264和h265
|
|
|
// TODO(chef): [opt] 当没有视频时，不应该返回h264的格式
|
|
|
// TODO(chef) 这里可以考虑换成只通知drain，由上层完成FragmentHeader的组装逻辑
|
|
|
//
|
|
|
OnPatPmt(b []byte)
|
|
|
|
|
|
OnPop(msg base.RtmpMsg)
|
|
|
}
|
|
|
|
|
|
// NewQueue @param maxMsgSize 最大缓存多少个包
|
|
|
func NewQueue(maxMsgSize int, observer IQueueObserver) *Queue {
|
|
|
return &Queue{
|
|
|
maxMsgSize: maxMsgSize,
|
|
|
data: make([]base.RtmpMsg, maxMsgSize)[0:0],
|
|
|
observer: observer,
|
|
|
audioCodecId: -1,
|
|
|
videoCodecId: -1,
|
|
|
done: false,
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// Push @param msg 函数调用结束后，内部不持有该内存块
|
|
|
func (q *Queue) Push(msg base.RtmpMsg) {
|
|
|
if q.done {
|
|
|
q.observer.OnPop(msg)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
q.data = append(q.data, msg.Clone())
|
|
|
|
|
|
switch msg.Header.MsgTypeId {
|
|
|
case base.RtmpTypeIdAudio:
|
|
|
q.audioCodecId = int(msg.Payload[0] >> 4)
|
|
|
case base.RtmpTypeIdVideo:
|
|
|
q.videoCodecId = int(msg.Payload[0] & 0xF)
|
|
|
}
|
|
|
|
|
|
if q.videoCodecId != -1 && q.audioCodecId != -1 {
|
|
|
q.drain()
|
|
|
return
|
|
|
}
|
|
|
|
|
|
if len(q.data) >= q.maxMsgSize {
|
|
|
q.drain()
|
|
|
return
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (q *Queue) drain() {
|
|
|
switch q.videoCodecId {
|
|
|
case int(base.RtmpCodecIdAvc):
|
|
|
q.observer.OnPatPmt(mpegts.FixedFragmentHeader)
|
|
|
case int(base.RtmpCodecIdHevc):
|
|
|
q.observer.OnPatPmt(mpegts.FixedFragmentHeaderHevc)
|
|
|
default:
|
|
|
// TODO(chef) 正确处理只有音频或只有视频的情况 #56
|
|
|
q.observer.OnPatPmt(mpegts.FixedFragmentHeader)
|
|
|
}
|
|
|
for i := range q.data {
|
|
|
q.observer.OnPop(q.data[i])
|
|
|
}
|
|
|
q.data = nil
|
|
|
|
|
|
q.done = true
|
|
|
}
|