apps/lkm
1
0
Fork 0
mirror of https://github.com/lkmio/lkm.git synced 2025-10-05 15:16:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

通知播放结束事件携带sink id

Browse Source
This commit is contained in:
yangjiechina
2024-11-05 19:46:15 +08:00
parent d66c724e0d
commit 4e809f9c26
3 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
stream/hook.go
View File

@@ -71,7 +71,7 @@ func Hook(event HookEvent, params string, body interface{}) (*http.Response, err
} }
func NewHookPlayEventInfo(sink Sink) eventInfo { func NewHookPlayEventInfo(sink Sink) eventInfo {
return eventInfo{Stream: sink.GetSourceID(), Protocol: sink.GetProtocol().String(), RemoteAddr: sink.String()} return eventInfo{Stream: sink.GetSourceID(), Protocol: sink.GetProtocol().String(), RemoteAddr: sink.RemoteAddr()}
} }
func NewHookPublishEventInfo(source Source) eventInfo { func NewHookPublishEventInfo(source Source) eventInfo {

14
stream/hook_sink.go
View File

@@ -27,14 +27,14 @@ func PreparePlaySinkWithReady(sink Sink, ok bool) (*http.Response, utils.HookSta
sink.SetReady(ok) sink.SetReady(ok)
source := SourceManager.Find(sink.GetSourceID()) source := SourceManager.Find(sink.GetSourceID())
if source == nil { if source == nil {
log.Sugar.Infof("添加sink到等待队列 sink:%s-%v source:%s", sink.GetProtocol().String(), sink.GetID(), sink.GetSourceID()) log.Sugar.Infof("添加%s sink到等待队列 id: %v source: %s", sink.GetProtocol().String(), sink.GetID(), sink.GetSourceID())
{ {
sink.Lock() sink.Lock()
defer sink.UnLock() defer sink.UnLock()
if SessionStateClosed == sink.GetState() { if SessionStateClosed == sink.GetState() {
log.Sugar.Warnf("添加到sink到等待队列失败, sink已经断开连接 %s", sink.GetID()) log.Sugar.Warnf("添加到%s sink到等待队列失败, sink已经断开连接 %s", sink.GetProtocol(), sink.GetID())
return response, utils.HookStateFailure return response, utils.HookStateFailure
} else { } else {
sink.SetState(SessionStateWaiting) sink.SetState(SessionStateWaiting)
@@ -52,7 +52,15 @@ func HookPlayDoneEvent(sink Sink) (*http.Response, bool) {
var response *http.Response var response *http.Response
if AppConfig.Hooks.IsEnableOnPlayDone() { if AppConfig.Hooks.IsEnableOnPlayDone() {
hook, err := Hook(HookEventPlayDone, sink.UrlValues().Encode(), NewHookPlayEventInfo(sink)) body := struct {
eventInfo
Sink string `json:"sink"`
}{
eventInfo: NewHookPlayEventInfo(sink),
Sink: SinkId2String(sink.GetID()),
}
hook, err := Hook(HookEventPlayDone, sink.UrlValues().Encode(), body)
if err != nil { if err != nil {
log.Sugar.Errorf("播放结束事件-通知失败 err: %s sink: %s-%v source: %s", err.Error(), sink.GetProtocol().String(), sink.GetID(), sink.GetSourceID()) log.Sugar.Errorf("播放结束事件-通知失败 err: %s sink: %s-%v source: %s", err.Error(), sink.GetProtocol().String(), sink.GetID(), sink.GetSourceID())
return hook, false return hook, false

5
stream/sink_utils.go
View File

@@ -2,6 +2,7 @@ package stream
import ( import (
"encoding/binary" "encoding/binary"
"fmt"
"net" "net"
"strconv" "strconv"
) )
@@ -48,3 +49,7 @@ func SinkId2String(id SinkID) string {
return id.(string) return id.(string)
} }
func CreateSinkDisconnectionMessage(sink Sink) string {
return fmt.Sprintf("%s sink断开连接. id: %s", sink.GetProtocol(), sink.GetID())
}