use read nonblock on Linux

gnufied · Jul 22, 2008 · aef42b4 · aef42b4
1 parent 053ccb6
commit aef42b4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/lib/packet.rb b/lib/packet.rb
@@ -29,5 +29,5 @@
 PACKET_APP = File.expand_path'../' unless defined?(PACKET_APP)
 
 module Packet
-  VERSION='0.1.8'
+  VERSION='0.1.9'
 end
diff --git a/lib/packet/packet_nbio.rb b/lib/packet/packet_nbio.rb
@@ -13,9 +13,15 @@ def gen_worker_key(worker_name,worker_key = nil)
     def read_data(t_sock)
       sock_data = []
       begin
-        while(t_data = t_sock.recv_nonblock((16*1024)-1))
-          raise DisconnectError.new(t_sock,sock_data.join) if t_data.empty?
-          sock_data << t_data
+        if RUBY_PLATFORM =~ /linux/i
+          while(t_data = t_sock.read_nonblock((16*1024)-1))
+            sock_data << t_data
+          end
+        else
+          while(t_data = t_sock.recv_nonblock((16*1024)-1))
+            raise DisconnectError.new(t_sock,sock_data.join) if t_data.empty?
+            sock_data << t_data
+          end
         end
       rescue Errno::EAGAIN
         return sock_data.join