cowtech-rails 2.4.0.0 → 2.5.0.0

data/app/models/cowtech/ruby_on_rails/models/scheduler.rb

@@ -14,12 +14,17 @@ module Cowtech
 				attr_accessor :definitions
 				attr_accessor :scheduler
 				
-				def initialize(application_class, log_file, pid_file, definitions)
+				def self.start(application_class, log_file, pid_file, &definitions)
+					self.new(application_class, log_file, pid_file, definitions).execute
+				end
+				
+				def initialize(application_class, log_file, pid_file, block = nil, &definitions)
 					@application = application_class
 					@logger = Logger.new(log_file)
 					@pid = pid_file.to_s
-					@definitions = definitions
+					@definitions = block || definitions
 					@scheduler = Rufus::Scheduler.start_new
+					@rake_loaded = false
 				end
 	
 				def log_string(message, options = {})
@@ -63,49 +68,47 @@ module Cowtech
 					end
 				end
 
-				def define_tasks
-					raise "Must be overriden by subclasses."
-				end
-
 				def execute
 					self.log("Scheduler started.", {:prefix => "MAIN"})
 					self.handle_plain
-
-					# if defined?(PhusionPassenger) then
-					# 	self.handle_phusion_passenger
-					# else
-					# 	self.handle_plain
-					# end
 				end
 				
 				def handle_phusion_passenger
-					File.delete(@pid) if FileTest.exists?(@pid)
+					if defined?(PhusionPassenger) then
+						File.delete(@pid) if FileTest.exists?(@pid)
 					
-				  PhusionPassenger.on_event(:starting_worker_process) do |forked|
-				    if forked && !FileTest.exists?(@pid) then
-							self.log("Starting process with PID #{$$}", {:prefix => ["WORKER", "START"]})
-				      File.open(@pid, "w") {|f| f.write($$) }
-				      self.handle_plain
-				    end
-				  end
+					  PhusionPassenger.on_event(:starting_worker_process) do |forked|
+					    if forked && !FileTest.exists?(@pid) then
+								self.log("Starting process with PID #{$$}", {:prefix => ["WORKER", "START"]})
+					      File.open(@pid, "w") {|f| f.write($$) }
+					      self.handle_plain
+					    end
+					  end
 				
-				  PhusionPassenger.on_event(:stopping_worker_process) do
-				    if FileTest.exists?(@pid) then
-				      if File.open(@pid, "r") {|f| pid = f.read.to_i} == $$ then
-								self.log("Stopped process with PID #{$$}", {:prefix => ["WORKER", "STOP"]})
-				        File.delete(@pid)
-				      end
-				    end
-				  end						
+					  PhusionPassenger.on_event(:stopping_worker_process) do
+					    if FileTest.exists?(@pid) then
+					      if File.open(@pid, "r") {|f| pid = f.read.to_i} == $$ then
+									self.log("Stopped process with PID #{$$}", {:prefix => ["WORKER", "STOP"]})
+					        File.delete(@pid)
+					      end
+					    end
+					  end						
+					else
+						self.handle_plain
+					end
 				end
 				
 				def handle_plain
-					@application.load_tasks
+					if !@rake_loaded then
+						@application.load_tasks
+						@rake_loaded = true
+					end
+					
 					@definitions.call(self)
-				end
+				end				
 				
-				def self.start(application_class, log_file, pid_file, &definitions)
-					self.new(application_class, log_file, pid_file, definitions).execute
+				def method_missing(method, *args, &block)  
+					self.scheduler.send(method, *args, &block)
 				end
       end
     end

data/lib/cowtech.rb

@@ -6,6 +6,7 @@
 
 dir = File.dirname(__FILE__) 
 
+require 'rake'
 require 'cowtech-extensions'
 require 'cowtech/extensions'
 require 'cowtech/monkey_patches'

data/lib/cowtech/tasks/log.rake

@@ -18,7 +18,7 @@ module Cowtech
         IO.popen(cmd) do |f| print f.gets end
       end
   
-      def self.rotate(email_class = nil)
+      def self.rotate(args)
         puts "Rotating log files..."
     
         # Get timestamp
@@ -42,7 +42,10 @@ module Cowtech
           FileUtils.mkdir_p(dir) if !File.directory?(dir)
       
           # Send file via mail
-          email_class.constantize.log_report(log_file).deliver if Rails.env == "production" && email_class.present?
+					if Rails.env == "production" && args[:email_class].present? then
+						puts "\tForwarding log file to requested email address..."
+	        	args[:email_class].constantize.log_report(log_file).deliver
+					end
 
           # Copy file
           FileUtils.cp(log_file, new_file)
@@ -75,7 +78,7 @@ end
 namespace :log do
   desc "Rotates log files"
   task :rotate, [:email_class] => [:environment] do |task, args|
-    Cowtech::RubyOnRails::LogUtils.rotate(args[:email_class])
+    Cowtech::RubyOnRails::LogUtils.rotate(args)
   end
   
   desc "Clean every log file"

data/lib/cowtech/tasks/mongodb.rake

@@ -6,12 +6,12 @@
 module Cowtech
   module RubyOnRails
     class MongoUtils
-      @@log_compressor_command = "tar cjvf"
+      @@log_compressor_command = "tar cjf"
       @@log_compressed_extension = "tbz2"
 
       def self.run_command(cmd); system(cmd) end
 
-      def self.backup
+      def self.backup(task_args)
         puts "--- Backupping MongoDB ..."
         
         # Get configuration
@@ -21,6 +21,7 @@ module Cowtech
 
         # Set output dir
         dest_file = Rails.root + "backups/mongodb/mongo-#{Time.now.strftime("%Y%m%d-%H%M%S")}"
+				final_file = dest_file.to_s + "." + @@log_compressed_extension
 
         # Create directory
         dir = File.dirname(dest_file)
@@ -39,9 +40,15 @@ module Cowtech
         
         # Compress
         puts "\t\tCompressing backup ..."
-        Cowtech::RubyOnRails::MongoUtils.run_command(@@log_compressor_command + " " + dest_file.to_s + "." + @@log_compressed_extension + " " + dest_file.to_s)
+        Cowtech::RubyOnRails::MongoUtils.run_command(@@log_compressor_command + " " + final_file + " " + dest_file.to_s)
         FileUtils.rm_rf(dest_file.to_s)
-  
+ 
+        # Send file via mail
+				if Rails.env == "production" && task_args[:email_class].present? then
+					puts "\tForwarding backup file to requested email address..."
+        	task_args[:email_class].constantize.backup(final_file).deliver
+				end
+
         puts "Backup saved in #{dest_file}.#{@@log_compressed_extension}"
       end
 
@@ -61,8 +68,8 @@ end
 
 namespace :mongodb do
   desc "Backups MongoDB collections"
-  task :backup do |task|
-    Cowtech::RubyOnRails::MongoUtils.backup
+  task :backup, [:email_class] => [:environment] do |task, args|
+    Cowtech::RubyOnRails::MongoUtils.backup(args)
   end
   
   desc "Clean every backup file"

data/lib/cowtech/tasks/sql.rake

@@ -12,9 +12,10 @@ module Cowtech
 
       def self.run_command(cmd); system(cmd) end
 
-      def self.mysql_execute(config)
-        dest_file = Rails.root + "backups/mysql/db-#{Time.now.strftime("%Y%m%d-%H%M%S")}.sql"
-    
+      def self.mysql_execute(config, task_args)
+        dest_file = Rails.root + "backups/mysql/mysql-#{Time.now.strftime("%Y%m%d-%H%M%S")}.sql"
+    		final_file = dest_file.to_s + "." + @@log_compressed_extension
+		
         dump_cmd = "mysqldump"
         dump_args = {"" => "-R -r \"#{dest_file}\"", "host" => "-h @@ARG@@", "username" => "-u @@ARG@@", "password" => "--password=\"@@ARG@@\"", "database" => "@@ARG@@"}
 
@@ -40,7 +41,13 @@ module Cowtech
         puts "\tCompressing backup ..."
         Cowtech::RubyOnRails::MysqlUtils.run_command(@@log_compressor_command + " " + dest_file.to_s)
     
-        puts "Backup saved in #{dest_file}.#{@@log_compressed_extension}"
+        # Send file via mail
+				if Rails.env == "production" && task_args[:email_class].present? then
+					puts "\tForwarding backup file to requested email address..."
+        	task_args[:email_class].constantize.backup(final_file).deliver
+				end
+
+        puts "Backup saved in #{final_file}"
       end
 
       # ALIAS
@@ -71,14 +78,14 @@ module Cowtech
         end 
       end
 
-      def self.backup
+      def self.backup(args)
         puts "--- Backupping database ..."
         # OTTENIAMO LA CONFIGURAZIONE
         db_config = YAML.load_file(Rails.root + "config/database.yml")
         env = Rails.env
 
         # ESEGUIAMO
-        Cowtech::RubyOnRails::MysqlUtils.send("#{db_config[env]["adapter"]}_execute", db_config[env])
+        Cowtech::RubyOnRails::MysqlUtils.send("#{db_config[env]["adapter"]}_execute", db_config[env], args)
       end
 
       def self.backup_clean
@@ -102,8 +109,8 @@ namespace :mysql do
   end 
   
   desc "Backups database"
-  task :backup do |task|
-    Cowtech::RubyOnRails::SqlUtils.backup
+  task :backup, [:email_class] => [:environment] do |task, args|
+    Cowtech::RubyOnRails::SqlUtils.backup(args)
   end
   
   desc "Clean every backup file"

data/lib/cowtech/version.rb

@@ -8,7 +8,7 @@ module Cowtech
   module Rails
     module Version
       MAJOR = 2
-      MINOR = 4
+      MINOR = 5
       PATCH = 0
       BUILD = 0
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cowtech-rails
 version: !ruby/object:Gem::Version
-  version: 2.4.0.0
+  version: 2.5.0.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-12-16 00:00:00.000000000Z
+date: 2011-12-17 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: cowtech-extensions
-  requirement: &70294263869080 !ruby/object:Gem::Requirement
+  requirement: &70342270472780 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,7 +21,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70294263869080
+  version_requirements: *70342270472780
 description: A general purpose Rails utility plugin.
 email: shogun_panda@me.com
 executables: []