-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathProgram.cs
114 lines (91 loc) · 4.32 KB
/
Program.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
using System.Net;
using Amazon.DynamoDBv2;
using Amazon.DynamoDBv2.DataModel;
using Amazon.DynamoDBv2.Model;
using DynamoDBGenerator.Attributes;
using Dynatello;
using Dynatello.Builders;
using Dynatello.Builders.Types;
ProductRepository productRepository = new ProductRepository("MY_TABLE", new AmazonDynamoDBClient());
public class ProductRepository
{
private readonly IAmazonDynamoDB _amazonDynamoDb;
private readonly GetRequestBuilder<string> _getProductByTable;
private readonly UpdateRequestBuilder<(string Id, decimal NewPrice, DateTime TimeStamp)> _updatePrice;
private readonly PutRequestBuilder<Product> _createProduct;
private readonly QueryRequestBuilder<decimal> _queryByPrice;
public ProductRepository(string tableName, IAmazonDynamoDB amazonDynamoDb)
{
_amazonDynamoDb = amazonDynamoDb;
_getProductByTable = Product.GetById
.OnTable(tableName)
.ToGetRequestBuilder(arg => arg); // Since the ArgumentType is set to string, we don't need to select a property.
_updatePrice = Product.UpdatePrice
.OnTable(tableName)
.WithUpdateExpression((db, arg) => $"SET {db.Price} = {arg.NewPrice}, {db.Metadata.ModifiedAt} = {arg.TimeStamp}") // Specify the update operation
.ToUpdateItemRequestBuilder((marshaller, arg) => marshaller.PartitionKey(arg.Id));
_createProduct = Product.Put
.OnTable(tableName)
.WithConditionExpression((db, arg) => $"{db.Id} <> {arg.Id}") // Ensure we don't have an existing Product in DynamoDB
.ToPutRequestBuilder();
_queryByPrice = Product.QueryByPrice
.OnTable(tableName)
.WithKeyConditionExpression((db, arg) => $"{db.Price} = {arg}")
.ToQueryRequestBuilder()
with
{
IndexName = Product.PriceIndex
};
}
public async Task<IReadOnlyList<Product>> SearchByPrice(decimal price)
{
QueryRequest request = _queryByPrice.Build(price);
QueryResponse? response = await _amazonDynamoDb.QueryAsync(request);
if (response.HttpStatusCode is not HttpStatusCode.OK)
throw new Exception("...");
return response.Items
.Select(x => Product.QueryByPrice.Unmarshall(x))
.ToArray();
}
public async Task Create(Product product)
{
PutItemRequest request = _createProduct.Build(product);
PutItemResponse response = await _amazonDynamoDb.PutItemAsync(request);
if (response.HttpStatusCode is not HttpStatusCode.OK)
throw new Exception("...");
}
public async Task<Product?> GetById(string id)
{
GetItemRequest request = _getProductByTable.Build(id);
GetItemResponse response = await _amazonDynamoDb.GetItemAsync(request);
if (response.HttpStatusCode is HttpStatusCode.NotFound)
return null;
if (response.HttpStatusCode is not HttpStatusCode.OK)
throw new Exception("...");
Product product = Product.GetById.Unmarshall(response.Item);
return product;
}
public async Task<Product?> UpdatePrice(string id, decimal price)
{
UpdateItemRequest request = _updatePrice.Build((id, price, DateTime.UtcNow));
UpdateItemResponse response = await _amazonDynamoDb.UpdateItemAsync(request);
if (response.HttpStatusCode is not HttpStatusCode.OK)
return null;
Product product = Product.UpdatePrice.Unmarshall(response.Attributes);
return product;
}
}
[DynamoDBMarshaller(typeof(Product), PropertyName = "Put")]
[DynamoDBMarshaller(typeof(Product), ArgumentType = typeof(string), PropertyName = "GetById")]
[DynamoDBMarshaller(typeof(Product), ArgumentType = typeof((string Id, decimal NewPrice, DateTime TimeStamp)), PropertyName = "UpdatePrice")]
[DynamoDBMarshaller(typeof(Product), ArgumentType = typeof(decimal), PropertyName = "QueryByPrice")]
public partial record Product(
[property: DynamoDBHashKey, DynamoDBGlobalSecondaryIndexRangeKey(Product.PriceIndex)] string Id,
[property: DynamoDBGlobalSecondaryIndexHashKey(Product.PriceIndex)] decimal Price,
string Description,
Product.MetadataEntity Metadata
)
{
public const string PriceIndex = "PriceIndex";
public record MetadataEntity(DateTime CreatedAt, DateTime ModifiedAt);
}