tubesock 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7bbfdedb374d8745921858be0f57db08061efa0a
-  data.tar.gz: 536094ed6152a6dc6fc394188ad3bcf4087b5fb6
+  metadata.gz: 8e34c2f3cd416319f09063d1420a92bb4de54f41
+  data.tar.gz: 35232dc37e37a9abd7b2595d4eeb8b9e9b252b39
 SHA512:
-  metadata.gz: 696025befa9b62e79e2a92a3f76817b2252e0ccb7d266b41c798f1ec553f93492e55a22f6c269eaa4ecc0fe54a03d04e5e81bb574254dea7412123b55eaa065e
-  data.tar.gz: dcdac25b4eadf8d0c13e7f56b804237356a58b7de0fdcea8c3161fb8133bed66aad2bdb60d57f2d62731c3d6f9dc6763ecdef01decfc43cfb5eaa7f247b647b7
+  metadata.gz: 81501d0966fd116753dfe71231626c3cda459672fef63bde5f12a0df305ed91d89fafba1179a29ea825562aa6052bf40c94d0e72c568e46b3bdba2d5daa417f2
+  data.tar.gz: 5a06e190adfd2967405e22af868f0967d671b20613c30682fa4bf3d925579e9f49df83d582c4729f640c61cdde314e86d86f309f726d0891e4370f3f36cbab87

data/lib/tubesock.rb

@@ -8,6 +8,10 @@ class Tubesock
   def initialize(socket, version)
     @socket     = socket
     @version    = version
+
+    @open_handlers    = []
+    @message_handlers = []
+    @close_handlers   = []
   end
 
   def self.hijack(env)
@@ -41,15 +45,15 @@ class Tubesock
   end
 
   def onopen(&block)
-    @openhandler = block
+    @open_handlers << block
   end
 
   def onmessage(&block)
-    @messagehandler = block
+    @message_handlers << block
   end
 
   def onclose(&block)
-    @closehandler = block
+    @close_handlers << block
   end
 
   def listen
@@ -57,6 +61,7 @@ class Tubesock
       Thread.current.abort_on_exception = true
       framebuffer = WebSocket::Frame::Incoming::Server.new(version: @version)
       running = true
+      @open_handlers.each(&:call)
       while running
         data, addrinfo = @socket.recvfrom(2000)
         running = false if data == ""
@@ -66,18 +71,12 @@ class Tubesock
           if data == ""
             running = false
           else
-            @messagehandler.call(HashWithIndifferentAccess.new(JSON.load(data))) if @messagehandler
+            @message_handlers.each{|h| h.call(HashWithIndifferentAccess.new(JSON.load(data))) }
           end
         end
       end
-      @closehandler.call if @closehandler
+      @close_handlers.each(&:call)
       @socket.close
     end
-    @openhandler.call if @openhandler
-  end
-
-  def close
-    @socket.close if @socket.open?
-    @closehandler.call
   end
 end

data/lib/tubesock/hijack.rb

@@ -6,9 +6,12 @@ class Tubesock
     included do
       def hijack
         if Tubesock.websocket?(env)
-          rocket = Tubesock.hijack(env)
-          yield rocket
-          rocket.listen
+          sock = Tubesock.hijack(env)
+          yield sock
+          sock.onclose do
+            ActiveRecord::Base.clear_active_connections!
+          end
+          sock.listen
           render text: nil, status: -1
         else
           render text: "Not found", status: :not_found

data/lib/tubesock/version.rb

@@ -1,3 +1,3 @@
 class Tubesock
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tubesock
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Nick Gauthier