PutItemOperationRequest

Author: irina-herciu

sdk/src/Services/DynamoDBv2/Custom/DocumentModel/Table.cs

@@ -1237,6 +1237,47 @@ internal Document PutItemHelper(Document doc, PutItemOperationConfig config)
             return ret;
         }
 
+        internal async Task<Document> PutItemHelperAsync(PutItemDocumentOperationRequest request, CancellationToken cancellationToken)
+        {
+            var req = MapPutItemDocumentOperationRequestToPutItemRequest(request);
+            this.UpdateRequestUserAgentDetails(req, isAsync: true);
+            var resp = await DDBClient.PutItemAsync(req, cancellationToken).ConfigureAwait(false);
+            request.Document.CommitChanges();
+            Document ret = null;
+            if (request.ReturnValues == ReturnValues.AllOldAttributes)
+            {
+                ret = this.FromAttributeMap(resp.Attributes);
+            }
+            return ret;
+        }
+        internal Document PutItemHelper(PutItemDocumentOperationRequest request)
+        {
+            var req = MapPutItemDocumentOperationRequestToPutItemRequest(request);
+
+#if NETSTANDARD
+            // Cast the IAmazonDynamoDB to the concrete client instead, so we can access the internal sync-over-async methods
+            var client = DDBClient as AmazonDynamoDBClient;
+            if (client == null)
+            {
+                throw new InvalidOperationException("Calling the synchronous PutItem from .NET or .NET Core requires initializing the Table " +
+                   "with an actual AmazonDynamoDBClient. You can use a mocked or substitute IAmazonDynamoDB when creating a Table via PutItemAsync instead.");
+            }
+#else
+            var client = DDBClient;
+#endif
+
+            var resp = client.PutItem(req);
+
+            request.Document.CommitChanges();
+            Document ret = null;
+            if (request.ReturnValues == ReturnValues.AllOldAttributes)
+            {
+                ret = this.FromAttributeMap(resp.Attributes);
+            }
+            return ret;
+        }
+
+
         internal async Task<Document> PutItemHelperAsync(Document doc, PutItemOperationConfig config, CancellationToken cancellationToken)
         {
             var currentConfig = config ?? new PutItemOperationConfig();
@@ -1282,6 +1323,31 @@ internal async Task<Document> PutItemHelperAsync(Document doc, PutItemOperationC
             return ret;
         }
 
+        private PutItemRequest MapPutItemDocumentOperationRequestToPutItemRequest(PutItemDocumentOperationRequest request)
+        {
+            if(request==null)
+                throw new ArgumentNullException(nameof(request));
+
+            if (request.Document == null)
+                throw new InvalidOperationException("The Document property of the PutItemDocumentOperationRequest cannot be null.");
+
+            PutItemRequest req = new PutItemRequest
+            {
+                TableName = TableName,
+                Item = this.ToAttributeMap(request.Document)
+            };
+
+            if (request.ReturnValues == ReturnValues.AllOldAttributes)
+                req.ReturnValues = EnumMapper.Convert(request.ReturnValues);
+
+            if (request.ConditionalExpression is { IsSet: true })
+            {
+                request.ConditionalExpression.ApplyExpression(req, this);
+            }
+
+            return req;
+        }
+
         #endregion
 
 

sdk/src/Services/DynamoDBv2/Custom/DocumentModel/_async/Table.Async.cs

@@ -36,7 +36,7 @@ public partial interface ITable
 
         /// <summary>
         /// Initiates the asynchronous execution of the PutItem operation.
-        /// <seealso cref="Amazon.DynamoDBv2.DocumentModel.Table.PutItem"/>
+        /// <seealso cref="Amazon.DynamoDBv2.DocumentModel.Table.PutItem(Document)"/>
         /// </summary>
         /// <param name="doc">Document to save.</param>
         /// <param name="cancellationToken">Token which can be used to cancel the task.</param>
@@ -45,14 +45,25 @@ public partial interface ITable
 
         /// <summary>
         /// Initiates the asynchronous execution of the PutItem operation.
-        /// <seealso cref="Amazon.DynamoDBv2.DocumentModel.Table.PutItem"/>
+        /// <seealso cref="Amazon.DynamoDBv2.DocumentModel.Table.PutItem(Document, PutItemOperationConfig)"/>
         /// </summary>
         /// <param name="doc">Document to save.</param>
         /// <param name="config">Configuration to use.</param>
         /// <param name="cancellationToken">Token which can be used to cancel the task.</param>
         /// <returns>A Task that can be used to poll or wait for results, or both.</returns>
         Task<Document> PutItemAsync(Document doc, PutItemOperationConfig config, CancellationToken cancellationToken = default(CancellationToken));
 
+        /// <summary>
+        /// Initiates the asynchronous execution of the PutItem operation.
+        /// <seealso cref="Table.PutItem(Amazon.DynamoDBv2.DocumentModel.Document,Amazon.DynamoDBv2.DocumentModel.PutItemOperationConfig)"/>
+        /// </summary>
+        /// <param name="request">The PutItemDocumentOperationRequest object containing all parameters for the PutItem operation.</param>
+        /// <param name="cancellationToken">Token which can be used to cancel the task.</param>
+        /// <returns>A Task that can be used to poll or wait for results, or both.</returns>
+        Task<Document> PutItemAsync(PutItemDocumentOperationRequest request, CancellationToken cancellationToken = default(CancellationToken));
+
+
+
         #endregion
 
         #region GetItemAsync
@@ -310,6 +321,16 @@ public partial class Table : ITable
             }
         }
 
+        /// <inheritdoc/>
+        public async Task<Document> PutItemAsync(PutItemDocumentOperationRequest request, CancellationToken cancellationToken = default(CancellationToken))
+        {
+            var operationName = DynamoDBTelemetry.ExtractOperationName(nameof(Table), nameof(PutItemAsync));
+            using (DynamoDBTelemetry.CreateSpan(TracerProvider, operationName, spanKind: SpanKind.CLIENT))
+            {
+                return await PutItemHelperAsync(request, cancellationToken).ConfigureAwait(false);
+            }
+        }
+
         #endregion
 
         #region GetItemAsync

sdk/src/Services/DynamoDBv2/Custom/DocumentModel/_bcl/Table.Sync.cs

@@ -39,6 +39,20 @@ public partial interface ITable
         /// <returns>True if put is successful or false if the condition in the config was not met.</returns>
         bool TryPutItem(Document doc, PutItemOperationConfig config = null);
 
+        /// <summary>
+        /// Puts a document into DynamoDB, using optional configs.
+        /// </summary>
+        /// <param name="request">The PutItemDocumentOperationRequest object containing all parameters for the PutItem operation.</param>
+        /// <returns>Null or updated attributes, depending on request config.</returns>
+        Document PutItem(PutItemDocumentOperationRequest request);
+
+        /// <summary>
+        /// Puts a document into DynamoDB, using optional configs.
+        /// </summary>
+        /// <param name="request">The PutItemDocumentOperationRequest object containing all parameters for the PutItem operation.</param>
+        /// <returns>True if put is successful or false if the condition was not met.</returns>
+        bool TryPutItem(PutItemDocumentOperationRequest request);
+
         #endregion
 
         #region GetItem
@@ -295,6 +309,35 @@ public bool TryPutItem(Document doc, PutItemOperationConfig config = null)
             }
         }
 
+        /// <inheritdoc/>
+        public Document PutItem(PutItemDocumentOperationRequest request)
+        {
+            var operationName = DynamoDBTelemetry.ExtractOperationName(nameof(Table), nameof(PutItem));
+            using (DynamoDBTelemetry.CreateSpan(TracerProvider, operationName, spanKind: SpanKind.CLIENT))
+            {
+                return PutItemHelper(request);
+            }
+        }
+
+        /// <inheritdoc/>
+        public bool TryPutItem(PutItemDocumentOperationRequest request)
+        {
+            var operationName = DynamoDBTelemetry.ExtractOperationName(nameof(Table), nameof(TryPutItem));
+            using (DynamoDBTelemetry.CreateSpan(TracerProvider, operationName, spanKind: SpanKind.CLIENT))
+            {
+                try
+                {
+                    PutItemHelper(request);
+                    return true;
+                }
+                catch (ConditionalCheckFailedException)
+                {
+                    return false;
+                }
+            }
+        }
+
+
         #endregion