chore: changed the specs to support aggregation pipeline

zookzook · Jul 30, 2023 · d7cb466 · d7cb466
1 parent f621a62
commit d7cb466
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/mongo.ex b/lib/mongo.ex
@@ -1051,7 +1051,7 @@ defmodule Mongo do
     * `:upsert` - if set to `true` creates a new document when no document
       matches the filter (default: `false`)
   """
-  @spec update_one(GenServer.server(), collection, BSON.document(), BSON.document(), Keyword.t()) ::
+  @spec update_one(GenServer.server(), collection, BSON.document(), [BSON.document()] | BSON.document(), Keyword.t()) ::
           result(Mongo.UpdateResult.t())
   def update_one(topology_pid, coll, filter, update, opts \\ []) do
     _ = modifier_docs(update, :update)
@@ -1061,7 +1061,7 @@ defmodule Mongo do
   @doc """
   Similar to `update_one/5` but unwraps the result and raises on error.
   """
-  @spec update_one!(GenServer.server(), collection, BSON.document(), BSON.document(), Keyword.t()) ::
+  @spec update_one!(GenServer.server(), collection, BSON.document(), [BSON.document()] | BSON.document(), Keyword.t()) ::
           result!(Mongo.UpdateResult.t())
   def update_one!(topology_pid, coll, filter, update, opts \\ []) do
     bangify(update_one(topology_pid, coll, filter, update, opts))
@@ -1074,7 +1074,7 @@ defmodule Mongo do
   please refer to the [MongoDB documentation](https://docs.mongodb.com/manual/reference/command/update/#dbcmd.update)
 
   """
-  @spec update_many(GenServer.server(), collection, BSON.document(), BSON.document(), Keyword.t()) ::
+  @spec update_many(GenServer.server(), collection, BSON.document(), [BSON.document()] | BSON.document(), Keyword.t()) ::
           result(Mongo.UpdateResult.t())
   def update_many(topology_pid, coll, filter, update, opts \\ []) do
     _ = modifier_docs(update, :update)
@@ -1088,7 +1088,7 @@ defmodule Mongo do
           GenServer.server(),
           collection,
           BSON.document(),
-          BSON.document(),
+          [BSON.document()] | BSON.document(),
           Keyword.t()
         ) :: result!(Mongo.UpdateResult.t())
   def update_many!(topology_pid, coll, filter, update, opts \\ []) do