apps/lkm
1
0
Fork 0
mirror of https://github.com/lkmio/lkm.git synced 2025-10-04 23:02:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化网络IO使用buffer读取

Browse Source
This commit is contained in:
yangjiechina
2024-06-15 19:31:17 +08:00
parent 17973b3e9e
commit 89de34bd98
22 changed files with 355 additions and 288 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
rtmp/rtmp_stream.go
View File

@@ -7,7 +7,7 @@ import (
"github.com/yangjiechina/lkm/stream"
)
type TransStream struct {
type transStream struct {
stream.BaseTransStream
chunkSize int
@@ -21,7 +21,7 @@ type TransStream struct {
mwBuffer stream.MergeWritingBuffer
}
func (t *TransStream) Input(packet utils.AVPacket) error {
func (t *transStream) Input(packet utils.AVPacket) error {
utils.Assert(t.BaseTransStream.Completed)
var data []byte
@@ -96,7 +96,7 @@ func (t *TransStream) Input(packet utils.AVPacket) error {
return nil
}
func (t *TransStream) AddSink(sink stream.Sink) error {
func (t *transStream) AddSink(sink stream.Sink) error {
utils.Assert(t.headerSize > 0)
t.BaseTransStream.AddSink(sink)
@@ -113,12 +113,10 @@ func (t *TransStream) AddSink(sink stream.Sink) error {
return nil
}
func (t *TransStream) WriteHeader() error {
func (t *transStream) WriteHeader() error {
utils.Assert(t.Tracks != nil)
utils.Assert(!t.BaseTransStream.Completed)
t.Init()
var audioStream utils.AVStream
var videoStream utils.AVStream
var audioCodecId utils.AVCodecID
@@ -181,7 +179,7 @@ func (t *TransStream) WriteHeader() error {
return nil
}
func (t *TransStream) Close() error {
func (t *transStream) Close() error {
//发送剩余的流
segment := t.mwBuffer.PopSegment()
if len(segment) > 0 {
@@ -191,7 +189,7 @@ func (t *TransStream) Close() error {
}
func NewTransStream(chunkSize int) stream.TransStream {
return &TransStream{chunkSize: chunkSize}
return &transStream{chunkSize: chunkSize}
}
func TransStreamFactory(source stream.Source, protocol stream.Protocol, streams []utils.AVStream) (stream.TransStream, error) {