rerun_task 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 rerun_task.gemspec
+gemspec

data/README.md

@@ -0,0 +1,32 @@
+Rerun Task
+==========
+
+
+Rerun task is gem for checking if your rake task was executed successfully. You have to place all your rake code to `WatchProcess.new("task_name").call{}`. I write below some examples.
+
+Examples
+====
+	task :dummy_task => :environment do
+		WatchProcess.new("task_name").call do
+			# your code write here 		
+		end
+	end
+	
+Now if you run rake task, then is created pid-files of your process. To your crontab place task, which every 5th minute check if all task done successfully.
+
+non-rails app
+====
+If you use as non-rails app, you have to copy /bin/rerun_task_crontab.rb from gem to your app
+
+	*/5 * * * * ruby /path/to/the/your-app/bin/rerun_task_crontab.rb
+
+rails app
+====
+	
+If you use it in rails app, you can use rails runner
+
+	*/5 * * * * cd path/to/your-app; bundle exec rails runner RerunTask::UnfinishedRunner.crontab_retry
+	
+Notes	
+====
+In this version is not allowed to run same process in same time.	

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/rerun_task_crontab.rb

@@ -0,0 +1,3 @@
+load "#{File.dirname(__FILE__)}/../lib/rerun_task.rb"
+
+RerunTask::UnfinishedRunner.crontab_retry

data/config/rerun_task.yml

@@ -0,0 +1 @@
+pid_dir: "/tmp"

data/lib/rerun_task/config.rb

@@ -0,0 +1,4 @@
+require 'yaml'
+module RerunTask
+  CONFIG = YAML.load_file("#{File.dirname(__FILE__)}/../../config/rerun_task.yml")
+end

data/lib/rerun_task/crontab.rb

@@ -0,0 +1,16 @@
+module RerunTask
+  class Crontab
+    attr_accessor :content
+    def initialize
+      @content = %x[crontab -l]
+      @content = @content.split("\n")
+    end
+
+    def find_task(task_name)
+      match = @content.reject{|i| !i.include?(task_name)}
+      return nil if match.size == 0
+      match.first.split(" ")[5,9].join(" ")
+    end
+
+  end
+end

data/lib/rerun_task/pids.rb

@@ -0,0 +1,22 @@
+module RerunTask
+  class Pids
+    def self.unfinished
+      res = []
+      Dir.glob("#{RerunTask::CONFIG['pid_dir']}/rerun_task/pids/*.pid").each do |file_path|
+        args = {}
+        File.open(file_path, 'r') do |f|
+          args = ProcessFile.parse_file(f.read)
+        end
+        next if args[:process_name].nil? || args[:pid].nil?
+        p = ProcessFile.load_file("#{RerunTask::CONFIG['pid_dir']}/rerun_task/pids", args[:process_name])
+        begin
+          p.process_exists?
+        rescue
+          res << p
+          next
+        end
+      end
+      res
+    end
+  end
+end

data/lib/rerun_task/process_file.rb

@@ -0,0 +1,79 @@
+require 'pathname'
+
+module RerunTask
+  class ProcessFile
+    attr_accessor :process_name, :pid
+
+    def initialize(pid_dir)
+      @pid_dir = pid_dir
+      FileUtils.mkdir_p(@pid_dir) unless Dir.exists?(@pid_dir)
+    end
+
+    def create_pid_file
+      if File.exists?(path)
+        puts "file exists #{path}"
+
+        process_from_file = ProcessFile.load_file(@pid_dir, process_name)
+        raise RuntimeError.new("Process #{process_name} is already running with pid #{process_from_file.pid}") if process_from_file.process_exists?
+      end
+
+      File.open(path, 'w+') do |f|
+        f.write(self.to_s)
+      end
+    end
+
+    def path
+      ProcessFile.create_file_path(@pid_dir, process_name)
+    end
+
+    def self.create_file_path(pid_dir, process_name)
+      basename = Pathname.new("#{pid_dir}/#{process_name}.pid").basename
+      "#{pid_dir}/#{basename}"
+    end
+
+    def process_name
+      @process_name = @process_name || extract_process_name(Process.pid)
+    end
+
+    def to_s
+      "#{Process.pid}\n#{process_name}"
+    end
+
+    def process_exists?
+      extract_process_name(pid) == @process_name
+    end
+
+    def self.load_file(pid_dir, process_name)
+      f = File.open(create_file_path(pid_dir, process_name), 'r')
+      p = ProcessFile.new(pid_dir)
+      args = parse_file(f.read)
+      p.process_name = process_name.nil? ? args[:process_name] : process_name
+      p.pid = args[:pid]
+      f.close
+      p
+    end
+
+    def self.parse_file(string)
+      a = string.split("\n")
+      { pid:  a.first, process_name: a.last }
+    end
+
+    def destroy
+      FileUtils.rm(path)
+    end
+
+    private
+
+    def extract_process_name(pid)
+      ps_res = %x[ps ax]
+      ps_res = ps_res.split("\n")
+      name = ""
+      ps_res.each do |line|
+        rows = line.split(" ")
+        name = rows.last if rows.first == pid.to_s
+      end
+      raise RuntimeError.new("process_name not found for pid #{pid}") if name === ""
+      name
+    end
+  end
+end

data/lib/rerun_task/unfinished_runner.rb

@@ -0,0 +1,14 @@
+module RerunTask
+  class UnfinishedRunner
+
+    def self.crontab_retry
+      crontab = Crontab.new()
+      Pids.unfinished.each do |process_file|
+        cmd = crontab.find_task(process_file.process_name)
+        next if cmd.nil?
+        puts "Running command #{cmd} again"
+        system("#{cmd} &")
+      end
+    end
+  end
+end

data/lib/rerun_task/version.rb

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

data/lib/rerun_task/watch_process.rb

@@ -0,0 +1,25 @@
+module RerunTask
+  class WatchProcess
+
+    def initialize(rake_task_name)
+      @name = rake_task_name
+      @pid_dir = "#{RerunTask::CONFIG['pid_dir']}/rerun_task/pids"
+      @process_file = ProcessFile.new(@pid_dir)
+    end
+
+    def call(&block)
+      start
+      block.call
+      finished
+    end
+
+    def start
+      @process_file.create_pid_file
+    end
+
+    def finished
+      @process_file.destroy
+    end
+
+  end
+end

data/lib/rerun_task.rb

@@ -0,0 +1,11 @@
+require "rerun_task/version"
+require "rerun_task/config"
+require "rerun_task/watch_process"
+require "rerun_task/process_file"
+require "rerun_task/pids"
+require "rerun_task/crontab"
+require "rerun_task/unfinished_runner"
+
+module RerunTask
+
+end

data/rerun_task.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "rerun_task/version"
+
+Gem::Specification.new do |s|
+  s.name        = "rerun_task"
+  s.version     = RerunTask::VERSION
+  s.authors     = ["lukasvotypka"]
+  s.email       = ["lukas.votypka@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{Re-run task if is not finished}
+  s.description = %q{This gem is developed for handling rake task and re-run them if didn't finished successfully}
+
+  s.rubyforge_project = "rerun_task"
+
+  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"]
+
+  s.add_development_dependency "test-unit"
+  s.add_development_dependency "mocha"
+end

data/test/watch_process_test.rb

@@ -0,0 +1,70 @@
+require 'test/unit'
+require "mocha/setup"
+require 'rerun_task'
+
+class WatchProcessTest < Test::Unit::TestCase
+
+  # Called before every test method runs. Can be used
+  # to set up fixture information.
+  def setup
+    @process_file = RerunTask::ProcessFile.new("#{RerunTask::CONFIG['pid_dir']}/rerun_task/pids")
+    begin
+      FileUtils.rm(@process_file.path)
+    rescue
+    end
+  end
+
+  # Called after every test method runs. Can be used to tear
+  # down fixture information.
+
+  def teardown
+    # Do nothing
+  end
+
+  def test_should_start
+    #assert_equal("/tmp/rerun_task/pids/watch_process_test.rb.pid", @process_file.path)
+    @w = RerunTask::WatchProcess.new("my_test")
+    assert_equal(false, File.exists?(@process_file.path))
+    @w.start
+    assert_equal(true, File.exists?(@process_file.path))
+    @w.finished
+    assert_equal(false, File.exists?(@process_file.path))
+
+    @w = RerunTask::WatchProcess.new("my_test")
+    @w.start
+    @w2 = RerunTask::WatchProcess.new("my_test")
+    assert_raise_kind_of(RuntimeError) do
+      @w2.start
+    end
+    @w.finished
+  end
+
+  def test_list_pids
+    path = "#{RerunTask::CONFIG['pid_dir']}/rerun_task/pids"
+    filename = "#{path}/test_example_task.rb.pid"
+    FileUtils.rm(filename) if File.exists?(filename)
+    File.open(filename, 'w') do |f|
+      f.write("1234\ntest_example_task.rb")
+    end
+
+    unfinished = RerunTask::Pids.unfinished
+    assert_equal(1, unfinished.size)
+    assert_equal('test_example_task.rb', unfinished.first.process_name)
+  end
+
+  def test_run_from_crontab
+    path = "#{RerunTask::CONFIG['pid_dir']}/rerun_task/pids"
+    filename = "#{path}/test_example_task.rb.pid"
+    FileUtils.rm(filename) if File.exists?(filename)
+    File.open(filename, 'w') do |f|
+      f.write("1234\ntest_example_task.rb")
+    end
+
+    crontab = RerunTask::Crontab.new()
+    crontab.content = ["0 0 1,5,9,13,17,21,25,29 * * ruby test_example_task.rb"]
+    res = crontab.find_task('test_example_task.rb')
+    assert_equal("ruby test_example_task.rb", res)
+
+    RerunTask::UnfinishedRunner.crontab_retry()
+  end
+end

metadata

@@ -0,0 +1,92 @@
+--- !ruby/object:Gem::Specification 
+name: rerun_task
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.1
+platform: ruby
+authors: 
+- lukasvotypka
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2013-12-23 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: test-unit
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: mocha
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :development
+  version_requirements: *id002
+description: This gem is developed for handling rake task and re-run them if didn't finished successfully
+email: 
+- lukas.votypka@gmail.com
+executables: 
+- rerun_task_crontab.rb
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- bin/rerun_task_crontab.rb
+- config/rerun_task.yml
+- lib/rerun_task.rb
+- lib/rerun_task/config.rb
+- lib/rerun_task/crontab.rb
+- lib/rerun_task/pids.rb
+- lib/rerun_task/process_file.rb
+- lib/rerun_task/unfinished_runner.rb
+- lib/rerun_task/version.rb
+- lib/rerun_task/watch_process.rb
+- rerun_task.gemspec
+- test/watch_process_test.rb
+homepage: ""
+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: rerun_task
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Re-run task if is not finished
+test_files: 
+- test/watch_process_test.rb
+has_rdoc: