zishuo/webrtc
1
0
Fork 0
mirror of https://github.com/pion/webrtc.git synced 2025-10-16 20:20:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix all golint errors

Browse Source
This commit is contained in:
Sean DuBois
2018-06-12 22:24:52 -07:00
parent c21bc319d2
commit ddbb6c8ba8
14 changed files with 86 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/save-to-disk/ivf-writer.go
View File

@@ -9,7 +9,7 @@ import (
"github.com/pions/webrtc/pkg/rtp/codecs"
)
type IVFWriter struct {
type ivfWriter struct {
fd *os.File
count uint64
currentFrame []byte
@@ -21,7 +21,7 @@ func panicWrite(fd *os.File, data []byte) {
}
}
func NewIVFWriter(fileName string) (*IVFWriter, error) {
func newIVFWriter(fileName string) (*ivfWriter, error) {
f, err := os.Create(fileName)
if err != nil {
return nil, err
@@ -41,11 +41,11 @@ func NewIVFWriter(fileName string) (*IVFWriter, error) {
panicWrite(f, header)
i := &IVFWriter{fd: f}
i := &ivfWriter{fd: f}
return i, nil
}
func (i *IVFWriter) AddPacket(packet *rtp.Packet) {
func (i *ivfWriter) addPacket(packet *rtp.Packet) {
vp8Packet := codecs.VP8Packet{}
err := vp8Packet.Unmarshal(packet)
@@ -66,14 +66,10 @@ func (i *IVFWriter) AddPacket(packet *rtp.Packet) {
binary.LittleEndian.PutUint32(frameHeader[0:], uint32(len(i.currentFrame))) // Frame length
binary.LittleEndian.PutUint64(frameHeader[4:], i.count) // PTS
i.count += 1
i.count++
panicWrite(i.fd, frameHeader)
panicWrite(i.fd, i.currentFrame)
i.currentFrame = nil
}
func (i *IVFWriter) Close() error {
return i.fd.Close()
}