From 699ec32683147ec4c7f89621acab6b124523983c Mon Sep 17 00:00:00 2001
From: Robert Andersson <inputfalken@gmail.com>
Date: Fri, 12 Jul 2024 11:59:59 +0200
Subject: [PATCH] Update sample

---
 samples/Repository/Program.cs                 | 20 ++++++++++++++-----
 .../Handlers/DeleteRequestHandler.cs          |  1 +
 2 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/samples/Repository/Program.cs b/samples/Repository/Program.cs
index 9c7b38c..86a975a 100644
--- a/samples/Repository/Program.cs
+++ b/samples/Repository/Program.cs
@@ -13,10 +13,11 @@
 
 public class ProductRepository
 {
-    private readonly IRequestHandler<string, Product?> _getProductByIdRequest;
+    private readonly IRequestHandler<string, Product?> _getById;
     private readonly IRequestHandler<(string Id, decimal NewPrice, DateTime TimeStamp), Product?> _updatePrice;
     private readonly IRequestHandler<Product, Product?> _createProduct;
     private readonly IRequestHandler<decimal, IReadOnlyList<Product>> _queryByPrice;
+    private readonly IRequestHandler<string, Product?> _deleteById;
 
     private class RequestLogger : IRequestPipeLine
     {
@@ -33,7 +34,7 @@ public async Task<AmazonWebServiceResponse> Invoke(RequestContext requestContext
     public ProductRepository(string tableName, IAmazonDynamoDB amazonDynamoDb)
     {
         var requestLogger = new RequestLogger();
-        _getProductByIdRequest = Product.GetById
+        _getById = Product.WithIdArgument
             .OnTable(tableName)
             .ToGetRequestHandler(
               x => x.ToGetRequestBuilder(),
@@ -45,6 +46,13 @@ public ProductRepository(string tableName, IAmazonDynamoDB amazonDynamoDb)
               }
             );
 
+        _deleteById = Product.WithIdArgument
+          .OnTable(tableName)
+          .ToDeleteRequestHandler(
+              x => x.ToDeleteRequestBuilder(),
+              x => x.AmazonDynamoDB = amazonDynamoDb
+          );
+
         _updatePrice = Product.UpdatePrice
             .OnTable(tableName)
             .ToUpdateRequestHandler(
@@ -74,7 +82,7 @@ public ProductRepository(string tableName, IAmazonDynamoDB amazonDynamoDb)
             );
 
         // You can also use a RequestBuilder if you want to handle the response yourself.
-        GetRequestBuilder<string> getProductByIdRequestBuilder = Product.GetById
+        GetRequestBuilder<string> getProductByIdRequestBuilder = Product.WithIdArgument
           .OnTable(tableName)
           .ToRequestBuilderFactory()
           .ToGetRequestBuilder();
@@ -84,14 +92,16 @@ public ProductRepository(string tableName, IAmazonDynamoDB amazonDynamoDb)
 
     public Task<Product?> Create(Product product) => _createProduct.Send(product, default);
 
-    public Task<Product?> GetById(string id) => _getProductByIdRequest.Send(id, default);
+    public Task<Product?> GetById(string id) => _getById.Send(id, default);
+    
+    public Task<Product?> DeleteById(string id) => _deleteById.Send(id, default);
 
     public Task<Product?> UpdatePrice(string id, decimal price) => _updatePrice.Send((id, price, DateTime.UtcNow), default);
 }
 
 // These attributes is what makes the source generator kick in. Make sure to have the class 'partial' as well.
 [DynamoDBMarshaller(AccessName = "Put")]
-[DynamoDBMarshaller(AccessName = "GetById", ArgumentType = typeof(string))]
+[DynamoDBMarshaller(AccessName = "WithIdArgument", ArgumentType = typeof(string))]
 [DynamoDBMarshaller(AccessName = "UpdatePrice", ArgumentType = typeof((string Id, decimal NewPrice, DateTime TimeStamp)))]
 [DynamoDBMarshaller(AccessName = "QueryByPrice", ArgumentType = typeof(decimal))]
 public partial record Product(
diff --git a/src/Dynatello/Handlers/DeleteRequestHandler.cs b/src/Dynatello/Handlers/DeleteRequestHandler.cs
index b505f6e..1313553 100644
--- a/src/Dynatello/Handlers/DeleteRequestHandler.cs
+++ b/src/Dynatello/Handlers/DeleteRequestHandler.cs
@@ -19,6 +19,7 @@ Func<Dictionary<string, AttributeValue>, T> createItem
         _createRequest = createRequest;
         _createItem = createItem;
     }
+    
     public async Task<T?> Send(TArg arg, CancellationToken cancellationToken)
     {
         var request = _createRequest(arg);