Implement Operation.ReadResponseAsync()

Author: cnblogs-dudu

Enyim.Caching/Enyim.Caching.csproj

@@ -25,5 +25,6 @@
     <PackageReference Include="Microsoft.Extensions.Caching.Abstractions" Version="2.1.2" />
     <PackageReference Include="Microsoft.AspNetCore.Http.Abstractions" Version="2.1.1" />
     <PackageReference Include="Newtonsoft.Json.Bson" Version="1.0.1" />
+    <PackageReference Include="System.Threading.Tasks.Extensions" Version="4.5.1" />
   </ItemGroup>
 </Project>

Enyim.Caching/Memcached/PooledSocket.cs

@@ -1,4 +1,3 @@
-using Microsoft.Extensions.Logging;
 using System;
 using System.Collections.Generic;
 using System.Diagnostics;
@@ -11,6 +10,7 @@
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
+using Microsoft.Extensions.Logging;
 
 namespace Enyim.Caching.Memcached
 {
@@ -208,6 +208,21 @@ public int ReadByte()
             }
         }
 
+        public int ReadByteAsync()
+        {
+            this.CheckDisposed();
+
+            try
+            {
+                return this.inputStream.ReadByte();
+            }
+            catch (IOException)
+            {
+                this.isAlive = false;
+                throw;
+            }
+        }
+
         public async Task<byte[]> ReadBytesAsync(int count)
         {
             var buffer = new ArraySegment<byte>(new byte[count], 0, count);

Enyim.Caching/Memcached/Protocol/Binary/BinaryOperation.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Threading.Tasks;
 
 namespace Enyim.Caching.Memcached.Protocol.Binary
 {
@@ -12,15 +13,10 @@ protected internal override IList<ArraySegment<byte>> GetBuffer()
             return this.Build().CreateBuffer();
         }
 
-        protected internal override System.Threading.Tasks.Task<Results.IOperationResult> ReadResponseAsync(PooledSocket socket)
-        {
-            throw new NotImplementedException();
-        }
-
         protected internal override bool ReadResponseAsync(PooledSocket socket, System.Action<bool> next)
         {
             throw new System.NotSupportedException();
-        }       
+        }
     }
 }
 

Enyim.Caching/Memcached/Protocol/Binary/BinarySingleItemOperation.cs

@@ -44,7 +44,7 @@ protected internal override IOperationResult ReadResponse(PooledSocket socket)
             return result;
         }
 
-        protected internal override async Task<IOperationResult> ReadResponseAsync(PooledSocket socket)
+        protected internal override async ValueTask<IOperationResult> ReadResponseAsync(PooledSocket socket)
         {
             var response = new BinaryResponse();
             var retval = await response.ReadAsync(socket);

Enyim.Caching/Memcached/Protocol/Binary/FlushOperation.cs

@@ -1,37 +1,54 @@
 using System;
 using System.Collections.Generic;
+using System.Threading.Tasks;
 using Enyim.Caching.Memcached.Results;
 using Enyim.Caching.Memcached.Results.Extensions;
 
 namespace Enyim.Caching.Memcached.Protocol.Binary
 {
-	public class FlushOperation : BinaryOperation, IFlushOperation
-	{
-		public FlushOperation() { }
-
-		protected override BinaryRequest Build()
-		{
-			var request = new BinaryRequest(OpCode.Flush);
-
-			return request;
-		}
-
-		protected internal override IOperationResult ReadResponse(PooledSocket socket)
-		{
-			var response = new BinaryResponse();
-			var retval = response.Read(socket);
-
-			this.StatusCode = StatusCode;
-			var result = new BinaryOperationResult()
-			{
-				Success = retval,
-				StatusCode = this.StatusCode
-			};
-
-			result.PassOrFail(retval, "Failed to read response");
-			return result;
-		}
-	}
+    public class FlushOperation : BinaryOperation, IFlushOperation
+    {
+        public FlushOperation() { }
+
+        protected override BinaryRequest Build()
+        {
+            var request = new BinaryRequest(OpCode.Flush);
+
+            return request;
+        }
+
+        protected internal override IOperationResult ReadResponse(PooledSocket socket)
+        {
+            var response = new BinaryResponse();
+            var retval = response.Read(socket);
+
+            this.StatusCode = StatusCode;
+            var result = new BinaryOperationResult()
+            {
+                Success = retval,
+                StatusCode = this.StatusCode
+            };
+
+            result.PassOrFail(retval, "Failed to read response");
+            return result;
+        }
+
+        protected internal override async ValueTask<IOperationResult> ReadResponseAsync(PooledSocket socket)
+        {
+            var response = new BinaryResponse();
+            var retval = await response.ReadAsync(socket);
+
+            this.StatusCode = StatusCode;
+            var result = new BinaryOperationResult()
+            {
+                Success = retval,
+                StatusCode = this.StatusCode
+            };
+
+            result.PassOrFail(retval, "Failed to read response");
+            return result;
+        }
+    }
 }
 
 #region [ License information          ]