quick_utils 0.0.1

data/.gitignore

@@ -0,0 +1,6 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+
+*.swp

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in quick_utils.gemspec
+gemspec

data/Rakefile

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

data/lib/quick_utils/rails_daemon.rb

@@ -0,0 +1,63 @@
+require 'rubygems'
+require 'daemons'
+require 'optparse'
+require 'logger'
+
+module QuickUtils
+	class RailsDaemon
+
+		@@proc_name = ''
+		@@log_file = ''
+		@@rails_root = Dir.pwd
+
+		def self.set_process_name(pname)
+			@@proc_name = pname
+			@@log_file = File.join(@@rails_root, 'log', "#{@@proc_name}.log")
+		end
+
+		def self.set_rails_root(dir)
+			@@rails_root = dir
+		end
+		
+		def initialize(args)
+			@options = {:worker_count => 1, :environment => :development, :delay => 5}
+			
+			optparse = OptionParser.new do |opts|
+				opts.banner = "Usage: #{File.basename($0)} [options] start|stop|restart|run"
+
+				opts.on('-h', '--help', 'Show this message') do
+					puts opts
+					exit 1
+				end
+				opts.on('-e', '--environment=NAME', 'Specifies the environment to run this apn_sender under ([development]/production).') do |e|
+					@options[:environment] = e
+				end
+				opts.on('-n', '--number-of-workers=WORKERS', "Number of unique workers to spawn") do |worker_count|
+					@options[:worker_count] = worker_count.to_i rescue 1
+				end
+				opts.on('-v', '--verbose', "Turn on verbose mode") do
+					@options[:verbose] = true
+				end
+				opts.on('-d', '--delay=D', "Delay between rounds of work (seconds)") do |d|
+					@options[:delay] = d
+				end
+			end
+			
+			# If no arguments, give help screen
+			@args = optparse.parse!(args.empty? ? ['-h'] : args)
+		end
+
+		def daemonize
+			@options[:worker_count].times do |worker_index|
+				process_name = @options[:worker_count] == 1 ? @@proc_name : "#{@@proc_name}.#{worker_index}"
+				Daemons.run_proc(process_name, :dir => "#{@@rails_root}/tmp/pids", :dir_mode => :normal, :ARGV => @args) do |*args|
+					run process_name
+				end
+			end
+		end
+
+		def run(worker_name = nil)
+
+		end
+	end
+end

data/lib/quick_utils/rake_daemon.rb

@@ -0,0 +1,19 @@
+module QuickUtils
+	class RakeDaemon < RailsDaemon
+
+		@@rake_task = ''
+
+		def self.set_rake_task(task)
+			@@rake_task = task
+		end
+
+		def run(worker_name = nil)
+
+			ENV['RAILS_ENV'] = @options[:environment].to_s
+			ENV['LOG_FILE'] = @@log_file
+
+			system "cd #{@@rails_root}; exec rake RAILS_ENV='#{@options[:environment].to_s}' LOG_FILE='#{@@log_file}' #{@@rake_task}"
+			
+		end
+	end
+end

data/lib/quick_utils/thread_runner.rb

@@ -0,0 +1,23 @@
+module QuickUtils
+  class ThreadRunner
+
+    def self.run(dataset, nthreads)
+      threads = []
+      count = dataset.size
+      nthreads = 1 if count < nthreads
+      bin_size = count / nthreads
+      puts "Dividing #{count} items into sizes of #{bin_size} for #{nthreads} threads."
+      t = 0
+      while t < nthreads do
+        subset = dataset[t*bin_size..t*bin_size+bin_size - 1]
+        threads << Thread.new(subset) do |cs|
+          yield cs
+        end
+        t = t + 1
+      end
+      threads.each {|t| t.join}
+    end
+
+  end
+end
+

data/lib/quick_utils/version.rb

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

data/lib/quick_utils/watcher.rb

@@ -0,0 +1,32 @@
+module QuickUtils
+	class Watcher < RailsDaemon
+		def self.set_app_name(app_name)
+			@@main_proc = app_name
+		end
+
+		def self.set_proc_list(plist)
+			@@processes = plist
+		end
+
+		def run(worker_name = nil)
+
+			logger = Logger.new(@@log_file)
+
+			while true
+				# get num thins running
+				thin_num = `ps -ef | grep "thin" | grep "#{@@main_proc}" | grep -v "grep" | wc | awk '{print $1}'`.to_i
+				needs_restart = false
+				@@processes.each do |proc|
+					pnum = `ps -ef | grep "#{proc}" | grep -v "grep" | wc | awk '{print $1}'`.to_i
+					needs_restart ||= (pnum == 0)
+				end
+				if thin_num == 0 || needs_restart
+					logger.info("Restarting #{main_proc} now...");
+					system "cd #{@@rails_root}; bundle exec script/restart;"
+				end
+				sleep @options[:delay]
+			end
+				
+		end
+	end
+end

data/lib/quick_utils.rb

@@ -0,0 +1,9 @@
+require "quick_utils/version"
+require "quick_utils/thread_runner"
+require "quick_utils/rails_daemon"
+require "quick_utils/rake_daemon"
+require "quick_utils/watcher"
+
+module QuickUtils
+  # Your code goes here...
+end

data/quick_utils.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "quick_utils/version"
+
+Gem::Specification.new do |s|
+  s.name        = "quick_utils"
+  s.version     = QuickUtils::VERSION
+  s.authors     = ["Alan Graham"]
+  s.email       = ["alangraham5@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{Utility classes for a rails-deployed application}
+  s.description = %q{Utility classes for a rails-deployed application}
+
+  s.rubyforge_project = "quick_utils"
+
+  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_dependency "daemons"
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification
+name: quick_utils
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Alan Graham
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-20 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: daemons
+  requirement: &70129914024040 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70129914024040
+description: Utility classes for a rails-deployed application
+email:
+- alangraham5@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- lib/quick_utils.rb
+- lib/quick_utils/rails_daemon.rb
+- lib/quick_utils/rake_daemon.rb
+- lib/quick_utils/thread_runner.rb
+- lib/quick_utils/version.rb
+- lib/quick_utils/watcher.rb
+- quick_utils.gemspec
+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: quick_utils
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Utility classes for a rails-deployed application
+test_files: []