Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1560 from rabbitmq/lukebakken/misc-changes-from-a…
…sync-session-shutdown Misc changes
- Loading branch information
Showing
13 changed files
with
396 additions
and
213 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
88 changes: 88 additions & 0 deletions
88
projects/RabbitMQ.Client/client/impl/AsyncEventingWrapper.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,88 @@ | ||
| using System; | ||
| using System.Linq; | ||
| using System.Threading.Tasks; | ||
| using RabbitMQ.Client.Events; | ||
|
|
||
| namespace RabbitMQ.Client.Impl | ||
| { | ||
| #nullable enable | ||
| internal struct AsyncEventingWrapper<T> | ||
| { | ||
| private event AsyncEventHandler<T>? _event; | ||
| private Delegate[]? _handlers; | ||
| private string? _context; | ||
| private Func<Exception, string, Task>? _onException; | ||
|
|
||
| public readonly bool IsEmpty => _event is null; | ||
|
|
||
| public AsyncEventingWrapper(string context, Func<Exception, string, Task> onException) | ||
| { | ||
| _event = null; | ||
| _handlers = null; | ||
| _context = context; | ||
| _onException = onException; | ||
| } | ||
|
|
||
| public void AddHandler(AsyncEventHandler<T>? handler) | ||
| { | ||
| _event += handler; | ||
| _handlers = null; | ||
| } | ||
|
|
||
| public void RemoveHandler(AsyncEventHandler<T>? handler) | ||
| { | ||
| _event -= handler; | ||
| _handlers = null; | ||
| } | ||
|
|
||
| // Do not make this function async! (This type is a struct that gets copied at the start of an async method => empty _handlers is copied) | ||
| public Task InvokeAsync(object sender, T parameter) | ||
| { | ||
| Delegate[]? handlers = _handlers; | ||
| if (handlers is null) | ||
| { | ||
| handlers = _event?.GetInvocationList(); | ||
| if (handlers is null) | ||
| { | ||
| return Task.CompletedTask; | ||
| } | ||
|
|
||
| _handlers = handlers; | ||
| } | ||
|
|
||
| return InternalInvoke(handlers, sender, parameter); | ||
| } | ||
|
|
||
| private readonly async Task InternalInvoke(Delegate[] handlers, object sender, T parameter) | ||
| { | ||
| foreach (AsyncEventHandler<T> action in handlers.Cast<AsyncEventHandler<T>>()) | ||
| { | ||
| try | ||
| { | ||
| await action(sender, parameter) | ||
| .ConfigureAwait(false); | ||
| } | ||
| catch (Exception exception) | ||
| { | ||
| if (_onException != null) | ||
| { | ||
| await _onException(exception, _context!) | ||
| .ConfigureAwait(false); | ||
| } | ||
| else | ||
| { | ||
| throw; | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| public void Takeover(in AsyncEventingWrapper<T> other) | ||
| { | ||
| _event = other._event; | ||
| _handlers = other._handlers; | ||
| _context = other._context; | ||
| _onException = other._onException; | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.