[patch] rtmp: PullSessionOption.TlsConfig and PushSessionOption.TlsConfig #208

pkg/rtmp/client_pull_session.go

@@ -9,6 +9,7 @@
 package rtmp
 
 import (
+	"crypto/tls"
 	"github.com/q191201771/lal/pkg/base"
 )
 
@@ -27,10 +28,17 @@ type PullSessionOption struct {
 	PullTimeoutMs int
 
 	ReadAvTimeoutMs            int
-	ReadBufSize                int // io层读取音视频数据时的缓冲大小，如果为0，则没有缓冲
-	HandshakeComplexFlag       bool
-	PeerWinAckSize             int
+	ReadBufSize                int  // io层读取音视频数据时的缓冲大小，如果为0，则没有缓冲
 	ReuseReadMessageBufferFlag bool // 接收Message时，是否复用内存块
+	PeerWinAckSize             int
+
+	HandshakeComplexFlag bool
+	// TlsConfig
+	// rtmps时使用。
+	// 不关心可以不填。
+	// 业务方可以通过这个字段自定义 tls.Config
+	// 注意，如果使用rtmps并且该字段为nil，那么内部会使用 base.DefaultTlsConfigClient 生成 tls.Config
+	TlsConfig *tls.Config
 }
 
 var defaultPullSessionOption = PullSessionOption{

pkg/rtmp/client_push_session.go

@@ -8,7 +8,10 @@
 
 package rtmp
 
-import "github.com/q191201771/lal/pkg/base"
+import (
+	"crypto/tls"
+	"github.com/q191201771/lal/pkg/base"
+)
 
 type PushSession struct {
 	IsFresh bool
@@ -21,10 +24,17 @@ type PushSessionOption struct {
 	// 如果为0，则没有超时时间
 	PushTimeoutMs int
 
-	WriteAvTimeoutMs     int
-	WriteBufSize         int // io层发送音视频数据的缓冲大小，如果为0，则没有缓冲
-	WriteChanSize        int // io层发送音视频数据的异步队列大小，如果为0，则同步发送
+	WriteAvTimeoutMs int
+	WriteBufSize     int // io层发送音视频数据的缓冲大小，如果为0，则没有缓冲
+	WriteChanSize    int // io层发送音视频数据的异步队列大小，如果为0，则同步发送
+
 	HandshakeComplexFlag bool
+	// TlsConfig
+	// rtmps时使用。
+	// 不关心可以不填。
+	// 业务方可以通过这个字段自定义 tls.Config
+	// 注意，如果使用rtmps并且该字段为nil，那么内部会使用 base.DefaultTlsConfigClient 生成 tls.Config
+	TlsConfig *tls.Config
 }
 
 var defaultPushSessionOption = PushSessionOption{

pkg/rtmp/client_session.go

@@ -74,12 +74,10 @@ type ClientSessionOption struct {
 	WriteBufSize  int // io层发送音视频数据的缓冲大小，如果为0，则没有缓冲
 	WriteChanSize int // io层发送音视频数据的异步队列大小，如果为0，则同步发送
 
-	HandshakeComplexFlag bool // 握手是否使用复杂模式
-
-	PeerWinAckSize int
-
 	ReuseReadMessageBufferFlag bool // 接收Message时，是否重用内存块
+	PeerWinAckSize             int
 
+	HandshakeComplexFlag bool // 握手是否使用复杂模式
 	// TlsConfig
 	// rtmps时使用。
 	// 不关心可以不填。