public override void ExecuteCmdlet()
{
PSVirtualNetworkRule virtualNetworkRule = new PSVirtualNetworkRule
{
Id = Id,
IgnoreMissingVNetServiceEndpoint = IgnoreMissingVNetServiceEndpoint
};
WriteObject(virtualNetworkRule);
return;
}
public override void ExecuteCmdlet()
{
IPSNetworkRule rule = null;
switch (this.ParameterSetName)
{
case ByVirtualNetworkRule:
rule = new PSVirtualNetworkRule(VirtualNetworkResourceId, Action);
break;
case ByIPRule:
rule = new PSIPRule(IPAddressOrRange, Action);
break;
default:
throw new PSArgumentException("Incorrect argument provided");
}
WriteObject(rule);
}
public override void ExecuteCmdlet()
{
DatabaseAccountGetResults databaseAccountGetResults = null;
try
{
databaseAccountGetResults = CosmosDBManagementClient.DatabaseAccounts.Get(ResourceGroupName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode != System.Net.HttpStatusCode.NotFound)
{
throw;
}
}
if (databaseAccountGetResults != null)
{
throw new ConflictingResourceException(message: string.Format(ExceptionMessage.Conflict, Name));
}
ConsistencyPolicy consistencyPolicy = base.PopoulateConsistencyPolicy(DefaultConsistencyLevel, MaxStalenessIntervalInSeconds, MaxStalenessPrefix);
string writeLocation = null;
Collection <Location> LocationCollection = new Collection <Location>();
if (Location != null && Location.Length > 0)
{
int failoverPriority = 0;
foreach (string l in Location)
{
Location loc = new Location(locationName: l, failoverPriority: failoverPriority);
LocationCollection.Add(loc);
if (failoverPriority == 0)
{
writeLocation = l;
}
failoverPriority++;
}
}
else if (LocationObject != null && LocationObject.Length > 0)
{
if (writeLocation != null)
{
WriteWarning("Cannot accept Location and LocationObject simultaneously as parameters");
return;
}
foreach (PSLocation psLocation in LocationObject)
{
LocationCollection.Add(PSLocation.ToSDKModel(psLocation));
if (psLocation.FailoverPriority == 0)
{
writeLocation = psLocation.LocationName;
}
}
}
Dictionary <string, string> tags = new Dictionary <string, string>();
if (Tag != null)
{
tags = base.PopulateTags(Tag);
}
Collection <VirtualNetworkRule> virtualNetworkRule = new Collection <VirtualNetworkRule>();
if (VirtualNetworkRule != null && VirtualNetworkRule.Length > 0)
{
foreach (string id in VirtualNetworkRule)
{
virtualNetworkRule.Add(new VirtualNetworkRule(id: id));
}
}
if (VirtualNetworkRuleObject != null && VirtualNetworkRuleObject.Length > 0)
{
foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
{
virtualNetworkRule.Add(PSVirtualNetworkRule.ToSDKModel(psVirtualNetworkRule));
}
}
DatabaseAccountCreateUpdateParameters databaseAccountCreateUpdateParameters = new DatabaseAccountCreateUpdateParameters(locations: LocationCollection, location: writeLocation, name: Name, tags: tags);
if (FromPointInTimeBackup)
{
PSRestoreParameters restoreParameters = restoreContext.GetRestoreParameters(CosmosDBManagementClient);
if (restoreParameters == null)
{
WriteWarning("Please provide valid parameters to restore");
return;
}
databaseAccountCreateUpdateParameters.CreateMode = CreateMode.Restore;
databaseAccountCreateUpdateParameters.RestoreParameters = restoreParameters.ToSDKModel();
}
databaseAccountCreateUpdateParameters.ConsistencyPolicy = consistencyPolicy;
databaseAccountCreateUpdateParameters.EnableMultipleWriteLocations = EnableMultipleWriteLocations;
databaseAccountCreateUpdateParameters.IsVirtualNetworkFilterEnabled = EnableVirtualNetwork;
databaseAccountCreateUpdateParameters.EnableAutomaticFailover = EnableAutomaticFailover;
databaseAccountCreateUpdateParameters.VirtualNetworkRules = virtualNetworkRule;
databaseAccountCreateUpdateParameters.DisableKeyBasedMetadataWriteAccess = DisableKeyBasedMetadataWriteAccess;
databaseAccountCreateUpdateParameters.PublicNetworkAccess = PublicNetworkAccess;
databaseAccountCreateUpdateParameters.EnableFreeTier = EnableFreeTier;
databaseAccountCreateUpdateParameters.EnableAnalyticalStorage = EnableAnalyticalStorage;
Collection <string> networkAclBypassResourceId = NetworkAclBypassResourceId != null ? new Collection <string>(NetworkAclBypassResourceId) : new Collection <string>();
databaseAccountCreateUpdateParameters.NetworkAclBypassResourceIds = networkAclBypassResourceId;
if (IpRule != null && IpRule.Length > 0)
{
databaseAccountCreateUpdateParameters.IpRules = base.PopulateIpRules(IpRule);
}
if (KeyVaultKeyUri != null)
{
databaseAccountCreateUpdateParameters.KeyVaultKeyUri = KeyVaultKeyUri;
}
if (NetworkAclBypass != null)
{
databaseAccountCreateUpdateParameters.NetworkAclBypass =
NetworkAclBypass == "AzureServices" ? SDKModel.NetworkAclBypass.AzureServices : SDKModel.NetworkAclBypass.None;
}
if (!string.IsNullOrEmpty(ApiKind))
{
if (ApiKind.Equals("MongoDB", StringComparison.OrdinalIgnoreCase))
{
if (ServerVersion != null)
{
databaseAccountCreateUpdateParameters.ApiProperties = new ApiProperties
{
ServerVersion = ServerVersion
};
}
}
else
{
switch (ApiKind)
{
case "Cassandra":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableCassandra"
}
};
break;
case "Gremlin":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableGremlin"
}
};
break;
case "Table":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableTable"
}
};
break;
case "Sql":
break;
}
ApiKind = null;
}
}
else
{
ApiKind = "GlobalDocumentDB";
}
databaseAccountCreateUpdateParameters.Kind = ApiKind;
if (!string.IsNullOrEmpty(BackupPolicyType))
{
PSBackupPolicy backupPolicy = new PSBackupPolicy()
{
BackupType = BackupPolicyType,
BackupIntervalInMinutes = BackupIntervalInMinutes,
BackupRetentionIntervalInHours = BackupRetentionIntervalInHours,
BackupStorageRedundancy = BackupStorageRedundancy
};
if (BackupPolicyType.Equals(PSBackupPolicy.ContinuousModeBackupType, StringComparison.OrdinalIgnoreCase) &&
(BackupIntervalInMinutes.HasValue || BackupRetentionIntervalInHours.HasValue))
{
WriteWarning("Cannot accept BackupInterval or BackupRetention parameters for ContinuousModeBackupType");
return;
}
databaseAccountCreateUpdateParameters.BackupPolicy = backupPolicy.ToSDKModel();
}
else if (BackupIntervalInMinutes.HasValue || BackupRetentionIntervalInHours.HasValue || !string.IsNullOrEmpty(BackupStorageRedundancy))
{
databaseAccountCreateUpdateParameters.BackupPolicy = new PeriodicModeBackupPolicy()
{
PeriodicModeProperties = new PeriodicModeProperties()
{
BackupIntervalInMinutes = BackupIntervalInMinutes,
BackupRetentionIntervalInHours = BackupRetentionIntervalInHours,
BackupStorageRedundancy = BackupStorageRedundancy
}
};
}
// Update analytical storage schema type.
databaseAccountCreateUpdateParameters.AnalyticalStorageConfiguration = CreateAnalyticalStorageConfiguration(AnalyticalStorageSchemaType);
if (ShouldProcess(Name, "Creating Database Account"))
{
DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.CreateOrUpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSDatabaseAccountGetResults(cosmosDBAccount));
}
return;
}
public override void ExecuteCmdlet()
{
if (!ParameterSetName.Equals(NameParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = null;
if (ParameterSetName.Equals(ResourceIdParameterSet, StringComparison.Ordinal))
{
resourceIdentifier = new ResourceIdentifier(ResourceId);
}
else if (ParameterSetName.Equals(ObjectParameterSet, StringComparison.Ordinal))
{
resourceIdentifier = new ResourceIdentifier(InputObject.Id);
}
ResourceGroupName = resourceIdentifier.ResourceGroupName;
Name = resourceIdentifier.ResourceName;
}
DatabaseAccountGetResults readDatabase = CosmosDBManagementClient.DatabaseAccounts.GetWithHttpMessagesAsync(ResourceGroupName, Name).GetAwaiter().GetResult().Body;
DatabaseAccountUpdateParameters databaseAccountUpdateParameters = new DatabaseAccountUpdateParameters(locations: readDatabase.Locations, location: readDatabase.WriteLocations.ElementAt(0).LocationName);
if (EnableMultipleWriteLocations != null)
{
databaseAccountUpdateParameters.EnableMultipleWriteLocations = EnableMultipleWriteLocations;
}
if (EnableVirtualNetwork != null)
{
databaseAccountUpdateParameters.IsVirtualNetworkFilterEnabled = EnableVirtualNetwork;
}
if (EnableAutomaticFailover != null)
{
databaseAccountUpdateParameters.EnableAutomaticFailover = EnableAutomaticFailover;
}
if (DisableKeyBasedMetadataWriteAccess != null)
{
databaseAccountUpdateParameters.DisableKeyBasedMetadataWriteAccess = DisableKeyBasedMetadataWriteAccess;
}
if (PublicNetworkAccess != null)
{
databaseAccountUpdateParameters.PublicNetworkAccess = PublicNetworkAccess;
}
if (KeyVaultKeyUri != null)
{
databaseAccountUpdateParameters.KeyVaultKeyUri = KeyVaultKeyUri;
}
if (EnableAnalyticalStorage != null)
{
databaseAccountUpdateParameters.EnableAnalyticalStorage = EnableAnalyticalStorage;
}
if (NetworkAclBypass != null)
{
databaseAccountUpdateParameters.NetworkAclBypass =
NetworkAclBypass == "AzureServices" ? SDKModel.NetworkAclBypass.AzureServices : SDKModel.NetworkAclBypass.None;
}
if (!string.IsNullOrEmpty(DefaultConsistencyLevel))
{
databaseAccountUpdateParameters.ConsistencyPolicy = base.PopoulateConsistencyPolicy(DefaultConsistencyLevel, MaxStalenessIntervalInSeconds, MaxStalenessPrefix);
}
if (Tag != null)
{
databaseAccountUpdateParameters.Tags = base.PopulateTags(Tag);
}
if (VirtualNetworkRule != null || VirtualNetworkRuleObject != null)
{
Collection <VirtualNetworkRule> virtualNetworkRule = new Collection <VirtualNetworkRule>();
if (VirtualNetworkRule != null && VirtualNetworkRule.Length > 0)
{
foreach (string id in VirtualNetworkRule)
{
virtualNetworkRule.Add(new VirtualNetworkRule(id: id));
}
}
if (VirtualNetworkRuleObject != null && VirtualNetworkRuleObject.Length > 0)
{
foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
{
virtualNetworkRule.Add(PSVirtualNetworkRule.ToSDKModel(psVirtualNetworkRule));
}
}
databaseAccountUpdateParameters.VirtualNetworkRules = virtualNetworkRule;
}
if (IpRule != null)
{
// not checking IpRules.Length > 0, to handle the removal of IpRules case
databaseAccountUpdateParameters.IpRules = base.PopulateIpRules(IpRule);
}
if (ServerVersion != null)
{
if (databaseAccountUpdateParameters.ApiProperties == null)
{
databaseAccountUpdateParameters.ApiProperties = new ApiProperties();
}
databaseAccountUpdateParameters.ApiProperties.ServerVersion = ServerVersion;
}
if (NetworkAclBypassResourceId != null)
{
Collection <string> networkAclBypassResourceId = new Collection <string>(NetworkAclBypassResourceId);
databaseAccountUpdateParameters.NetworkAclBypassResourceIds = networkAclBypassResourceId;
}
if (BackupIntervalInMinutes.HasValue || BackupRetentionIntervalInHours.HasValue || !string.IsNullOrEmpty(BackupStorageRedundancy))
{
if (BackupPolicyType == "Continuous")
{
WriteWarning("Cannot set BackupPolicyType along with BackupInterval or BackupRetention parameters");
return;
}
if (readDatabase.BackupPolicy is PeriodicModeBackupPolicy)
{
databaseAccountUpdateParameters.BackupPolicy = new PeriodicModeBackupPolicy()
{
PeriodicModeProperties = new PeriodicModeProperties()
{
BackupIntervalInMinutes = BackupIntervalInMinutes,
BackupRetentionIntervalInHours = BackupRetentionIntervalInHours,
BackupStorageRedundancy = BackupStorageRedundancy
}
};
}
else
{
WriteWarning("Can accept BackupInterval or BackupRetention parameters only for accounts with PeriodicMode backup policy");
return;
}
}
// Update backup policy to ContinuousModeBackupPolicy
if (BackupPolicyType == "Continuous" && readDatabase.BackupPolicy is PeriodicModeBackupPolicy)
{
databaseAccountUpdateParameters.BackupPolicy = new ContinuousModeBackupPolicy();
}
// Update analytical storage schema type.
databaseAccountUpdateParameters.AnalyticalStorageConfiguration = CreateAnalyticalStorageConfiguration(AnalyticalStorageSchemaType);
if (ShouldProcess(Name, "Updating Database Account"))
{
DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.UpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSDatabaseAccountGetResults(cosmosDBAccount));
}
return;
}
public override void ExecuteCmdlet()
{
if (!ParameterSetName.Equals(NameParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = null;
if (ParameterSetName.Equals(ResourceIdParameterSet, StringComparison.Ordinal))
{
resourceIdentifier = new ResourceIdentifier(ResourceId);
}
else if (ParameterSetName.Equals(ObjectParameterSet, StringComparison.Ordinal))
{
resourceIdentifier = new ResourceIdentifier(InputObject.Id);
}
ResourceGroupName = resourceIdentifier.ResourceGroupName;
Name = resourceIdentifier.ResourceName;
}
DatabaseAccountGetResults readDatabase = CosmosDBManagementClient.DatabaseAccounts.GetWithHttpMessagesAsync(ResourceGroupName, Name).GetAwaiter().GetResult().Body;
DatabaseAccountUpdateParameters databaseAccountUpdateParameters = new DatabaseAccountUpdateParameters(locations: readDatabase.Locations, location: readDatabase.WriteLocations.ElementAt(0).LocationName);
if (EnableMultipleWriteLocations != null)
{
databaseAccountUpdateParameters.EnableMultipleWriteLocations = EnableMultipleWriteLocations;
}
if (EnableVirtualNetwork != null)
{
databaseAccountUpdateParameters.IsVirtualNetworkFilterEnabled = EnableVirtualNetwork;
}
if (EnableAutomaticFailover != null)
{
databaseAccountUpdateParameters.EnableAutomaticFailover = EnableAutomaticFailover;
}
if (DisableKeyBasedMetadataWriteAccess != null)
{
databaseAccountUpdateParameters.DisableKeyBasedMetadataWriteAccess = DisableKeyBasedMetadataWriteAccess;
}
if (PublicNetworkAccess != null)
{
databaseAccountUpdateParameters.PublicNetworkAccess = PublicNetworkAccess;
}
if (!string.IsNullOrEmpty(DefaultConsistencyLevel))
{
ConsistencyPolicy consistencyPolicy = new ConsistencyPolicy();
{
switch (DefaultConsistencyLevel)
{
case "Strong":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Strong;
break;
case "Session":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Session;
break;
case "Eventual":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Eventual;
break;
case "ConsistentPrefix":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.ConsistentPrefix;
break;
case "BoundedStaleness":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.BoundedStaleness;
consistencyPolicy.MaxIntervalInSeconds = MaxStalenessIntervalInSeconds;
consistencyPolicy.MaxStalenessPrefix = MaxStalenessPrefix;
break;
default:
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Session;
break;
}
databaseAccountUpdateParameters.ConsistencyPolicy = consistencyPolicy;
}
}
if (Tag != null)
{
Dictionary <string, string> tags = new Dictionary <string, string>();
foreach (string key in Tag.Keys)
{
tags.Add(key, Tag[key].ToString());
}
databaseAccountUpdateParameters.Tags = tags;
}
if (VirtualNetworkRule != null || VirtualNetworkRuleObject != null)
{
Collection <VirtualNetworkRule> virtualNetworkRule = new Collection <VirtualNetworkRule>();
if (VirtualNetworkRule != null && VirtualNetworkRule.Length > 0)
{
foreach (string id in VirtualNetworkRule)
{
virtualNetworkRule.Add(new VirtualNetworkRule(id: id));
}
}
if (VirtualNetworkRuleObject != null && VirtualNetworkRuleObject.Length > 0)
{
foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
{
virtualNetworkRule.Add(PSVirtualNetworkRule.ConvertPSVirtualNetworkRuleToVirtualNetworkRule(psVirtualNetworkRule));
}
}
databaseAccountUpdateParameters.VirtualNetworkRules = virtualNetworkRule;
}
if (IpRangeFilter != null)
{
string IpRangeFilterAsString = IpRangeFilter?.Aggregate(string.Empty, (output, next) => string.Concat(output, (!string.IsNullOrWhiteSpace(output) && !string.IsNullOrWhiteSpace(next) ? "," : string.Empty), next)) ?? string.Empty;
databaseAccountUpdateParameters.IpRangeFilter = IpRangeFilterAsString;
}
if (ShouldProcess(Name, "Updating Database Account"))
{
DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.UpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSDatabaseAccount(cosmosDBAccount));
}
return;
}
public override void ExecuteCmdlet()
{
ConsistencyPolicy consistencyPolicy = new ConsistencyPolicy();
{
switch (DefaultConsistencyLevel)
{
case "Strong":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Strong;
break;
case "Session":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Session;
break;
case "Eventual":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Eventual;
break;
case "ConsistentPrefix":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.ConsistentPrefix;
break;
case "BoundedStaleness":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.BoundedStaleness;
consistencyPolicy.MaxIntervalInSeconds = MaxStalenessIntervalInSeconds;
consistencyPolicy.MaxStalenessPrefix = MaxStalenessPrefix;
break;
default:
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Session;
break;
}
}
if (!string.IsNullOrEmpty(ApiKind))
{
if (!ApiKind.Equals("GlobalDocumentDB", StringComparison.OrdinalIgnoreCase) && !ApiKind.Equals("MongoDB", StringComparison.OrdinalIgnoreCase))
{
WriteError(new ErrorRecord(new PSArgumentException("Gremlin, Cassandra and Table account creation not supported" +
"in Azure Powershell"), string.Empty, ErrorCategory.CloseError, null));
return;
}
}
else
{
ApiKind = "GlobalDocumentDB";
}
string writeLocation = null;
Collection <Location> LocationCollection = new Collection <Location>();
if (Location != null && Location.Length > 0)
{
int failoverPriority = 0;
foreach (string l in Location)
{
Location loc = new Location(locationName: l, failoverPriority: failoverPriority);
LocationCollection.Add(loc);
if (failoverPriority == 0)
{
writeLocation = l;
}
failoverPriority++;
}
}
else if (LocationObject != null && LocationObject.Length > 0)
{
if (writeLocation != null)
{
WriteError(new ErrorRecord(new PSArgumentException("Cannot accept Location and LocationObject simultaneously as parameters"), string.Empty, ErrorCategory.CloseError, null));
return;
}
foreach (PSLocation psLocation in LocationObject)
{
LocationCollection.Add(PSLocation.ConvertPSLocationToLocation(psLocation));
if (psLocation.FailoverPriority == 0)
{
writeLocation = psLocation.LocationName;
}
}
}
if (string.IsNullOrEmpty(writeLocation))
{
WriteError(new ErrorRecord(new PSArgumentException("Cannot create Account without a Write Location."), string.Empty, ErrorCategory.CloseError, null));
return;
}
Dictionary <string, string> tags = new Dictionary <string, string>();
if (Tag != null)
{
foreach (string key in Tag.Keys)
{
tags.Add(key, Tag[key].ToString());
}
}
Collection <VirtualNetworkRule> virtualNetworkRule = new Collection <VirtualNetworkRule>();
if ((VirtualNetworkRule != null && VirtualNetworkRule.Length > 0) ||
(VirtualNetworkRuleObject != null && VirtualNetworkRuleObject.Length > 0))
{
foreach (string id in VirtualNetworkRule)
{
virtualNetworkRule.Add(new VirtualNetworkRule(id: id));
}
foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
{
virtualNetworkRule.Add(PSVirtualNetworkRule.ConvertPSVirtualNetworkRuleToVirtualNetworkRule(psVirtualNetworkRule));
}
}
string IpRangeFilterAsString = null;
if (IpRangeFilter != null && IpRangeFilter.Length > 0)
{
for (int i = 0; i < IpRangeFilter.Length; i++)
{
if (i == 0)
{
IpRangeFilterAsString = IpRangeFilter[0];
}
else
{
IpRangeFilterAsString = string.Concat(IpRangeFilterAsString, ",", IpRangeFilter[i]);
}
}
}
DatabaseAccountCreateUpdateParameters databaseAccountCreateUpdateParameters = new DatabaseAccountCreateUpdateParameters(locations: LocationCollection, location: writeLocation, name: Name, kind: ApiKind, consistencyPolicy: consistencyPolicy, tags: tags, ipRangeFilter: IpRangeFilterAsString);
databaseAccountCreateUpdateParameters.EnableMultipleWriteLocations = EnableMultipleWriteLocations;
databaseAccountCreateUpdateParameters.IsVirtualNetworkFilterEnabled = EnableVirtualNetwork;
databaseAccountCreateUpdateParameters.EnableAutomaticFailover = EnableAutomaticFailover;
databaseAccountCreateUpdateParameters.VirtualNetworkRules = virtualNetworkRule;
if (ShouldProcess(Name, "Creating Database Account"))
{
DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.CreateOrUpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSDatabaseAccount(cosmosDBAccount));
}
return;
}
public override void ExecuteCmdlet()
{
ConsistencyPolicy consistencyPolicy = new ConsistencyPolicy();
{
switch (DefaultConsistencyLevel)
{
case "Strong":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Strong;
break;
case "Session":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Session;
break;
case "Eventual":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Eventual;
break;
case "ConsistentPrefix":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.ConsistentPrefix;
break;
case "BoundedStaleness":
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.BoundedStaleness;
consistencyPolicy.MaxIntervalInSeconds = MaxStalenessIntervalInSeconds;
consistencyPolicy.MaxStalenessPrefix = MaxStalenessPrefix;
break;
default:
consistencyPolicy.DefaultConsistencyLevel = Management.CosmosDB.Models.DefaultConsistencyLevel.Session;
break;
}
}
string writeLocation = null;
Collection <Location> LocationCollection = new Collection <Location>();
if (Location != null && Location.Length > 0)
{
int failoverPriority = 0;
foreach (string l in Location)
{
Location loc = new Location(locationName: l, failoverPriority: failoverPriority);
LocationCollection.Add(loc);
if (failoverPriority == 0)
{
writeLocation = l;
}
failoverPriority++;
}
}
else if (LocationObject != null && LocationObject.Length > 0)
{
if (writeLocation != null)
{
WriteWarning("Cannot accept Location and LocationObject simultaneously as parameters");
return;
}
foreach (PSLocation psLocation in LocationObject)
{
LocationCollection.Add(PSLocation.ConvertPSLocationToLocation(psLocation));
if (psLocation.FailoverPriority == 0)
{
writeLocation = psLocation.LocationName;
}
}
}
if (string.IsNullOrEmpty(writeLocation))
{
WriteWarning("Cannot create Account without a Write Location.");
return;
}
Dictionary <string, string> tags = new Dictionary <string, string>();
if (Tag != null)
{
foreach (string key in Tag.Keys)
{
tags.Add(key, Tag[key].ToString());
}
}
Collection <VirtualNetworkRule> virtualNetworkRule = new Collection <VirtualNetworkRule>();
if (VirtualNetworkRule != null && VirtualNetworkRule.Length > 0)
{
foreach (string id in VirtualNetworkRule)
{
virtualNetworkRule.Add(new VirtualNetworkRule(id: id));
}
}
if (VirtualNetworkRuleObject != null && VirtualNetworkRuleObject.Length > 0)
{
foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
{
virtualNetworkRule.Add(PSVirtualNetworkRule.ConvertPSVirtualNetworkRuleToVirtualNetworkRule(psVirtualNetworkRule));
}
}
string IpRangeFilterAsString = null;
if (IpRangeFilter != null && IpRangeFilter.Length > 0)
{
IpRangeFilterAsString = IpRangeFilter?.Aggregate(string.Empty, (output, next) => string.Concat(output, (!string.IsNullOrWhiteSpace(output) && !string.IsNullOrWhiteSpace(next) ? "," : string.Empty), next)) ?? string.Empty;
}
DatabaseAccountCreateUpdateParameters databaseAccountCreateUpdateParameters = new DatabaseAccountCreateUpdateParameters(locations: LocationCollection, location: writeLocation, name: Name, consistencyPolicy: consistencyPolicy, tags: tags, ipRangeFilter: IpRangeFilterAsString);
databaseAccountCreateUpdateParameters.EnableMultipleWriteLocations = EnableMultipleWriteLocations;
databaseAccountCreateUpdateParameters.IsVirtualNetworkFilterEnabled = EnableVirtualNetwork;
databaseAccountCreateUpdateParameters.EnableAutomaticFailover = EnableAutomaticFailover;
databaseAccountCreateUpdateParameters.VirtualNetworkRules = virtualNetworkRule;
databaseAccountCreateUpdateParameters.DisableKeyBasedMetadataWriteAccess = DisableKeyBasedMetadataWriteAccess;
databaseAccountCreateUpdateParameters.IpRangeFilter = IpRangeFilterAsString;
databaseAccountCreateUpdateParameters.PublicNetworkAccess = PublicNetworkAccess;
if (!string.IsNullOrEmpty(ApiKind))
{
if (!ApiKind.Equals("MongoDB", StringComparison.OrdinalIgnoreCase))
{
switch (ApiKind)
{
case "Cassandra":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableCassandra"
}
};
break;
case "Gremlin":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableGremlin"
}
};
break;
case "Table":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableTable"
}
};
break;
case "Sql":
break;
}
ApiKind = null;
}
}
else
{
ApiKind = "GlobalDocumentDB";
}
databaseAccountCreateUpdateParameters.Kind = ApiKind;
if (ShouldProcess(Name, "Creating Database Account"))
{
DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.CreateOrUpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSDatabaseAccount(cosmosDBAccount));
}
return;
}
public override void ExecuteCmdlet()
{
if (!ParameterSetName.Equals(NameParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = null;
if (ParameterSetName.Equals(ResourceIdParameterSet, StringComparison.Ordinal))
{
resourceIdentifier = new ResourceIdentifier(ResourceId);
}
else if (ParameterSetName.Equals(ObjectParameterSet, StringComparison.Ordinal))
{
resourceIdentifier = new ResourceIdentifier(InputObject.Id);
}
ResourceGroupName = resourceIdentifier.ResourceGroupName;
Name = resourceIdentifier.ResourceName;
}
DatabaseAccountGetResults readDatabase = CosmosDBManagementClient.DatabaseAccounts.GetWithHttpMessagesAsync(ResourceGroupName, Name).GetAwaiter().GetResult().Body;
DatabaseAccountUpdateParameters databaseAccountUpdateParameters = new DatabaseAccountUpdateParameters(locations: readDatabase.Locations, location: readDatabase.WriteLocations.ElementAt(0).LocationName);
if (EnableMultipleWriteLocations != null)
{
databaseAccountUpdateParameters.EnableMultipleWriteLocations = EnableMultipleWriteLocations;
}
if (EnableVirtualNetwork != null)
{
databaseAccountUpdateParameters.IsVirtualNetworkFilterEnabled = EnableVirtualNetwork;
}
if (EnableAutomaticFailover != null)
{
databaseAccountUpdateParameters.EnableAutomaticFailover = EnableAutomaticFailover;
}
if (DisableKeyBasedMetadataWriteAccess != null)
{
databaseAccountUpdateParameters.DisableKeyBasedMetadataWriteAccess = DisableKeyBasedMetadataWriteAccess;
}
if (PublicNetworkAccess != null)
{
databaseAccountUpdateParameters.PublicNetworkAccess = PublicNetworkAccess;
}
if (KeyVaultKeyUri != null)
{
databaseAccountUpdateParameters.KeyVaultKeyUri = KeyVaultKeyUri;
}
if (EnableAnalyticalStorage != null)
{
databaseAccountUpdateParameters.EnableAnalyticalStorage = EnableAnalyticalStorage;
}
if (!string.IsNullOrEmpty(DefaultConsistencyLevel))
{
databaseAccountUpdateParameters.ConsistencyPolicy = base.PopoulateConsistencyPolicy(DefaultConsistencyLevel, MaxStalenessIntervalInSeconds, MaxStalenessPrefix);
}
if (Tag != null)
{
databaseAccountUpdateParameters.Tags = base.PopulateTags(Tag);
}
if (VirtualNetworkRule != null || VirtualNetworkRuleObject != null)
{
Collection <VirtualNetworkRule> virtualNetworkRule = new Collection <VirtualNetworkRule>();
if (VirtualNetworkRule != null && VirtualNetworkRule.Length > 0)
{
foreach (string id in VirtualNetworkRule)
{
virtualNetworkRule.Add(new VirtualNetworkRule(id: id));
}
}
if (VirtualNetworkRuleObject != null && VirtualNetworkRuleObject.Length > 0)
{
foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
{
virtualNetworkRule.Add(PSVirtualNetworkRule.ToSDKModel(psVirtualNetworkRule));
}
}
databaseAccountUpdateParameters.VirtualNetworkRules = virtualNetworkRule;
}
if (IpRule != null)
{
// not checking IpRules.Length > 0, to handle the removal of IpRules case
databaseAccountUpdateParameters.IpRules = base.PopulateIpRules(IpRule);
}
if (ShouldProcess(Name, "Updating Database Account"))
{
DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.UpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSDatabaseAccountGetResults(cosmosDBAccount));
}
return;
}
public override void ExecuteCmdlet()
{
ConsistencyPolicy consistencyPolicy = base.PopoulateConsistencyPolicy(DefaultConsistencyLevel, MaxStalenessIntervalInSeconds, MaxStalenessPrefix);
string writeLocation = null;
Collection <Location> LocationCollection = new Collection <Location>();
if (Location != null && Location.Length > 0)
{
int failoverPriority = 0;
foreach (string l in Location)
{
Location loc = new Location(locationName: l, failoverPriority: failoverPriority);
LocationCollection.Add(loc);
if (failoverPriority == 0)
{
writeLocation = l;
}
failoverPriority++;
}
}
else if (LocationObject != null && LocationObject.Length > 0)
{
if (writeLocation != null)
{
WriteWarning("Cannot accept Location and LocationObject simultaneously as parameters");
return;
}
foreach (PSLocation psLocation in LocationObject)
{
LocationCollection.Add(PSLocation.ToSDKModel(psLocation));
if (psLocation.FailoverPriority == 0)
{
writeLocation = psLocation.LocationName;
}
}
}
if (string.IsNullOrEmpty(writeLocation))
{
WriteWarning("Cannot create Account without a Write Location.");
return;
}
Dictionary <string, string> tags = new Dictionary <string, string>();
if (Tag != null)
{
tags = base.PopulateTags(Tag);
}
Collection <VirtualNetworkRule> virtualNetworkRule = new Collection <VirtualNetworkRule>();
if (VirtualNetworkRule != null && VirtualNetworkRule.Length > 0)
{
foreach (string id in VirtualNetworkRule)
{
virtualNetworkRule.Add(new VirtualNetworkRule(id: id));
}
}
if (VirtualNetworkRuleObject != null && VirtualNetworkRuleObject.Length > 0)
{
foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
{
virtualNetworkRule.Add(PSVirtualNetworkRule.ToSDKModel(psVirtualNetworkRule));
}
}
DatabaseAccountCreateUpdateParameters databaseAccountCreateUpdateParameters = new DatabaseAccountCreateUpdateParameters(locations: LocationCollection, location: writeLocation, name: Name, consistencyPolicy: consistencyPolicy, tags: tags);
databaseAccountCreateUpdateParameters.EnableMultipleWriteLocations = EnableMultipleWriteLocations;
databaseAccountCreateUpdateParameters.IsVirtualNetworkFilterEnabled = EnableVirtualNetwork;
databaseAccountCreateUpdateParameters.EnableAutomaticFailover = EnableAutomaticFailover;
databaseAccountCreateUpdateParameters.VirtualNetworkRules = virtualNetworkRule;
databaseAccountCreateUpdateParameters.DisableKeyBasedMetadataWriteAccess = DisableKeyBasedMetadataWriteAccess;
databaseAccountCreateUpdateParameters.PublicNetworkAccess = PublicNetworkAccess;
databaseAccountCreateUpdateParameters.EnableFreeTier = EnableFreeTier;
databaseAccountCreateUpdateParameters.EnableAnalyticalStorage = EnableAnalyticalStorage;
if (IpRule != null && IpRule.Length > 0)
{
databaseAccountCreateUpdateParameters.IpRules = base.PopulateIpRules(IpRule);
}
if (KeyVaultKeyUri != null)
{
databaseAccountCreateUpdateParameters.KeyVaultKeyUri = KeyVaultKeyUri;
}
if (!string.IsNullOrEmpty(ApiKind))
{
if (ApiKind.Equals("MongoDB", StringComparison.OrdinalIgnoreCase))
{
if (ServerVersion != null)
{
databaseAccountCreateUpdateParameters.ApiProperties = new ApiProperties
{
ServerVersion = ServerVersion
};
}
}
else
{
switch (ApiKind)
{
case "Cassandra":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableCassandra"
}
};
break;
case "Gremlin":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableGremlin"
}
};
break;
case "Table":
databaseAccountCreateUpdateParameters.Capabilities = new List <Capability> {
new Capability {
Name = "EnableTable"
}
};
break;
case "Sql":
break;
}
ApiKind = null;
}
}
else
{
ApiKind = "GlobalDocumentDB";
}
databaseAccountCreateUpdateParameters.Kind = ApiKind;
if (ShouldProcess(Name, "Creating Database Account"))
{
DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.CreateOrUpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSDatabaseAccountGetResults(cosmosDBAccount));
}
return;
}
