Merge pull request #99 from kt81/var-virtual-node

Make DefaultNodeLocator take the ServerAddressMutations as a parameter

Author: cnblogs-dudu

Enyim.Caching/Memcached/Locators/DefaultNodeLocator.cs

@@ -11,7 +11,7 @@ namespace Enyim.Caching.Memcached
 	/// </summary>
 	public sealed class DefaultNodeLocator : IMemcachedNodeLocator, IDisposable
 	{
-		private const int ServerAddressMutations = 100;
+		private readonly int serverAddressMutations;
 
 		// holds all server keys for mapping an item key to the server consistently
 		private uint[] keys;
@@ -21,31 +21,36 @@ public sealed class DefaultNodeLocator : IMemcachedNodeLocator, IDisposable
 		private List<IMemcachedNode> allServers;
 		private ReaderWriterLockSlim serverAccessLock;
 
-		public DefaultNodeLocator()
+		public DefaultNodeLocator() : this(100)
+		{
+		}
+
+		public DefaultNodeLocator(int serverAddressMutations)
 		{
 			this.servers = new Dictionary<uint, IMemcachedNode>(new UIntEqualityComparer());
 			this.deadServers = new Dictionary<IMemcachedNode, bool>();
 			this.allServers = new List<IMemcachedNode>();
 			this.serverAccessLock = new ReaderWriterLockSlim();
+			this.serverAddressMutations = serverAddressMutations;
 		}
 
 		private void BuildIndex(List<IMemcachedNode> nodes)
 		{
-			var keys = new uint[nodes.Count * DefaultNodeLocator.ServerAddressMutations];
+			var keys = new uint[nodes.Count * this.serverAddressMutations];
 
 			int nodeIdx = 0;
 
 			foreach (IMemcachedNode node in nodes)
 			{
-				var tmpKeys = DefaultNodeLocator.GenerateKeys(node, DefaultNodeLocator.ServerAddressMutations);
+				var tmpKeys = DefaultNodeLocator.GenerateKeys(node, this.serverAddressMutations);
 
 				for (var i = 0; i < tmpKeys.Length; i++)
 				{
 					this.servers[tmpKeys[i]] = node;
 				}
 
 				tmpKeys.CopyTo(keys, nodeIdx);
-				nodeIdx += DefaultNodeLocator.ServerAddressMutations;
+				nodeIdx += this.serverAddressMutations;
 			}
 
 			Array.Sort<uint>(keys);

Enyim.Caching/Memcached/Locators/Factories/DefaultNodeLocatorFactory.cs

@@ -0,0 +1,26 @@
+using System.Collections.Generic;
+
+namespace Enyim.Caching.Memcached.LocatorFactories
+{
+    /// <summary>
+    /// Create DefaultNodeLocator with any ServerAddressMutations
+    /// </summary>
+    public class DefaultNodeLocatorFactory :IProviderFactory<IMemcachedNodeLocator>
+    {
+        private readonly int serverAddressMutations;
+
+        public DefaultNodeLocatorFactory(int serverAddressMutations)
+        {
+            this.serverAddressMutations = serverAddressMutations;
+        }
+
+        public IMemcachedNodeLocator Create()
+        {
+            return new DefaultNodeLocator(serverAddressMutations);
+        }
+
+        public void Initialize(Dictionary<string, string> parameters)
+        {
+        }
+    }
+}