Elara6331/lrpc
Archived
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Skip LRPCChannel send(), done(), and close() if already closed

Browse Source
This commit is contained in:
Elara 2022-06-03 12:33:03 -07:00
parent 368c7333c5
commit 1e627b833e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
client/web/lrpc.rb
View File

@ -100,6 +100,7 @@ class LRPCChannel
# Set self variables # Set self variables
@client = client @client = client
@id = id @id = id
@closed = false
# Set function variables to no-ops # Set function variables to no-ops
@onMessage = proc {|fn|} @onMessage = proc {|fn|}
@onClose = proc {} @onClose = proc {}
@ -112,6 +113,7 @@ class LRPCChannel
# send sends a value on the channel. This should not # send sends a value on the channel. This should not
# be called by the consumer of the channel. # be called by the consumer of the channel.
def send(val) def send(val)
return if @closed
fn = @onMessage fn = @onMessage
fn(val) fn(val)
end end
@ -119,6 +121,7 @@ class LRPCChannel
# done cancels the context corresponding to the channel # done cancels the context corresponding to the channel
# on the server side and closes the channel. # on the server side and closes the channel.
def done() def done()
return if @closed
@client.callMethod("lrpc", "ChannelDone", @id) @client.callMethod("lrpc", "ChannelDone", @id)
self.close() self.close()
@client._callMap.delete(@id) @client._callMap.delete(@id)
@ -144,5 +147,6 @@ class LRPCChannel
def close() def close()
fn = @onClose fn = @onClose
fn() fn()
@closed = true
end end
end end