diamondburned/arikawa
1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2025-11-23 13:14:16 +00:00
Code Issues Releases Activity

Add GatewayOpts.ErrorIsFatalClose

Browse source
This commit is contained in:
diamondburned 2022-11-17 22:07:52 -08:00
parent 87c479a2dc
commit 78ad477b83
No known key found for this signature in database
GPG key ID: D78C4471CE776659
1 changed files with 23 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
utils/ws/gateway.go
View file

@ -89,6 +89,23 @@ var DefaultGatewayOpts = GatewayOpts{
AlwaysCloseGracefully: true,
}
// ErrorIsFatalClose returns true if the error is a fatal close error. It uses
// opts.FatalCloseCodes to check for the codes.
func (opts GatewayOpts) ErrorIsFatalClose(err error) bool {
var closeErr *CloseEvent
if !errors.As(err, &closeErr) {
return false
}
for _, code := range opts.FatalCloseCodes {
if code == closeErr.Code {
return true
}
}
return false
}
// Gateway describes an instance that handles the Discord gateway. It is
// basically an abstracted concurrent event loop that the user could signal to
// start connecting to the Discord gateway server.
@ -306,8 +323,7 @@ func (g *Gateway) spin(ctx context.Context, h Handler) {
switch data := op.Data.(type) {
case *CloseEvent:
for _, code := range g.opts.FatalCloseCodes {
if code == data.Code {
if g.opts.ErrorIsFatalClose(data) {
// Don't wrap the error, but instead, just pipe it as-is
// through the channel.
g.outer.ch <- op
@ -315,7 +331,6 @@ func (g *Gateway) spin(ctx context.Context, h Handler) {
return
}
}
}
ok = h.OnOp(ctx, op)
g.outer.ch <- op