resque-hitman 0.0.1

Sign up to get free protection for your applications and to get access to all the features.
@@ -0,0 +1,17 @@
1
+ *.gem
2
+ *.rbc
3
+ .bundle
4
+ .config
5
+ .yardoc
6
+ Gemfile.lock
7
+ InstalledFiles
8
+ _yardoc
9
+ coverage
10
+ doc/
11
+ lib/bundler/man
12
+ pkg
13
+ rdoc
14
+ spec/reports
15
+ test/tmp
16
+ test/version_tmp
17
+ tmp
data/Gemfile ADDED
@@ -0,0 +1,9 @@
1
+ source 'https://rubygems.org'
2
+
3
+ # Specify your gem's dependencies in resque-hitman.gemspec
4
+ gemspec
5
+
6
+ group :test do
7
+ gem "rspec"
8
+ gem "rake"
9
+ end
data/LICENSE ADDED
@@ -0,0 +1,22 @@
1
+ Copyright (c) 2012 Andy Delcambre
2
+
3
+ MIT License
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining
6
+ a copy of this software and associated documentation files (the
7
+ "Software"), to deal in the Software without restriction, including
8
+ without limitation the rights to use, copy, modify, merge, publish,
9
+ distribute, sublicense, and/or sell copies of the Software, and to
10
+ permit persons to whom the Software is furnished to do so, subject to
11
+ the following conditions:
12
+
13
+ The above copyright notice and this permission notice shall be
14
+ included in all copies or substantial portions of the Software.
15
+
16
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17
+ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18
+ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19
+ NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
20
+ LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
21
+ OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
22
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
@@ -0,0 +1,29 @@
1
+ # Resque::Hitman
2
+
3
+ TODO: Write a gem description
4
+
5
+ ## Installation
6
+
7
+ Add this line to your application's Gemfile:
8
+
9
+ gem 'resque-hitman'
10
+
11
+ And then execute:
12
+
13
+ $ bundle
14
+
15
+ Or install it yourself as:
16
+
17
+ $ gem install resque-hitman
18
+
19
+ ## Usage
20
+
21
+ TODO: Write usage instructions here
22
+
23
+ ## Contributing
24
+
25
+ 1. Fork it
26
+ 2. Create your feature branch (`git checkout -b my-new-feature`)
27
+ 3. Commit your changes (`git commit -am 'Added some feature'`)
28
+ 4. Push to the branch (`git push origin my-new-feature`)
29
+ 5. Create new Pull Request
@@ -0,0 +1,6 @@
1
+ #!/usr/bin/env rake
2
+ require "bundler/gem_tasks"
3
+ require "rspec/core/rake_task"
4
+
5
+ task :default => :spec
6
+ RSpec::Core::RakeTask.new(:spec)
@@ -0,0 +1 @@
1
+ require "resque/hitman"
@@ -0,0 +1,58 @@
1
+ require 'resque/hitman/version'
2
+
3
+ module Resque
4
+ module Hitman
5
+ module_function
6
+
7
+ def activate!(interval=5.0)
8
+ @interval = Float(interval)
9
+ old_before_fork = Resque.before_fork
10
+ Resque.before_fork = Proc.new do |job|
11
+ $hitman_monitor_thread ||= monitor(job.worker)
12
+ if old_before_fork
13
+ old_before_fork.call()
14
+ end
15
+ end
16
+ end
17
+
18
+ def deactivate!
19
+ Resque.before_fork = nil
20
+ if $hitman_monitor_thread
21
+ $hitman_monitor_thread.kill
22
+ $hitman_monitor_thread = nil
23
+ end
24
+ end
25
+
26
+ def kill!(worker)
27
+ id = if worker.respond_to?(:job)
28
+ worker.id
29
+ else
30
+ worker
31
+ end
32
+ set(key(id), 1)
33
+ end
34
+
35
+ def key(worker_id)
36
+ "resque:hitman:kill:#{worker_id}"
37
+ end
38
+
39
+ def check(worker)
40
+ unless redis.del(key(worker.id)) == 0
41
+ worker.kill_child
42
+ end
43
+ end
44
+
45
+ def monitor(worker)
46
+ Thread.new do
47
+ loop do
48
+ check(worker)
49
+ sleep @interval
50
+ end
51
+ end
52
+ end
53
+
54
+ def redis
55
+ Resque.redis
56
+ end
57
+ end
58
+ end
@@ -0,0 +1,5 @@
1
+ module Resque
2
+ module Hitman
3
+ VERSION = "0.0.1"
4
+ end
5
+ end
@@ -0,0 +1,17 @@
1
+ # -*- encoding: utf-8 -*-
2
+ require File.expand_path('../lib/resque/hitman/version', __FILE__)
3
+
4
+ Gem::Specification.new do |gem|
5
+ gem.authors = ["Andy Delcambre"]
6
+ gem.email = ["adelcambre@engineyard.com"]
7
+ gem.description = %q{A resque plugin to kill jobs}
8
+ gem.summary = %q{This is a resque plugin that uses redis as an api to kill a job that a worker is currently working on.}
9
+ gem.homepage = "http://github.com/adelcambre/resque-hitman"
10
+
11
+ gem.files = `git ls-files`.split($\)
12
+ gem.name = "resque-hitman"
13
+ gem.require_paths = ["lib"]
14
+ gem.version = Resque::Hitman::VERSION
15
+
16
+ gem.add_dependency "resque"
17
+ end
@@ -0,0 +1,46 @@
1
+ require 'resque'
2
+ require 'resque/hitman'
3
+
4
+ class HitmanJob
5
+ @queue = :test
6
+ @runs = []
7
+
8
+ def self.perform(param = "sample")
9
+ @runs << param
10
+ end
11
+ end
12
+
13
+ describe Resque::Hitman do
14
+ after do
15
+ Resque::Hitman.deactivate!
16
+ end
17
+
18
+ context "not-activated" do
19
+ it "doesn't add a before_fork hook" do
20
+ Resque.before_fork.should be_nil
21
+ end
22
+
23
+ it "calls the previously registered before_fork hook" do
24
+ $hook_called = false
25
+ Resque.before_fork = Proc.new { $hook_called = true }
26
+ Resque::Hitman.activate!
27
+
28
+ Resque.enqueue(HitmanJob)
29
+
30
+ worker = Resque::Worker.new("*")
31
+ worker.work(0)
32
+
33
+ $hook_called.should be_true
34
+ end
35
+ end
36
+
37
+ context "activated" do
38
+ before do
39
+ Resque::Hitman.activate!
40
+ end
41
+
42
+ it "adds a before_fork hook" do
43
+ Resque.before_fork.should_not be_nil
44
+ end
45
+ end
46
+ end
metadata ADDED
@@ -0,0 +1,73 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: resque-hitman
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.0.1
5
+ prerelease:
6
+ platform: ruby
7
+ authors:
8
+ - Andy Delcambre
9
+ autorequire:
10
+ bindir: bin
11
+ cert_chain: []
12
+ date: 2012-05-18 00:00:00.000000000 Z
13
+ dependencies:
14
+ - !ruby/object:Gem::Dependency
15
+ name: resque
16
+ requirement: !ruby/object:Gem::Requirement
17
+ none: false
18
+ requirements:
19
+ - - ! '>='
20
+ - !ruby/object:Gem::Version
21
+ version: '0'
22
+ type: :runtime
23
+ prerelease: false
24
+ version_requirements: !ruby/object:Gem::Requirement
25
+ none: false
26
+ requirements:
27
+ - - ! '>='
28
+ - !ruby/object:Gem::Version
29
+ version: '0'
30
+ description: A resque plugin to kill jobs
31
+ email:
32
+ - adelcambre@engineyard.com
33
+ executables: []
34
+ extensions: []
35
+ extra_rdoc_files: []
36
+ files:
37
+ - .gitignore
38
+ - Gemfile
39
+ - LICENSE
40
+ - README.md
41
+ - Rakefile
42
+ - lib/resque-hitman.rb
43
+ - lib/resque/hitman.rb
44
+ - lib/resque/hitman/version.rb
45
+ - resque-hitman.gemspec
46
+ - spec/hitman_spec.rb
47
+ homepage: http://github.com/adelcambre/resque-hitman
48
+ licenses: []
49
+ post_install_message:
50
+ rdoc_options: []
51
+ require_paths:
52
+ - lib
53
+ required_ruby_version: !ruby/object:Gem::Requirement
54
+ none: false
55
+ requirements:
56
+ - - ! '>='
57
+ - !ruby/object:Gem::Version
58
+ version: '0'
59
+ required_rubygems_version: !ruby/object:Gem::Requirement
60
+ none: false
61
+ requirements:
62
+ - - ! '>='
63
+ - !ruby/object:Gem::Version
64
+ version: '0'
65
+ requirements: []
66
+ rubyforge_project:
67
+ rubygems_version: 1.8.19
68
+ signing_key:
69
+ specification_version: 3
70
+ summary: This is a resque plugin that uses redis as an api to kill a job that a worker
71
+ is currently working on.
72
+ test_files: []
73
+ has_rdoc: