Fixing things here and there, not like anyone is reading these messages

bin/jmandy-hadoop

@@ -1,6 +1,6 @@
 #!/usr/bin/env ruby
 require "rubygems"
-require "jmandy"
+require 'jmandy'
 require 'optparse'
 require 'ostruct'
 require 'uri'
@@ -31,6 +31,12 @@ def set_env(opts_string)
   end
 end
 
+def check_env
+  %w(JAVA_HOME JRUBY_HOME HADOOP_HOME).each do |env|
+    raise "#{env} not found" unless ENV[env]
+  end
+end
+
 file   = File.expand_path(ARGV[0])
 inputs  = ARGV[1].split(",")
 input = inputs.map {|path| "#{path}"}.join(" ")
@@ -40,6 +46,7 @@ config = absolute_path(options.config || 'cluster.xml')
 puts "Packaging code for distribution..."
 cmdenv = options.cmdenv
 set_env(cmdenv)
+check_env
 
 at_exit do
   puts

examples/word_count.rb

@@ -7,6 +7,9 @@
   map_tasks 5
   reduce_tasks 5
 
+  setup do
+  end
+
   map do |key, value|
     words = {}
     value.split(' ').each do |word|

lib/jmandy/job.rb

@@ -86,11 +86,17 @@ def reducer_class
     end
 
     def compile_map
-      @map.is_a?(Proc) ? JMandy::Mappers::Base.compile({}, &@map) : @map
+      args = {}
+      args[:setup] = @setup if @setup
+      args[:teardown] = @teardown if @teardown
+      @map.is_a?(Proc) ? JMandy::Mappers::Base.compile(args, &@map) : @map
     end
 
     def compile_reduce
-      @reduce.is_a?(Proc) ? JMandy::Reducers::Base.compile({}, &@reduce) : @reduce
+      args = {}
+      args[:setup] = @setup if @setup
+      args[:teardown] = @teardown if @teardown
+      @reduce.is_a?(Proc) ? JMandy::Reducers::Base.compile(args, &@reduce) : @reduce
     end
   end
 end

lib/jmandy/mappers/base_mapper.rb

@@ -10,19 +10,23 @@ def self.compile(opts={}, &blk)
         Class.new(JMandy::Mappers::Base) do 
           self.class_eval do
             define_method(:mapper, blk) if blk
+            define_method(:setup, opts[:setup]) if opts[:setup]
+            define_method(:teardown, opts[:teardown]) if opts[:teardown]
           end
         end
       end
 
       def execute
+        setup if self.respond_to?(:setup)
         mapper(@key, @value)
+        teardown if self.respond_to?(:teardown)
       end
-
+      
       private
-
+      
       def mapper(key,value)
         #nil
       end
     end
   end
-end
+end

lib/jmandy/reducers/base_reducer.rb

@@ -10,14 +10,18 @@ def self.compile(opts={}, &blk)
         Class.new(JMandy::Reducers::Base) do 
           self.class_eval do
             define_method(:reducer, blk) if blk
+            define_method(:setup, opts[:setup]) if opts[:setup]
+            define_method(:teardown, opts[:teardown]) if opts[:teardown]
           end
         end
       end
-
+      
       def execute
+        setup if self.respond_to?(:setup)
         reducer(@key, @values)
+        teardown if self.respond_to?(:teardown)
       end
-
+      
       private
 
       def reducer(key,values)