Lift up context to downstreamConn.handleMessage
This commit is contained in:
parent
e28332a5aa
commit
eb47f0dcd7
@ -615,22 +615,22 @@ func (dc *downstreamConn) marshalMessage(msg *irc.Message, net *network) *irc.Me
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (dc *downstreamConn) handleMessage(msg *irc.Message) error {
|
func (dc *downstreamConn) handleMessage(msg *irc.Message) error {
|
||||||
|
ctx, cancel := context.WithTimeout(context.TODO(), handleDownstreamMessageTimeout)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
switch msg.Command {
|
switch msg.Command {
|
||||||
case "QUIT":
|
case "QUIT":
|
||||||
return dc.Close()
|
return dc.Close()
|
||||||
default:
|
default:
|
||||||
if dc.registered {
|
if dc.registered {
|
||||||
return dc.handleMessageRegistered(msg)
|
return dc.handleMessageRegistered(ctx, msg)
|
||||||
} else {
|
} else {
|
||||||
return dc.handleMessageUnregistered(msg)
|
return dc.handleMessageUnregistered(ctx, msg)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dc *downstreamConn) handleMessageUnregistered(msg *irc.Message) error {
|
func (dc *downstreamConn) handleMessageUnregistered(ctx context.Context, msg *irc.Message) error {
|
||||||
ctx, cancel := context.WithTimeout(context.TODO(), handleDownstreamMessageTimeout)
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
switch msg.Command {
|
switch msg.Command {
|
||||||
case "NICK":
|
case "NICK":
|
||||||
var nick string
|
var nick string
|
||||||
@ -1416,10 +1416,7 @@ func (dc *downstreamConn) runUntilRegistered() error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dc *downstreamConn) handleMessageRegistered(msg *irc.Message) error {
|
func (dc *downstreamConn) handleMessageRegistered(ctx context.Context, msg *irc.Message) error {
|
||||||
ctx, cancel := context.WithTimeout(context.TODO(), handleDownstreamMessageTimeout)
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
switch msg.Command {
|
switch msg.Command {
|
||||||
case "CAP":
|
case "CAP":
|
||||||
var subCmd string
|
var subCmd string
|
||||||
|
Loading…
Reference in New Issue
Block a user