add reset association

Mapotempo · May 28, 2024 · 9355151 · 9355151
1 parent cff1218
commit 9355151
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/lib/couchbase-orm/associations.rb b/lib/couchbase-orm/associations.rb
@@ -53,6 +53,10 @@ def belongs_to(name, **options)
 
                     instance_variable_set(instance_var, value)
                 end
+
+                define_method(:"#{name}_reset") do
+                    remove_instance_variable(instance_var) if instance_variable_defined?(instance_var)
+                end
             end
 
             def has_and_belongs_to_many(name, **options)
@@ -106,6 +110,10 @@ def has_and_belongs_to_many(name, **options)
                     instance_variable_set(instance_var, value)
                 end
 
+                define_method(:"#{name}_reset") do
+                    self.remove_instance_variable(instance_var) if self.instance_variable_defined?(instance_var)
+                end
+
                 return unless options[:autosave]
 
                 save_method = :"autosave_associated_records_for_#{name}"

diff --git a/lib/couchbase-orm/utilities/has_many.rb b/lib/couchbase-orm/utilities/has_many.rb
@@ -58,6 +58,10 @@ class #{class_name} < CouchbaseOrm::Base
                 end
             end
 
+            define_method(:"#{model}_reset") do
+                self.remove_instance_variable(instance_var) if self.instance_variable_defined?(instance_var)
+            end
+
             @associations ||= []
             @associations << [model, options[:dependent]]
         end