Tweak services.AddEnyimMemcached() and app.UseEnyimMemcached()

Author: cnblogs-dudu

Enyim.Caching/EnyimMemcachedApplicationBuilderExtensions.cs

@@ -5,23 +5,27 @@
 using System.Threading.Tasks;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
+using Enyim.Caching.Configuration;
+using Enyim.Caching.Memcached;
 
 namespace Microsoft.AspNetCore.Builder
 {
     public static class EnyimMemcachedApplicationBuilderExtensions
     {
         public static IApplicationBuilder UseEnyimMemcached(this IApplicationBuilder app)
         {
+            var logger = app.ApplicationServices.GetService<ILogger<IMemcachedClient>>();
             try
             {
-                app.ApplicationServices.GetService<IMemcachedClient>()
-                    .GetAsync<string>("EnyimMemcached").Wait();
-                Console.WriteLine("EnyimMemcached Started.");
-            } catch (Exception ex)
+                var client = app.ApplicationServices.GetRequiredService<IMemcachedClient>();
+                client.GetValueAsync<string>("UseEnyimMemcached").Wait();
+                Console.WriteLine("EnyimMemcached connected memcached servers.");
+            }
+            catch (Exception ex)
             {
-                app.ApplicationServices.GetService<ILogger<IMemcachedClient>>()
-                    .LogError(new EventId(), ex, "EnyimMemcached Failed.");
+                logger.LogError(ex, "Failed in UseEnyimMemcached");
             }
+
             return app;
         }
     }

Enyim.Caching/EnyimMemcachedServiceCollectionExtensions.cs

@@ -86,7 +86,7 @@ private static IServiceCollection AddEnyimMemcachedInternal(IServiceCollection s
             services.TryAddSingleton<ITranscoder, DefaultTranscoder>();
             services.TryAddSingleton<IMemcachedKeyTransformer, DefaultKeyTransformer>();
             services.TryAddTransient<IMemcachedClientConfiguration, MemcachedClientConfiguration>();
-            services.AddSingleton<MemcachedClient, MemcachedClient>();
+            services.AddSingleton<MemcachedClient>();
 
             services.AddSingleton<IMemcachedClient>(factory => factory.GetService<MemcachedClient>());
             services.AddSingleton<IDistributedCache>(factory => factory.GetService<MemcachedClient>());