add deleteContainer operation

Author: amine-mejaouel

src/FSharp.CosmosDb/Cosmos.fs

@@ -47,89 +47,72 @@ module Cosmos =
           Query = None
           Parameters = [] }
 
-    let query<'T> query op : ContainerOperation<'T> =
-        Query
-            { defaultQueryOp () with
-                  Query = Some query
-                  Connection = op }
+    let query<'T> query op : QueryOp<'T> =
+        { defaultQueryOp () with
+              Query = Some query
+              Connection = op }
 
     let parameters arr op =
-        match op with
-        | Query q ->
-            Query
-                { q with
-                      Parameters = q.Parameters @ arr }
-        | _ -> failwith "Only the Query discriminated union supports parameters"
+        { op with QueryOp.Parameters = op.Parameters @ arr }
 
     // --- INSERT --- //
 
     let insertMany<'T> (values: 'T list) op =
-        Insert { Connection = op; Values = values }
+        { InsertOp.Connection = op; Values = values }
 
     let insert<'T> (value: 'T) op =
-        Insert { Connection = op; Values = [ value ] }
+        { InsertOp.Connection = op; Values = [ value ] }
 
     // --- INSERT --- //
 
     let upsertMany<'T> (values: 'T list) op =
-        Upsert { Connection = op; Values = values }
+        { UpsertOp.Connection = op; Values = values }
 
     let upsert<'T> (value: 'T) op =
-        Upsert { Connection = op; Values = [ value ] }
+        { UpsertOp.Connection = op; Values = [ value ] }
 
     // --- UPDATE --- //
 
     let update<'T> id partitionKey (updater: 'T -> 'T) op =
-        Update
-            { Connection = op
-              Id = id
-              PartitionKey = partitionKey
-              Updater = updater }
+        { UpdateOp.Connection = op
+          Id = id
+          PartitionKey = partitionKey
+          Updater = updater }
 
-    // --- DELETE --- //
+    // --- DELETE ITEM --- //
 
-    let delete<'T> id partitionKey op =
-        Delete
-            { Connection = op
-              Id = id
-              PartitionKey = partitionKey }
+    let deleteItem<'T> id partitionKey op =
+        { DeleteItemOp.Connection = op
+          Id = id
+          PartitionKey = partitionKey }
+            
+    // --- DELETE CONTAINER --- //
+
+    let deleteContainer<'T> op : DeleteContainerOp<'T> =
+        { DeleteContainerOp.Connection = op }
 
     // --- READ --- //
 
     let read id partitionKey op =
-        Read
-            { Connection = op
-              Id = id
-              PartitionKey = partitionKey }
+        { ReadOp.Connection = op
+          Id = id
+          PartitionKey = partitionKey }
 
     // --- REPLACE --- //
 
     let replace<'T> (item: 'T) op =
-        Replace { Connection = op; Item = item }
+        { ReplaceOp.Connection = op; Item = item }
 
     // --- Execute --- //
 
     let private getClient (connInfo: ConnectionOperation) = connInfo.GetClient()
 
     let dispose (connInfo: ConnectionOperation) = (connInfo :> IDisposable).Dispose()
 
-    let execAsync<'T> (op: ContainerOperation<'T>) =
-        match op with
-        | Query op -> OperationHandling.execQuery getClient op
-        | Insert op -> OperationHandling.execInsert getClient op
-        | Update op -> OperationHandling.execUpdate getClient op
-        | Delete op -> OperationHandling.execDelete getClient op
-        | Upsert op -> OperationHandling.execUpsert getClient op
-        | Read op -> OperationHandling.execRead getClient op
-        | Replace op -> OperationHandling.execReplace getClient op
-
-    let execBatchAsync<'T> batchSize (op: ContainerOperation<'T>) =
-        match op with
-        | Query op ->
-            let queryOps = QueryRequestOptions()
-            queryOps.MaxItemCount <- batchSize
-            OperationHandling.execQueryBatch getClient op queryOps
-        | _ -> failwith "Batch return operation only supported with query operations, use `execAsync` instead."
+    let execBatchAsync<'T> batchSize op =
+        let queryOps = QueryRequestOptions()
+        queryOps.MaxItemCount <- batchSize
+        OperationHandling.execQueryBatch getClient op queryOps
 
     // --- Access Cosmos APIs directly --- //
 
@@ -233,3 +216,14 @@ module Cosmos =
                 processor.Build()
             | None ->
                 failwith "Unable to connect the change feed. Ensure the container and lease container info is all set"
+                
+type Cosmos =
+    static member private getClient (connInfo: ConnectionOperation) = connInfo.GetClient()
+    static member execAsync (op: QueryOp<'T>) = OperationHandling.execQuery Cosmos.getClient op
+    static member execAsync op = OperationHandling.execInsert Cosmos.getClient op
+    static member execAsync op = OperationHandling.execUpdate Cosmos.getClient op
+    static member execAsync op = OperationHandling.execDeleteItem Cosmos.getClient op
+    static member execAsync op = OperationHandling.execDeleteContainer Cosmos.getClient op
+    static member execAsync op = OperationHandling.execUpsert Cosmos.getClient op
+    static member execAsync op = OperationHandling.execRead Cosmos.getClient op
+    static member execAsync op = OperationHandling.execReplace Cosmos.getClient op

src/FSharp.CosmosDb/OperationHandling.fs

@@ -189,7 +189,7 @@ let execUpdate (getClient: ConnectionOperation -> CosmosClient) (op: UpdateOp<'T
 
     | None -> failwith "Unable to read from the container to get the item for updating"
 
-let execDelete (getClient: ConnectionOperation -> CosmosClient) (op: DeleteOp<'T>) =
+let execDeleteItem (getClient: ConnectionOperation -> CosmosClient) (op: DeleteItemOp<'T>) =
     let connInfo = op.Connection
     let client = getClient connInfo
 
@@ -216,7 +216,29 @@ let execDelete (getClient: ConnectionOperation -> CosmosClient) (op: DeleteOp<'T
         |> AsyncSeq.ofSeqAsync
 
     | None -> failwith "Unable to read from the container to get the item for updating"
+    
+let execDeleteContainer (getClient: ConnectionOperation -> CosmosClient) (op: DeleteContainerOp<'T>) =
+    let connInfo = op.Connection
+    let client = getClient connInfo
+
+    let result =
+        maybe {
+            let! databaseId = connInfo.DatabaseId
+            let! containerName = connInfo.ContainerName
+
+            let db = client.GetDatabase databaseId
+
+            let container = db.GetContainer containerName
+
+            return
+                container.DeleteContainerAsync ()
+                |> Async.AwaitTask
+        }
 
+    match result with
+    | Some result -> result
+    | None -> failwith "Unable to delete container"
+    
 let execRead (getClient: ConnectionOperation -> CosmosClient) (op: ReadOp<'T>) =
     let connInfo = op.Connection
     let client = getClient connInfo

src/FSharp.CosmosDb/Types.fs

@@ -1,5 +1,6 @@
 namespace FSharp.CosmosDb
 
+open FSharp.CosmosDb
 open Microsoft.Azure.Cosmos
 open System.Threading
 open System.Threading.Tasks
@@ -100,10 +101,13 @@ type UpdateOp<'T> =
       PartitionKey: string
       Updater: 'T -> 'T }
 
-type DeleteOp<'T> =
+type DeleteItemOp<'T> =
     { Connection: ConnectionOperation
       Id: string
       PartitionKey: string }
+    
+type DeleteContainerOp<'T> =
+    { Connection: ConnectionOperation }
 
 type ReadOp<'T> =
     { Connection: ConnectionOperation
@@ -114,15 +118,6 @@ type ReplaceOp<'T> =
     { Connection: ConnectionOperation
       Item: 'T }
 
-type ContainerOperation<'T> =
-    | Query of QueryOp<'T>
-    | Insert of InsertOp<'T>
-    | Update of UpdateOp<'T>
-    | Delete of DeleteOp<'T>
-    | Upsert of UpsertOp<'T>
-    | Read of ReadOp<'T>
-    | Replace of ReplaceOp<'T>
-
 type ChangeFeedOptions<'T> =
     { Connection: ConnectionOperation
       Processor: string