@@ -15,7 +15,7 @@ public ValueTask<Dictionary<string, List<string>>> GetSubscriptions(string? agen
15
15
}
16
16
return new ValueTask < Dictionary < string , List < string > > > ( _subscriptions ) ;
17
17
}
18
- public ValueTask Subscribe ( string agentType , string topic )
18
+ public async ValueTask SubscribeAsync ( string agentType , string topic )
19
19
{
20
20
if ( ! _subscriptions . TryGetValue ( topic , out var subscriptions ) )
21
21
{
@@ -27,11 +27,9 @@ public ValueTask Subscribe(string agentType, string topic)
27
27
}
28
28
_subscriptions [ topic ] = subscriptions ;
29
29
state . State . Subscriptions = _subscriptions ;
30
- state . WriteStateAsync ( ) ;
31
-
32
- return ValueTask . CompletedTask ;
30
+ await state . WriteStateAsync ( ) . ConfigureAwait ( false ) ;
33
31
}
34
- public ValueTask Unsubscribe ( string agentType , string topic )
32
+ public async ValueTask UnsubscribeAsync ( string agentType , string topic )
35
33
{
36
34
if ( ! _subscriptions . TryGetValue ( topic , out var subscriptions ) )
37
35
{
@@ -43,8 +41,7 @@ public ValueTask Unsubscribe(string agentType, string topic)
43
41
}
44
42
_subscriptions [ topic ] = subscriptions ;
45
43
state . State . Subscriptions = _subscriptions ;
46
- state . WriteStateAsync ( ) ;
47
- return ValueTask . CompletedTask ;
44
+ await state . WriteStateAsync ( ) ;
48
45
}
49
46
}
50
47
public sealed class SubscriptionsState
0 commit comments