| | | 1 | | namespace TeleFlow.Framework.States; |
| | | 2 | | |
| | | 3 | | public sealed class UpdateStateData |
| | | 4 | | { |
| | | 5 | | private readonly IStateDataStore _store; |
| | | 6 | | private readonly IStateDataSerializer _serializer; |
| | | 7 | | |
| | | 8 | | public UpdateStateData( |
| | | 9 | | IStateDataStore store, |
| | | 10 | | IStateDataSerializer serializer, |
| | | 11 | | StateKey key) |
| | | 12 | | { |
| | 16 | 13 | | ArgumentNullException.ThrowIfNull(store); |
| | 16 | 14 | | ArgumentNullException.ThrowIfNull(serializer); |
| | | 15 | | |
| | 16 | 16 | | _store = store; |
| | 16 | 17 | | _serializer = serializer; |
| | | 18 | | Key = key; |
| | 16 | 19 | | } |
| | | 20 | | |
| | | 21 | | public StateKey Key { get; } |
| | | 22 | | |
| | | 23 | | public async ValueTask<TValue?> GetAsync<TValue>( |
| | | 24 | | string key, |
| | | 25 | | CancellationToken cancellationToken = default) |
| | | 26 | | { |
| | 7 | 27 | | ValidateDataKey(key); |
| | | 28 | | |
| | 6 | 29 | | var value = await _store.GetDataAsync(Key, key, cancellationToken).ConfigureAwait(false); |
| | 6 | 30 | | return value is null |
| | 6 | 31 | | ? default |
| | 6 | 32 | | : _serializer.Deserialize<TValue>(value); |
| | 6 | 33 | | } |
| | | 34 | | |
| | | 35 | | public async ValueTask<TValue> GetRequiredAsync<TValue>( |
| | | 36 | | string key, |
| | | 37 | | CancellationToken cancellationToken = default) |
| | | 38 | | { |
| | 8 | 39 | | ValidateDataKey(key); |
| | | 40 | | |
| | 8 | 41 | | var value = await _store.GetDataAsync(Key, key, cancellationToken).ConfigureAwait(false); |
| | 8 | 42 | | if (value is null) |
| | | 43 | | { |
| | 1 | 44 | | throw new KeyNotFoundException($"State data key '{key}' was not found."); |
| | | 45 | | } |
| | | 46 | | |
| | 7 | 47 | | var result = _serializer.Deserialize<TValue>(value); |
| | 7 | 48 | | if (result is null) |
| | | 49 | | { |
| | 0 | 50 | | throw new InvalidOperationException($"State data key '{key}' deserialized to null."); |
| | | 51 | | } |
| | | 52 | | |
| | 7 | 53 | | return result; |
| | 7 | 54 | | } |
| | | 55 | | |
| | | 56 | | public ValueTask SetAsync<TValue>( |
| | | 57 | | string key, |
| | | 58 | | TValue value, |
| | | 59 | | CancellationToken cancellationToken = default) |
| | | 60 | | { |
| | 12 | 61 | | ValidateDataKey(key); |
| | 12 | 62 | | ArgumentNullException.ThrowIfNull(value); |
| | | 63 | | |
| | 11 | 64 | | var serialized = _serializer.Serialize(value); |
| | 11 | 65 | | return _store.SetDataAsync(Key, key, serialized, cancellationToken); |
| | | 66 | | } |
| | | 67 | | |
| | | 68 | | public ValueTask RemoveAsync( |
| | | 69 | | string key, |
| | | 70 | | CancellationToken cancellationToken = default) |
| | | 71 | | { |
| | 0 | 72 | | ValidateDataKey(key); |
| | 0 | 73 | | return _store.RemoveDataAsync(Key, key, cancellationToken); |
| | | 74 | | } |
| | | 75 | | |
| | | 76 | | public ValueTask ClearAsync(CancellationToken cancellationToken = default) |
| | | 77 | | { |
| | 9 | 78 | | return _store.ClearDataAsync(Key, cancellationToken); |
| | | 79 | | } |
| | | 80 | | |
| | | 81 | | private static void ValidateDataKey(string key) |
| | | 82 | | { |
| | 27 | 83 | | ArgumentException.ThrowIfNullOrWhiteSpace(key); |
| | 26 | 84 | | } |
| | | 85 | | } |