lib/rom/sql/extensions/postgres/commands.rb in rom-sql-1.3.1 vs lib/rom/sql/extensions/postgres/commands.rb in rom-sql-1.3.2

@@ -9,37 +9,37 @@
           # Executes insert statement and returns inserted tuples
           #
           # @api private
           def insert(tuples)
             dataset = tuples.map do |tuple|
-              relation.dataset.returning(*relation.columns).insert(tuple)
+              relation.dataset.returning.insert(tuple)
             end.flatten(1)
 
             wrap_dataset(dataset)
           end
 
           # Executes multi_insert statement and returns inserted tuples
           #
           # @api private
           def multi_insert(tuples)
-            relation.dataset.returning(*relation.columns).multi_insert(tuples)
+            relation.dataset.returning.multi_insert(tuples)
           end
 
           # Executes upsert statement (INSERT with ON CONFLICT clause)
           # and returns inserted/updated tuples
           #
           # @api private
           def upsert(tuple, opts = EMPTY_HASH)
-            relation.dataset.returning(*relation.columns).insert_conflict(opts).insert(tuple)
+            relation.dataset.returning.insert_conflict(opts).insert(tuple)
           end
         end
 
         module Update
           # Executes update statement and returns updated tuples
           #
           # @api private
           def update(tuple)
-            dataset = relation.dataset.returning(*relation.columns).update(tuple)
+            dataset = relation.dataset.returning.update(tuple)
             wrap_dataset(dataset)
           end
         end
 
         # Upsert command