apps/plugin-hdl
1
0
Fork 0
mirror of https://github.com/Monibuca/plugin-hdl.git synced 2025-10-05 16:56:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修改设置父级Context和IO的方式

Browse Source
This commit is contained in:
dexter
2022-02-27 18:59:37 +08:00
parent 16575235a1
commit 99f0214e91
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.go
View File

@@ -86,8 +86,8 @@ func (*HDLConfig) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "video/x-flv") w.Header().Set("Content-Type", "video/x-flv")
sub := &HDLSubscriber{} sub := &HDLSubscriber{}
sub.ID = r.RemoteAddr sub.ID = r.RemoteAddr
sub.OnEvent(r.Context()) //注入父级Context sub.SetParentCtx(r.Context())
sub.OnEvent(w) //注入Writer sub.SetIO(w)
if err := plugin.Subscribe(streamPath, sub); err == nil { if err := plugin.Subscribe(streamPath, sub); err == nil {
at, vt := sub.AudioTrack, sub.VideoTrack at, vt := sub.AudioTrack, sub.VideoTrack
hasVideo := at != nil hasVideo := at != nil

4
pull.go
View File

@@ -20,12 +20,12 @@ func (puller *HDLPuller) connect() (err error) {
if strings.HasPrefix(puller.RemoteURL, "http") { if strings.HasPrefix(puller.RemoteURL, "http") {
var res *http.Response var res *http.Response
if res, err = http.Get(puller.RemoteURL); err == nil { if res, err = http.Get(puller.RemoteURL); err == nil {
puller.OnEvent(res.Body) puller.SetIO(res.Body)
} }
} else { } else {
var res *os.File var res *os.File
if res, err = os.Open(puller.RemoteURL); err == nil { if res, err = os.Open(puller.RemoteURL); err == nil {
puller.OnEvent(res) puller.SetIO(res)
} }
} }
if err != nil { if err != nil {