mirrors/ctrl
mirrors/ctrl
1
0
Fork 0
mirror of https://github.com/postmannen/ctrl.git synced 2024-12-14 12:37:31 +00:00
Code Activity

added select for ctx.Done for methodREQRelay

Browse source
This commit is contained in:
postmannen 2022-02-08 05:27:39 +01:00
parent 75ed40877b
commit afe24716db
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
requests.go
View file

@ -1798,7 +1798,10 @@ func (m methodREQRelay) getKind() Event {
func (m methodREQRelay) handler(proc process, message Message, node string) ([]byte, error) { func (m methodREQRelay) handler(proc process, message Message, node string) ([]byte, error) {
// relay the message here to the actual host here. // relay the message here to the actual host here.
proc.processes.wg.Add(1)
go func() { go func() {
defer proc.processes.wg.Done()
message.ToNode = message.RelayToNode message.ToNode = message.RelayToNode
message.FromNode = Node(node) message.FromNode = Node(node)
message.Method = message.RelayOriginalMethod message.Method = message.RelayOriginalMethod
@ -1811,7 +1814,10 @@ func (m methodREQRelay) handler(proc process, message Message, node string) ([]b
return return
} }
proc.toRingbufferCh <- []subjectAndMessage{sam} select {
case proc.toRingbufferCh <- []subjectAndMessage{sam}:
case <-proc.ctx.Done():
}
}() }()
// Send back an ACK message. // Send back an ACK message.