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

📦 NEW: 新增SubscribeExist功能

Browse Source
This commit is contained in:
dexter
2022-11-30 13:18:21 +08:00
parent 7a23da59c6
commit 4438376b20
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
plugin.go
View File

@@ -213,6 +213,24 @@ func (opt *Plugin) Publish(streamPath string, pub IPublisher) error {
return pub.receive(streamPath, pub, conf.GetPublishConfig())
}
var ErrStreamNotExist = errors.New("stream not exist")
// SubscribeExist 订阅已经存在的流
func (opt *Plugin) SubscribeExist(streamPath string, sub ISubscriber) error {
opt.Info("subscribe exit", zap.String("path", streamPath))
path, _, _ := strings.Cut(streamPath, "?")
if !Streams.Has(path) {
opt.Warn("stream not exist", zap.String("path", streamPath))
return ErrStreamNotExist
}
conf, ok := opt.Config.(config.SubscribeConfig)
if !ok {
conf = EngineConfig
}
return sub.receive(streamPath, sub, conf.GetSubscribeConfig())
}
// Subscribe 订阅一个流，如果流不存在则创建一个等待流
func (opt *Plugin) Subscribe(streamPath string, sub ISubscriber) error {
opt.Info("subscribe", zap.String("path", streamPath))
conf, ok := opt.Config.(config.SubscribeConfig)
@@ -222,6 +240,7 @@ func (opt *Plugin) Subscribe(streamPath string, sub ISubscriber) error {
return sub.receive(streamPath, sub, conf.GetSubscribeConfig())
}
// SubscribeBlock 阻塞订阅一个流，直到订阅结束
func (opt *Plugin) SubscribeBlock(streamPath string, sub ISubscriber, t byte) (err error) {
if err = opt.Subscribe(streamPath, sub); err == nil {
sub.PlayBlock(t)