public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
AccountName = resourceIdentifier.ResourceName;
}
else if (ParameterSetName.Equals(ObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(InputObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
Name = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
SqlDatabaseGetResults readSqlDatabaseGetResults = null;
try
{
readSqlDatabaseGetResults = CosmosDBManagementClient.SqlResources.GetSqlDatabase(ResourceGroupName, AccountName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode == System.Net.HttpStatusCode.NotFound)
{
throw new ResourceNotFoundException(message: string.Format(ExceptionMessage.NotFound, Name), innerException: e);
}
}
CreateUpdateOptions options = new CreateUpdateOptions();
if (Throughput != null)
{
options.Throughput = Throughput.ToString();
}
SqlDatabaseCreateUpdateParameters sqlDatabaseCreateUpdateParameters = new SqlDatabaseCreateUpdateParameters
{
Resource = new SqlDatabaseResource
{
Id = Name
},
Options = options
};
if (ShouldProcess(Name, "Updating an existing CosmosDB Sql Database"))
{
SqlDatabaseGetResults sqlDatabaseGetResults = CosmosDBManagementClient.SqlResources.CreateUpdateSqlDatabaseWithHttpMessagesAsync(ResourceGroupName, AccountName, Name, sqlDatabaseCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSSqlDatabaseGetResults(sqlDatabaseGetResults));
}
return;
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
AccountName = resourceIdentifier.ResourceName;
}
GremlinDatabaseGetResults readGremlinDatabaseGetResults = null;
try
{
readGremlinDatabaseGetResults = CosmosDBManagementClient.GremlinResources.GetGremlinDatabase(ResourceGroupName, AccountName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode != System.Net.HttpStatusCode.NotFound)
{
throw;
}
}
if (readGremlinDatabaseGetResults != null)
{
throw new ConflictingResourceException(message: string.Format(ExceptionMessage.Conflict, Name));
}
CreateUpdateOptions options = new CreateUpdateOptions();
if (Throughput != null)
{
options.Throughput = Throughput.ToString();
}
GremlinDatabaseCreateUpdateParameters gremlinDatabaseCreateUpdateParameters = new GremlinDatabaseCreateUpdateParameters
{
Resource = new GremlinDatabaseResource
{
Id = Name
},
Options = options
};
if (ShouldProcess(Name, "Creating a new CosmosDB Gremlin Database"))
{
GremlinDatabaseGetResults gremlinDatabaseGetResults = CosmosDBManagementClient.GremlinResources.CreateUpdateGremlinDatabaseWithHttpMessagesAsync(ResourceGroupName, AccountName, Name, gremlinDatabaseCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSGremlinDatabaseGetResults(gremlinDatabaseGetResults));
}
return;
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
AccountName = resourceIdentifier.ResourceName;
}
else if (ParameterSetName.Equals(ObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(InputObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
Name = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
MongoDBDatabaseGetResults readMongoDBDatabaseGetResults = null;
try
{
readMongoDBDatabaseGetResults = CosmosDBManagementClient.MongoDBResources.GetMongoDBDatabase(ResourceGroupName, AccountName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode == System.Net.HttpStatusCode.NotFound)
{
throw new ResourceNotFoundException(message: string.Format(ExceptionMessage.NotFound, Name), innerException: e);
}
}
CreateUpdateOptions options = ThroughputHelper.PopulateCreateUpdateOptions(Throughput, AutoscaleMaxThroughput);
MongoDBDatabaseCreateUpdateParameters mongoDBDatabaseCreateUpdateParameters = new MongoDBDatabaseCreateUpdateParameters
{
Resource = new MongoDBDatabaseResource
{
Id = Name
},
Options = options
};
if (ShouldProcess(Name, "Updating an existing CosmosDB MongoDB Database"))
{
MongoDBDatabaseGetResults mongoDBDatabaseGetResults = CosmosDBManagementClient.MongoDBResources.CreateUpdateMongoDBDatabase(ResourceGroupName, AccountName, Name, mongoDBDatabaseCreateUpdateParameters);
WriteObject(new PSMongoDBDatabaseGetResults(mongoDBDatabaseGetResults));
}
return;
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
AccountName = resourceIdentifier.ResourceName;
}
CassandraKeyspaceGetResults readCassandraKeyspaceGetResults = null;
try
{
readCassandraKeyspaceGetResults = CosmosDBManagementClient.CassandraResources.GetCassandraKeyspace(ResourceGroupName, AccountName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode != System.Net.HttpStatusCode.NotFound)
{
throw;
}
}
if (readCassandraKeyspaceGetResults != null)
{
throw new ConflictingResourceException(message: string.Format(ExceptionMessage.Conflict, Name));
}
CreateUpdateOptions options = ThroughputHelper.PopulateCreateUpdateOptions(Throughput, AutoscaleMaxThroughput);
CassandraKeyspaceCreateUpdateParameters cassandraKeyspaceCreateUpdateParameters = new CassandraKeyspaceCreateUpdateParameters
{
Resource = new CassandraKeyspaceResource
{
Id = Name
},
Options = options
};
if (ShouldProcess(Name, "Creating a new CosmosDB Cassandra Keyspace"))
{
CassandraKeyspaceGetResults cassandraKeyspaceGetResults = CosmosDBManagementClient.CassandraResources.CreateUpdateCassandraKeyspaceWithHttpMessagesAsync(ResourceGroupName, AccountName, Name, cassandraKeyspaceCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSCassandraKeyspaceGetResults(cassandraKeyspaceGetResults));
}
return;
}
public static CreateUpdateOptions PopulateCreateUpdateOptions(int?throughput = null, int?autoscaleMaxThroughput = null)
{
Throughput = throughput;
AutoscaleMaxThroughput = autoscaleMaxThroughput;
ThroughputHelper.ThroughputValidation();
CreateUpdateOptions createUpdateOptions = new CreateUpdateOptions();
if (Throughput != null)
{
createUpdateOptions.Throughput = Throughput.Value;
}
else if (autoscaleMaxThroughput != null)
{
createUpdateOptions.AutoscaleSettings = new AutoscaleSettings {
MaxThroughput = AutoscaleMaxThroughput.Value
};
}
return(createUpdateOptions);
}
/// <summary>
/// Transform options object to query parameteres
/// </summary>
/// <param name="url">The URL.</param>
/// <param name="options">The options.</param>
/// <returns>System.String.</returns>
internal string PrepareUrlWithOptions(string url, CreateUpdateOptions options)
{
string urlWithOptions = "";
List <string> parameters = new List <string>();
if (options.Silent)
{
parameters.Add("silent=1");
}
if (!options.Hook)
{
parameters.Add("hook=false");
}
if (options.Fields != null && options.Fields.Any())
{
parameters.Add(string.Join(",", options.Fields.Select(s => s).ToArray()));
}
urlWithOptions = parameters.Any() ? url + "?" + string.Join("&", parameters.ToArray()) : url;
return(urlWithOptions);
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
AccountName = resourceIdentifier.ResourceName;
}
else if (ParameterSetName.Equals(ObjectParameterSet))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(InputObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
Name = resourceIdentifier.ResourceName;
}
CreateUpdateOptions options = new CreateUpdateOptions();
if (Throughput != null)
{
options.Throughput = Throughput.ToString();
}
GremlinDatabaseCreateUpdateParameters gremlinDatabaseCreateUpdateParameters = new GremlinDatabaseCreateUpdateParameters
{
Resource = new GremlinDatabaseResource
{
Id = Name
},
Options = options
};
if (ShouldProcess(Name, "Updating CosmosDB Gremlin Database"))
{
GremlinDatabaseGetResults gremlinDatabaseGetResults = CosmosDBManagementClient.GremlinResources.CreateUpdateGremlinDatabaseWithHttpMessagesAsync(ResourceGroupName, AccountName, Name, gremlinDatabaseCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSGremlinDatabaseGetResults(gremlinDatabaseGetResults));
}
return;
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(InputObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
KeyspaceName = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
CreateUpdateOptions options = new CreateUpdateOptions();
if (Throughput != null)
{
options.Throughput = Throughput.ToString();
}
CassandraTableResource cassandraTableResource = new CassandraTableResource
{
Id = Name,
DefaultTtl = TtlInSeconds
};
cassandraTableResource.Schema = PSCassandraSchema.ConvertPSCassandraSchemaToCassandraSchema(Schema);
CassandraTableCreateUpdateParameters cassandraTableCreateUpdateParameters = new CassandraTableCreateUpdateParameters
{
Resource = cassandraTableResource,
Options = options
};
if (ShouldProcess(Name, "Setting CosmosDB Cassandra Table"))
{
CassandraTableGetResults cassandraTableGetResults = CosmosDBManagementClient.CassandraResources.CreateUpdateCassandraTableWithHttpMessagesAsync(ResourceGroupName, AccountName, KeyspaceName, Name, cassandraTableCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSCassandraTableGetResults(cassandraTableGetResults));
}
return;
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
DatabaseName = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
SqlContainerGetResults readSqlContainerGetResults = null;
try
{
readSqlContainerGetResults = CosmosDBManagementClient.SqlResources.GetSqlContainer(ResourceGroupName, AccountName, DatabaseName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode != System.Net.HttpStatusCode.NotFound)
{
throw;
}
}
if (readSqlContainerGetResults != null)
{
throw new ConflictingResourceException(message: string.Format(ExceptionMessage.Conflict, Name));
}
List <string> Paths = new List <string>();
foreach (string path in PartitionKeyPath)
{
Paths.Add(path);
}
SqlContainerResource sqlContainerResource = new SqlContainerResource
{
Id = Name,
PartitionKey = new ContainerPartitionKey
{
Kind = PartitionKeyKind,
Paths = Paths,
Version = PartitionKeyVersion
}
};
if (UniqueKeyPolicy != null)
{
sqlContainerResource.UniqueKeyPolicy = PSUniqueKeyPolicy.ConvertPSUniqueKeyPolicyToUniqueKeyPolicy(UniqueKeyPolicy);
}
if (TtlInSeconds != null)
{
sqlContainerResource.DefaultTtl = TtlInSeconds;
}
if (ConflictResolutionPolicy != null)
{
sqlContainerResource.ConflictResolutionPolicy = PSConflictResolutionPolicy.ConvertPSConflictResolutionPolicyToConflictResolutionPolicy(ConflictResolutionPolicy);
}
if (ConflictResolutionPolicyMode != null)
{
ConflictResolutionPolicy conflictResolutionPolicy = new ConflictResolutionPolicy
{
Mode = ConflictResolutionPolicyMode
};
if (ConflictResolutionPolicyMode.Equals(ConflictResolutionMode.LastWriterWins, StringComparison.OrdinalIgnoreCase))
{
conflictResolutionPolicy.ConflictResolutionPath = ConflictResolutionPolicyPath;
}
else if (ConflictResolutionPolicyMode.Equals(ConflictResolutionMode.Custom, StringComparison.OrdinalIgnoreCase))
{
conflictResolutionPolicy.ConflictResolutionProcedure = ConflictResolutionPolicyProcedure;
}
sqlContainerResource.ConflictResolutionPolicy = conflictResolutionPolicy;
}
if (IndexingPolicy != null)
{
sqlContainerResource.IndexingPolicy = PSIndexingPolicy.ConvertPSIndexingToIndexingPolicy(IndexingPolicy);
}
CreateUpdateOptions options = new CreateUpdateOptions();
if (Throughput != null)
{
options.Throughput = Throughput.ToString();
}
SqlContainerCreateUpdateParameters sqlContainerCreateUpdateParameters = new SqlContainerCreateUpdateParameters
{
Resource = sqlContainerResource,
Options = options
};
if (ShouldProcess(Name, "Creating a new CosmosDB Sql Container"))
{
SqlContainerGetResults sqlContainerGetResults = CosmosDBManagementClient.SqlResources.CreateUpdateSqlContainerWithHttpMessagesAsync(ResourceGroupName, AccountName, DatabaseName, Name, sqlContainerCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSSqlContainerGetResults(sqlContainerGetResults));
}
return;
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(InputObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
DatabaseName = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
List <string> Paths = new List <string>();
foreach (string path in PartitionKeyPath)
{
Paths.Add(path);
}
GremlinGraphResource gremlinGraphResource = new GremlinGraphResource
{
Id = Name,
PartitionKey = new ContainerPartitionKey
{
Kind = PartitionKeyKind,
Paths = Paths,
Version = PartitionKeyVersion
}
};
if (UniqueKeyPolicy != null)
{
UniqueKeyPolicy uniqueKeyPolicy = new UniqueKeyPolicy
{
UniqueKeys = new List <UniqueKey>()
};
foreach (PSUniqueKey uniqueKey in UniqueKeyPolicy.UniqueKeys)
{
UniqueKey key = new UniqueKey
{
Paths = new List <string>()
};
foreach (string path in uniqueKey.Paths)
{
key.Paths.Add(path);
}
uniqueKeyPolicy.UniqueKeys.Add(key);
}
gremlinGraphResource.UniqueKeyPolicy = uniqueKeyPolicy;
}
if (TtlInSeconds != null)
{
gremlinGraphResource.DefaultTtl = TtlInSeconds;
}
if (ConflictResolutionPolicy != null)
{
ConflictResolutionPolicyMode = ConflictResolutionPolicy.Mode;
if (ConflictResolutionPolicy.ConflictResolutionPath != null)
{
ConflictResolutionPolicyPath = ConflictResolutionPolicy.ConflictResolutionPath;
}
if (ConflictResolutionPolicy.ConflictResolutionProcedure != null)
{
ConflictResolutionPolicyProcedure = ConflictResolutionPolicy.ConflictResolutionProcedure;
}
}
if (ConflictResolutionPolicyMode != null)
{
ConflictResolutionPolicy conflictResolutionPolicy = new ConflictResolutionPolicy
{
Mode = ConflictResolutionPolicyMode
};
if (ConflictResolutionPolicyMode.Equals("LastWriterWins", StringComparison.OrdinalIgnoreCase))
{
conflictResolutionPolicy.ConflictResolutionPath = ConflictResolutionPolicyPath;
}
else if (ConflictResolutionPolicyMode.Equals("Custom", StringComparison.OrdinalIgnoreCase))
{
conflictResolutionPolicy.ConflictResolutionProcedure = ConflictResolutionPolicyProcedure;
}
gremlinGraphResource.ConflictResolutionPolicy = conflictResolutionPolicy;
}
if (IndexingPolicy != null)
{
IndexingPolicy indexingPolicy = new IndexingPolicy
{
Automatic = IndexingPolicy.Automatic,
IndexingMode = IndexingPolicy.IndexingMode,
};
if (IndexingPolicy.IncludedPaths != null)
{
IList <IncludedPath> includedPaths = new List <IncludedPath>();
foreach (PSIncludedPath pSIncludedPath in IndexingPolicy.IncludedPaths)
{
includedPaths.Add(new IncludedPath
{
Path = pSIncludedPath.Path,
Indexes = PSIncludedPath.ConvertPSIndexesToIndexes(pSIncludedPath.Indexes)
});
}
indexingPolicy.IncludedPaths = new List <IncludedPath>(includedPaths);
}
if (IndexingPolicy.ExcludedPaths != null && IndexingPolicy.ExcludedPaths.Count > 0)
{
IList <ExcludedPath> excludedPaths = new List <ExcludedPath>();
foreach (PSExcludedPath pSExcludedPath in IndexingPolicy.ExcludedPaths)
{
excludedPaths.Add(new ExcludedPath {
Path = pSExcludedPath.Path
});
}
indexingPolicy.ExcludedPaths = new List <ExcludedPath>(excludedPaths);
}
if (IndexingPolicy.CompositeIndexes != null)
{
IList <IList <CompositePath> > compositeIndexes = new List <IList <CompositePath> >();
foreach (IList <PSCompositePath> pSCompositePathList in IndexingPolicy.CompositeIndexes)
{
IList <CompositePath> compositePathList = new List <CompositePath>();
foreach (PSCompositePath pSCompositePath in pSCompositePathList)
{
compositePathList.Add(new CompositePath {
Order = pSCompositePath.Order, Path = pSCompositePath.Path
});
}
compositeIndexes.Add(compositePathList);
}
indexingPolicy.CompositeIndexes = new List <IList <CompositePath> >(compositeIndexes);
}
if (IndexingPolicy.SpatialIndexes != null && IndexingPolicy.SpatialIndexes.Count > 0)
{
IList <SpatialSpec> spatialIndexes = new List <SpatialSpec>();
foreach (PSSpatialSpec pSSpatialSpec in IndexingPolicy.SpatialIndexes)
{
spatialIndexes.Add(new SpatialSpec {
Path = pSSpatialSpec.Path, Types = pSSpatialSpec.Types
});
}
indexingPolicy.SpatialIndexes = new List <SpatialSpec>(spatialIndexes);
}
gremlinGraphResource.IndexingPolicy = indexingPolicy;
}
CreateUpdateOptions options = new CreateUpdateOptions();
if (Throughput != null)
{
options.Throughput = Throughput.ToString();
}
GremlinGraphCreateUpdateParameters gremlinGraphCreateUpdateParameters = new GremlinGraphCreateUpdateParameters
{
Resource = gremlinGraphResource,
Options = options
};
if (ShouldProcess(Name, "Setting CosmosDB Gremlin Graph"))
{
GremlinGraphGetResults gremlinGraphGetResults = CosmosDBManagementClient.GremlinResources.CreateUpdateGremlinGraphWithHttpMessagesAsync(ResourceGroupName, AccountName, DatabaseName, Name, gremlinGraphCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSGremlinGraphGetResults(gremlinGraphGetResults));
}
return;
}
SqlContainer.Update.IUpdate SqlContainer.Update.IWithChildResource.WithTrigger(string name, string body, TriggerType type, TriggerOperation operation, CreateUpdateOptions options)
{
return(this.WithTrigger(name, body, type, operation, options));
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
DatabaseName = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
else if (ParameterSetName.Equals(ObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(InputObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
Name = resourceIdentifier.ResourceName;
DatabaseName = ResourceIdentifierExtensions.GetSqlDatabaseName(resourceIdentifier);
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
SqlContainerGetResults readSqlContainerGetResults = null;
try
{
readSqlContainerGetResults = CosmosDBManagementClient.SqlResources.GetSqlContainer(ResourceGroupName, AccountName, DatabaseName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode == System.Net.HttpStatusCode.NotFound)
{
throw new ResourceNotFoundException(message: string.Format(ExceptionMessage.NotFound, Name), innerException: e);
}
}
SqlContainerResource sqlContainerResource = UpdateAzCosmosDBSqlContainer.PopulateSqlContainerResource(readSqlContainerGetResults.Resource);
if (PartitionKeyPath != null)
{
List <string> Paths = new List <string>(PartitionKeyPath);
sqlContainerResource.PartitionKey = new ContainerPartitionKey
{
Kind = PartitionKeyKind,
Paths = Paths,
Version = PartitionKeyVersion
};
}
if (UniqueKeyPolicy != null)
{
sqlContainerResource.UniqueKeyPolicy = PSUniqueKeyPolicy.ToSDKModel(UniqueKeyPolicy);
}
if (TtlInSeconds != null)
{
sqlContainerResource.DefaultTtl = TtlInSeconds;
}
if (ConflictResolutionPolicy != null)
{
sqlContainerResource.ConflictResolutionPolicy = PSConflictResolutionPolicy.ToSDKModel(ConflictResolutionPolicy);
}
else if (ConflictResolutionPolicyMode != null)
{
ConflictResolutionPolicy conflictResolutionPolicy = new ConflictResolutionPolicy
{
Mode = ConflictResolutionPolicyMode
};
if (ConflictResolutionPolicyMode.Equals(ConflictResolutionMode.LastWriterWins, StringComparison.OrdinalIgnoreCase))
{
conflictResolutionPolicy.ConflictResolutionPath = ConflictResolutionPolicyPath;
}
else if (ConflictResolutionPolicyMode.Equals(ConflictResolutionMode.Custom, StringComparison.OrdinalIgnoreCase))
{
conflictResolutionPolicy.ConflictResolutionProcedure = ConflictResolutionPolicyProcedure;
}
sqlContainerResource.ConflictResolutionPolicy = conflictResolutionPolicy;
}
if (IndexingPolicy != null)
{
sqlContainerResource.IndexingPolicy = PSIndexingPolicy.ToSDKModel(IndexingPolicy);
}
if (AnalyticalStorageTtl != null)
{
sqlContainerResource.AnalyticalStorageTtl = AnalyticalStorageTtl;
}
CreateUpdateOptions options = ThroughputHelper.PopulateCreateUpdateOptions(Throughput, AutoscaleMaxThroughput);
SqlContainerCreateUpdateParameters sqlContainerCreateUpdateParameters = new SqlContainerCreateUpdateParameters
{
Resource = sqlContainerResource,
Options = options
};
if (ShouldProcess(Name, "Updating an existing CosmosDB Sql Container"))
{
SqlContainerGetResults sqlContainerGetResults = CosmosDBManagementClient.SqlResources.CreateUpdateSqlContainer(ResourceGroupName, AccountName, DatabaseName, Name, sqlContainerCreateUpdateParameters);
WriteObject(new PSSqlContainerGetResults(sqlContainerGetResults));
}
return;
}
public SqlContainerImpl WithTrigger(string name, SqlTriggerResource resource = default(SqlTriggerResource), CreateUpdateOptions options = default(CreateUpdateOptions))
{
var parameter = new SqlTriggerCreateUpdateParameters()
{
Location = this.Location,
Resource = resource ?? new SqlTriggerResource(),
Options = options ?? new CreateUpdateOptions(),
};
parameter.Resource.Id = name;
this.triggerToUpdate.Add(name, parameter);
return(this);
}
public SqlContainerImpl WithUserDefinedFunction(string name, SqlUserDefinedFunctionResource resource = default(SqlUserDefinedFunctionResource), CreateUpdateOptions options = default(CreateUpdateOptions))
{
var parameter = new SqlUserDefinedFunctionCreateUpdateParameters()
{
Location = this.Location,
Resource = resource ?? new SqlUserDefinedFunctionResource(),
Options = options ?? new CreateUpdateOptions(),
};
parameter.Resource.Id = name;
this.userDefinedFunctionToUpdate.Add(name, parameter);
return(this);
}
public SqlContainerImpl WithStoredProcedure(string name, SqlStoredProcedureResource resource = default(SqlStoredProcedureResource), CreateUpdateOptions options = default(CreateUpdateOptions))
{
var parameter = new SqlStoredProcedureCreateUpdateParameters()
{
Location = this.Location,
Resource = resource ?? new SqlStoredProcedureResource(),
Options = options ?? new CreateUpdateOptions(),
};
parameter.Resource.Id = name;
this.storedProcedureToUpdate.Add(name, parameter);
return(this);
}
SqlContainer.Definition.IWithAttach <UpdateParentT> SqlContainer.Definition.IWithChildResource <UpdateParentT> .WithTrigger(string name, Models.SqlTriggerResource resource, CreateUpdateOptions options)
{
return(this.WithTrigger(name, resource, options));
}
SqlContainer.Definition.IWithAttach <UpdateParentT> SqlContainer.Definition.IWithChildResource <UpdateParentT> .WithTrigger(string name, string body, TriggerType type, TriggerOperation operation, CreateUpdateOptions options)
{
return(this.WithTrigger(name, body, type, operation, options));
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
DatabaseName = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
MongoDBCollectionGetResults readMongoDBCollectionGetResults = null;
try
{
readMongoDBCollectionGetResults = CosmosDBManagementClient.MongoDBResources.GetMongoDBCollection(ResourceGroupName, AccountName, DatabaseName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode != System.Net.HttpStatusCode.NotFound)
{
throw;
}
}
if (readMongoDBCollectionGetResults != null)
{
throw new ConflictingResourceException(message: string.Format(ExceptionMessage.Conflict, Name));
}
MongoDBCollectionResource mongoDBCollectionResource = new MongoDBCollectionResource
{
Id = Name
};
if (Shard != null)
{
mongoDBCollectionResource.ShardKey = new Dictionary <string, string> {
{ Shard, "Hash" }
};
}
if (Index != null)
{
List <MongoIndex> Indexes = new List <MongoIndex>();
foreach (PSMongoIndex psMongoIndex in Index)
{
Indexes.Add(PSMongoIndex.ConvertPSMongoIndexToMongoIndex(psMongoIndex));
}
mongoDBCollectionResource.Indexes = Indexes;
}
CreateUpdateOptions options = new CreateUpdateOptions();
if (Throughput != null)
{
options.Throughput = Throughput.ToString();
}
MongoDBCollectionCreateUpdateParameters mongoDBCollectionCreateUpdateParameters = new MongoDBCollectionCreateUpdateParameters
{
Resource = mongoDBCollectionResource,
Options = options
};
if (ShouldProcess(Name, "Creating a new CosmosDB MongoDB Collection"))
{
MongoDBCollectionGetResults mongoDBCollectionGetResults = CosmosDBManagementClient.MongoDBResources.CreateUpdateMongoDBCollectionWithHttpMessagesAsync(ResourceGroupName, AccountName, DatabaseName, Name, mongoDBCollectionCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSMongoDBCollectionGetResults(mongoDBCollectionGetResults));
}
return;
}
public SqlContainerImpl WithStoredProcedure(string name, string body, CreateUpdateOptions options)
{
return(this.WithStoredProcedure(name, new SqlStoredProcedureResource(id: name, body: body), options));
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
KeyspaceName = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
else if (ParameterSetName.Equals(ObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(InputObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
Name = resourceIdentifier.ResourceName;
KeyspaceName = ResourceIdentifierExtensions.GetCassandraKeyspaceName(resourceIdentifier);
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
CassandraTableGetResults readCassandraTableGetResults = null;
try
{
readCassandraTableGetResults = CosmosDBManagementClient.CassandraResources.GetCassandraTable(ResourceGroupName, AccountName, KeyspaceName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode == System.Net.HttpStatusCode.NotFound)
{
throw new ResourceNotFoundException(message: string.Format(ExceptionMessage.NotFound, Name), innerException: e);
}
}
CassandraTableResource cassandraTableResource = PopulateCassandraTableResource(readCassandraTableGetResults.Resource);
if (TtlInSeconds != null)
{
cassandraTableResource.DefaultTtl = TtlInSeconds;
}
if (Schema != null)
{
cassandraTableResource.Schema = PSCassandraSchema.ConvertPSCassandraSchemaToCassandraSchema(Schema);
}
CreateUpdateOptions options = new CreateUpdateOptions();
if (Throughput != null)
{
options.Throughput = Throughput.ToString();
}
CassandraTableCreateUpdateParameters cassandraTableCreateUpdateParameters = new CassandraTableCreateUpdateParameters
{
Resource = cassandraTableResource,
Options = options,
};
if (ShouldProcess(Name, "Updating an existing CosmosDB Cassandra Table"))
{
CassandraTableGetResults cassandraTableGetResults = CosmosDBManagementClient.CassandraResources.CreateUpdateCassandraTableWithHttpMessagesAsync(ResourceGroupName, AccountName, KeyspaceName, Name, cassandraTableCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSCassandraTableGetResults(cassandraTableGetResults));
}
return;
}
public SqlContainerImpl WithUserDefinedFunction(string name, string body, CreateUpdateOptions options)
{
return(this.WithUserDefinedFunction(name, new SqlUserDefinedFunctionResource(id: name, body: body), options));
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
DatabaseName = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
else if (ParameterSetName.Equals(ObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(InputObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
Name = resourceIdentifier.ResourceName;
DatabaseName = ResourceIdentifierExtensions.GetMongoDBDatabaseName(resourceIdentifier);
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
MongoDBCollectionGetResults readMongoDBCollectionGetResults = null;
try
{
readMongoDBCollectionGetResults = CosmosDBManagementClient.MongoDBResources.GetMongoDBCollection(ResourceGroupName, AccountName, DatabaseName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode == System.Net.HttpStatusCode.NotFound)
{
throw new ResourceNotFoundException(message: string.Format(ExceptionMessage.NotFound, Name), innerException: e);
}
}
MongoDBCollectionResource mongoDBCollectionResource = PopulateMongoDBResource(readMongoDBCollectionGetResults.Resource);
if (Shard != null)
{
mongoDBCollectionResource.ShardKey = new Dictionary <string, string> {
{ Shard, "Hash" }
};
}
if (Index != null)
{
List <MongoIndex> Indexes = new List <MongoIndex>();
foreach (PSMongoIndex psMongoIndex in Index)
{
Indexes.Add(PSMongoIndex.ToSDKModel(psMongoIndex));
}
mongoDBCollectionResource.Indexes = Indexes;
}
if (AnalyticalStorageTtl != null)
{
mongoDBCollectionResource.AnalyticalStorageTtl = AnalyticalStorageTtl;
}
CreateUpdateOptions options = ThroughputHelper.PopulateCreateUpdateOptions(Throughput, AutoscaleMaxThroughput);
MongoDBCollectionCreateUpdateParameters mongoDBCollectionCreateUpdateParameters = new MongoDBCollectionCreateUpdateParameters
{
Resource = mongoDBCollectionResource,
Options = options
};
if (ShouldProcess(Name, "Updating an existing CosmosDB MongoDB Collection"))
{
MongoDBCollectionGetResults mongoDBCollectionGetResults = CosmosDBManagementClient.MongoDBResources.CreateUpdateMongoDBCollectionWithHttpMessagesAsync(ResourceGroupName, AccountName, DatabaseName, Name, mongoDBCollectionCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSMongoDBCollectionGetResults(mongoDBCollectionGetResults));
}
return;
}
public SqlContainerImpl WithTrigger(string name, string body, TriggerType type, TriggerOperation operation, CreateUpdateOptions options)
{
return(this.WithTrigger(name, new SqlTriggerResource(id: name, body: body, triggerType: type, triggerOperation: operation), options));
}
public override void ExecuteCmdlet()
{
if (ParameterSetName.Equals(ParentObjectParameterSet, StringComparison.Ordinal))
{
ResourceIdentifier resourceIdentifier = new ResourceIdentifier(ParentObject.Id);
ResourceGroupName = resourceIdentifier.ResourceGroupName;
DatabaseName = resourceIdentifier.ResourceName;
AccountName = ResourceIdentifierExtensions.GetDatabaseAccountName(resourceIdentifier);
}
GremlinGraphGetResults readGremlinGraphGetResults = null;
try
{
readGremlinGraphGetResults = CosmosDBManagementClient.GremlinResources.GetGremlinGraph(ResourceGroupName, AccountName, DatabaseName, Name);
}
catch (CloudException e)
{
if (e.Response.StatusCode != System.Net.HttpStatusCode.NotFound)
{
throw;
}
}
if (readGremlinGraphGetResults != null)
{
throw new ConflictingResourceException(message: string.Format(ExceptionMessage.Conflict, Name));
}
List <string> Paths = new List <string>(PartitionKeyPath);
GremlinGraphResource gremlinGraphResource = new GremlinGraphResource
{
Id = Name,
PartitionKey = new ContainerPartitionKey
{
Kind = PartitionKeyKind,
Paths = Paths,
Version = PartitionKeyVersion
}
};
if (UniqueKeyPolicy != null)
{
gremlinGraphResource.UniqueKeyPolicy = PSUniqueKeyPolicy.ToSDKModel(UniqueKeyPolicy);
}
if (TtlInSeconds != null)
{
gremlinGraphResource.DefaultTtl = TtlInSeconds;
}
if (ConflictResolutionPolicy != null)
{
gremlinGraphResource.ConflictResolutionPolicy = PSConflictResolutionPolicy.ToSDKModel(ConflictResolutionPolicy);
}
else if (ConflictResolutionPolicyMode != null)
{
ConflictResolutionPolicy conflictResolutionPolicy = new ConflictResolutionPolicy
{
Mode = ConflictResolutionPolicyMode
};
if (ConflictResolutionPolicyMode.Equals(ConflictResolutionMode.LastWriterWins, StringComparison.OrdinalIgnoreCase))
{
conflictResolutionPolicy.ConflictResolutionPath = ConflictResolutionPolicyPath;
}
else if (ConflictResolutionPolicyMode.Equals(ConflictResolutionMode.Custom, StringComparison.OrdinalIgnoreCase))
{
conflictResolutionPolicy.ConflictResolutionProcedure = ConflictResolutionPolicyProcedure;
}
gremlinGraphResource.ConflictResolutionPolicy = conflictResolutionPolicy;
}
if (IndexingPolicy != null)
{
gremlinGraphResource.IndexingPolicy = PSIndexingPolicy.ToSDKModel(IndexingPolicy);
}
CreateUpdateOptions options = ThroughputHelper.PopulateCreateUpdateOptions(Throughput, AutoscaleMaxThroughput);
GremlinGraphCreateUpdateParameters gremlinGraphCreateUpdateParameters = new GremlinGraphCreateUpdateParameters
{
Resource = gremlinGraphResource,
Options = options
};
if (ShouldProcess(Name, "Creating a new CosmosDB Gremlin Graph"))
{
GremlinGraphGetResults gremlinGraphGetResults = CosmosDBManagementClient.GremlinResources.CreateUpdateGremlinGraphWithHttpMessagesAsync(ResourceGroupName, AccountName, DatabaseName, Name, gremlinGraphCreateUpdateParameters).GetAwaiter().GetResult().Body;
WriteObject(new PSGremlinGraphGetResults(gremlinGraphGetResults));
}
return;
}
public async Task GremlinGraphCreateUpdateTests()
{
IList <IncludedPath> includedPath = new List <IncludedPath> {
new IncludedPath {
Path = "/*"
}
};
IList <ExcludedPath> excludedPaths = new List <ExcludedPath> {
new ExcludedPath {
Path = "/pathToNotIndex/*"
}
};
IList <IList <CompositePath> > compositeIndexes = new List <IList <CompositePath> >
{
new List <CompositePath>
{
new CompositePath {
Path = "/orderByPath1", Order = CompositePathSortOrder.Ascending
},
new CompositePath {
Path = "/orderByPath2", Order = CompositePathSortOrder.Descending
}
},
new List <CompositePath>
{
new CompositePath {
Path = "/orderByPath3", Order = CompositePathSortOrder.Ascending
},
new CompositePath {
Path = "/orderByPath4", Order = CompositePathSortOrder.Descending
}
}
};
IList <SpatialSpec> spatialIndexes = new List <SpatialSpec> {
new SpatialSpec("/*", new List <SpatialType> {
new SpatialType("Point")
})
};
IndexingPolicy indexingPolicy = new IndexingPolicy(true, IndexingMode.Consistent, includedPath, excludedPaths, compositeIndexes, spatialIndexes);
ContainerPartitionKey containerPartitionKey = new ContainerPartitionKey(new List <string> {
"/address"
}, "Hash", null);
IList <string> paths = new List <string>()
{
"/testpath"
};
UniqueKey uk = new UniqueKey(paths);
IList <UniqueKey> uniqueKeys = new List <UniqueKey>();
uniqueKeys.Add(uk);
UniqueKeyPolicy uniqueKeyPolicy = new UniqueKeyPolicy(uniqueKeys);
ConflictResolutionPolicy conflictResolutionPolicy = new ConflictResolutionPolicy(new ConflictResolutionMode("LastWriterWins"), "/path", "");
CreateUpdateOptions createUpdateOptions = new CreateUpdateOptions(sampleThroughput, new AutoscaleSettings());
GremlinGraphCreateUpdateParameters gremlinGraphCreateUpdateParameters = new GremlinGraphCreateUpdateParameters(new GremlinGraphResource(gremlinGraphName, indexingPolicy, containerPartitionKey, -1, uniqueKeyPolicy, conflictResolutionPolicy), createUpdateOptions);
Response <GremlinGraphGetResults> gremlinResponse = await WaitForCompletionAsync(await CosmosDBManagementClient.GremlinResources.StartCreateUpdateGremlinGraphAsync(resourceGroupName, databaseAccountName, databaseName, gremlinGraphName, gremlinGraphCreateUpdateParameters));
Assert.NotNull(gremlinResponse);
GremlinGraphGetResults gremlinGraphGetResults = gremlinResponse.Value;
Assert.NotNull(gremlinGraphGetResults);
VerifyGremlinGraphCreation(gremlinGraphGetResults, gremlinGraphCreateUpdateParameters);
var throughputResponse = CosmosDBManagementClient.GremlinResources.GetGremlinGraphThroughputAsync(resourceGroupName, databaseAccountName, databaseName, gremlinGraphName);
ThroughputSettingsGetResults throughputSettingsGetResults = throughputResponse.ConfigureAwait(false).GetAwaiter().GetResult();
Assert.NotNull(throughputSettingsGetResults);
Assert.NotNull(throughputSettingsGetResults.Name);
Assert.AreEqual(throughputSettingsGetResults.Resource.Throughput, sampleThroughput);
Assert.AreEqual(gremlinGraphsThroughputType, throughputSettingsGetResults.Type);
CreateUpdateOptions createUpdateOptions2 = new CreateUpdateOptions(sampleThroughput2, new AutoscaleSettings());
GremlinGraphCreateUpdateParameters gremlinGraphCreateUpdateParameters2 = new GremlinGraphCreateUpdateParameters(new GremlinGraphResource(gremlinGraphName, indexingPolicy, containerPartitionKey, -1, uniqueKeyPolicy, conflictResolutionPolicy), createUpdateOptions2);
Response <GremlinGraphGetResults> gremlinResponse2 = await WaitForCompletionAsync(await CosmosDBManagementClient.GremlinResources.StartCreateUpdateGremlinGraphAsync(resourceGroupName, databaseAccountName, databaseName, gremlinGraphName, gremlinGraphCreateUpdateParameters2));
Assert.NotNull(gremlinResponse2);
GremlinGraphGetResults gremlinGraphGetResults2 = gremlinResponse2.Value;
Assert.NotNull(gremlinGraphGetResults2);
VerifyGremlinGraphCreation(gremlinGraphGetResults2, gremlinGraphCreateUpdateParameters2);
var throughputResponse2 = CosmosDBManagementClient.GremlinResources.GetGremlinGraphThroughputAsync(resourceGroupName, databaseAccountName, databaseName, gremlinGraphName);
ThroughputSettingsGetResults throughputSettingsGetResults2 = throughputResponse2.ConfigureAwait(false).GetAwaiter().GetResult();
Assert.NotNull(throughputSettingsGetResults2);
Assert.NotNull(throughputSettingsGetResults2.Name);
Assert.AreEqual(throughputSettingsGetResults2.Resource.Throughput, sampleThroughput2);
Assert.AreEqual(gremlinGraphsThroughputType, throughputSettingsGetResults2.Type);
}
SqlContainer.Update.IUpdate SqlContainer.Update.IWithChildResource.WithTrigger(string name, Models.SqlTriggerResource resource, CreateUpdateOptions options)
{
return(this.WithTrigger(name, resource, options));
}
