Add CreateCustomGrainDirectoryCache in GrainDirectoryCacheFactory to …

…get IGrainDirectoryCache from DI or create the default LRUBasedGrainDirectoryCache.

Update CachedGrainLocator to use GrainDirectoryCacheFactory.CreateCustomGrainDirectoryCache to create inner cache.

src/Orleans.Runtime/GrainDirectory/CachedGrainLocator.cs

@@ -38,7 +38,7 @@ public CachedGrainLocator(
         {
             this.grainDirectoryResolver = grainDirectoryResolver;
             this.clusterMembershipService = clusterMembershipService;
-            this.cache = GrainDirectoryCacheFactory.CreateGrainDirectoryCache(serviceProvider, grainDirectoryOptions.Value);
+            this.cache = GrainDirectoryCacheFactory.CreateCustomGrainDirectoryCache(serviceProvider, grainDirectoryOptions.Value);
         }
 
         public async ValueTask<GrainAddress> Lookup(GrainId grainId)

src/Orleans.Runtime/GrainDirectory/GrainDirectoryCacheFactory.cs

@@ -36,6 +36,25 @@ public static IGrainDirectoryCache CreateGrainDirectoryCache(IServiceProvider se
             }
         }
 
+        /// <summary>
+        /// Creates <see cref="IGrainDirectoryCache"/> from DI or defaults to <see cref="LRUBasedGrainDirectoryCache"/> if no <see cref="IGrainDirectoryCache"/> found in DI.
+        /// </summary>
+        /// <param name="services">The services.</param>
+        /// <param name="options">The options.</param>
+        /// <returns>The newly created <see cref="IGrainDirectoryCache"/> instance.</returns>
+        public static IGrainDirectoryCache CreateCustomGrainDirectoryCache(IServiceProvider services, GrainDirectoryOptions options)
+        {
+            var grainDirectoryCache = services.GetService<IGrainDirectoryCache>();
+            if (grainDirectoryCache != null)
+            {
+                return grainDirectoryCache;
+            }
+            else
+            {
+                return new LRUBasedGrainDirectoryCache(options.CacheSize, options.MaximumCacheTTL);
+            }
+        }
+
         internal static AdaptiveDirectoryCacheMaintainer CreateGrainDirectoryCacheMaintainer(
             LocalGrainDirectory router,
             IGrainDirectoryCache cache,