public Task <bool> SaveAsync(IUpdateEntry entry, CancellationToken cancellationToken = default)
{
var id = entry.GetCurrentValue <string>(_entityType.FindProperty(StoreKeyConvention.IdPropertyName));
switch (entry.EntityState)
{
case EntityState.Added:
return(_cosmosClient.CreateDocumentAsync(_collectionId, CreateDocument(entry), cancellationToken));
case EntityState.Modified:
var jObjectProperty = _entityType.FindProperty(StoreKeyConvention.JObjectPropertyName);
var document = jObjectProperty != null ? (JObject)entry.GetCurrentValue(jObjectProperty) : null;
if (document != null)
{
UpdateDocument(document, entry);
}
else
{
document = CreateDocument(entry);
// Set Discriminator Property for updates
document[_entityType.CosmosSql().DiscriminatorProperty.Name] =
JToken.FromObject(_entityType.CosmosSql().DiscriminatorValue);
}
return(_cosmosClient.ReplaceDocumentAsync(_collectionId, id, document, cancellationToken));
case EntityState.Deleted:
return(_cosmosClient.DeleteDocumentAsync(_collectionId, id, cancellationToken));
}
return(Task.FromResult(false));
}