add deleteContainerIfExists

also adds these operations:
- containerExists
- getContainerProperties
- databaseExists

Author: amine-mejaouel

samples/FSharp.CosmosDb.Samples/Program.fs

@@ -138,6 +138,13 @@ let main argv =
             |> Cosmos.deleteContainer
             |> Cosmos.execAsync
             |> Async.Ignore
+            
+        do!
+            conn
+            |> Cosmos.container "Family"
+            |> Cosmos.deleteContainerIfExists
+            |> Cosmos.execAsync
+            |> Async.Ignore
 
         return 0 // return an integer exit code
     }

src/FSharp.CosmosDb/Cosmos.fs

@@ -54,6 +54,10 @@ module Cosmos =
 
     let parameters arr op =
         { op with QueryOp.Parameters = op.Parameters @ arr }
+        
+    // --- DATABASE EXISTS --- //
+    let databaseExists<'T> op =
+        { CheckIfDatabaseExistsOp.Connection = op }
 
     // --- INSERT --- //
 
@@ -85,12 +89,25 @@ module Cosmos =
         { DeleteItemOp.Connection = op
           Id = id
           PartitionKey = partitionKey }
+        
+    // --- GET CONTAINER PROPERTIES --- //
+    let getContainerProperties op =
+        { GetContainerPropertiesOp.Connection = op }
+        
+    // --- CONTAINER EXISTS --- //
+    let containerExists op =
+        { CheckIfContainerExistsOp.Connection = op }
 
     // --- DELETE CONTAINER --- //
 
     let deleteContainer<'T> op : DeleteContainerOp<'T> =
         { DeleteContainerOp.Connection = op }
 
+    // --- DELETE CONTAINER IF EXISTS --- //
+
+    let deleteContainerIfExists op : DeleteContainerIfExistsOp =
+        { DeleteContainerIfExistsOp.Connection = op }
+
     // --- READ --- //
 
     let read id partitionKey op =
@@ -220,10 +237,14 @@ module Cosmos =
 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.execCheckIfDatabaseExists 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.execGetContainerProperties Cosmos.getClient op
+    static member execAsync op = OperationHandling.execCheckIfContainerExists Cosmos.getClient op
     static member execAsync op = OperationHandling.execDeleteContainer Cosmos.getClient op
+    static member execAsync op = OperationHandling.execDeleteContainerIfExists 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

@@ -66,6 +66,29 @@ let execQueryBatch (getClient: ConnectionOperation -> CosmosClient) (op: QueryOp
     | Some result -> result |> AsyncSeq.ofAsyncFeedIterator
     | None ->
         failwith "Unable to construct a query as some values are missing across the database, container name and query"
+        
+let execCheckIfDatabaseExists (getClient: ConnectionOperation -> CosmosClient) (op: CheckIfDatabaseExistsOp) =
+    let connInfo = op.Connection
+    let client = getClient connInfo
+    
+    use iterator = client.GetDatabaseQueryIterator<DatabaseProperties>()
+    
+    match connInfo.DatabaseId with
+    | Some databaseId ->
+            iterator
+            |> AsyncSeq.unfold (fun t ->
+                if iterator.HasMoreResults then
+                    Some (iterator.ReadNextAsync(), iterator)
+                else
+                    None)
+            |> AsyncSeq.collect (fun i ->
+                asyncSeq {
+                    let! c = i |> Async.AwaitTask
+                    for x in c do
+                        yield x
+                })
+            |> AsyncSeq.exists (fun i -> i.Id = databaseId)
+    | None -> failwith "failed to check if database exists"
 
 let execInsert (getClient: ConnectionOperation -> CosmosClient) (op: InsertOp<'T>) =
     let connInfo = op.Connection
@@ -217,6 +240,45 @@ let execDeleteItem (getClient: ConnectionOperation -> CosmosClient) (op: DeleteI
 
     | None -> failwith "Unable to read from the container to get the item for updating"
 
+let execGetContainerProperties (getClient: ConnectionOperation -> CosmosClient) (op: GetContainerPropertiesOp) =
+    let connInfo = op.Connection
+    let client = getClient connInfo
+    
+    let containerName =
+        match connInfo.ContainerName with
+        | None -> failwith "ContainerName is not provided"
+        | Some containerName -> containerName
+    
+    use iterator =
+        match connInfo.DatabaseId with
+        | None -> failwith "DatabaseId is not provided"
+        | Some databaseId ->
+            client
+                .GetDatabase(databaseId)
+                .GetContainerQueryIterator<ContainerProperties>()
+    
+    iterator
+    |> AsyncSeq.unfold (fun t ->
+        if iterator.HasMoreResults then
+            Some (iterator.ReadNextAsync(), iterator)
+        else
+            None)
+    |> AsyncSeq.collect (fun i ->
+        asyncSeq {
+            let! c = i |> Async.AwaitTask
+            for x in c do
+                yield x
+        })
+    |> AsyncSeq.tryFind (fun i -> i.Id = containerName)
+    
+let execCheckIfContainerExists (getClient: ConnectionOperation -> CosmosClient) (op: CheckIfContainerExistsOp) =
+    async {
+        let! containerProperties = execGetContainerProperties getClient { Connection= op.Connection }
+        
+        return containerProperties
+               |> Option.isSome
+    }
+    
 let execDeleteContainer (getClient: ConnectionOperation -> CosmosClient) (op: DeleteContainerOp<'T>) =
     let connInfo = op.Connection
     let client = getClient connInfo
@@ -238,6 +300,15 @@ let execDeleteContainer (getClient: ConnectionOperation -> CosmosClient) (op: De
     match result with
     | Some result -> result
     | None -> failwith "Unable to delete container"
+
+let execDeleteContainerIfExists (getClient: ConnectionOperation -> CosmosClient) (op: DeleteContainerIfExistsOp) =
+    async {
+        let! databaseExists = execCheckIfDatabaseExists getClient { Connection= op.Connection }
+        let! containerExists = execCheckIfContainerExists getClient { Connection= op.Connection }
+        if databaseExists && containerExists then
+            do! execDeleteContainer getClient { Connection= op.Connection }
+                |> Async.Ignore
+    }
 
 let execRead (getClient: ConnectionOperation -> CosmosClient) (op: ReadOp<'T>) =
     let connInfo = op.Connection

src/FSharp.CosmosDb/Types.fs

@@ -87,6 +87,9 @@ type QueryOp<'T> =
       Query: string option
       Parameters: (string * obj) list }
 
+type CheckIfDatabaseExistsOp =
+    { Connection: ConnectionOperation }
+
 type InsertOp<'T> =
     { Connection: ConnectionOperation
       Values: 'T list }
@@ -106,9 +109,18 @@ type DeleteItemOp<'T> =
       Id: string
       PartitionKey: string }
 
+type GetContainerPropertiesOp = 
+    { Connection: ConnectionOperation }
+    
+type CheckIfContainerExistsOp =
+    { Connection: ConnectionOperation }
+    
 type DeleteContainerOp<'T> =
     { Connection: ConnectionOperation }
 
+type DeleteContainerIfExistsOp =
+    { Connection: ConnectionOperation }
+    
 type ReadOp<'T> =
     { Connection: ConnectionOperation
       Id: string