act_as_fire_record_beta 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7fa5ffce208135ba537f2eba3ae37caf239e6b49446f2d31c77668263ec178c3
-  data.tar.gz: c4705cc8018288e15a766bd7efde203fdb03d701db208a66cbb46409732b1077
+  metadata.gz: 631c8a9418186db9bba40eeed7d122996c161c6f44bba58a230840e3ee063b73
+  data.tar.gz: 82a5ae9c0f6aa62920cdba71fb8e31f279aea21fdb8c989b297f15e04e1e06d7
 SHA512:
-  metadata.gz: 382202705f5383d954bfd336137b6d74e0bf66879a3a07e8a480e4ec3e1d9427bd35e5d23180a929c39c74ab8d7d2607eff70802a8f8c86f6825a7c6e3fedcfc
-  data.tar.gz: '01158cd356bae1b962f62ba6575d52fb9e5648d83dc501ab8e3d0aaf2c19bffe6a5f1501aa17206a3f6f8f71a3f841370b5f7295aa8c19a07faecc249de319ab'
+  metadata.gz: acab4a53dd1d91888490a537e4b85234c451ac2c57af072a2cb383a63cd7009db6f7130ceb8d2b710b9f012f660727873c897419bd64e98c57e8003cf6118418
+  data.tar.gz: c3b124d95d7e1dab694fcb86fb83ed2e90efc4502121a6d87e4263b7b7bf7590f0a1505c47e7d08b6cff49230d1ea40d26eadf98a906a3ab55cc1d9ccf8e7ad3

data/lib/act_as_fire_record_beta/google/cloud/firestore/query.rb

@@ -5,13 +5,13 @@ module Google
         def get_records(limit: nil)
           scope = limit ? limit(limit) : self
           scope.get.map do |data|
-            model_lass = ActAsFireRecordBeta.class_mapping[query.from[0].collection_id]
-            model_lass.to_instance(data)
+            fire_record_class.to_instance(data)
           end
         end
 
         def destroy_all
-          get_records.each(&:destroy)
+          doc_refs = get_records.map(&:doc_ref)
+          fire_record_class.delete_in_batch(doc_refs)
         end
 
         def first(limit = 1)
@@ -22,6 +22,10 @@ module Google
         def exists?
           !!first
         end
+
+        def fire_record_class
+          @_fire_record_class ||= ActAsFireRecordBeta.class_mapping[query.from[0].collection_id]
+        end
       end
     end
   end

data/lib/act_as_fire_record_beta/version.rb

@@ -1,3 +1,3 @@
 module ActAsFireRecordBeta
-  VERSION = "0.0.2"
+  VERSION = "0.0.4"
 end

data/lib/act_as_fire_record_beta.rb

@@ -23,6 +23,8 @@ module ActAsFireRecordBeta
     attribute :created_at, :time
     attribute :updated_at, :time
 
+    attr_accessor :doc_ref
+
     ActAsFireRecordBeta.class_mapping[collection_name_with_env] = self
   end
 
@@ -30,9 +32,16 @@ module ActAsFireRecordBeta
     extend Forwardable
     delegate :logger => Rails
 
-    def firestore_attributes(*names)
+    # Copied from activemodel-7.0.4/lib/active_model/attributes.rb
+    NO_DEFAULT_PROVIDED = Object.new
+
+    def firestore_attribute(name, cast_type = nil, default: NO_DEFAULT_PROVIDED, **options)
+      attribute(name, cast_type, default: default, **options)
+      firestore_attributes << name.to_sym
+    end
+
+    def firestore_attributes
       @_firestore_attributes ||= []
-      @_firestore_attributes += names.map(&:to_sym)
     end
 
     def col
@@ -91,7 +100,20 @@ module ActAsFireRecordBeta
     end
 
     def destroy_all
-      col.list_documents.each(&:delete)
+      delete_in_batch(col.list_documents)
+    end
+
+    def delete_in_batch(doc_refs)
+      max_batch_size = 500
+      nested_doc_refs = doc_refs.each_slice(max_batch_size)
+      firestore = Google::Cloud::Firestore.new
+      nested_doc_refs.each do |doc_refs|
+        firestore.batch do |b|
+          doc_refs.each do |doc_ref|
+            b.delete(doc_ref)
+          end
+        end
+      end
     end
 
     def save_params(record)
@@ -107,7 +129,7 @@ module ActAsFireRecordBeta
       @_firestore_attributes.each do |key|
         params[key] = data[key]
       end
-      new(params)
+      new(params).tap { |record| record.doc_ref = data }
     end
 
     private
@@ -165,6 +187,13 @@ module ActAsFireRecordBeta
     true
   end
 
+  # Override
+  def inspect
+    hash = attributes.transform_keys(&:to_sym)
+    hash_text = hash.to_s[1..-2]
+    "#<#{self.class} #{hash_text}>"
+  end
+
   private
 
   def save_params

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: act_as_fire_record_beta
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Junichi Ito
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-10-22 00:00:00.000000000 Z
+date: 2022-10-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -76,7 +76,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.7
+rubygems_version: 3.2.33
 signing_key:
 specification_version: 4
 summary: ActiveRecord like interface for Firestore