-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
format_mpegts_segment.go
65 lines (50 loc) · 1.08 KB
/
format_mpegts_segment.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package record
import (
"os"
"path/filepath"
"time"
"github.com/bluenviron/mediamtx/internal/logger"
)
type formatMPEGTSSegment struct {
f *formatMPEGTS
startDTS time.Duration
startNTP time.Time
lastFlush time.Duration
path string
fi *os.File
}
func (s *formatMPEGTSSegment) initialize() {
s.lastFlush = s.startDTS
s.f.dw.setTarget(s)
}
func (s *formatMPEGTSSegment) close() error {
err := s.f.bw.Flush()
if s.fi != nil {
s.f.a.agent.Log(logger.Debug, "closing segment %s", s.path)
err2 := s.fi.Close()
if err == nil {
err = err2
}
if err2 == nil {
s.f.a.agent.OnSegmentComplete(s.path)
}
}
return err
}
func (s *formatMPEGTSSegment) Write(p []byte) (int, error) {
if s.fi == nil {
s.path = Path{Start: s.startNTP}.Encode(s.f.a.pathFormat)
s.f.a.agent.Log(logger.Debug, "creating segment %s", s.path)
err := os.MkdirAll(filepath.Dir(s.path), 0o755)
if err != nil {
return 0, err
}
fi, err := os.Create(s.path)
if err != nil {
return 0, err
}
s.f.a.agent.OnSegmentCreate(s.path)
s.fi = fi
}
return s.fi.Write(p)
}