mirror of
https://github.com/langhuihui/monibuca.git
synced 2026-05-08 08:31:10 +08:00
langhuihui
74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package m7s
|
|
|
|
import (
|
|
"m7s.live/m7s/v5/pkg"
|
|
"m7s.live/m7s/v5/pkg/config"
|
|
"m7s.live/m7s/v5/pkg/util"
|
|
"time"
|
|
)
|
|
|
|
type Connection struct {
|
|
util.MarcoTask
|
|
Plugin *Plugin
|
|
StreamPath string // 对应本地流
|
|
RemoteURL string // 远程服务器地址（用于推拉）
|
|
ConnectProxy string // 连接代理
|
|
}
|
|
|
|
type IPuller interface {
|
|
util.ITask
|
|
GetPullContext() *PullContext
|
|
}
|
|
|
|
type Puller = func() IPuller
|
|
|
|
type PullContext struct {
|
|
Connection
|
|
Publisher *Publisher
|
|
publishConfig *config.Publish
|
|
config.Pull
|
|
puller IPuller
|
|
}
|
|
|
|
func (p *PullContext) GetPullContext() *PullContext {
|
|
return p
|
|
}
|
|
|
|
func (p *PullContext) Init(puller IPuller, plugin *Plugin, streamPath string, url string) *PullContext {
|
|
publishConfig := plugin.config.Publish
|
|
publishConfig.PublishTimeout = 0
|
|
p.Pull = plugin.config.Pull
|
|
p.publishConfig = &publishConfig
|
|
p.Plugin = plugin
|
|
p.ConnectProxy = plugin.config.Pull.Proxy
|
|
p.RemoteURL = url
|
|
p.StreamPath = streamPath
|
|
p.Logger = p.Logger.With("pullURL", url, "streamPath", streamPath)
|
|
p.puller = puller
|
|
puller.SetRetry(plugin.config.Pull.RePull, time.Second*5)
|
|
return p
|
|
}
|
|
|
|
func (p *PullContext) GetKey() string {
|
|
return p.StreamPath
|
|
}
|
|
|
|
func (p *PullContext) Publish() (err error) {
|
|
p.Publisher, err = p.Plugin.PublishWithConfig(p.puller.GetTask().Context, p.StreamPath, *p.publishConfig)
|
|
return
|
|
}
|
|
|
|
func (p *PullContext) Start() (err error) {
|
|
s := p.Plugin.Server
|
|
if _, ok := s.Pulls.Get(p.GetKey()); ok {
|
|
return pkg.ErrStreamExist
|
|
}
|
|
s.Pulls.Add(p)
|
|
s.AddTask(p.puller)
|
|
return
|
|
}
|
|
|
|
func (p *PullContext) Dispose() {
|
|
p.Plugin.Server.Pulls.Remove(p)
|
|
}
|