thefoxon/joy4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

flv: flush bufio.Writer when WriteTrailer

Browse Source
This commit is contained in:
nareix 2016-07-10 23:49:55 +08:00
parent 53d1f9ef98
commit 55302701af
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
format/flv/flv.go
View File

@ -15,12 +15,14 @@ import (
type Muxer struct { type Muxer struct {
pw *pio.Writer pw *pio.Writer
bw *bufio.Writer
streams []av.CodecData streams []av.CodecData
} }
func NewMuxer(w io.Writer) *Muxer { func NewMuxer(w io.Writer) *Muxer {
self := &Muxer{} self := &Muxer{}
self.pw = pio.NewWriter(bufio.NewWriterSize(w, pio.RecommendBufioSize)) self.bw = bufio.NewWriterSize(w, pio.RecommendBufioSize)
self.pw = pio.NewWriter(self.bw)
return self return self
} }
@ -103,6 +105,9 @@ func (self *Muxer) WritePacket(pkt av.Packet) (err error) {
} }
func (self *Muxer) WriteTrailer() (err error) { func (self *Muxer) WriteTrailer() (err error) {
if err = self.bw.Flush(); err != nil {
return
}
return return
} }