kunmeng/vdk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

testing

Browse Source
This commit is contained in:
deepch
2024-04-29 13:11:11 +03:00
parent 413b746b2f
commit f6346a05aa
1 changed files with 43 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
format/nvr/muxer.go
View File

@@ -11,7 +11,6 @@ import (
"github.com/deepch/vdk/codec/h264parser" "github.com/deepch/vdk/codec/h264parser"
"github.com/deepch/vdk/format/mp4" "github.com/deepch/vdk/format/mp4"
"github.com/google/uuid" "github.com/google/uuid"
"github.com/moby/sys/mountinfo"
"github.com/shirou/gopsutil/v3/disk" "github.com/shirou/gopsutil/v3/disk"
"os" "os"
"path/filepath" "path/filepath"
@@ -48,6 +47,7 @@ type Muxer struct {
pstart, pend time.Duration pstart, pend time.Duration
started bool started bool
serverID, streamName, channelName, streamID, channelID, hostLong, hostShort string serverID, streamName, channelName, streamID, channelID, hostLong, hostShort string
handleFileChange func(string, string, int64, time.Time, time.Time, time.Duration)
} }
type Gof struct { type Gof struct {
@@ -75,26 +75,27 @@ func init() {
} }
func NewMuxer(serverID, streamName, channelName, streamID, channelID string, mpoint []string, patch, format string, limit int) (m *Muxer, err error) { func NewMuxer(serverID, streamName, channelName, streamID, channelID string, mpoint []string, patch, format string, limit int, c func(string, string, int64, time.Time, time.Time, time.Duration)) (m *Muxer, err error) {
hostLong, _ := os.Hostname() hostLong, _ := os.Hostname()
var hostShort string var hostShort string
if p, _, ok := strings.Cut(hostLong, "."); ok { if p, _, ok := strings.Cut(hostLong, "."); ok {
hostShort = p hostShort = p
} }
m = &Muxer{ m = &Muxer{
mpoint: mpoint, mpoint: mpoint,
patch: patch, patch: patch,
h: -1, h: -1,
gof: &Gof{}, gof: &Gof{},
format: format, format: format,
limit: limit, limit: limit,
serverID: serverID, serverID: serverID,
streamName: streamName, streamName: streamName,
channelName: channelName, channelName: channelName,
streamID: streamID, streamID: streamID,
channelID: channelID, channelID: channelID,
hostLong: hostLong, hostLong: hostLong,
hostShort: hostShort, hostShort: hostShort,
handleFileChange: c,
} }
return return
} }
@@ -236,14 +237,14 @@ func (m *Muxer) filePatch() (string, error) {
) )
for i, i2 := range m.mpoint { for i, i2 := range m.mpoint {
if m, err := mountinfo.Mounted(i2); err == nil && m { //if m, err := mountinfo.Mounted(i2); err == nil && m {
if d, err := disk.Usage(i2); err == nil { if d, err := disk.Usage(i2); err == nil {
if d.UsedPercent < mu { if d.UsedPercent < mu {
ui = i ui = i
mu = d.UsedPercent mu = d.UsedPercent
}
} }
} }
// }
} }
if ui == -1 { if ui == -1 {
@@ -325,6 +326,15 @@ func (m *Muxer) filePatch() (string, error) {
return ts, nil return ts, nil
} }
func (m *Muxer) Codecs() string {
var codecs []string
for _, stream := range m.gof.Streams {
codecs = append(codecs, stream.Type().String())
}
return strings.Join(codecs, ",")
}
func (m *Muxer) WriteTrailer() (err error) { func (m *Muxer) WriteTrailer() (err error) {
if m.muxer != nil { if m.muxer != nil {
m.muxer.WriteTrailer() m.muxer.WriteTrailer()
@@ -341,6 +351,18 @@ func (m *Muxer) WriteTrailer() (err error) {
if err = os.Rename(m.d.Name(), filepath.Join(filepath.Dir(m.d.Name()), filepath.Base(p))); err != nil { if err = os.Rename(m.d.Name(), filepath.Join(filepath.Dir(m.d.Name()), filepath.Base(p))); err != nil {
return err return err
} }
var size int64
if fi, err := m.d.Stat(); err == nil {
size = fi.Size()
}
m.handleFileChange(
m.Codecs(),
filepath.Join(filepath.Dir(m.d.Name()), filepath.Base(p)),
size,
m.start,
m.end,
m.dur,
)
} }
err = m.d.Close() err = m.d.Close()
} }