コード例 #1
0
        public DefaultLoadBalancerContext(LoadBalancerManager loadBalancerManager, string loadBalancerId, LoadBalancerConfig loadBalancerConfig)
        {
            LoadBalancerManager  = loadBalancerManager;
            ConfigurationManager = loadBalancerManager.LoadBalancerManagerConfig.ConfigurationManager;
            EventMetricManager   = loadBalancerManager.LoadBalancerManagerConfig.EventMetricManager;
            AuditMetricManager   = loadBalancerManager.LoadBalancerManagerConfig.AuditMetricManager;
            StatusMetricManager  = loadBalancerManager.LoadBalancerManagerConfig.StatusMetricManager;

            ManagerId      = loadBalancerManager.ManagerId;
            LoadBalancerId = loadBalancerId;
            Ping           = loadBalancerConfig.Ping;
            ServerSource   = loadBalancerConfig.ServerSource;
            AdditionalInfo = new Dictionary <string, string>()
            {
                { "LoadBalancerKey", LoadBalancerKey }
            };

            Initialize(loadBalancerConfig);
        }
コード例 #2
0
        public void WeightedRoundRobinTest()
        {
            int    serverCount    = 10;
            int    repeatTimes    = 100;
            int    repeatInterval = 0;
            int    threadCount    = 20;
            int    serverSumCount = 0;
            string loadBalancerId = MethodInfo.GetCurrentMethod().Name;

            IConfiguration configuration        = ObjectFactory.CreateAppSettingConfiguration();
            var            configurationSource  = ObjectFactory.CreateDefaultConfigurationSource(0, "App.config", configuration);
            var            configurationManager = ObjectFactory.CreateDefaultConfigurationManager(configurationSource);

            Dictionary <int, AtomicInteger> indexCountMapping = new Dictionary <int, AtomicInteger>();
            List <ServerGroup> serverGroups = new List <ServerGroup>();

            for (int i = 0; i < serverCount; i++)
            {
                var metadata = new Dictionary <string, string>();
                metadata["Index"] = i.ToString();
                var server = new Server("Server_" + i, metadata);
                serverGroups.Add(new ServerGroup(server.ServerId, i, new List <Server>()
                {
                    server
                }));
                indexCountMapping[i] = 0;
                serverSumCount      += i;
            }

            var ping                = new TruePing();
            var serverSource        = new DefaultDynamicServerSource(serverGroups);
            var factory             = LoadBalancerManager.GetManager("soa", new LoadBalancerManagerConfig(configurationManager));
            var loadBalancerConfig  = new LoadBalancerConfig(ping, serverSource);
            var loadBalancer        = factory.GetLoadBalancer(loadBalancerId, loadBalancerConfig);
            var defaultLoadBalancer = ((DefaultLoadBalancer)loadBalancer).LoadBalancerContext;

            Action action = () =>
            {
                for (int i = 0; i < repeatTimes; i++)
                {
                    for (int j = 0; j < serverSumCount; j++)
                    {
                        var requestContext = loadBalancer.GetRequestContext(requestConfig);
                        int index          = int.Parse(requestContext.Server.Metadata["Index"]);
                        indexCountMapping[index].IncrementAndGet();
                        if (repeatInterval > 0)
                        {
                            Thread.Sleep(repeatInterval);
                        }
                    }
                }
            };

            List <Task> tasks = new List <Task>();

            for (int i = 0; i < threadCount; i++)
            {
                tasks.Add(Task.Factory.StartNew(action));
            }
            Task.WaitAll(tasks.ToArray());

            foreach (var item in indexCountMapping)
            {
                Assert.AreEqual(item.Key * repeatTimes * threadCount, item.Value.Value, item.Key.ToString());
            }
        }
コード例 #3
0
        public void WeightedRoundRobinWithSpecificWeightTest()
        {
            int    serverCount    = 10;
            int    repeatTimes    = 100;
            int    repeatInterval = 0;
            int    threadCount    = 20;
            int    serverSumCount = 0;
            string loadBalancerId = MethodInfo.GetCurrentMethod().Name;

            IConfiguration configuration        = ObjectFactory.CreateAppSettingConfiguration();
            var            configurationSource  = ObjectFactory.CreateDefaultConfigurationSource(0, "App.config", configuration);
            var            configurationManager = ObjectFactory.CreateDefaultConfigurationManager(configurationSource);

            Dictionary <int, AtomicInteger> indexCountMapping = new Dictionary <int, AtomicInteger>();
            List <ServerGroup> serverGroups = new List <ServerGroup>();
            List <Server>      servers      = new List <Server>();

            for (int i = 0; i < serverCount; i++)
            {
                var metadata = new Dictionary <string, string>();
                metadata["Index"] = i.ToString();
                var server = new Server("Server_" + i, metadata);
                servers.Add(server);
                serverGroups.Add(new ServerGroup(server.ServerId, i, new List <Server>()
                {
                    server
                }));
                indexCountMapping[i] = 0;
                serverSumCount      += i;
            }
            LoadBalancerRoute route0 = new LoadBalancerRoute("default", serverGroups);

            ServerGroup route1Group0 = new ServerGroup("group0", 2, new List <Server>()
            {
                servers[0]
            });
            ServerGroup route1Group1 = new ServerGroup("group1", 1, new List <Server>()
            {
                servers[1]
            });
            LoadBalancerRoute route1 = new LoadBalancerRoute("route1", new List <ServerGroup>()
            {
                route1Group0, route1Group1
            });

            ServerGroup route2Group0 = new ServerGroup("group0", 2, new List <Server>()
            {
                servers[0]
            });
            ServerGroup route2Group1 = new ServerGroup("group1", 3, new List <Server>()
            {
                servers[1]
            });
            LoadBalancerRoute route2 = new LoadBalancerRoute("route2", new List <ServerGroup>()
            {
                route2Group0, route2Group1
            });

            LoadBalancerRequestConfig route1Config = new LoadBalancerRequestConfig(new List <LoadBalancerRouteConfig>()
            {
                new LoadBalancerRouteConfig("route1", 1, false)
            });
            LoadBalancerRequestConfig route2Config = new LoadBalancerRequestConfig(new List <LoadBalancerRouteConfig>()
            {
                new LoadBalancerRouteConfig("route2", 1, false)
            });

            int server0Count       = route1Group0.Weight + route2Group0.Weight;
            int server1Count       = route1Group1.Weight + route2Group1.Weight;
            int route1RequestCount = route1Group0.Weight + route1Group1.Weight;
            int route2RequestCount = route2Group0.Weight + route2Group1.Weight;

            var ping         = new TruePing();
            var serverSource = new DefaultDynamicServerSource(new List <LoadBalancerRoute>()
            {
                route0, route1, route2
            });
            var factory             = LoadBalancerManager.GetManager("soa", new LoadBalancerManagerConfig(configurationManager));
            var loadBalancerConfig  = new LoadBalancerConfig(ping, serverSource);
            var loadBalancer        = factory.GetLoadBalancer(loadBalancerId, loadBalancerConfig);
            var defaultLoadBalancer = ((DefaultLoadBalancer)loadBalancer).LoadBalancerContext;

            Action action = () =>
            {
                for (int i = 0; i < repeatTimes; i++)
                {
                    for (int j = 0; j < serverSumCount; j++)
                    {
                        var requestContext = loadBalancer.GetRequestContext(requestConfig);
                        int index          = int.Parse(requestContext.Server.Metadata["Index"]);
                        indexCountMapping[index].IncrementAndGet();
                        if (repeatInterval > 0)
                        {
                            Thread.Sleep(repeatInterval);
                        }
                    }

                    for (int j = 0; j < route1RequestCount; j++)
                    {
                        var requestContext = loadBalancer.GetRequestContext(route1Config);
                        int index          = int.Parse(requestContext.Server.Metadata["Index"]);
                        indexCountMapping[index].IncrementAndGet();
                        if (repeatInterval > 0)
                        {
                            Thread.Sleep(repeatInterval);
                        }
                    }

                    for (int j = 0; j < route2RequestCount; j++)
                    {
                        var requestContext = loadBalancer.GetRequestContext(route2Config);
                        int index          = int.Parse(requestContext.Server.Metadata["Index"]);
                        indexCountMapping[index].IncrementAndGet();
                        if (repeatInterval > 0)
                        {
                            Thread.Sleep(repeatInterval);
                        }
                    }
                }
            };

            List <Task> tasks = new List <Task>();

            for (int i = 0; i < threadCount; i++)
            {
                tasks.Add(Task.Factory.StartNew(action));
            }
            Task.WaitAll(tasks.ToArray());

            foreach (var item in indexCountMapping)
            {
                if (item.Key == 0)
                {
                    Assert.AreEqual((item.Key + server0Count) * repeatTimes * threadCount, item.Value.Value, item.Key.ToString());
                }
                else if (item.Key == 1)
                {
                    Assert.AreEqual((item.Key + server1Count) * repeatTimes * threadCount, item.Value.Value, item.Key.ToString());
                }
                else
                {
                    Assert.AreEqual(item.Key * repeatTimes * threadCount, item.Value.Value, item.Key.ToString());
                }
            }
        }