rdkit 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 10a0bd157c17b8098e43a87046384ebc7e9ae534
-  data.tar.gz: 7ad179971146d74db65ffbc8a6999b1b9b8bb034
+  metadata.gz: 88ca08b9ed9e2a240465d7d1a7c6c94bbe16e0cd
+  data.tar.gz: 700e931ec7471e9c9dee5f28e91a4510642df5bf
 SHA512:
-  metadata.gz: f8fd1464e00be7ce386d4eefb1132984cbfb7632ff498eb7df6ddd97fd07127122f27b8828c1722ad22b515b606a686286fc5a5e550adc964c00409e7950aa62
-  data.tar.gz: 2d71bef8dfac46a9b7d4274ab755e83a3f944a05a401def43b1bda6932d69c1a2cef181a229a015faf5299d73d22361778502e9653c4e52df61d251572e304a6
+  metadata.gz: d764b39f08556c73f39aa34f9a479a55945953bd8eba366ec299560246b0b80a2b0e836c780c14d43bbc2d2a3f2cf9495f9212c214b8617a8ff2cb92a83dfd55
+  data.tar.gz: f308794848e69d7f80cb1af00fbb9dd57ec1ee98f0925a948c65758f6b3a45fab4271f07497d1bd48a34ffea72533390ed34d596a8d69e4ebad343651cdc0bd6

data/README.md

@@ -417,6 +417,12 @@ $ redis-benchmark -p 3721 -n 10 nonblock
 
 See the difference between blocking and non-blocking commands?
 
+#### Implementing blocked commands
+
+Since RDKit version 0.1.5, it allows injection of additional IO handlers into the main loop.
+
+For examples, please refer to `examples/ioinject` for an injected UDP echo server.
+
 ### Implemented Redis Commands
 
 | command     | support                              | note                                         |

data/example/ioinject.rb

@@ -0,0 +1,10 @@
+require 'rdkit'
+
+require_relative 'ioinject/version'
+require_relative 'ioinject/core'
+require_relative 'ioinject/resp_responder'
+require_relative 'ioinject/server'
+
+server = IOInject::Server.new
+
+server.start

data/example/ioinject/core.rb

@@ -0,0 +1,41 @@
+require 'socket'
+
+module IOInject
+  class Core < RDKit::Core
+    def initialize(server)
+      ############ injection starts here ############
+      @injection_called = 0
+
+      udp_socket = UDPSocket.new
+      host, port = 'localhost', 8080
+      udp_socket.bind(host, port)
+
+      puts "UDP echo server injection code running at #{host}:#{port}"
+
+      server.inject_io_handler(udp_socket) do
+        begin
+          msg, sender_addrinfo, _, *controls = udp_socket.recvmsg_nonblock
+
+          p [msg, sender_addrinfo, _, *controls]
+
+          # echo back
+          udp_socket.send(msg, 0, sender_addrinfo)
+
+          @injection_called += 1
+        rescue IO::WaitReadable
+          # simply pass
+        end
+      end
+      ############ injection ends here ############
+    end
+
+    def tick!; end
+
+    def introspection
+      {
+        ioinject_version: IOInject::VERSION,
+        injection_called: @injection_called
+      }
+    end
+  end
+end

data/example/ioinject/resp_responder.rb

@@ -0,0 +1,6 @@
+module IOInject
+  class RESPResponder < RDKit::RESPResponder
+    def initialize
+    end
+  end
+end

data/example/ioinject/server.rb

@@ -0,0 +1,17 @@
+module IOInject
+  class Server < RDKit::Server
+    def initialize
+      super('0.0.0.0', 3721)
+
+      # @core is required by RDKit
+      @core = Core.new(self)
+
+      # @responder is also required by RDKit
+      @responder = RESPResponder.new
+    end
+
+    def introspection
+      super.merge(ioinject: @core.introspection)
+    end
+  end
+end

data/example/ioinject/version.rb

@@ -0,0 +1,3 @@
+module IOInject
+  VERSION = '0.0.1'
+end

data/lib/rdkit/server.rb

@@ -51,6 +51,12 @@ module RDKit
       reader, writer = create_pipe
       @selfpipe      = { :reader => reader, :writer => writer }
       @signal_queue  = []
+
+      @additional_io_handlers = {}
+    end
+
+    def inject_io_handler(another_io, &block)
+      @additional_io_handlers[another_io] = block
     end
 
     def start
@@ -313,7 +319,7 @@ module RDKit
     end
 
     def process_clients
-      readable, _ = IO.select([@server_socket, @selfpipe[:reader], @clients.keys].flatten, nil, nil, 1.0 / HZ)
+      readable, _ = IO.select([@server_socket, @selfpipe[:reader], @clients.keys, @additional_io_handlers.keys].flatten, nil, nil, 1.0 / HZ)
 
       if readable
         readable.each do |socket|
@@ -321,6 +327,8 @@ module RDKit
             add_client
           elsif socket == @selfpipe[:reader]
             handle_signals
+          elsif block = @additional_io_handlers[socket]
+            block.call
           else
             process(socket)
           end

data/lib/rdkit/version.rb

@@ -1,3 +1,3 @@
 module RDKit
-  VERSION = '0.1.4'
+  VERSION = '0.1.5'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rdkit
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Forrest Ye
@@ -200,6 +200,11 @@ files:
 - example/http/core.rb
 - example/http/responder.rb
 - example/http/server.rb
+- example/ioinject.rb
+- example/ioinject/core.rb
+- example/ioinject/resp_responder.rb
+- example/ioinject/server.rb
+- example/ioinject/version.rb
 - lib/rdkit.rb
 - lib/rdkit/callbacks.rb
 - lib/rdkit/client.rb