如何在 Java 中利用数组实现简单的权重轮询算法以分配不同服务器的访问流量

云雪小哥_6481

云雪小哥_6481

2026-05-19

616人浏览

原创

权重轮询通过累计权重与取模计算实现O(1)空间调度,避免重复存储;用servers[]、weights[]数组和AtomicInteger维护currentIndex,按权重比例循环返回服务器。

如何在 java 中利用数组实现简单的权重轮询算法以分配不同服务器的访问流量

权重轮询(Weighted Round Robin)是一种常见的负载均衡策略,它根据服务器的权重值按比例分配请求。在 Java 中,不依赖第三方框架,仅用数组就能高效实现——核心在于用累计权重替代重复存储,避免内存浪费,同时保证分配结果可预测、可复用。

理解权重轮询的核心逻辑

假设有三台服务器:A(权重3)、B(权重2)、C(权重=1),理想分配序列应为 A,A,A,B,B,C(共6次循环),即每个服务器被选中的频次与其权重严格成正比。但直接构造长度为总权重的数组(如长度6的字符串数组)在权重很大时(如 1000、5000)会浪费内存。更优做法是:维护当前指针 + 累计权重判断,用 O(1) 空间完成 O(1) 单次调度。

用两个数组实现轻量级权重轮询

定义两个平行数组:

  • servers[]:存储服务器标识(如 IP 或名称)
  • weights[]:对应位置的整数权重(必须 ≥ 1)

再维护一个全局变量 currentIndex(初始为 0)和一个预计算的 totalWeight(所有权重之和)。每次调用 nextServer() 时:

java-api-doc-generator
java-api-doc-generator

分析 Java Spring Boot 后端 Controller 代码,自动生成 Markdown 格式的接口文档。当用户说“生成 Java 接口文档”、“生成 API 文档”、“从 Spring 代码生成文档”时触发。

下载
  • 对 currentIndex 取模 totalWeight,得到本轮偏移量 pos
  • 遍历 weights[],累加权重直到累计和 > pos,返回对应下标的服务器
  • 将 currentIndex 自增 1

例如:weights = [3,2,1] → totalWeight = 6;当 currentIndex = 4,pos = 4 % 6 = 4;累加:3(A)→ 小于 4;+2=5(B)→ 大于 4 → 返回 B。

代码实现(无状态、线程安全建议)

以下为简洁可运行示例(使用 AtomicInteger 保障多线程下的 currentIndex 安全):

public class WeightedRoundRobin {
    private final String[] servers;
    private final int[] weights;
    private final int totalWeight;
    private final AtomicInteger currentIndex = new AtomicInteger(0);

    public WeightedRoundRobin(String[] servers, int[] weights) {
        if (servers.length == 0 || servers.length != weights.length) {
            throw new IllegalArgumentException("Servers and weights must be non-empty and same length");
        }
        for (int w : weights) {
            if (w = 1");
        }
        this.servers = servers;
        this.weights = weights;
        this.totalWeight = Arrays.stream(weights).sum();
    }

    public String nextServer() {
        int pos = currentIndex.getAndIncrement() % totalWeight;
        int sum = 0;
        for (int i = 0; i 

注意事项与优化方向

该实现时间复杂度为 O(n),n 是服务器数量,适用于几十台以内的场景。若服务器数极多(如数百),可预构建前缀和数组 + 二分查找,将单次调度降至 O(log n);若需支持运行时权重动态变更,应加锁或改用 CAS 更新整个权重快照;另外,真实生产中建议配合健康检查剔除失效节点,避免流量打到宕机机器。

Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

11417

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

8522

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

7332

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1224

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

988

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1456

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

3009

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

20251

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1315

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习