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

HDL插件导出

Browse Source
This commit is contained in:
dexter
2022-09-04 11:53:45 +08:00
parent b109dd48a6
commit e0fb371fd9
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
main.go
View File

@@ -28,8 +28,8 @@ func (c *HDLConfig) OnEvent(event any) {
case FirstConfig:
if c.PullOnStart {
for streamPath, url := range c.PullList {
if err := plugin.Pull(streamPath, url, new(HDLPuller), false); err != nil {
plugin.Error("pull", zap.String("streamPath", streamPath), zap.String("url", url), zap.Error(err))
if err := HDLPlugin.Pull(streamPath, url, new(HDLPuller), false); err != nil {
HDLPlugin.Error("pull", zap.String("streamPath", streamPath), zap.String("url", url), zap.Error(err))
}
}
}
@@ -37,8 +37,8 @@ func (c *HDLConfig) OnEvent(event any) {
if c.PullOnSubscribe {
for streamPath, url := range c.PullList {
if streamPath == v.Path {
if err := plugin.Pull(streamPath, url, new(HDLPuller), false); err != nil {
plugin.Error("pull", zap.String("streamPath", streamPath), zap.String("url", url), zap.Error(err))
if err := HDLPlugin.Pull(streamPath, url, new(HDLPuller), false); err != nil {
HDLPlugin.Error("pull", zap.String("streamPath", streamPath), zap.String("url", url), zap.Error(err))
}
break
}
@@ -48,9 +48,11 @@ func (c *HDLConfig) OnEvent(event any) {
}
func (c *HDLConfig) API_Pull(rw http.ResponseWriter, r *http.Request) {
err := plugin.Pull(r.URL.Query().Get("streamPath"), r.URL.Query().Get("target"), new(HDLPuller), r.URL.Query().Has("save"))
err := HDLPlugin.Pull(r.URL.Query().Get("streamPath"), r.URL.Query().Get("target"), new(HDLPuller), r.URL.Query().Has("save"))
if err != nil {
http.Error(rw, err.Error(), http.StatusBadRequest)
} else {
rw.Write([]byte("ok"))
}
}
@@ -60,7 +62,7 @@ func (*HDLConfig) API_List(rw http.ResponseWriter, r *http.Request) {
// 确保HDLConfig实现了PullPlugin接口
var plugin = InstallPlugin(new(HDLConfig))
var HDLPlugin = InstallPlugin(new(HDLConfig))
type HDLSubscriber struct {
Subscriber
@@ -124,7 +126,7 @@ func (*HDLConfig) ServeHTTP(w http.ResponseWriter, r *http.Request) {
sub.ID = r.RemoteAddr
sub.SetParentCtx(r.Context())
sub.SetIO(w)
if err := plugin.SubscribeBlock(streamPath, sub, SUBTYPE_FLV); err != nil {
if err := HDLPlugin.SubscribeBlock(streamPath, sub, SUBTYPE_FLV); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
}
}

4
pull.go
View File

@@ -14,7 +14,7 @@ import (
)
func (puller *HDLPuller) Connect() (err error) {
plugin.Info("connect", zap.String("remoteURL", puller.RemoteURL))
HDLPlugin.Info("connect", zap.String("remoteURL", puller.RemoteURL))
if strings.HasPrefix(puller.RemoteURL, "http") {
var res *http.Response
if res, err = http.Get(puller.RemoteURL); err == nil {
@@ -72,4 +72,4 @@ type HDLPuller struct {
Publisher
Puller
absTS uint32 //绝对时间戳
}
}