如何保证AtomicLong addAndGet结果正确性？多线程下载百分比计算异常排查

嘿，我来帮你揪出这个问题的根源！你遇到的核心问题其实出在变量的作用域上——每次调用getDownloadSize()时，你都重新创建了totalWriteCount和lastDownloadSize这两个AtomicLong实例，这意味着每次计算都是从零开始累加，完全没有保留上一次下载的状态，自然第二次下载时的数值不符合预期。

问题代码的核心缺陷 #

你看，在当前的getDownloadSize()方法里，totalWriteCount和lastDownloadSize是方法内部的局部变量：

private long getDownloadSize() {
    synchronized (this) {
        // 每次调用方法都会新建这两个变量，之前的状态完全丢失！
        final AtomicLong totalWriteCount = new AtomicLong(0);
        final AtomicLong lastDownloadSize = new AtomicLong(0);
        // ... 累加逻辑
    }
}

每次执行这个方法，这两个变量都会被初始化为0，遍历任务累加后的值只会在本次方法调用中有效，方法执行完就被销毁了，根本没法保存上一次下载的writeCount和lastDownloadSize数据。

修复方案：将变量改为类成员变量 #

要解决这个问题，你需要把这两个原子变量提升为类的成员变量，这样它们的状态会在对象的生命周期内持续保留，而不是每次调用方法都重置。同时调整方法内的逻辑，避免重复初始化：

// 把这两个变量改成类的成员变量，保证状态可以持久化
private final AtomicLong totalWriteCount = new AtomicLong(0);
private final AtomicLong lastDownloadSize = new AtomicLong(0);

private long getDownloadSize() {
    synchronized (this) {
        // 临时变量存储本次遍历得到的累加值
        long currentWriteCount = 0;
        long currentDownloadSize = 0;
        
        for (DownloadTask task : downloadTasks) {
            currentWriteCount += task.getWriteCount();
            currentDownloadSize += task.getPosition().getDownloadSize();
        }
        
        // 更新成员变量：累积本次的写入量
        totalWriteCount.addAndGet(currentWriteCount);
        // 根据你的需求：lastDownloadSize = 上次的lastDownloadSize + 本次的writeCount
        lastDownloadSize.addAndGet(currentWriteCount);
        
        System.out.println("===== writeCount : " + totalWriteCount + " lastDownloadSize : " + lastDownloadSize);
        // 返回总下载量，根据你的业务需求调整计算逻辑
        return totalWriteCount.get() + lastDownloadSize.get();
    }
}

额外注意事项 #

1. 线程安全保障：你已经用synchronized (this)来包裹核心逻辑，这能保证多个线程调用getDownloadSize()时不会出现竞态条件，配合AtomicLong的原子操作，能很好地保证线程安全。
2. 业务逻辑对齐：你提到“第二次下载时的lastDownloadSize本应等于上次下载的writeCount与lastDownloadSize之和”，请确认currentWriteCount确实是本次新增的写入量，如果不是，需要调整lastDownloadSize的更新逻辑，确保和你的业务需求匹配。
3. 初始化时机：如果是支持续传的场景，记得在下载开始前（比如从本地读取已下载的大小）初始化totalWriteCount和lastDownloadSize的初始值，而不是默认的0。

内容的提问来源于stack exchange，提问作者luckylo