lib/rom/sql/extensions/postgres/commands.rb in rom-sql-1.2.2 vs lib/rom/sql/extensions/postgres/commands.rb in rom-sql-1.3.0

@@ -8,13 +8,15 @@
         module Create
           # Executes insert statement and returns inserted tuples
           #
           # @api private
           def insert(tuples)
-            tuples.map do |tuple|
+            dataset = tuples.map do |tuple|
               relation.dataset.returning(*relation.columns).insert(tuple)
             end.flatten(1)
+
+            wrap_dataset(dataset)
           end
 
           # Executes multi_insert statement and returns inserted tuples
           #
           # @api private
@@ -34,10 +36,11 @@
         module Update
           # Executes update statement and returns updated tuples
           #
           # @api private
           def update(tuple)
-            relation.dataset.returning(*relation.columns).update(tuple)
+            dataset = relation.dataset.returning(*relation.columns).update(tuple)
+            wrap_dataset(dataset)
           end
         end
 
         # Upsert command
         #