Connect-CosmosDB.ps1
|
function Connect-CosmosDB { <# .SYNOPSIS Connects to a CosmosDB .DESCRIPTION Connects and sets a script wide connection variable hashtable that is used by the other functins in this module Follow the instructions here to create a CosmosDB (or do some fancy Azure Powershell stuff :-)): https://docs.microsoft.com/en-us/azure/cosmos-db/create-documentdb-dotnet .PARAMETER URI Something like: https://<Your DB Name>.documents.azure.com .PARAMETER Key The (primary) key, a guid formattede like: 9bc7fb04-2992-4033-844f-139eb9c2fe93, for your CosmosDB .PARAMETER KeyType master or resource - depending on your setup .PARAMETER tokenVersion 1.0 - until further notice from Microsoft .EXAMPLE Connect-CosmosDB -URI https://MyPrivateCosmosDB.documents.azure.com -Key 9bc7fb04-2992-4033-844f-139eb9c2fe93 You do not need to set the variables KeyType if you use the masterkey. Likewise with tokenVersion .NOTES This cmdlet should be run before you run the other cmdlets in this module, and every time you need to access a new database account (URI). #> [CmdletBinding()] param ( [Parameter(Mandatory=$True, HelpMessage='https://<Your DB Name>.documents.azure.com')] [string]$URI, [Parameter(Mandatory=$True, HelpMessage="DocumentDB Account key")] [string]$Key, [ValidateSet('master','resource')] [string]$KeyType = 'master', [string]$tokenVersion = '1.0' ) begin { $URI = $URI.Trim().TrimEnd('/') #Setup CosmosDBVariables $PSParameters = 'Key','URI','KeyType','tokenVersion' $Script:CosmosDBVariables = @{} foreach ($ParameterName in $PSParameters) { $Value = Get-Variable $ParameterName | Select-Object -expand value Write-Verbose "Setting $ParameterName to $Value in CosmosDBVariables" $Script:CosmosDBVariables[$ParameterName] = $Value } $Header = New-CosmosDBHeader -Verb GET -resourceType dbs $dbsUri = '{0}/dbs' -f $URI try { $Databases = Invoke-RestMethod -Uri $dbsUri -Headers $Header -Method GET -ContentType "application/json" } catch { Write-Warning -Message $_.Exception.Message continue } } process { $script:CosmosDBConnection=@{} foreach ($Database in $Databases.Databases) { Write-Verbose "Adding Database $($Database.id) to CosmosDBConnection" $script:CosmosDBConnection[$($Database.id + '_db')] = $Database # Get all Collections $CollsUri = '{0}/{1}/colls' -f $URI,$Database._self $header = New-CosmosDBHeader -resourceId $Database._rid -resourceType colls -Verb get $Collections = Invoke-RestMethod -Uri $CollsUri -Headers $Header -Method get -ContentType "application/json" $CollHash = @{} foreach ($Collection in $Collections.DocumentCollections) { Write-Verbose "Adding Collection $($Collection.id) to CosmosDBConnection" $CollHash[$Collection.id] = $Collection } $script:CosmosDBConnection[$Database.id] = $CollHash } } end { } } |