mirror of
https://github.com/gowvp/gb28181.git
synced 2026-04-22 15:07:10 +08:00
xugo
340 lines
13 KiB
Go
340 lines
13 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"net/url"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gowvp/owl/internal/conf"
|
|
"github.com/gowvp/owl/internal/core/ipc"
|
|
"github.com/gowvp/owl/internal/core/recording"
|
|
"github.com/gowvp/owl/internal/core/sms"
|
|
"github.com/gowvp/owl/pkg/gbs"
|
|
"github.com/ixugo/goddd/pkg/orm"
|
|
"github.com/ixugo/goddd/pkg/web"
|
|
)
|
|
|
|
type WebHookAPI struct {
|
|
smsCore sms.Core
|
|
ipcCore ipc.Core
|
|
recordingCore recording.Core
|
|
conf *conf.Bootstrap
|
|
log *slog.Logger
|
|
gbs *gbs.Server
|
|
uc *Usecase
|
|
|
|
protocols map[string]ipc.Protocoler
|
|
}
|
|
|
|
func NewWebHookAPI(core sms.Core, conf *conf.Bootstrap, gbs *gbs.Server, ipcBundle IPCBundle, recordingCore recording.Core) WebHookAPI {
|
|
return WebHookAPI{
|
|
smsCore: core,
|
|
ipcCore: ipcBundle.Core,
|
|
recordingCore: recordingCore,
|
|
conf: conf,
|
|
log: slog.With("hook", "zlm"),
|
|
gbs: gbs,
|
|
protocols: ipcBundle.Protocols,
|
|
}
|
|
}
|
|
|
|
func registerZLMWebhookAPI(r gin.IRouter, api WebHookAPI, handler ...gin.HandlerFunc) {
|
|
{
|
|
group := r.Group("/webhook", handler...)
|
|
group.POST("/on_server_started", web.WrapH(api.onServerStarted))
|
|
group.POST("/on_server_keepalive", web.WrapH(api.onServerKeepalive))
|
|
group.POST("/on_stream_changed", web.WrapH(api.onStreamChanged))
|
|
group.POST("/on_publish", web.WrapH(api.onPublish))
|
|
group.POST("/on_play", web.WrapH(api.onPlay))
|
|
group.POST("/on_stream_none_reader", web.WrapH(api.onStreamNoneReader))
|
|
group.POST("/on_rtp_server_timeout", web.WrapH(api.onRTPServerTimeout))
|
|
group.POST("/on_stream_not_found", web.WrapH(api.onStreamNotFound))
|
|
group.POST("/on_record_mp4", web.WrapH(api.onRecordMP4))
|
|
}
|
|
}
|
|
|
|
// getChannelType 通过 app+stream 查询通道获取类型
|
|
// 支持自定义 app/stream 的 RTMP/RTSP 通道：先按 app+stream 查询，查不到再按 id=stream 查询
|
|
// 如果都找不到，则回退到使用 stream 前缀判断类型
|
|
func (w WebHookAPI) getChannelType(ctx context.Context, app, stream string) string {
|
|
ch, err := w.ipcCore.GetChannelByAppStreamOrID(ctx, app, stream)
|
|
if err == nil {
|
|
return ch.Type
|
|
}
|
|
// 回退：使用 stream 前缀判断类型（兼容旧逻辑）
|
|
return ipc.GetType(stream)
|
|
}
|
|
|
|
func (w WebHookAPI) onServerStarted(c *gin.Context, _ *struct{}) (DefaultOutput, error) {
|
|
w.log.InfoContext(c.Request.Context(), "webhook onServerStarted")
|
|
// 所有 rtmp 通道离线
|
|
if err := w.ipcCore.BatchOfflineRTMP(context.Background()); err != nil {
|
|
w.log.ErrorContext(c.Request.Context(), "webhook onServerStarted", "err", err)
|
|
}
|
|
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
|
|
// onServerKeepalive 服务器定时上报时间，上报间隔可配置，默认 10s 上报一次
|
|
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_16%E3%80%81on-server-keepalive
|
|
func (w WebHookAPI) onServerKeepalive(_ *gin.Context, in *onServerKeepaliveInput) (DefaultOutput, error) {
|
|
// TODO: 仅支持默认
|
|
w.smsCore.Keepalive(sms.DefaultMediaServerID)
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
|
|
// onPublish rtsp/rtmp/rtp 推流鉴权事件。
|
|
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_7%E3%80%81on-publish
|
|
func (w WebHookAPI) onPublish(c *gin.Context, in *onPublishInput) (*onPublishOutput, error) {
|
|
ctx := c.Request.Context()
|
|
w.log.Info("webhook onPublish", "app", in.App, "stream", in.Stream, "schema", in.Schema, "mediaServerID", in.MediaServerID)
|
|
|
|
// 通过 app+stream 查询通道获取类型，支持自定义 app/stream
|
|
channelType := w.getChannelType(ctx, in.App, in.Stream)
|
|
|
|
// 获取协议适配器，检查是否实现了 OnPublisher 接口
|
|
protocol, ok := w.protocols[channelType]
|
|
if !ok {
|
|
return &onPublishOutput{DefaultOutput: newDefaultOutputOK()}, nil
|
|
}
|
|
|
|
publisher, ok := protocol.(ipc.OnPublisher)
|
|
if !ok {
|
|
// 协议不需要推流鉴权，直接通过
|
|
return &onPublishOutput{DefaultOutput: newDefaultOutputOK()}, nil
|
|
}
|
|
|
|
// 解析参数
|
|
params, err := url.ParseQuery(in.Params)
|
|
if err != nil {
|
|
return &onPublishOutput{DefaultOutput: DefaultOutput{Code: 1, Msg: err.Error()}}, nil
|
|
}
|
|
|
|
// 将 url.Values 转换为 map[string]string
|
|
paramsMap := make(map[string]string)
|
|
for k, v := range params {
|
|
if len(v) > 0 {
|
|
paramsMap[k] = v[0]
|
|
}
|
|
}
|
|
paramsMap["media_server_id"] = in.MediaServerID
|
|
|
|
// 调用协议适配器的 OnPublish 方法
|
|
allowed, err := publisher.OnPublish(ctx, in.App, in.Stream, paramsMap)
|
|
if err != nil {
|
|
return &onPublishOutput{DefaultOutput: DefaultOutput{Code: 1, Msg: err.Error()}}, nil
|
|
}
|
|
if !allowed {
|
|
return &onPublishOutput{DefaultOutput: DefaultOutput{Code: 1, Msg: "鉴权失败"}}, nil
|
|
}
|
|
|
|
return &onPublishOutput{DefaultOutput: newDefaultOutputOK()}, nil
|
|
}
|
|
|
|
// onStreamChanged rtsp/rtmp 流注册或注销时触发此事件；此事件对回复不敏感。
|
|
// 流注册时自动启动录制，流注销时停止录制并更新通道状态
|
|
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_12%E3%80%81on-stream-changed
|
|
func (w WebHookAPI) onStreamChanged(c *gin.Context, in *onStreamChangedInput) (DefaultOutput, error) {
|
|
ctx := c.Request.Context()
|
|
w.log.InfoContext(ctx, "webhook onStreamChanged", "app", in.App, "stream", in.Stream, "schema", in.Schema, "mediaServerID", in.MediaServerID, "regist", in.Regist)
|
|
|
|
stream := in.Stream
|
|
app := in.App
|
|
// lalmax 兼容
|
|
if in.StreamName != "" {
|
|
stream = in.StreamName
|
|
app = in.AppName
|
|
}
|
|
|
|
// 通过 app+stream 查询通道获取类型，支持自定义 app/stream
|
|
channelType := w.getChannelType(ctx, app, stream)
|
|
|
|
if in.Regist {
|
|
// 流注册时根据录像模式决定是否启动录制
|
|
ch, err := w.ipcCore.GetChannelByAppStreamOrID(ctx, app, stream)
|
|
if err != nil {
|
|
w.log.WarnContext(ctx, "获取通道信息失败，尝试启动录制", "stream", stream, "err", err)
|
|
// 找不到通道时仍尝试按旧逻辑启动录制
|
|
if err := w.recordingCore.StartRecording(ctx, channelType, app, stream); err != nil {
|
|
w.log.WarnContext(ctx, "启动录制失败", "stream", stream, "err", err)
|
|
}
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
|
|
if !ch.Ext.IsNoneRecord() {
|
|
// always 模式：自动启动录制
|
|
if err := w.recordingCore.StartRecording(ctx, channelType, app, stream); err != nil {
|
|
w.log.WarnContext(ctx, "启动录制失败", "stream", stream, "err", err)
|
|
}
|
|
w.log.InfoContext(ctx, "自动启动录制（always模式）", "stream", stream)
|
|
}
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
|
|
// 流注销时停止录制
|
|
if err := w.recordingCore.StopRecording(ctx, app, stream); err != nil {
|
|
w.log.WarnContext(ctx, "停止录制失败", "stream", stream, "err", err)
|
|
}
|
|
|
|
// 流注销时通过 Protocoler 接口统一处理所有协议的状态更新
|
|
// 每个协议适配器在 OnStreamChanged 中处理自己的状态逻辑
|
|
protocol, ok := w.protocols[channelType]
|
|
if ok {
|
|
if err := protocol.OnStreamChanged(ctx, app, stream); err != nil {
|
|
slog.ErrorContext(ctx, "webhook onStreamChanged", "err", err)
|
|
}
|
|
}
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
|
|
// onPlay rtsp/rtmp/http-flv/ws-flv/hls 播放触发播放器身份验证事件。
|
|
// 播放流时会触发此事件。如果流不存在，则首先触发 on_play 事件，然后触发 on_stream_not_found 事件。
|
|
// 播放rtsp流时，如果该流开启了rtsp专用认证（on_rtsp_realm），则不会触发on_play事件。
|
|
// https://docs.zlmediakit.com/guide/media_server/web_hook_api.html#_6-on-play
|
|
func (w WebHookAPI) onPlay(c *gin.Context, in *onPublishInput) (DefaultOutput, error) {
|
|
ctx := c.Request.Context()
|
|
w.log.InfoContext(ctx, "webhook onPlay", "app", in.App, "stream", in.Stream, "schema", in.Schema)
|
|
|
|
// 更新通道的播放状态（所有协议统一处理）
|
|
if _, err := w.ipcCore.EditChannelPlaying(ctx, in.Stream, true); err != nil {
|
|
w.log.WarnContext(ctx, "更新播放状态失败", "stream", in.Stream, "err", err)
|
|
}
|
|
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
|
|
// onStreamNoneReader 流无人观看时事件，用户可以通过此事件选择是否关闭无人看的流。
|
|
// 一个直播流注册上线了，如果一直没人观看也会触发一次无人观看事件，触发时的协议 schema 是随机的，
|
|
// 看哪种协议最晚注册(一般为 hls)。
|
|
// 后续从有人观看转为无人观看，触发协议 schema 为最后一名观看者使用何种协议。
|
|
// 目前 mp4/hls 录制不当做观看人数(mp4 录制可以通过配置文件 mp4_as_player 控制，
|
|
// 但是 rtsp/rtmp/rtp 转推算观看人数，也会触发该事件。
|
|
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_12%E3%80%81on-stream-changed
|
|
func (w WebHookAPI) onStreamNoneReader(c *gin.Context, in *onStreamNoneReaderInput) (onStreamNoneReaderOutput, error) {
|
|
ctx := c.Request.Context()
|
|
w.log.InfoContext(ctx, "webhook onStreamNoneReader", "app", in.App, "stream", in.Stream, "mediaServerID", in.MediaServerID)
|
|
|
|
// 更新通道的播放状态为未播放（所有协议统一处理）
|
|
if _, err := w.ipcCore.EditChannelPlaying(ctx, in.Stream, false); err != nil {
|
|
w.log.WarnContext(ctx, "更新播放状态失败", "stream", in.Stream, "err", err)
|
|
}
|
|
|
|
// 根据录像模式判断是否关闭流：
|
|
// - none(不录制): 无人观看时关闭流
|
|
// - always/ai(有录像计划): 无人观看时保持流不关闭
|
|
ch, err := w.ipcCore.GetChannelByAppStreamOrID(ctx, in.App, in.Stream)
|
|
if err != nil {
|
|
// 找不到通道时默认关闭流
|
|
w.log.WarnContext(ctx, "获取通道失败，默认关闭流", "stream", in.Stream, "err", err)
|
|
return onStreamNoneReaderOutput{Close: true}, nil
|
|
}
|
|
|
|
// 如果录像模式为 none，则关闭流；否则保持流不关闭以继续录制
|
|
shouldClose := ch.Ext.IsNoneRecord()
|
|
w.log.InfoContext(ctx, "无人观看判断", "stream", in.Stream, "record_mode", ch.Ext.GetRecordMode(), "close", shouldClose)
|
|
|
|
return onStreamNoneReaderOutput{Close: shouldClose}, nil
|
|
}
|
|
|
|
// onRTPServerTimeout RTP 服务器超时事件
|
|
// 调用 openRtpServer 接口，rtp server 长时间未收到数据,执行此 web hook,对回复不敏感
|
|
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_17%E3%80%81on-rtp-server-timeout
|
|
func (w WebHookAPI) onRTPServerTimeout(c *gin.Context, in *onRTPServerTimeoutInput) (DefaultOutput, error) {
|
|
w.log.InfoContext(c.Request.Context(), "webhook onRTPServerTimeout", "local_port", in.LocalPort, "ssrc", in.SSRC, "stream_id", in.StreamID, "mediaServerID", in.MediaServerID)
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
|
|
// onStreamNotFound 流不存在事件
|
|
// TODO: 重启后立即播放，会出发 "channel not exist" 待处理
|
|
func (w WebHookAPI) onStreamNotFound(c *gin.Context, in *onStreamNotFoundInput) (DefaultOutput, error) {
|
|
ctx := c.Request.Context()
|
|
w.log.InfoContext(ctx, "webhook onStreamNotFound", "app", in.App, "stream", in.Stream, "schema", in.Schema, "mediaServerID", in.MediaServerID)
|
|
|
|
stream := in.StreamName
|
|
app := in.AppName
|
|
// 确保不是 lalmax 的流
|
|
if in.StreamName == "" {
|
|
stream = in.Stream
|
|
app = in.App
|
|
if !(in.Schema == "rtmp" || in.Schema == "rtsp") {
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
}
|
|
|
|
// 通过 app+stream 查询通道获取类型，支持自定义 app/stream
|
|
channelType := w.getChannelType(ctx, app, stream)
|
|
protocol, ok := w.protocols[channelType]
|
|
if ok {
|
|
if err := protocol.OnStreamNotFound(ctx, app, stream); err != nil {
|
|
slog.InfoContext(ctx, "webhook onStreamNotFound", "err", err)
|
|
}
|
|
}
|
|
|
|
return newDefaultOutputOK(), nil
|
|
}
|
|
|
|
// onRecordMP4 录制 mp4 完成后通知事件
|
|
// ZLM 在 MP4 切片完成时会触发此回调，将录像信息入库
|
|
// https://docs.zlmediakit.com/zh/guide/media_server/web_hook_api.html#_8%E3%80%81on-record-mp4
|
|
func (w WebHookAPI) onRecordMP4(c *gin.Context, in *onRecordMP4Input) (DefaultOutput, error) {
|
|
ctx := c.Request.Context()
|
|
w.log.InfoContext(ctx, "webhook onRecordMP4",
|
|
"app", in.App,
|
|
"stream", in.Stream,
|
|
"file_path", in.FilePath,
|
|
"file_size", in.FileSize,
|
|
"time_len", in.TimeLen,
|
|
"start_time", in.StartTime,
|
|
)
|
|
|
|
// 计算相对路径：从配置的存储目录开始
|
|
relativePath := in.FilePath
|
|
if w.conf.Server.Recording.StorageDir != "" {
|
|
// 尝试提取相对路径
|
|
storageDir := w.conf.Server.Recording.StorageDir
|
|
if idx := strings.Index(in.FilePath, storageDir); idx >= 0 {
|
|
relativePath = in.FilePath[idx:]
|
|
} else {
|
|
// 使用 URL 字段作为相对路径
|
|
relativePath = in.URL
|
|
}
|
|
}
|
|
|
|
// 计算开始和结束时间
|
|
startTime := time.Unix(in.StartTime, 0)
|
|
endTime := startTime.Add(time.Duration(in.TimeLen * float64(time.Second)))
|
|
|
|
// 通过 app+stream 查找 channel ID，支持自定义 app/stream
|
|
var cid string
|
|
ch, err := w.ipcCore.GetChannelByAppStreamOrID(ctx, in.App, in.Stream)
|
|
if err == nil {
|
|
cid = ch.ID
|
|
} else {
|
|
// 如果找不到通道，使用 stream 作为 CID 的标识
|
|
cid = in.Stream
|
|
w.log.WarnContext(ctx, "未找到对应通道，使用 stream 作为 CID", "app", in.App, "stream", in.Stream)
|
|
}
|
|
|
|
// 入库
|
|
_, err = w.recordingCore.AddRecording(ctx, &recording.AddRecordingInput{
|
|
CID: cid,
|
|
App: in.App,
|
|
Stream: in.Stream,
|
|
StartedAt: orm.Time{Time: startTime},
|
|
EndedAt: orm.Time{Time: endTime},
|
|
Duration: in.TimeLen,
|
|
Path: filepath.Clean(relativePath),
|
|
Size: in.FileSize,
|
|
})
|
|
if err != nil {
|
|
w.log.ErrorContext(ctx, "录像入库失败", "err", err)
|
|
// 仍返回成功，避免 ZLM 重试
|
|
}
|
|
|
|
return newDefaultOutputOK(), nil
|
|
}
|