active_record_calculator 0.0.1

data/.gitignore

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

data/Gemfile

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

data/Rakefile

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

data/active_record_calculator.gemspec

@@ -0,0 +1,31 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "active_record_calculator/version"
+
+Gem::Specification.new do |s|
+  s.name        = "active_record_calculator"
+  s.version     = ActiveRecordCalculator::VERSION
+  s.authors     = ["Grady Griffin"]
+  s.email       = ["gradyg@izea.com"]
+  s.homepage    = ""
+  s.summary     = %q{ActiveRecord Calculations done faster}
+  s.description = %q{active_record_calculator does groupable aggregate functions in one sql call for better performance}
+
+  s.rubyforge_project = "active_record_calculator"
+
+  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"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+  if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+    s.add_runtime_dependency('activerecord')
+    s.add_development_dependency("rspec")
+  else
+    s.add_dependency('activerecord')
+    s.add_development_dependency("rspec")
+  end
+end

data/lib/active_record_calculator/calculator_proxy.rb

@@ -0,0 +1,71 @@
+module ActiveRecordCalculator
+  class CalculatorProxy
+    def initialize(klass, finder_options = {})
+      @klass = klass
+      @operations = []
+      @columns = []
+      @finder_options = finder_options.except(:order, :select)
+    end
+    
+    def col(column_name, as = nil)
+      add_column(column_name, as)
+    end
+    alias :column :col
+    
+    def cnt(column_name, as, options = {})
+      add_operation(:count, column_name, as, options)
+    end
+    alias :count :cnt
+    
+    def sum(column_name, as, conditions = {})
+      add_operation(:sum, column_name, as, options)
+    end
+    
+    def avg(column_name, as, conditions = {})
+      add_operation(:avg, column_name, as, options)
+    end
+    alias :average :avg
+    
+    def max(column_name, as, conditions = {})
+      add_operation(:max, column_name, as, options)
+    end
+    alias :maximum :max
+    
+    def min(column_name, as, conditions = {})
+      add_operation(:min, column_name, as, options)
+    end
+    alias :minimum :min
+    
+    def calculate
+      sql = @klass.send(:construct_finder_sql, @finder_options)
+      sql.gsub!(/^SELECT \*/, select)
+      @operations = []
+      @klass.find_by_sql(sql)
+    end
+    
+    def set_finder_options(finder_options = {})
+      @finder_options = finder_options.except(:select)
+    end
+    
+    def select
+      s = ["SELECT\n"]
+      s += @columns.join(', ') + "\n"
+      s += @operations.collect {|op| op.build_select(@klass)}.join(",\n")
+    end
+    
+    private
+    
+    def add_column(column_name, as)
+      if as
+        @columns << column_name
+      else
+        @columns << "#{column_name} AS #{as}"
+      end
+    end
+    
+    def add_operation(op, column_name, as, options)
+      options = {:conditions => options} if options.is_a?(String) 
+      @operations << Operation.new(op, column_name, as, options)
+    end
+  end
+end

data/lib/active_record_calculator/operation.rb

@@ -0,0 +1,36 @@
+module ActiveRecordCalculator
+  class Operation
+    
+    def initialize(op, column, as, options)
+      @op = op
+      @column = column
+      @as = as
+      @options = options
+    end
+    
+    def build_select(klass)
+      mock = klass.send(:construct_calculation_sql, @op, @column, @options)
+      all, sel, cond = if mock =~ /\s+WHERE\s+/
+        mock.match(/^SELECT\s+(.*)\s+AS.*\s+WHERE\s?(.*)/).to_a
+      else  
+        mock.match(/^SELECT\s+(.*)\s+AS/).to_a
+      end
+      return "#{sel} AS #{@as}" unless cond
+      i = sel.index('(')
+      start = sel[0..i]
+      mid = "CASE #{cond} WHEN 1 THEN #{sel[i+1..-2]} ELSE #{default} END"
+      fin = ") AS #{@as}"
+      "#{start}#{mid}#{fin}"
+    end
+    
+    def inspect
+      "#{@as} => #{@op}"
+    end
+    
+    private
+    
+    def default
+      @op == :sum ? 0 : 'NULL'
+    end
+  end
+end

data/lib/active_record_calculator/version.rb

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

data/lib/active_record_calculator.rb

@@ -0,0 +1,21 @@
+require "active_record_calculator/version"
+
+module ActiveRecordCalculator
+  def self.included(base)
+    base.extend ClassMethods
+  end
+  
+  module ClassMethods
+    def calculate_many(options = {}, &blk)
+      calculator = CalculatorProxy.new(self, options)
+      yield calculator
+      calculator.calculate
+    end
+    
+    def calculator
+      CalculatorProxy.new(self)
+    end
+  end
+end
+
+::ActiveRecord::Base.send :include, ActiveRecordCalculator

metadata

@@ -0,0 +1,100 @@
+--- !ruby/object:Gem::Specification 
+name: active_record_calculator
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Grady Griffin
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-01-27 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: activerecord
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+description: active_record_calculator does groupable aggregate functions in one sql call for better performance
+email: 
+- gradyg@izea.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Rakefile
+- active_record_calculator.gemspec
+- lib/active_record_calculator.rb
+- lib/active_record_calculator/calculator_proxy.rb
+- lib/active_record_calculator/operation.rb
+- lib/active_record_calculator/version.rb
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: active_record_calculator
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: ActiveRecord Calculations done faster
+test_files: []
+