time_limits 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in time_limits.gemspec
+gemspec

data/README

@@ -0,0 +1,17 @@
+Apply timeouts to methods via meta programming
+
+Usage:
+
+class Foo
+
+  def my_possibly_slow_method() sleep(rand 10); end
+  def my_possibly_slow_query() sleep(rand 10); rand; end
+  
+  include TimeLimits
+  time_limit 1.second, :my_possibly_slow_method
+  time_limit 1.second, :my_possibly_slow_method, :rescue => false
+
+end
+
+a = Foo.new.my_possibly_slow_method # may throw Timeout
+b = Foo.new.my_possibly_slow_query # may return false

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/lib/time_limits/class_methods.rb

@@ -0,0 +1,32 @@
+module TimeLimits
+  module ClassMethods
+    
+    def time_limit( timeout, *names )
+      raise unless timeout.to_i > 0
+      options = names.last.kind_of?(Hash) ? names.pop : {}
+      raise unless names.size > 0
+      names.each do |name|
+        original = instance_method(name)
+        if options.has_key?(:rescue) then
+          rescue_value = options[:rescue]
+          define_method( name ) do |*args| # &block not supported in 1.8
+            TimeLimits::Timeout.in_time( timeout ) do
+              begin
+                original.bind(self).call( *args ) # &block not supported in 1.8
+              rescue ::Timeout::Error
+                rescue_value
+              end
+            end
+          end
+        else
+          define_method( name ) do |*args| # &block not supported in 1.8
+            TimeLimits::Timeout.in_time( timeout ) do
+              original.bind(self).call( *args ) # &block not supported in 1.8
+            end
+          end
+        end
+      end
+    end
+    
+  end
+end

data/lib/time_limits/timeout.rb

@@ -0,0 +1,33 @@
+if RUBY_VERSION =~ /^1\.8/ then
+  require "rubygems"
+  require "system_timer"
+end
+
+module TimeLimits
+  module Timeout
+    
+    if RUBY_VERSION =~ /^1\.8/ then
+      
+      def in_time( duration = 1, &block )
+        SystemTimer.timeout_after(duration) { yield }
+      end
+    
+    else
+      
+      def in_time( duration = 1, &block )
+        Timeout::timeout(duration) { yield }
+      end
+      
+    end
+    
+    # options :rescue value
+    def in_time_or_else( duration = 1, options = {}, &block )
+      in_time( duration, &block )
+    rescue ::Timeout::Error => x
+      return options[:rescue]
+    end
+    
+    extend Timeout
+    
+  end
+end

data/lib/time_limits/version.rb

@@ -0,0 +1,3 @@
+module TimeLimits
+  VERSION = "0.0.1"
+end

data/lib/time_limits.rb

@@ -0,0 +1,10 @@
+require "time_limits/timeout" # TimeLimits::Timeout.timeout( duration = 1.second, &block )
+require "time_limits/class_methods" # time_limit timeout, *method_names, :rescue => value
+
+module TimeLimits
+  class << self
+    def included( base )
+      base.send( :extend, ClassMethods )
+    end
+  end
+end

data/test/test.rb

@@ -0,0 +1,42 @@
+#!/usr/bin/env ruby
+$:.push File.expand_path("../../lib", __FILE__)
+#require "rubygems"
+#require "ruby-debug"
+require "time_limits"
+
+class X
+  def slow() sleep(2); end
+  def slow2() sleep(2); end
+  def slow3() sleep(3); end
+  def slow4() sleep(4); end
+  include TimeLimits
+  time_limit 1, :slow, :slow2, :slow3, :slow4
+end
+
+begin
+  X.new.slow2
+  raise "Failed"
+rescue Timeout::Error => x
+  puts "Ok"
+end
+
+class X2
+  def slow() sleep(2); true; end
+  def slow2() sleep(2); end
+  include TimeLimits
+  time_limit 1, :slow, :slow2, :rescue => false
+end
+
+ok = X2.new.slow2
+raise "Failed" unless ok == false
+
+begin
+  TimeLimits::Timeout.in_time { sleep 2 }
+  raise "Failed"
+rescue Timeout::Error => x
+  puts "Ok"
+end
+
+ok = TimeLimits::Timeout.in_time_or_else( 1, :rescue => false ) { sleep 2 }
+raise unless ok == false
+puts "Ok"

data/time_limits.gemspec

@@ -0,0 +1,20 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "time_limits/version"
+
+Gem::Specification.new do |s|
+  s.name        = "time_limits"
+  s.version     = TimeLimits::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Mark Lanett"]
+  s.email       = ["mark.lanett@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{Apply timeouts to methods via meta programming}
+  
+  s.rubyforge_project = "time_limits"
+  
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

metadata

@@ -0,0 +1,74 @@
+--- !ruby/object:Gem::Specification 
+name: time_limits
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Mark Lanett
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-04-08 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: 
+- mark.lanett@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README
+- Rakefile
+- lib/time_limits.rb
+- lib/time_limits/class_methods.rb
+- lib/time_limits/timeout.rb
+- lib/time_limits/version.rb
+- test/test.rb
+- time_limits.gemspec
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: time_limits
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Apply timeouts to methods via meta programming
+test_files: 
+- test/test.rb