resque-heartbeat 0.0.1

data/lib/resque/heartbeat.rb

@@ -0,0 +1,70 @@
+require 'resque'
+
+module Resque
+  def self.prune_dead_workers
+    begin
+      Worker.all.each { |worker| worker.prune_if_dead }
+    rescue Exception => e
+      p e
+    end
+  end
+
+  class Worker
+    def startup_with_heartbeat
+      startup_without_heartbeat
+      heart.run
+    end
+    alias startup_without_heartbeat startup
+    alias startup startup_with_heartbeat
+
+    def heart
+      @heart ||= Heart.new(self)
+    end
+
+    def prune_if_dead
+      unregister_worker if heart.last_beat_before?(5)
+    end
+
+    class Heart
+      attr_reader :worker
+
+      def initialize(worker)
+        @worker = worker
+      end
+
+      def run
+        Thread.new { loop { sleep(2) && beat! } }
+      end
+
+      def redis
+        @redis && connected? ? @redis : @redis = connect
+      end
+
+      def connect
+        # apparently the Redis connection is not thread-safe, so we connect another instance
+        # see https://github.com/ezmobius/redis-rb/issues#issue/75
+        redis = Redis.connect(:url => Travis.config['redis']['url']).tap { |redis| redis.client.connect }
+        Redis::Namespace.new(:resque, :redis => redis)
+      end
+
+      def connected?
+        @redis.client.connected?
+      end
+
+      def beat!
+        redis.sadd(:workers, worker)
+        redis.set("worker:#{worker}:heartbeat", Time.now.to_s)
+      rescue Exception => e
+        p e
+      end
+
+      def last_beat_before?(seconds)
+        Time.parse(last_beat).utc < (Time.now.utc - seconds) rescue true
+      end
+
+      def last_beat
+        Resque.redis.get("worker:#{worker}:heartbeat") || worker.started
+      end
+    end
+  end
+end

data/lib/resque_heartbeat/version.rb

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

data/lib/resque_heartbeat.rb

@@ -0,0 +1 @@
+require 'resque/heartbeat'

metadata

@@ -0,0 +1,49 @@
+--- !ruby/object:Gem::Specification
+name: resque-heartbeat
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Sven Fuchs
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-06-06 00:00:00.000000000 +02:00
+default_executable: 
+dependencies: []
+description: ! '[description]'
+email: svenfuchs@artweb-design.de
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/resque/heartbeat.rb
+- lib/resque_heartbeat.rb
+- lib/resque_heartbeat/version.rb
+has_rdoc: true
+homepage: http://github.com/svenfuchs/resque-heartbeat
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: ! '[none]'
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: ! '[summary]'
+test_files: []