使用Gin+WebSocket在HTML中无插件播放RTSP

项目地址：gin-rtsp

在后台的开发中遇到了对接显示摄像头视频流的需求。目前获取海康及大华等主流的摄像头的视频流使用的基本都是RTSP协议。不过HTML页面并不能直接播放RTSP协议的视频流，查询了一番各种网页播放RTSP的资料，有如下的一些方案：

• 插件开发播放：使用ActiveX等浏览器插件的方式来播放，海康和大华的浏览器管理页面便是通过安装浏览器插件来播放视频的。视频播放稳定，延时短，但是对技术要求较高，对于chrome等现代浏览器也存在兼容性问题，并不想考虑。

• RTSP 转 HLS：使用FFMPEG将RTSP转为HLS，推流到流服务器，如安装了nginx-rtmp-module模块的nginx，用这个方案测试了下，HLS协议在PC端和移动端的浏览器的播放都很稳，但是用HLS协议的直播流延时很大，至少有15秒左右，对于低延时视频的需求只能PASS。

• RTSP 转 RTMP：与上一方案类似，使用FFMPEG将RTSP转为RTMP推到流服务器分发播放，相比HLS延时很低，本来已经准备使用这个方案了，但是前端使用的video.js库总是会偶现无法加载视频的问题，而且播放RTMP需要使用到Flash，在chrome等浏览器中已经默认禁止加载逐步淘汰，只能抛弃。

• WebSocket：最终在万能的Github上翻到了一个JSMpeg项目，采用FFMPEG转为MPEG1 Video通过WebSocket代理推送到前端直接解码播放的方案。测试了下，延迟低，无需插件，画面质量也可以根据需要调整，效果很不错。

JSMpeg项目示例的WebSocket代理使用的是JS，简单实现了单个视频源的播放功能。我们的后台使用的是golang的Gin框架，会有多个网页客户端播放多个视频流。好在看了下JS的代码，这个WebSocket代理的原理并不难，在Gin中集成WebSocket也很方便。这里记录下我的集成方案。

主要模块

• API 接口：接收FFMPEG的推流数据和客户端的HTTP请求，将客户端需要播放的RTSP地址转换为一个对应的WebSocket地址，客户端通过这个WebSocket地址便可以直接播放视频，为了及时释放不再观看的视频流，这里设计为客户端播放时需要在每隔60秒的时间里循环请求这个接口，超过指定时间没有收到请求的话后台便会关闭这个视频流。

• FFMPEG 视频转换：收到前端的请求后，启动一个Goroutine调用系统的FFMPEG命令转换指定的RTSP视频流并推送到后台对应的接口，自动结束已超时转换任务。

• WebSocket Manager：管理WebSocket客户端，将请求同一WebSocket地址的客户端添加到一个Group中，向各个Group广播对应的RTSP视频流，删除Group中已断开连接的客户端，释放空闲的Group。

这里大致介绍下这三个主要模块的实现要点。

API 接口

API接收客户端发送的包含了需要播放RTSP流地址的Json数据，格式如：

{
    "url":"rtsp://admin:admin@192.168.1.11:554/cam/realmonitor?channel=1&subtype=0"
}

在有多个客户端需要播放相同的RTSP流地址时，需要保证返回对应的WebSocket地址相同，这里使用了UUID v3来将RTSP地址散列化保证返回的地址相同。

service/rtsptrans.go

processCh := uuid.NewV3(uuid.NamespaceURL, splitList[1]).String()
playURL := fmt.Sprintf("/stream/live/%s", processCh)

FFMPEG转换的视频数据也会通过HTTP协议传回服务端，每帧byte数据会以'\n'结束，在go语言中可以通过bufio模块来读出这样的数据。

api/rtsp.go

bodyReader := bufio.NewReader(c.Request.Body)

for {
	data, err := bodyReader.ReadBytes('\n')
	if err != nil {
		break
	}
}

FFMPEG 视频转换

视频转换模块会在收到需要转换的RTSP流地址后，启动一个FFMPEG子进程来转换RTSP视频流，这里是使用exec.Command来完成：

service/rtsptrans.go

params := []string{
	"-rtsp_transport",
	"tcp",
	"-re",
	"-i",
	rtsp,
	"-q",
	"5",
	"-f",
	"mpegts",
	"-fflags",
	"nobuffer",
	"-c:v",
	"mpeg1video",
	"-an",
	"-s",
	"960x540",
	fmt.Sprintf("http://127.0.0.1:3000/stream/upload/%s", playCh),
}

cmd := exec.Command("ffmpeg", params...)
cmd.Stdout = nil
cmd.Stderr = nil
stdin, err := cmd.StdinPipe()

通过FFMPEG的 -q 和 -s 参数可以调试视频的质量和分辨率。为了简便，命令的stdout和stderr都赋值为了nil，实际项目中可以保存到日志中方便排查问题。为了及时释放不再播放的资源，客户端停止请求超过一定时间后，FFMPEG子进程会自动关闭，通过golang的select可以很方便的实现这个功能。

service/rtsptrans.go

for {
	select {
	case <-*ch:
		util.Log().Info("reflush channel %s rtsp %v", playCh, rtsp)

	case <-time.After(60 * time.Second):
		stdin.Write([]byte("q"))
		err = cmd.Wait()
		if err != nil {
			util.Log().Error("Run ffmpeg err:%v", err.Error)
		}
		return
	}
}

这里的*ch channel通过一个map和每个子进程关联，子进程关闭时需要从map中清除，需要考虑并发的问题，可以使用sync.Map来保证线程安全。

WebSocket Manager

WebSocket Manager 负责对页面上请求视频数据的 ws 客户端进行管理，在Gin中，主要是使用github.com/gorilla/websocket这个库来开发相关功能。JSMpeg库连接WebSocket时使用到了Sec-WebSocket-Protocol这个header，需要对其处理：

upgrader := websocket.Upgrader{
	// cross origin domain
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
	// 处理 Sec-WebSocket-Protocol Header
	Subprotocols: []string{ctx.GetHeader("Sec-WebSocket-Protocol")},
}
conn, err := upgrader.Upgrade(ctx.Writer, ctx.Request, nil)

ws 客户端连接后，会分配一个唯一的UUID，放入到URL对应的Group中，相同Group下的客户端会收到同一视频流的数据。客户端断开连接后，需要从Group中删除，同时释放掉已经为空的Group。这个过程同样需要考虑到并发的问题，WebSocket Manager通过单独启动一个Goroutine监听注册，断开连接，广播的三个对应的golang的channel，来统一管理各个Group，可以很好的解决这个问题。具体实现在 service/wsservice.go#L75，代码比较长就不贴了。

测试

项目需要运行在安装有FFMPEG程序的环境中。通过编写了一份Dockerfile已经封装好了需要的环境，可以使用Docker build后，以Docker的方式运行。

$ docker build -t ginrtsp .
$ docker run -td -p 3000:3000 ginrtsp

使用内置的FFMPEG转换

将需要播放的RTSP流地址提交到 /stream/play 接口，例如：

POST /stream/play
{
   "url": "rtsp://admin:password@192.168.3.10:554/cam/realmonitor?channel=1&subtype=0"
}

后台可以正常转换此RTSP地址时便会返回一个对应的地址，例如：

{
    "code": 0,
    "data": {
        "path": "/stream/live/5b96bff4-bdb2-3edb-9d6e-f96eda03da56"
    },
    "msg": "success"
}

编辑html文件夹下view-stream.html文件，将script部分的url修改为此地址，在浏览器中打开，便可以看到视频了。

手动运行FFMPEG

由于后台转换RTSP的进程在超过60秒没有请求后便会停止，也可以通过手动运行ffmpeg命令，来更方便地在测试状态下查看视频。

ffmpeg -rtsp_transport tcp -re -i 'rtsp://admin:password@192.168.3.10:554/cam/realmonitor?channel=1&subtype=0' -q 0 -f mpegts -c:v mpeg1video -an -s 960x540 http://127.0.0.1:3000/stream/upload/test

通过如上命令，运行之后在view-stream.html文件的url中填入对应的地址为/stream/upload/test，在浏览器中打开查看视频。

显示效果

总结

得益于JSMpeg项目的强大，实现一个WebSocket的在网页上播放RTSP视频流还是很简单的了。随着golang语言日渐成熟，基于现成的库也可以方便的在Gin中添加WebSocket功能。需要注意主要是并发时，对FFMPEG子进程，WebSocket客户端的增删问题，好在golang天生对并发有良好的支持，gouroutine，channel，sync库这些golang核心知识掌握好了便可很好的应对这些问题。

首发自个人博客 某中二的黑科技研究中心 ，欢迎访问交流。