couch_potato 1.14.0 → 1.15.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dd4dede2152745db1f3f0bf7ff3a94dfea3c7eb427beb02292034866dd7dac22
-  data.tar.gz: 29ca77512b6bdeebcdd31847deac4957792946519dda4f658f60dd90c9dd83e4
+  metadata.gz: ffcaae6b2815d237b48cc3e4ad553f66920fe4df2c84c59ae38177260949ba6b
+  data.tar.gz: bd266a1033fa83fcba822a19e3de86b2c1aac2d80ba2dd1d3aef0920d4819d8f
 SHA512:
-  metadata.gz: 21728691cd7eb4b23bd6600899655cc7fa1dbb6a15a6d40c669b53941fab30bf38cb8a95fa9a3e8f85e05d44af116f58cd20a424eedd60d417972db893c07aff
-  data.tar.gz: d2d26373afd0bcce487b35703be3bd7bd08cc7e4e108f98e8279bf81c41715b30f5590fdf8080c4b136ec95e356c857b5c50bef123b6971e67294e79cdeb180a
+  metadata.gz: 16f3e1748e3b3314f1cf3d658fb175dd90c5e7c7cd717ab42477961c364bac527b262642a29f208c4aad200e9a31ce7470c0a538db8bf8d641e13da92f85debb
+  data.tar.gz: '089f1c9e5dc7238b659a4d1b2d0f3d341a084fe52f6d9c2d058ea8440f197f7d64b6f5a9d78c8cebbd19b131aede038329ab6c0f79b503d220f1f20d2e7d8a20'

data/CHANGES.md

@@ -1,5 +1,10 @@
 ## Changes
 
+# 1.15.0
+
+- cache loading multiple documents
+- keep the database cache when switching to another database and back to the original one
+
 # 1.14.0
 
 - add database_collection to models to help avoid n+1 requests

data/lib/couch_potato/database.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'active_support/core_ext/enumerable'
+
 module CouchPotato
   class Database
     class ValidationsFailedError < ::StandardError; end
@@ -143,7 +145,9 @@ module CouchPotato
     # returns nil if the single document could not be found. when passing an array and some documents
     # could not be found these are omitted from the returned array
     def load_document(id)
-      cached = cache && id.is_a?(String) && cache[id]
+      return load_documents(id) if id.is_a?(Array)
+      
+      cached = cache && cache[id]
       if cache
         if cached
           ActiveSupport::Notifications.instrument('couch_potato.load.cached') do
@@ -159,6 +163,23 @@ module CouchPotato
     end
     alias load load_document
 
+    def load_documents(ids)
+      return [] if ids.empty?
+
+      uncached_ids = ids - (cache&.keys || [])
+      uncached_docs_by_id = bulk_load(uncached_ids).index_by {|doc| doc.id if doc.respond_to?(:id) }
+      if cache
+        uncached_ids.each do |id|
+          doc = uncached_docs_by_id[id]
+          cache[id] = doc if doc
+        end
+      end
+      cached_docs_by_id = ActiveSupport::Notifications.instrument('couch_potato.load.cached') do
+        cache&.slice(*ids) || {}
+      end
+      ids.filter_map { |id| (cached_docs_by_id[id]) || uncached_docs_by_id[id] }
+    end
+
     # loads one or more documents by its id(s)
     # behaves like #load except it raises a CouchPotato::NotFound if any of the documents could not be found
     def load!(id)
@@ -176,6 +197,8 @@ module CouchPotato
     # returns the underlying CouchRest::Database instance
     attr_reader :couchrest_database
 
+    attr_accessor :switched_databases
+
     # returns a new database instance connected to the CouchDB database
     # with the given name. the name is passed through the
     # additional_databases configuration to resolve it to a database
@@ -183,21 +206,29 @@ module CouchPotato
     # if the current database has a cache, the new database will receive
     # a cleared copy of it.
     def switch_to(database_name)
-      resolved_database_name = CouchPotato.resolve_database_name(database_name)
-      self
+      self.switched_databases ||= {}
+      if (db = switched_databases[database_name || :default])
+        return db
+      end
+      switched_databases[name || :default] ||= self
+
+
+      resolved_database_name = CouchPotato.resolve_database_name(database_name) unless database_name == :default
+      couchrest_database = resolved_database_name ? CouchPotato.couchrest_database_for_name(resolved_database_name) : CouchPotato.couchrest_database
+      new_db = self
         .class
-        .new(CouchPotato.couchrest_database_for_name(resolved_database_name), name: database_name)
+        .new(couchrest_database, name: database_name == :default ? nil : database_name)
         .tap(&copy_clear_cache_proc)
+
+      new_db.switched_databases = switched_databases
+      new_db
     end
 
     # returns a new database instance connected to the default CouchDB database.
     # if the current database has a cache, the new database will receive
     # a cleared copy of it.
     def switch_to_default
-      self
-        .class
-        .new(CouchPotato.couchrest_database)
-        .tap(&copy_clear_cache_proc)
+      switch_to(:default)
     end
 
     private
@@ -248,14 +279,10 @@ module CouchPotato
       raise "Can't load a document without an id (got nil)" if id.nil?
 
       ActiveSupport::Notifications.instrument('couch_potato.load') do
-        if id.is_a?(Array)
-          bulk_load id
-        else
-          instance = couchrest_database.get(id)
-          instance.database = self if instance
-          instance
-        end
-      end
+        instance = couchrest_database.get(id)
+        instance.database = self if instance
+        instance
+    end
     end
 
     def view_cache_id(spec)
@@ -294,11 +321,13 @@ module CouchPotato
     def bulk_load(ids)
       return [] if ids.empty?
 
-      response = couchrest_database.bulk_load ids
-      docs = response['rows'].map { |row| row['doc'] }.compact
-      docs.each do |doc|
-        doc.database = self if doc.respond_to?(:database=)
-        doc.database_collection = docs if doc.respond_to?(:database_collection=)
+      ActiveSupport::Notifications.instrument('couch_potato.load') do
+        response = couchrest_database.bulk_load ids
+        docs = response['rows'].map { |row| row['doc'] }.compact
+        docs.each do |doc|
+          doc.database = self if doc.respond_to?(:database=)
+          doc.database_collection = docs if doc.respond_to?(:database_collection=)
+        end
       end
     end
 

data/lib/couch_potato/version.rb

@@ -1,4 +1,4 @@
 module CouchPotato
-  VERSION = '1.14.0'.freeze
+  VERSION = '1.15.0'.freeze
   RSPEC_VERSION = '4.1.0'.freeze
 end

data/spec/unit/caching_spec.rb

@@ -18,41 +18,92 @@ RSpec.describe 'database caching' do
     {}
   end
 
-  it 'gets an object from the cache the 2nd time via #load_documemt' do
-    expect(couchrest_db).to receive(:get).with('1').exactly(1).times
+  context 'for a single document' do
+    it 'gets an object from the cache the 2nd time via #load_documemt' do
+      expect(couchrest_db).to receive(:get).with('1').exactly(1).times
 
-    db.load_document '1'
-    db.load_document '1'
-  end
+      db.load_document '1'
+      db.load_document '1'
+    end
 
-  it 'gets an object from the cache the 2nd time via #load' do
-    expect(couchrest_db).to receive(:get).with('1').exactly(1).times
+    it 'gets an object from the cache the 2nd time via #load' do
+      expect(couchrest_db).to receive(:get).with('1').exactly(1).times
 
-    db.load '1'
-    db.load '1'
-  end
+      db.load '1'
+      db.load '1'
+    end
 
-  it 'gets an object from the cache the 2nd time via #load!' do
-    expect(couchrest_db).to receive(:get).with('1').exactly(1).times
+    it 'gets an object from the cache the 2nd time via #load!' do
+      expect(couchrest_db).to receive(:get).with('1').exactly(1).times
 
-    db.load! '1'
-    db.load! '1'
-  end
+      db.load! '1'
+      db.load! '1'
+    end
 
-  it 'returns the correct object' do
-    doc = double(:doc, 'database=': nil)
-    allow(couchrest_db).to receive_messages(get: doc)
+    it 'returns the correct object' do
+      doc = double(:doc, 'database=': nil)
+      allow(couchrest_db).to receive_messages(get: doc)
 
-    db.load_document '1'
-    expect(db.load_document('1')).to eql(doc)
+      db.load_document '1'
+      expect(db.load_document('1')).to eql(doc)
+    end
   end
 
-  it 'does not cache bulk loads' do
-    allow(couchrest_db).to receive_messages(bulk_load: {'rows' => []})
-    expect(couchrest_db).to receive(:bulk_load).with(['1']).exactly(2).times
+  context 'for multiple documents' do
+    let(:doc1) { double(:doc1, 'database=': nil, id: '1') }
+    let(:doc2) { double(:doc12, 'database=': nil, id: '2') }
+
+    it 'only loads uncached documents' do
+      allow(couchrest_db).to receive(:bulk_load).with(['1']).and_return('rows' => [{'doc' => doc1}])
+      allow(couchrest_db).to receive(:bulk_load).with(['2']).and_return('rows' => [{'doc' => doc2}])
+      
+
+      db.load_document(['1'])
+      db.load_document(['1', '2'])
+
+      expect(couchrest_db).to have_received(:bulk_load).with(['1']).exactly(1).times
+      expect(couchrest_db).to have_received(:bulk_load).with(['2']).exactly(1).times
+    end
+
+    it 'loads nothing if all documents are cached' do
+      allow(couchrest_db).to receive(:bulk_load).with(['1', '2'])
+        .and_return('rows' => [{'doc' => doc1}, {'doc' => doc2}])
+
+      db.load_document(['1', '2'])
+      db.load_document(['1', '2'])
+
+      expect(couchrest_db).to have_received(:bulk_load).with(['1', '2']).exactly(1).times
+    end
+
+    it 'returns all requested documents' do
+      allow(couchrest_db).to receive(:bulk_load).with(['1']).and_return('rows' => [{'doc' => doc1}])
+      allow(couchrest_db).to receive(:bulk_load).with(['2']).and_return('rows' => [{'doc' => doc2}])
+      
+
+      db.load_document(['1'])
+      result = db.load_document(['1', '2'])
+
+      expect(result).to eql([doc1, doc2])
+    end
+
+    it 'does not cache documents that do not respond to id' do
+      doc1 = {
+        'id' => '1',
+      }
+      doc2 = {
+        'id' => '2',
+      }
+      allow(couchrest_db).to receive(:bulk_load).with(['1', '2'])
+        .and_return('rows' => [{'doc' => doc1}, {'doc' => doc1}])
+
+      db.load_document(['1', '2'])
+      db.load_document(['1', '2'])
+
+      expect(couchrest_db).to have_received(:bulk_load).with(['1', '2']).exactly(2).times
+    end
+  end
 
-    db.load_document ['1']
-    db.load_document ['1']
+  context 'when switching the database' do
   end
 
   it 'clears the cache when destroying a document via #destroy_document' do

data/spec/unit/database_spec.rb

@@ -72,8 +72,8 @@ describe CouchPotato::Database, 'load' do
   end
 
   context 'when several ids given' do
-    let(:doc1) { DbTestUser.new }
-    let(:doc2) { DbTestUser.new }
+    let(:doc1) { DbTestUser.new(id: '1') }
+    let(:doc2) { DbTestUser.new(id: '2') }
     let(:response) do
       { 'rows' => [{ 'doc' => nil }, { 'doc' => doc1 }, { 'doc' => doc2 }] }
     end
@@ -98,7 +98,7 @@ describe CouchPotato::Database, 'load' do
     end
 
     it 'does not write itself to a document that has no database= method' do
-      doc1 = double(:doc1)
+      doc1 = double(:doc1, id: '1')
       allow(doc1).to receive(:respond_to?) { false }
       allow(couchrest_db).to receive(:bulk_load) do
         { 'rows' => [{ 'doc' => doc1 }] }
@@ -116,7 +116,7 @@ describe CouchPotato::Database, 'load' do
     end
 
     it 'does not set database_collection on a document that has no database_collection= method' do
-      doc1 = double(:doc1)
+      doc1 = double(:doc1, id: '1')
       allow(doc1).to receive(:respond_to?) { false }
       allow(couchrest_db).to receive(:bulk_load) do
         { 'rows' => [{ 'doc' => doc1 }] }
@@ -571,6 +571,14 @@ describe CouchPotato::Database, '#switch_to' do
 
     expect(new_db.cache).to be_nil
   end
+
+  it 're-uses the original cache when switching back to the original database' do
+    db.cache = { key: 'value' }
+    new_db = db.switch_to('db2')
+    original_db = new_db.switch_to_default
+
+    expect(original_db.cache).to have_key(:key)
+  end
 end
 
 describe CouchPotato::Database, '#switch_to_default' do

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: couch_potato
 version: !ruby/object:Gem::Version
-  version: 1.14.0
+  version: 1.15.0
 platform: ruby
 authors:
 - Alexander Lang
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-02-25 00:00:00.000000000 Z
+date: 2024-03-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activemodel