diff options
| -rw-r--r-- | activerecord/lib/active_record/relation.rb | 1 | ||||
| -rw-r--r-- | activerecord/lib/active_record/relation/query_methods.rb | 35 | ||||
| -rw-r--r-- | activerecord/test/cases/statement_cache_test.rb | 45 |
3 files changed, 78 insertions, 3 deletions
diff --git a/activerecord/lib/active_record/relation.rb b/activerecord/lib/active_record/relation.rb index 94ab465cc1..6516c4be5c 100644 --- a/activerecord/lib/active_record/relation.rb +++ b/activerecord/lib/active_record/relation.rb @@ -202,6 +202,7 @@ module ActiveRecord # Please see further details in the # {Active Record Query Interface guide}[http://guides.rubyonrails.org/active_record_querying.html#running-explain]. def explain + #TODO: Fix for binds. exec_explain(collecting_queries_for_explain { exec_queries }) end diff --git a/activerecord/lib/active_record/relation/query_methods.rb b/activerecord/lib/active_record/relation/query_methods.rb index d020f1ba52..4628a5811b 100644 --- a/activerecord/lib/active_record/relation/query_methods.rb +++ b/activerecord/lib/active_record/relation/query_methods.rb @@ -377,6 +377,9 @@ module ActiveRecord end end + #For bind param caching. TODO: VALIDATE AND CORRECT THIS + self.bind_values = [] + #end self end @@ -804,7 +807,7 @@ module ActiveRecord build_joins(arel, joins_values) unless joins_values.empty? - collapse_wheres(arel, (where_values - ['']).uniq) + collapse_wheres(arel, (where_values - [''])) #TODO: Add uniq with real value comparison / ignore uniqs that have binds arel.having(*having_values.uniq.reject{|h| h.blank?}) unless having_values.empty? @@ -893,8 +896,12 @@ module ActiveRecord when String, Array [@klass.send(:sanitize_sql, other.empty? ? opts : ([opts] + other))] when Hash - attributes = @klass.send(:expand_hash_conditions_for_aggregates, opts) + temp_opts = opts.dup + create_binds(temp_opts) + temp_opts = substitute_opts(temp_opts) + + attributes = @klass.send(:expand_hash_conditions_for_aggregates, temp_opts) attributes.values.grep(ActiveRecord::Relation) do |rel| self.bind_values += rel.bind_values end @@ -905,6 +912,30 @@ module ActiveRecord end end + def create_binds(temp_opts) + binds = [] + temp_opts.map do |column, value| + case value + when String, Integer + if @klass.column_names.include? column.to_s + binds.push([@klass.columns_hash[column.to_s], value]) + end + end + end + self.bind_values += binds + end + + def substitute_opts(temp_opts) + temp_opts = temp_opts.each_with_index do |(column,value), index| + if @klass.columns_hash[column.to_s] != nil + case value + when String, Integer + temp_opts[column] = connection.substitute_at(column, index) + end + end + end + end + def build_from opts, name = from_value case opts diff --git a/activerecord/test/cases/statement_cache_test.rb b/activerecord/test/cases/statement_cache_test.rb index 76da49707f..ae34b174db 100644 --- a/activerecord/test/cases/statement_cache_test.rb +++ b/activerecord/test/cases/statement_cache_test.rb @@ -10,6 +10,49 @@ module ActiveRecord @connection = ActiveRecord::Base.connection end + #Cache v 1.1 tests + def test_statement_cache + Book.create(name: "my book") + Book.create(name: "my other book") + + cache = StatementCache.new do + Book.where(:name => "my book") + end + + b = cache.execute name: "my book" + assert_equal "my book", b[0].name + b = cache.execute name: "my other book" + assert_equal "my other book", b[0].name + end + + + #Validate primary key binding + def test_statement_cache_id + Book.create(name: "my book") + Book.create(name: "my other book") + + cache = StatementCache.new do + Book.where(id: "1") + end + + b = cache.execute id: "1" + assert_equal "my book", b[0].name + b = cache.execute id: "2" + assert_equal "my other book", b[0].name + end + + def test_find_or_create_by + Book.create(name: "my book") + + a = Book.find_or_create_by(name: "my book") + b = Book.find_or_create_by(name: "my other book") + + assert_equal("my book", a.name) + assert_equal("my other book", b.name) + end + + #End + def test_statement_cache_with_simple_statement cache = ActiveRecord::StatementCache.new do Book.where(name: "my book").where("author_id > 3") @@ -61,4 +104,4 @@ module ActiveRecord assert first_books != additional_books end end -end +end
\ No newline at end of file |