gregfitz23-chrono_trigger 0.0.3 → 0.0.4

data/History.txt

@@ -1,3 +1,7 @@
+== 0.0.4 2009-04-09
+* 1 enhancement
+  * Added chrono_trigger binary and supporting files
+
 == 0.0.2 2009-04-02
 
 * 1 minor enhancement:

data/Manifest.txt

@@ -3,8 +3,11 @@ Manifest.txt
 PostInstall.txt
 README.rdoc
 Rakefile
+bin/chrono_trigger
 lib/chrono_trigger.rb
 lib/chrono_trigger/cron_entry.rb
+lib/chrono_trigger/process.rb
+lib/chrono_trigger/runner.rb
 lib/chrono_trigger/shell.rb
 lib/chrono_trigger/tasks.rb
 lib/chrono_trigger/trigger.rb

data/PostInstall.txt

@@ -1,3 +1 @@
-# Remember to add chrono_trigger.rake to lib/tasks in your rails project containing:
-# require 'chrono_trigger/tasks'
 

data/README.rdoc

@@ -10,8 +10,6 @@ A cron framework for defining cron tasks using a readable DSL.
 
 == SYNOPSIS:
 
-After installation, create lib/tasks/chrono_trigger.rake containing: require 'chrono_trigger/tasks'
-
 Create trigger files (by default in the lib/triggers) directory.
 Triggers should follow the pattern:
 
@@ -22,7 +20,7 @@ trigger "name" do
 	at :hour=>9, :minute=>[30,50]
 end
 
-Execute rake RAILS_ENV={env} chrono_trigger:run
+Run chrono_trigger -t{full path to trigger file}
 
 == REQUIREMENTS:
 

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
-:patch: 3
+:patch: 4
 :major: 0
 :minor: 0

data/bin/chrono_trigger

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+
+require 'chrono_trigger/runner'
+
+ChronoTrigger::Runner.run
+

data/lib/chrono_trigger/process.rb

@@ -0,0 +1,22 @@
+module ChronoTrigger
+  
+  class Process
+
+    def run(options={})
+      @t = Thread.new do
+        shell = ChronoTrigger::Shell.new
+        shell.load_triggers(options[:trigger_file])
+        loop do
+          shell.execute_triggers
+          sleep 1.minute.to_i
+        end
+      end
+      
+      @t.join
+    end
+    
+    def stop
+      @t.exit
+    end
+  end    
+end

data/lib/chrono_trigger/runner.rb

@@ -0,0 +1,276 @@
+require File.join(File.dirname(__FILE__), 'process')
+require "chrono_trigger"
+require 'optparse'
+require 'yaml'
+
+module ChronoTrigger
+  class Runner
+
+    attr_accessor :options
+    private :options, :options=
+
+    def self.run
+      new
+    end
+    
+    def self.shutdown
+      @@instance.shutdown
+    end
+
+    def initialize
+      @@instance = self
+      parse_options
+
+      @process = ProcessHelper.new(options[:logger], options[:pid_file], options[:user], options[:group])
+
+      if options[:stop]
+        @process.kill 
+        exit(1) 
+      end
+
+      pid = @process.running?
+      if pid
+        if options[:force]
+          STDERR.puts "Shutting down existing ChronoTrigger."
+          @process.kill
+          @process = ProcessHelper.new(options[:logger], options[:pid_file], options[:user], options[:group])
+        else
+          STDERR.puts "There is already a ChronoTrigger process running (pid #{pid}), exiting."
+          exit(1)
+        end
+      elsif pid.nil?
+        STDERR.puts "Cleaning up stale pidfile at #{options[:pid_file]}."
+      end
+
+      start
+    end
+
+    def parse_options
+      self.options =  { 
+                       :log_level => Logger::INFO,
+                       :daemonize => false,
+                       :pid_file => File.join('', 'var', 'run', 'chrono_trigger.pid')
+                      }
+
+      OptionParser.new do |opts|
+        opts.summary_width = 25
+
+        opts.banner = "ChronoTrigger\n\n",
+                      "usage: chrono_trigger [options...]\n",
+                      "       chrono_trigger --help\n",
+                      "       chrono_trigger --version\n"
+
+        opts.separator ""
+        opts.separator ""; opts.separator "ChronoTrigger Options:"
+        
+        opts.on("-tTRIGGER_FILE", "--triggers TRIGGERS", "Path to file specifying triggers to be executed") do |trigger_file|
+          options[:trigger_file] = trigger_file
+        end
+        
+        opts.on("-f", "--force", "Force restart of ChronoTrigger process.") do
+          options[:force] = true
+        end
+        
+        opts.on("-s", "--stop", "Stop a currently running ChronoTrigger process.") do
+          options[:stop] = true
+        end
+        
+        opts.separator ""
+        opts.separator ""; opts.separator "Process:"
+
+        opts.on("-PFILE", "--pid FILENAME", "save PID in FILENAME when using -d option.", "(default: #{options[:pid_file]})") do |pid_file|
+          options[:pid_file] = File.expand_path(pid_file)
+        end
+
+        opts.on("-u", "--user USER", "User to run as") do |user|
+          options[:user] = user.to_i == 0 ? Etc.getpwnam(user).uid : user.to_i
+        end
+
+        opts.on("-gGROUP", "--group GROUP", "Group to run as") do |group|
+          options[:group] = group.to_i == 0 ? Etc.getgrnam(group).gid : group.to_i
+        end
+
+        opts.separator ""; opts.separator "Logging:"
+
+        opts.on("-L", "--log [FILE]", "Path to print debugging information.") do |log_path|
+          options[:logger] = File.expand_path(log_path)
+        end
+
+        opts.on("-v", "Increase logging verbosity (may be used multiple times).") do
+          options[:log_level] -= 1
+        end
+
+        opts.on("-d", "Run as a daemon.") do
+          options[:daemonize] = true
+        end
+      end.parse!
+    end
+
+    def start
+      drop_privileges
+
+      @process.daemonize if options[:daemonize]
+
+      setup_signal_traps
+      @process.write_pid_file
+
+      STDOUT.puts "Starting ChronoTrigger."
+      @chrono_trigger_process = ChronoTrigger::Process.new
+      @chrono_trigger_process.run(options)
+
+      @process.remove_pid_file
+    end
+
+    def drop_privileges
+      ::Process.egid = options[:group] if options[:group]
+      ::Process.euid = options[:user] if options[:user]
+    end
+
+    def shutdown
+      begin
+        STDOUT.puts "Shutting down."
+        @chrono_trigger_process.stop
+        exit(1)
+      rescue Object => e
+        STDERR.puts "There was an error shutting down: #{e}"
+        exit(70)
+      end
+    end
+
+    def setup_signal_traps
+      Signal.trap("INT") { shutdown }
+      Signal.trap("TERM") { shutdown }
+    end
+  end
+
+  class ProcessHelper
+
+    def initialize(log_file = nil, pid_file = nil, user = nil, group = nil)
+      @log_file = log_file
+      @pid_file = pid_file
+      @user = user
+      @group = group
+    end
+
+    def safefork
+      begin
+        if pid = fork
+          return pid
+        end
+      rescue Errno::EWOULDBLOCK
+        sleep 5
+        retry
+      end
+    end
+
+    def daemonize
+      sess_id = detach_from_terminal
+      exit if pid = safefork
+
+      Dir.chdir("/")
+      File.umask 0000
+
+      close_io_handles
+      redirect_io
+
+      return sess_id
+    end
+
+    def detach_from_terminal
+      srand
+      safefork and exit
+
+      unless sess_id = ::Process.setsid
+        raise "Couldn't detach from controlling terminal."
+      end
+
+      trap 'SIGHUP', 'IGNORE'
+
+      sess_id
+    end
+
+    def close_io_handles
+      ObjectSpace.each_object(IO) do |io|
+        unless [STDIN, STDOUT, STDERR].include?(io)
+          begin
+            io.close unless io.closed?
+          rescue Exception
+          end
+        end
+      end
+    end
+
+    def redirect_io
+      begin; STDIN.reopen('/dev/null'); rescue Exception; end
+
+      if @log_file
+        begin
+          STDOUT.reopen(@log_file, "a")
+          STDOUT.sync = true
+        rescue Exception
+          begin; STDOUT.reopen('/dev/null'); rescue Exception; end
+        end
+      else
+        begin; STDOUT.reopen('/dev/null'); rescue Exception; end
+      end
+
+      begin; STDERR.reopen(STDOUT); rescue Exception; end
+      STDERR.sync = true
+    end
+
+    def rescue_exception
+      begin
+        yield
+      rescue Exception
+      end
+    end
+
+    def write_pid_file
+      return unless @pid_file
+      FileUtils.mkdir_p(File.dirname(@pid_file))
+      File.open(@pid_file, "w") { |f| f.write(::Process.pid) }
+      File.chmod(0644, @pid_file)
+    end
+
+    def remove_pid_file
+      return unless @pid_file
+      File.unlink(@pid_file) if File.exists?(@pid_file)
+    end
+
+    def running?
+      return false unless @pid_file
+
+      pid = File.read(@pid_file).chomp.to_i rescue nil
+      pid = nil if pid == 0
+      return false unless pid
+
+      begin
+        ::Process.kill(0, pid)
+        return pid
+      rescue Errno::ESRCH
+        return nil
+      rescue Errno::EPERM
+        return pid
+      end
+    end
+    
+    def kill
+      return false unless @pid_file
+
+      pid = File.read(@pid_file).chomp.to_i rescue nil
+      pid = nil if pid == 0
+      return false unless pid
+
+      begin
+        ::Process.kill("TERM", pid)
+        remove_pid_file
+        return pid
+      rescue Errno::ESRCH
+        return nil
+      rescue Errno::EPERM
+        return pid
+      end
+      
+    end
+  end
+end

data/lib/chrono_trigger/shell.rb

@@ -25,6 +25,7 @@ module ChronoTrigger
     #Run execute on any trigger who's cron entry matches the current time.
     def execute_triggers
       now = Time.now
+      STDOUT.puts "triggers: #{triggers}"
       triggers.map {|trigger| trigger.execute_on_match(now)}
     end
     

data/lib/chrono_trigger/trigger.rb

@@ -55,6 +55,7 @@ module ChronoTrigger
     end
     
     def execute
+      STDOUT.puts "executing #{@exec_block}"
       @exec_block.call
     end
   

data/lib/triggers/test_triggers.rb

@@ -1,5 +1,5 @@
 trigger "trigger1" do
-  runs { puts "trigger 1 runs every 1 minutes; executed at #{Time.now}"}
+  runs { File.open("/tmp/test_#{Time.now}.txt", 'w') {|file| file.puts "trigger 1 runs every 1 minutes; executed at #{Time.now}"}}
   every :minutes=>1  
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: gregfitz23-chrono_trigger
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors: 
 - Greg Fitzgerald
@@ -9,14 +9,14 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-04-06 00:00:00 -07:00
-default_executable: 
+date: 2009-04-09 00:00:00 -07:00
+default_executable: chrono_trigger
 dependencies: []
 
 description: TODO
 email: greg_fitz@yahoo.com
-executables: []
-
+executables: 
+- chrono_trigger
 extensions: []
 
 extra_rdoc_files: 
@@ -27,8 +27,11 @@ files:
 - PostInstall.txt
 - README.rdoc
 - VERSION.yml
+- bin/chrono_trigger
 - lib/chrono_trigger
 - lib/chrono_trigger/cron_entry.rb
+- lib/chrono_trigger/process.rb
+- lib/chrono_trigger/runner.rb
 - lib/chrono_trigger/shell.rb
 - lib/chrono_trigger/tasks.rb
 - lib/chrono_trigger/trigger.rb