【山大会议】创建 WebRTC Peer

最新推荐文章于 2024-05-11 00:15:00 发布

孤名为義

最新推荐文章于 2024-05-11 00:15:00 发布

阅读量227

点赞数

分类专栏：山大会议文章标签： webrtc 音视频服务器 golang

本文链接：https://blog.csdn.net/weixin_45922876/article/details/125132592

版权

山大会议专栏收录该内容

7 篇文章 0 订阅

订阅专栏

本文详细介绍了WebRTC的通信架构，包括Mesh、MCU和SFU三种模式。重点讲解了SFU的工作原理，其中Peer作为连接的核心，负责媒体流的接收和发送。在SFU模式下，服务端创建接收器和发送器响应发布者和订阅者。此外，还展示了如何通过Peer发送关键帧来处理丢包问题。

摘要由CSDN通过智能技术生成

WebRTC 的通信架构

Mesh：多个终端之间两两进行连接。
MCU（Multipoint Conferencing Unit）：各终端将自己要共享的音视频流发送给服务器，服务器端会将在同一个房间中的所有终端的音视频流进行混合，最终生成一个混合后的音视频流再发给各个终端。
SFU（Selective Forwarding Unit）：该方案也是由一个服务器和多个终端组成，不对音视频进行混流，服务器收到某个终端共享的音视频流后，就直接将该音视频流转发给房间内的其他终端。

Peer 的作用

本系统采用 WebRTC 架构为 SFU，当用户要发布媒体流的时候，服务端会创建一个接收器，当用户要订阅媒体流的时候，服务端会创建一个发送器。一个 Peer 记录了一个连接，用来响应发布者或订阅者，并为之创建接收器或发送器。

定义 Peer 结构

type WebRTCPeer struct {
	id         int
	PC         *webrtc.PeerConnection
	VideoTrack *webrtc.Track
	AudioTrack *webrtc.Track
	stop       chan int
	pli        chan int
}

PC表示 WebRTC 连接，它会与实现相应协议的另一个端点建立 p2p 通信。VideoTrack和AudioTrack表示视频和音频轨道。stop是一个管道，当需要停止传输流时，只需要向stop中发送消息即可。pli用于关键帧丢包重传。

响应客户端

var engine *WebRTCEngine

func init() {
	engine = NewWebRTCEngine()
}

/ AnswerPublisher 响应发布者
func (p *WebRTCPeer) AnswerPublisher(offer webrtc.SessionDescription) (
	answer webrtc.SessionDescription, err error) {
	
	return engine.CreateMediaStreamReceiver(offer, &p.PC, &p.VideoTrack, &p.AudioTrack, p.stop, p.pli)
}

// AnswerSubscriber 响应订阅者
func (p *WebRTCPeer) AnswerSubscriber(offer webrtc.SessionDescription,
	addVideoTrack, addAudioTrack **webrtc.Track) (answer webrtc.SessionDescription, err error) {
	
	return engine.CreateMediaStreamSender(offer, &p.PC, addVideoTrack, addAudioTrack)
}

发送关键帧

func (p *WebRTCPeer) SendPLI() {
	go func() {
		defer func() {
			if r := recover(); r != nil {
				global.Logger.Error(r)
			}
		}()

		ticker := time.NewTicker(time.Second)
		i := 0
		for {
			select {
			case <-ticker.C:
				p.pli <- 1
				if i > 3 {
					return
				}
				i++
			case <-p.stop:
				return
			}
		}
	}()
}