ethnt · February 24, 2017 19:39 · Feb 24, 2017 · Feb 24, 2017 · Feb 24, 2017
diff --git a/referee.rb b/referee.rb
@@ -1,3 +1,5 @@
+require 'redis'
+
 module Referee
   class Timeout < StandardError; end
 
@@ -7,6 +9,8 @@ def initialize(key, limit = 5, &block)
       @key = key
       @limit = limit
 
+      @redis.set(@key, @limit)
+
       start
 
       @work = Thread.new { yield self }

diff --git a/killer.rb b/killer.rb
@@ -0,0 +1,5 @@
+require 'redis'
+
+@redis = Redis.new
+
+@redis.set('timeout', 0)
diff --git a/referee.rb b/referee.rb
@@ -2,13 +2,13 @@ module Referee
   class Timeout < StandardError; end
 
   class Timer
-    def initialize(limit, &block)
+    def initialize(key, limit = 5, &block)
+      @redis = Redis.new
+      @key = key
       @limit = limit
 
       start
 
-      @timed_out = false
-
       @work = Thread.new { yield self }
       @work.join
 
@@ -17,7 +17,21 @@ def initialize(limit, &block)
 
     def start
       @timer = Thread.new do
-        sleep(@limit)
+        @counter = 0
+
+        # TODO: Use promises to make sure we have the limit from redis before
+        #   incrementing the counter.
+        loop do
+          @limit = @redis.get(@key).to_i
+
+          @counter += 1
+
+          if @counter > @limit
+            break
+          end
+
+          sleep(1)
+        end
 
         @timed_out = true
 

diff --git a/referee.rb b/referee.rb
@@ -0,0 +1,39 @@
+module Referee
+  class Timeout < StandardError; end
+
+  class Timer
+    def initialize(limit, &block)
+      @limit = limit
+
+      start
+
+      @timed_out = false
+
+      @work = Thread.new { yield self }
+      @work.join
+
+      @timer.join if @timed_out
+    end
+
+    def start
+      @timer = Thread.new do
+        sleep(@limit)
+
+        @timed_out = true
+
+        @work.kill
+
+        raise Referee::Timeout
+      end
+    end
+
+    def stop
+      @timer.kill
+    end
+
+    def reset
+      stop
+      start
+    end
+  end
+end
diff --git a/worker.rb b/worker.rb
@@ -0,0 +1,12 @@
+require_relative 'referee'
+
+begin
+  Referee::Timer.new(5) do |timer|
+    sleep(1)
+    puts "yup"
+    sleep(5)
+    puts "nope"
+  end
+rescue Referee::Timeout
+  puts "timed out!"
+end
No results found