You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
lal/pkg/logic/server_manager__notify.go

107 lines
3.2 KiB
Go

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

// Copyright 2023, Chef. All rights reserved.
// https://github.com/q191201771/lal
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)
package logic
import (
"github.com/q191201771/lal/pkg/base"
"github.com/q191201771/naza/pkg/taskpool"
)
// server_manager__notify.go
//
// NotifyHandler部分。
//
// 注意所有Notify回调都异步化在单独的协程中执行。
// 使得外部的NotifyHandler实现不会影响到内部逻辑的执行。比如避免回调在锁内触发回调中调用API又再次拿锁的情况。
//
func (sm *ServerManager) nhInitNotifyHandler() {
// TODO(chef): [opt] 这里已经做了异步化处理http notify那边的异步可以去掉 202304
// 如果外部没有传入则使用默认的http notify handler
if sm.option.NotifyHandler == nil {
sm.option.NotifyHandler = NewHttpNotify(sm.config.HttpNotifyConfig, sm.config.ServerId)
}
sm.notifyHandlerThread, _ = taskpool.NewPool(func(option *taskpool.Option) {
option.InitWorkerNum = 1
option.MaxWorkerNum = 1
})
}
func (sm *ServerManager) nhOnServerStart(info base.LalInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.LalInfo)
sm.option.NotifyHandler.OnServerStart(p)
}, info)
}
func (sm *ServerManager) nhOnUpdate(info base.UpdateInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.UpdateInfo)
sm.option.NotifyHandler.OnUpdate(p)
}, info)
}
func (sm *ServerManager) nhOnPubStart(info base.PubStartInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.PubStartInfo)
sm.option.NotifyHandler.OnPubStart(p)
}, info)
}
func (sm *ServerManager) nhOnPubStop(info base.PubStopInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.PubStopInfo)
sm.option.NotifyHandler.OnPubStop(p)
}, info)
}
func (sm *ServerManager) nhOnSubStart(info base.SubStartInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.SubStartInfo)
sm.option.NotifyHandler.OnSubStart(p)
}, info)
}
func (sm *ServerManager) nhOnSubStop(info base.SubStopInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.SubStopInfo)
sm.option.NotifyHandler.OnSubStop(p)
}, info)
}
func (sm *ServerManager) nhOnRelayPullStart(info base.PullStartInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.PullStartInfo)
sm.option.NotifyHandler.OnRelayPullStart(p)
}, info)
}
func (sm *ServerManager) nhOnRelayPullStop(info base.PullStopInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.PullStopInfo)
sm.option.NotifyHandler.OnRelayPullStop(p)
}, info)
}
func (sm *ServerManager) nhOnRtmpConnect(info base.RtmpConnectInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.RtmpConnectInfo)
sm.option.NotifyHandler.OnRtmpConnect(p)
}, info)
}
func (sm *ServerManager) nhOnHlsMakeTs(info base.HlsMakeTsInfo) {
sm.notifyHandlerThread.Go(func(param ...interface{}) {
p := param[0].(base.HlsMakeTsInfo)
sm.option.NotifyHandler.OnHlsMakeTs(p)
}, info)
}