magent 0.0.3 → 0.1.0

data/examples/simple/bot.rb

@@ -24,6 +24,9 @@ class Bot
     $stderr.puts "RESULT: #{result}"
   end
 
+  at_least_every 15 do
+    puts "Hi there!, you'll see this message again in ~15 seconds"
+  end
 end
 
 Magent.register(Bot.new)

data/lib/magent.rb

@@ -10,7 +10,7 @@ require 'magent/actor'
 require 'magent/processor'
 
 module Magent
-  VERSION = '0.0.3'
+  VERSION = '0.1.0'
 
   def self.connection
     @@connection ||= Mongo::Connection.new(nil, nil, :auto_reconnect => true)
@@ -30,3 +30,4 @@ module Magent
 end
 
 Magent.database = 'magent'
+

data/lib/magent/actor.rb

@@ -3,6 +3,7 @@ module Magent
     def self.included(klass)
       klass.class_eval do
         extend Actor::ClassMethods
+        include Actor::InstanceMethods
       end
     end
 
@@ -32,6 +33,35 @@ module Magent
           Channel.new(channel_name)
         end
       end
+
+      def tasks
+        @tasks ||= []
+      end
+
+      def at_least_every(seconds, &block)
+        tasks << {:every => seconds, :last_time => Time.now, :block => block}
+      end
+    end
+
+    module InstanceMethods
+      def _run_tasks
+        tasks = self.class.tasks
+
+        return false if tasks.empty?
+        performed = false
+
+        tasks.each do |task|
+          delta = Time.now - task[:last_time]
+
+          if delta >= task[:every]
+            task[:last_time] = Time.now
+            instance_eval(&task[:block])
+            performed = true
+          end
+        end
+
+        performed
+      end
     end
   end # Actor
 

data/lib/magent/processor.rb

@@ -4,6 +4,7 @@ module Magent
 
     def initialize(actor)
       @actor = actor
+      @shutdown = false
 
       @actor.class.actions.each do |action|
         if !@actor.respond_to?(action)
@@ -13,30 +14,55 @@ module Magent
     end
 
     def run!
+      processed_messages = 0
       delay = 0
+
+      trap('TERM') { shutdown!; exit 0 }
+      trap('SIGINT') { shutdown!; exit 0 }
+
       loop do
-        method, payload = @actor.class.channel.dequeue
+        break if @shutdown
+
+        delay = 0 if @actor._run_tasks
 
-        if method.nil?
+        @method, @payload = @actor.class.channel.dequeue
+
+        if @method.nil?
           delay += 0.1 if delay <= 5
         else
           delay = 0
-          $stderr.puts "#{@actor.class}##{method}(#{payload.inspect})"
+          $stderr.puts "#{@actor.class}##{@method}(#{@payload.inspect})"
           begin
-            if @actor.class.actions.include?(method)
-              @actor.send(method, payload)
+            if @actor.class.actions.include?(@method)
+              processed_messages += 1
+              @actor.send(@method, @payload)
+
+              if processed_messages > 20
+                processed_messages = 0
+                GC.start
+              end
             else
-              $stderr.puts "Unknown action: #{method} (payload=#{payload.inspect})"
+              $stderr.puts "Unknown action: #{@method} (payload=#{@payload.inspect})"
             end
+          rescue SystemExit
           rescue Exception => e
-            $stderr.puts "Error while executing #{method.inspect} #{payload.inspect}"
-            $stderr.puts "#{e.to_s} #{e.backtrace.join("\t\n")}"
-            @actor.class.channel.failed(:message => e.message, :method => method, :payload => payload, :backtrace => e.backtrace)
+            $stderr.puts "Error while executing #{@method.inspect} #{@payload.inspect}"
+            $stderr.puts "#{e.to_s}\n#{e.backtrace.join("\t\n")}"
+            @actor.class.channel.failed(:message => e.message, :method => @method, :payload => @payload, :backtrace => e.backtrace, :date => Time.now.utc)
+          ensure
+            @method, @payload = nil
           end
         end
-
         sleep delay
       end
     end
+
+    def shutdown!
+      @shutdown = true
+      $stderr.puts "Shutting down..."
+      if @method
+        @actor.class.channel.enqueue(@method, @payload)
+      end
+    end
   end #Processor
 end

data/magent.gemspec

@@ -2,11 +2,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{magent}
-  s.version = "0.0.3"
+  s.version = "0.1.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["David Cuadrado"]
-  s.date = %q{2009-10-11}
+  s.date = %q{2009-10-14}
   s.default_executable = %q{magent}
   s.description = %q{Simple job queue system based on mongodb}
   s.email = ["krawek@gmail.com"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: magent
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors: 
 - David Cuadrado
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-10-11 00:00:00 -05:00
+date: 2009-10-14 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency