-
Notifications
You must be signed in to change notification settings - Fork 1
/
conn_options.go
51 lines (42 loc) · 1.25 KB
/
conn_options.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package rtmp
import (
"context"
)
type connOptions struct {
connInitializers []func(Conn)
onConnectValidators []func(
ctx context.Context,
connect Connect,
) ConnectError
onPublishValidators []func(
ctx context.Context,
publish Publish,
) (errorInfo map[string]interface{})
}
type ConnOption func(*connOptions)
func WithConnInitializers(connInitializers ...func(Conn)) ConnOption {
return func(o *connOptions) {
o.connInitializers = append(o.connInitializers, connInitializers...)
}
}
func WithOnConnectValidators(onConnectValidators ...func(ctx context.Context, connect Connect) ConnectError) ConnOption {
return func(o *connOptions) {
o.onConnectValidators = append(o.onConnectValidators, onConnectValidators...)
}
}
func WithOnPublishValidators(onPublishValidators ...func(ctx context.Context, publish Publish) (errorInfo map[string]interface{})) ConnOption {
return func(o *connOptions) {
o.onPublishValidators = append(o.onPublishValidators, onPublishValidators...)
}
}
func (o connOptions) Apply(c *defaultConn) {
if len(o.onConnectValidators) > 0 {
c.onConnectValidators = o.onConnectValidators
}
if len(o.onPublishValidators) > 0 {
c.onPublishValidators = o.onPublishValidators
}
for _, f := range o.connInitializers {
f(c)
}
}