Close socket, terminate reactor, closes #222.

slack-ruby · Sep 7, 2018 · 115942e · 115942e
1 parent b1b681a
commit 115942e
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/lib/slack/real_time/concurrency/async.rb b/lib/slack/real_time/concurrency/async.rb
@@ -6,7 +6,7 @@ module Concurrency
       module Async
         class Client < ::Async::WebSocket::Client
           extend ::Forwardable
-          def_delegators :@driver, :on, :text, :binary
+          def_delegators :@driver, :on, :text, :binary, :emit
         end
 
         class Socket < Slack::RealTime::Socket
@@ -26,11 +26,13 @@ def connect!
           end
 
           def close
-            @driver.close
+            @closing = true
+            @driver.close if @driver
             super
           end
 
           def run_loop
+            @closing = false
             while @driver && @driver.next_event
               # $stderr.puts event.inspect
             end
@@ -50,8 +52,13 @@ def build_endpoint
             endpoint
           end
 
+          def connect_socket
+            build_endpoint.connect
+          end
+
           def connect
-            @driver = Client.new(build_endpoint.connect, url)
+            @socket = connect_socket
+            @driver = Client.new(@socket, url)
           end
         end
       end