Merge pull request #1684 from danielmarbach/disposable

AsyncDisposable

Author: lukebakken

projects/RabbitMQ.Client.OAuth2/RabbitMQ.Client.OAuth2.csproj

@@ -31,7 +31,7 @@
       https://learn.microsoft.com/en-us/answers/questions/1371494/for-net-standard-2-0-library-why-add-net-core-3-1
       https://devblogs.microsoft.com/dotnet/embracing-nullable-reference-types/#what-should-library-authors-do
     -->
-    <LangVersion>8.0</LangVersion>
+    <LangVersion>9.0</LangVersion>
     <Nullable>enable</Nullable>
   </PropertyGroup>
 

projects/RabbitMQ.Client.OpenTelemetry/RabbitMQ.Client.OpenTelemetry.csproj

@@ -27,7 +27,7 @@
     <GeneratePackageOnBuild>true</GeneratePackageOnBuild>
     <PackageOutputPath>../../packages</PackageOutputPath>
     <PackageReadmeFile>README.md</PackageReadmeFile>
-    <LangVersion>7.3</LangVersion>
+    <LangVersion>9.0</LangVersion>
   </PropertyGroup>
 
   <PropertyGroup Condition="'$(Configuration)' == 'Release' And '$(CI)' == 'true'">

projects/RabbitMQ.Client/IChannel.cs

@@ -42,10 +42,10 @@ namespace RabbitMQ.Client
     /// functionality offered by versions 0-8, 0-8qpid, 0-9 and 0-9-1 of AMQP.
     /// </summary>
     /// <remarks>
-    /// Extends the <see cref="IDisposable"/> interface, so that the "using"
+    /// Extends the <see cref="IDisposable"/> interface and the <see cref="IAsyncDisposable"/> interface, so that the "using"
     /// statement can be used to scope the lifetime of a channel when appropriate.
     /// </remarks>
-    public interface IChannel : IDisposable
+    public interface IChannel : IAsyncDisposable, IDisposable
     {
         /// <summary>
         /// Channel number, unique per connections.

projects/RabbitMQ.Client/IConnection.cs

@@ -50,12 +50,12 @@ namespace RabbitMQ.Client
     ///  Alternatively, an API tutorial can be found in the User Guide.
     /// </para>
     /// <para>
-    /// Extends the <see cref="IDisposable"/> interface, so that the "using"
+    /// Extends the <see cref="IDisposable"/> and the <see cref="IAsyncDisposable"/> interface, so that the "using"
     /// statement can be used to scope the lifetime of a connection when
     /// appropriate.
     /// </para>
     /// </remarks>
-    public interface IConnection : INetworkConnection, IDisposable
+    public interface IConnection : INetworkConnection, IAsyncDisposable, IDisposable
     {
         /// <summary>
         /// The maximum channel number this connection supports (0 if unlimited).

projects/RabbitMQ.Client/Impl/AutorecoveringChannel.cs

@@ -257,7 +257,9 @@ await _connection.DeleteRecordedChannelAsync(this,
         public override string ToString()
             => InnerChannel.ToString();
 
-        public void Dispose()
+        public void Dispose() => DisposeAsync().AsTask().GetAwaiter().GetResult();
+
+        public async ValueTask DisposeAsync()
         {
             if (_disposed)
             {
@@ -266,7 +268,8 @@ public void Dispose()
 
             if (IsOpen)
             {
-                this.AbortAsync().GetAwaiter().GetResult();
+                await this.AbortAsync()
+                    .ConfigureAwait(false);
             }
 
             _recordedConsumerTags.Clear();

projects/RabbitMQ.Client/Impl/AutorecoveringConnection.Recovery.cs

@@ -299,11 +299,12 @@ private async ValueTask RecoverExchangesAsync(IConnection connection,
             {
                 try
                 {
-                    using (IChannel ch = await connection.CreateChannelAsync(cancellationToken: cancellationToken).ConfigureAwait(false))
+                    IChannel channel = await connection.CreateChannelAsync(cancellationToken: cancellationToken).ConfigureAwait(false);
+                    await using (channel.ConfigureAwait(false))
                     {
-                        await recordedExchange.RecoverAsync(ch, cancellationToken)
+                        await recordedExchange.RecoverAsync(channel, cancellationToken)
                             .ConfigureAwait(false);
-                        await ch.CloseAsync(cancellationToken)
+                        await channel.CloseAsync(cancellationToken)
                             .ConfigureAwait(false);
                     }
                 }
@@ -351,11 +352,12 @@ private async Task RecoverQueuesAsync(IConnection connection,
                 try
                 {
                     string newName = string.Empty;
-                    using (IChannel ch = await connection.CreateChannelAsync(cancellationToken: cancellationToken).ConfigureAwait(false))
+                    IChannel channel = await connection.CreateChannelAsync(cancellationToken: cancellationToken).ConfigureAwait(false);
+                    await using (channel.ConfigureAwait(false))
                     {
-                        newName = await recordedQueue.RecoverAsync(ch, cancellationToken)
+                        newName = await recordedQueue.RecoverAsync(channel, cancellationToken)
                             .ConfigureAwait(false);
-                        await ch.CloseAsync(cancellationToken)
+                        await channel.CloseAsync(cancellationToken)
                             .ConfigureAwait(false);
                     }
                     string oldName = recordedQueue.Name;
@@ -463,11 +465,12 @@ private async ValueTask RecoverBindingsAsync(IConnection connection,
             {
                 try
                 {
-                    using (IChannel ch = await connection.CreateChannelAsync(cancellationToken: cancellationToken).ConfigureAwait(false))
+                    IChannel channel = await connection.CreateChannelAsync(cancellationToken: cancellationToken).ConfigureAwait(false);
+                    await using (channel.ConfigureAwait(false))
                     {
-                        await binding.RecoverAsync(ch, cancellationToken)
+                        await binding.RecoverAsync(channel, cancellationToken)
                             .ConfigureAwait(false);
-                        await ch.CloseAsync(cancellationToken)
+                        await channel.CloseAsync(cancellationToken)
                             .ConfigureAwait(false);
                     }
                 }

projects/RabbitMQ.Client/Impl/AutorecoveringConnection.cs

@@ -264,7 +264,9 @@ await RecordChannelAsync(channel, channelsSemaphoreHeld: false, cancellationToke
             return channel;
         }
 
-        public void Dispose()
+        public void Dispose() => DisposeAsync().AsTask().GetAwaiter().GetResult();
+
+        public async ValueTask DisposeAsync()
         {
             if (_disposed)
             {
@@ -273,7 +275,8 @@ public void Dispose()
 
             try
             {
-                _innerConnection.Dispose();
+                await _innerConnection.DisposeAsync()
+                    .ConfigureAwait(false);
             }
             catch (OperationInterruptedException)
             {

projects/RabbitMQ.Client/Impl/ChannelBase.cs

@@ -568,6 +568,26 @@ protected virtual void Dispose(bool disposing)
             }
         }
 
+        public async ValueTask DisposeAsync()
+        {
+            await DisposeAsyncCore()
+                .ConfigureAwait(false);
+
+            Dispose(false);
+        }
+
+        protected virtual async ValueTask DisposeAsyncCore()
+        {
+            if (IsOpen)
+            {
+                await this.AbortAsync().ConfigureAwait(false);
+            }
+
+            ConsumerDispatcher.Dispose();
+            _rpcSemaphore.Dispose();
+            _confirmSemaphore?.Dispose();
+        }
+
         public Task ConnectionTuneOkAsync(ushort channelMax, uint frameMax, ushort heartbeat, CancellationToken cancellationToken)
         {
             var method = new ConnectionTuneOk(channelMax, frameMax, heartbeat);

projects/RabbitMQ.Client/Impl/Connection.cs

@@ -485,7 +485,9 @@ internal ValueTask WriteAsync(RentedMemory frames, CancellationToken cancellatio
             return _frameHandler.WriteAsync(frames, cancellationToken);
         }
 
-        public void Dispose()
+        public void Dispose() => DisposeAsync().AsTask().GetAwaiter().GetResult();
+
+        public async ValueTask DisposeAsync()
         {
             if (_disposed)
             {
@@ -496,7 +498,8 @@ public void Dispose()
             {
                 if (IsOpen)
                 {
-                    this.AbortAsync().GetAwaiter().GetResult();
+                    await this.AbortAsync()
+                        .ConfigureAwait(false);
                 }
 
                 _session0.Dispose();

projects/Test/Applications/CreateChannel/CreateChannel.csproj

@@ -14,6 +14,7 @@
 
   <PropertyGroup>
     <OutputType>Exe</OutputType>
+    <LangVersion>9.0</LangVersion>
   </PropertyGroup>
 
   <ItemGroup>