From 55302701af7b88f1819d9b6cd63cd3640be05327 Mon Sep 17 00:00:00 2001
From: nareix <hi.nareix@gmail.com>
Date: Sun, 10 Jul 2016 23:49:55 +0800
Subject: [PATCH] flv: flush bufio.Writer when WriteTrailer

---
 format/flv/flv.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/format/flv/flv.go b/format/flv/flv.go
index d2ca161..2e04bd3 100644
--- a/format/flv/flv.go
+++ b/format/flv/flv.go
@@ -15,12 +15,14 @@ import (
 
 type Muxer struct {
 	pw *pio.Writer
+	bw *bufio.Writer
 	streams []av.CodecData
 }
 
 func NewMuxer(w io.Writer) *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
 }
 
@@ -103,6 +105,9 @@ func (self *Muxer) WritePacket(pkt av.Packet) (err error) {
 }
 
 func (self *Muxer) WriteTrailer() (err error) {
+	if err = self.bw.Flush(); err != nil {
+		return
+	}
 	return
 }