reductor 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 276380fee6ca82990688b8563dded3e1a3917344
-  data.tar.gz: e3a96aa7bafa4c87f30326a66188e48bb75a3be8
+  metadata.gz: 6a5a238cb24cc1f8de7500be93a91693486367ec
+  data.tar.gz: ddeb998004dd0be173e73a358d9b11428431cd28
 SHA512:
-  metadata.gz: 0c3d0d2d0424d3ed3a10ddb2c17dcd0b1cda06e5185b0a3cbed2c1824eff3f90e0d0af41b1e865d8feeb690ffda7c20943dc53e40bc0d57196d0f74b9b61bac1
-  data.tar.gz: 9745420bc9af25e64f96647120f9ea27a4b791dd98b16eeb46137e3ede8dd71b9d4e1625cbb27f493a1165f9b7801c9e34c3f5964cf56ff6e5f92a4a99d34aba
+  metadata.gz: 32bc32f45554932f300f2893c701edc7ef1029d5e56bc3d49193577f8819fac8761ff1602a769348117ce826c897af56adc285f7e6d22bfc66a5b5ecf138f948
+  data.tar.gz: eb5b6999b97240ad15f65212506cf125c4cd83ec65fd6d1292a783c740d8b5c3c554ade963ecd10dd11fd0a60c642103c5ddea33493e106da8889b5cfae08704

data/.gitignore

@@ -0,0 +1 @@
+*.gem

data/README.md

@@ -5,7 +5,10 @@ Collection of common map_reduce operations, implemented with mongoid.
 ## Basic sample:
 
 ```
-pry(main)> Project.all.reduction(:counts_by_day)
+pry(main)>Project.all.reduct :sum do
+pry(main)>  by :project_id, { created_at: :by_day }
+pry(main)>  value :duration
+pry(main)>end
 => {
                         nil => 12.0,
     2012-05-10 22:00:00 UTC => 1.0,

data/lib/reductor.rb

@@ -1,27 +1,54 @@
 module Mongoid
   class Criteria
 
-    def reduction(reduction_type, out: :inline, options: {})
-      reductor = Reductor.new(self, reduction_type, options)
-      Hash[reductor.reduce.out(inline: true).map{|e| [e['_id'], e["value"]] }]
+    def reduct(reduction_type, &block)
+      reductor = Reductor.new(self, reduction_type, &block)
+      Hash[reductor.reduct.map{|e| [e['_id'], e["value"]] }]
     end
 
   end
 end
 
 require 'reductor/reductions'
+require 'reductor/simple_field'
+require 'reductor/converted_field'
 
 class Reductor
 
-  def initialize(collection, type, options)
-    @collection = collection
-    @type = type
-    @options = options
+  attr_accessor :collection, :type, :block, :fields
+
+  def initialize(collection, type, &block)
+    self.collection = collection
+    self.type = type
+    self.block = block
+  end
+
+  def keys(*keys)
+    self.fields = keys.map do |f|
+      if f.is_a?(Hash)
+        f.map{ |k,v| Reductor::ConvertedField.new(k, v) }
+      else
+        Reductor::SimpleField.new(f)
+      end
+    end.flatten
   end
 
-  def reduce
+  def map
+    %Q{
+      function() {
+        #{emit}
+      }
+    }
+  end
+
+  def reduct
     self.extend reduction_module
-    @collection.map_reduce(map, reduce)
+    instance_eval &block
+    @collection.map_reduce(map, reduce).finalize(finalize).out(@out || { inline: true })
+  end
+
+  def out arg
+    @out = arg
   end
 
   def reduction_module

data/lib/reductor/converted_field.rb

@@ -0,0 +1,31 @@
+class Reductor
+  class ConvertedField < Struct.new(:field_name, :translation_type)
+
+    def to_s
+      "#{field_name}: (#{translated})"
+    end
+
+    def translated
+      send(translation_type, field_name)
+    end
+
+    private
+
+    def to_hour(field_name)
+      "this.#{field_name} ? new Date(this.#{field_name}.getFullYear(), this.#{field_name}.getMonth(), this.#{field_name}.getDate(), this.#{field_name}.getHours()) : this.#{field_name}"
+    end
+
+    def to_day(field_name)
+      "this.#{field_name} ? new Date(this.#{field_name}.getFullYear(), this.#{field_name}.getMonth(), this.#{field_name}.getDate()) : this.#{field_name}"
+    end
+
+    def to_month(field_name)
+      "this.#{field_name} ? new Date(this.#{field_name}.getFullYear(), this.#{field_name}.getMonth()) : this.#{field_name}"
+    end
+
+    def to_year(field_name)
+      "this.#{field_name} ? new Date(this.#{field_name}.getFullYear()) : this.#{field_name}"
+    end
+
+  end
+end

data/lib/reductor/reductions.rb

@@ -1,4 +1,5 @@
-require 'reductor/reductions/counts_by_day'
+require 'reductor/reductions/count'
+require 'reductor/reductions/sum'
 class Reductor
   module Reductions
     # cats

data/lib/reductor/reductions/count.rb

@@ -0,0 +1,40 @@
+class Reductor
+  module Reductions
+    module Count
+
+      def emit
+        if fields.size > 1
+          "emit({#{fields.join(", ")}}, 1);"
+        else
+          "emit(#{fields.first.translated}, 1);"
+        end
+      end
+
+      def count options
+        @as = options[:as]
+      end
+
+      def value_field_name
+        @as || 'count';
+      end
+
+      def reduce
+        %Q{
+          function(key, values) {
+            return Array.sum(values);
+          }
+        }
+      end
+
+
+      def finalize
+        %Q{
+          function(key, values) {
+            return { #{value_field_name}: values };
+          }
+        }
+      end
+
+    end
+  end
+end

data/lib/reductor/reductions/sum.rb

@@ -0,0 +1,40 @@
+class Reductor
+  module Reductions
+    module Sum
+
+      def emit
+        if fields.size > 1
+          "emit({ #{fields.join(", ")} } , #{translated_sum_field});"
+        else
+          "emit( #{fields.first.translated} , #{translated_sum_field});"
+        end
+      end
+
+      def translated_sum_field
+        "this.#{@sum_field}"
+      end
+
+      def sum field, options = {}
+        @sum_field = field
+        @sum_field_name = options[:as] || @sum_field
+      end
+
+      def reduce
+        %Q{
+          function(key, values) {
+            return Array.sum(values);
+          }
+        }
+      end
+
+      def finalize
+        %Q{
+          function(key, values) {
+            return { #{@sum_field_name}: values };
+          }
+        }
+      end
+
+    end
+  end
+end

data/lib/reductor/simple_field.rb

@@ -0,0 +1,13 @@
+class Reductor
+  class SimpleField < Struct.new(:field_name)
+
+    def to_s
+      "#{field_name}: #{translated}"
+    end
+
+    def translated
+      "this.#{field_name}"
+    end
+
+  end
+end

data/reductor.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name        = 'reductor'
-  s.version     = '0.0.1'
+  s.version     = '0.0.2'
   s.date        = '2013-07-12'
   s.summary     = "Map-reduce with ease."
   s.description = "Collection of common map_reduce operations, implemented with mongoid."

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: reductor
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Marcin Stecki
@@ -16,10 +16,14 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- .gitignore
 - README.md
 - lib/reductor.rb
+- lib/reductor/converted_field.rb
 - lib/reductor/reductions.rb
-- lib/reductor/reductions/counts_by_day.rb
+- lib/reductor/reductions/count.rb
+- lib/reductor/reductions/sum.rb
+- lib/reductor/simple_field.rb
 - reductor.gemspec
 homepage: http://rubygems.org/gems/hola
 licenses: []
@@ -45,3 +49,4 @@ signing_key:
 specification_version: 4
 summary: Map-reduce with ease.
 test_files: []
+has_rdoc: 

data/lib/reductor/reductions/counts_by_day.rb

@@ -1,32 +0,0 @@
-class Reductor
-  module Reductions
-    module CountsByDay
-
-      def datetime_field
-        @options[:field] || 'created_at'
-      end
-
-      def map
-        %Q{
-          function() {
-            if(this.#{datetime_field}){
-              date = new Date(this.#{datetime_field}.getFullYear(), this.#{datetime_field}.getMonth(), this.#{datetime_field}.getDate());
-              } else {
-                date = null;
-              }
-              emit(date , 1);
-            }
-        }
-      end
-
-      def reduce
-        %Q{
-          function(key, values) {
-            return Array.sum(values);
-          }
-        }
-      end
-
-    end
-  end
-end