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;
}