Browse Source

feat: allow slash chat at flv saving

pull/237/head
Douglas Goulart 1 year ago
parent
commit
1259edc89e
  1. 10
      container/flv/muxer.go

10
container/flv/muxer.go

@ -150,15 +150,11 @@ func (writer *FLVWriter) Info() (ret av.Info) {
type FlvDvr struct{} type FlvDvr struct{}
func (f *FlvDvr) GetWriter(info av.Info) av.WriteCloser { func (f *FlvDvr) GetWriter(info av.Info) av.WriteCloser {
paths := strings.SplitN(info.Key, "/", 2) paths := strings.Split(info.Key, "/")
if len(paths) != 2 {
log.Warning("invalid info")
return nil
}
flvDir := configure.Config.GetString("flv_dir") flvDir := configure.Config.GetString("flv_dir")
err := os.MkdirAll(path.Join(flvDir, paths[0]), 0755) err := os.MkdirAll(path.Join(append([]string{flvDir}, paths[:len(paths)-1]...)...), 0755)
if err != nil { if err != nil {
log.Error("mkdir error: ", err) log.Error("mkdir error: ", err)
return nil return nil
@ -172,7 +168,7 @@ func (f *FlvDvr) GetWriter(info av.Info) av.WriteCloser {
return nil return nil
} }
writer := NewFLVWriter(paths[0], paths[1], info.URL, w) writer := NewFLVWriter(strings.Join(paths[:len(paths)-1], "/"), paths[len(paths)-1], info.URL, w)
log.Debug("new flv dvr: ", writer.Info()) log.Debug("new flv dvr: ", writer.Info())
return writer return writer
} }

Loading…
Cancel
Save