-
Notifications
You must be signed in to change notification settings - Fork 9
/
event.go
267 lines (233 loc) · 8.4 KB
/
event.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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
package acfundanmu
import (
"log"
"sync"
)
// 弹幕信息类型
type eventType int
const (
stopDanmu eventType = iota
commentDanmu
likeDanmu
enterRoomDanmu
followAuthorDanmu
throwBananaDanmu
giftDanmu
richTextDanmu
joinClubDanmu
shareLiveDanmu
bananaCountEvent
displayEvent
topUsersEvent
recentCommentEvent
chatCallEvent
chatAcceptEvent
chatReadyEvent
chatEndEvent
authorChatCallEvent
authorChatAcceptEvent
authorChatReadyEvent
authorChatEndEvent
authorChatChangeSoundConfigEvent
redpackListEvent
kickedOutEvent
violationAlertEvent
managerStateEvent
)
// 事件handler
type eventHandler func(*AcFunLive, any)
// 事件handler的map
type handlerMap struct {
sync.RWMutex
listMap map[eventType][]eventHandler
}
// 将f加入到t对应的事件handler列表里
func (h *handlerMap) add(t eventType, f eventHandler) {
h.Lock()
defer h.Unlock()
h.listMap[t] = append(h.listMap[t], f)
}
// 调用事件handler列表里的handler
func (ac *AcFunLive) callEvent(t eventType, i any) {
ac.handlerMap.RLock()
defer ac.handlerMap.RUnlock()
list, ok := ac.handlerMap.listMap[t]
if ok {
for _, f := range list {
go func(f eventHandler) {
defer func() {
if err := recover(); err != nil {
log.Printf("dispatchEvent() %v goroutine error: %v", t, err)
}
}()
f(ac, i)
}(f)
}
}
}
// OnDanmuStop 处理获取弹幕结束，有可能是网络原因导致连接超时无法获取弹幕，直播不一定结束，可以多次调用
func (ac *AcFunLive) OnDanmuStop(handler func(*AcFunLive, error)) {
ac.handlerMap.add(stopDanmu, func(ac *AcFunLive, i any) {
if i == nil {
handler(ac, nil)
} else {
handler(ac, i.(error))
}
})
}
// OnComment 处理评论弹幕，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnComment(handler func(*AcFunLive, *Comment)) {
ac.handlerMap.add(commentDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*Comment))
})
}
// OnLike 处理点赞弹幕，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnLike(handler func(*AcFunLive, *Like)) {
ac.handlerMap.add(likeDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*Like))
})
}
// OnEnterRoom 处理用户进场，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnEnterRoom(handler func(*AcFunLive, *EnterRoom)) {
ac.handlerMap.add(enterRoomDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*EnterRoom))
})
}
// OnFollowAuthor 处理用户关注主播，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnFollowAuthor(handler func(*AcFunLive, *FollowAuthor)) {
ac.handlerMap.add(followAuthorDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*FollowAuthor))
})
}
// OnThrowBanana 处理用户投蕉，现在基本用 OnGift 代替，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnThrowBanana(handler func(*AcFunLive, *ThrowBanana)) {
ac.handlerMap.add(throwBananaDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*ThrowBanana))
})
}
// OnGift 处理用户赠送礼物，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnGift(handler func(*AcFunLive, *Gift)) {
ac.handlerMap.add(giftDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*Gift))
})
}
// OnRichText 处理富文本，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnRichText(handler func(*AcFunLive, *RichText)) {
ac.handlerMap.add(richTextDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*RichText))
})
}
// OnJoinClub 处理用户加入主播守护团，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnJoinClub(handler func(*AcFunLive, *JoinClub)) {
ac.handlerMap.add(joinClubDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*JoinClub))
})
}
// OnShareLive 处理分享直播间到其他平台的弹幕，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnShareLive(handler func(*AcFunLive, *ShareLive)) {
ac.handlerMap.add(shareLiveDanmu, func(ac *AcFunLive, i any) {
handler(ac, i.(*ShareLive))
})
}
// OnBananaCount 处理直播间获得的香蕉数，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnBananaCount(handler func(ac *AcFunLive, allBananaCount string)) {
ac.handlerMap.add(bananaCountEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(string))
})
}
// OnDisplayInfo 处理直播间的一些数据，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnDisplayInfo(handler func(*AcFunLive, *DisplayInfo)) {
ac.handlerMap.add(displayEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*DisplayInfo))
})
}
// OnTopUsers 处理直播间礼物榜在线前三的信息，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnTopUsers(handler func(*AcFunLive, []TopUser)) {
ac.handlerMap.add(topUsersEvent, func(ac *AcFunLive, i any) {
handler(ac, i.([]TopUser))
})
}
// OnRecentComment 处理APP进直播间时显示的最近发的弹幕，可以多次调用
func (ac *AcFunLive) OnRecentComment(handler func(*AcFunLive, []Comment)) {
ac.handlerMap.add(recentCommentEvent, func(ac *AcFunLive, i any) {
handler(ac, i.([]Comment))
})
}
// OnChatCall 处理主播发起连麦，可以多次调用
func (ac *AcFunLive) OnChatCall(handler func(*AcFunLive, *ChatCall)) {
ac.handlerMap.add(chatCallEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*ChatCall))
})
}
// OnChatAccept 处理用户接受连麦，可以多次调用
func (ac *AcFunLive) OnChatAccept(handler func(*AcFunLive, *ChatAccept)) {
ac.handlerMap.add(chatAcceptEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*ChatAccept))
})
}
// OnChatReady 处理用户接受连麦的信息，可以多次调用
func (ac *AcFunLive) OnChatReady(handler func(*AcFunLive, *ChatReady)) {
ac.handlerMap.add(chatReadyEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*ChatReady))
})
}
// OnChatEnd 处理连麦结束，可以多次调用
func (ac *AcFunLive) OnChatEnd(handler func(*AcFunLive, *ChatEnd)) {
ac.handlerMap.add(chatEndEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*ChatEnd))
})
}
// OnAuthorChatCall 处理主播发起连麦，可以多次调用
func (ac *AcFunLive) OnAuthorChatCall(handler func(*AcFunLive, *AuthorChatCall)) {
ac.handlerMap.add(authorChatCallEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*AuthorChatCall))
})
}
// OnAuthorChatAccept 处理主播接受连麦，可以多次调用
func (ac *AcFunLive) OnAuthorChatAccept(handler func(*AcFunLive, *AuthorChatAccept)) {
ac.handlerMap.add(authorChatAcceptEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*AuthorChatAccept))
})
}
// OnAuthorChatReady 处理主播接受连麦的信息，可以多次调用
func (ac *AcFunLive) OnAuthorChatReady(handler func(*AcFunLive, *AuthorChatReady)) {
ac.handlerMap.add(authorChatReadyEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*AuthorChatReady))
})
}
// OnAuthorChatEnd 处理连麦结束，可以多次调用
func (ac *AcFunLive) OnAuthorChatEnd(handler func(*AcFunLive, *AuthorChatEnd)) {
ac.handlerMap.add(authorChatEndEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*AuthorChatEnd))
})
}
// OnAuthorChatChangeSoundConfig 处理主播连麦声音设置更改，可以多次调用
func (ac *AcFunLive) OnAuthorChatChangeSoundConfig(handler func(*AcFunLive, *AuthorChatChangeSoundConfig)) {
ac.handlerMap.add(authorChatChangeSoundConfigEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(*AuthorChatChangeSoundConfig))
})
}
// OnRedpackList 处理直播间的红包列表，handler需要支持并行处理，可以多次调用
func (ac *AcFunLive) OnRedpackList(handler func(*AcFunLive, []Redpack)) {
ac.handlerMap.add(redpackListEvent, func(ac *AcFunLive, i any) {
handler(ac, i.([]Redpack))
})
}
// OnKickedOut 处理被踢出直播间，可以多次调用
func (ac *AcFunLive) OnKickedOut(handler func(ac *AcFunLive, kickedOutReason string)) {
ac.handlerMap.add(kickedOutEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(string))
})
}
// OnViolationAlert 处理直播间警告，可以多次调用
func (ac *AcFunLive) OnViolationAlert(handler func(ac *AcFunLive, violationContent string)) {
ac.handlerMap.add(violationAlertEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(string))
})
}
// OnManagerState 处理登陆帐号的房管状态，可以多次调用
func (ac *AcFunLive) OnManagerState(handler func(*AcFunLive, ManagerState)) {
ac.handlerMap.add(managerStateEvent, func(ac *AcFunLive, i any) {
handler(ac, i.(ManagerState))
})
}