1 files changed, 24 insertions, 0 deletions
diff --git a/lib/dtas/nonblock.rb b/lib/dtas/nonblock.rb
new file mode 100644
index 0000000..c8beecd
--- /dev/null
+++ b/lib/dtas/nonblock.rb
@@ -0,0 +1,24 @@
+# Copyright (C) 2015 all contributors <dtas-all@nongnu.org>
+# License: GPL-3.0+ (https://www.gnu.org/licenses/gpl-3.0.txt)
+
+class DTAS::Nonblock < IO
+  if RUBY_VERSION.to_f <= 2.0
+    EX = {}.freeze
+    def read_nonblock(len, buf = nil, opts = EX)
+      super(len, buf)
+    rescue IO::WaitReadable
+      raise if opts[:exception]
+      :wait_readable
+    rescue EOFError
+      raise if opts[:exception]
+      nil
+    end
+
+    def write_nonblock(buf, opts = EX)
+      super(buf)
+    rescue IO::WaitWritable
+      raise if opts[:exception]
+      :wait_writable
+    end
+  end
+end