0

0

优化Java数据批量获取:利用Stream API避免共享可变性

心靈之曲

心靈之曲

发布时间:2025-10-01 12:46:08

|

851人浏览过

|

来源于php中文网

原创

优化Java数据批量获取:利用Stream API避免共享可变性

本文探讨了在Java中从数据库批量获取数据时,如何通过重构代码来避免共享可变性问题。针对数据库参数限制,传统forEach循环结合addAll操作会导致外部列表的副作用。教程将展示如何利用Java Stream API的map、flatMap和collect操作,以声明式、无副作用的方式高效地聚合分批查询结果,从而提升代码的纯净性、可读性和并发安全性。

1. 问题背景与共享可变性挑战

在实际的软件开发中,我们经常需要从数据库中获取大量数据。然而,数据库通常对单个查询中允许的参数数量有限制(例如,sql in 子句可能限制在500个参数)。这意味着我们不能一次性查询所有数据,而需要将大的查询列表分割成多个小批次进行查询。

考虑以下场景:我们需要根据一个包含5000个数字的列表,分批从数据库中获取猫(Cat)和狗(Dog)的信息,每次查询最多接受500个参数。一个常见的初始实现方式可能如下:

// 假设 Cat 和 Dog 是实体类,catRepo 和 dogRepo 是数据访问层接口
// CatRepo 和 DogRepo 都有 fetchCats(List<Integer>) 和 fetchDogs(List<Integer>) 方法

AtomicInteger counter = new AtomicInteger();
List<Cat> catList = new ArrayList<>(); // 共享的可变列表
List<Dog> dogList = new ArrayList<>(); // 共享的可变列表

// 模拟生成5000个数字作为查询键
List<Integer> numbers = Stream.iterate(1, e -> e + 1)
    .limit(5000)
    .collect(Collectors.toList());

// 将数字列表分割成每批次500个的小列表
Collection<List<Integer>> partitionedListOfNumbers = numbers.stream()
    .collect(Collectors.groupingBy(num -> counter.getAndIncrement() / 500))
    .values(); // 得到 List<List<Integer>> 结构

// 遍历分批的列表,执行查询并累加结果
partitionedListOfNumbers.stream()
    .forEach(list -> {
        List<Cat> interimCatList = catRepo.fetchCats(list); // 从数据库获取猫列表
        catList.addAll(interimCatList); // 修改外部的 catList
        List<Dog> interimDogList = dogRepo.fetchDogs(list); // 从数据库获取狗列表
        dogList.addAll(interimDogList); // 修改外部的 dogList
    });

// 此时 catList 和 dogList 包含了所有查询结果

上述代码虽然能够完成任务,但存在一个明显的问题:catList 和 dogList 是在 forEach 循环外部声明的,并且在循环内部通过 addAll 方法被修改。这种模式被称为共享可变性(Shared Mutability),即多个操作(或线程)共享并修改同一个可变状态。

共享可变性在函数式编程中被视为一种“副作用”,它带来了以下弊端:

  • 难以推理:代码的行为不再仅仅取决于其输入,还取决于外部状态,使得理解和预测程序行为变得复杂。
  • 并发安全问题:在多线程环境中,多个线程同时修改共享列表可能导致数据不一致或竞态条件,需要额外的同步机制(如 Collections.synchronizedList 或 CopyOnWriteArrayList),增加了复杂性。
  • 可测试性差:由于依赖外部状态,单元测试变得困难,需要模拟或设置复杂的上下文。

为了提高代码的纯净性、可读性和并发安全性,我们应该尽量避免共享可变性。

立即学习Java免费学习笔记(深入)”;

2. 基于Stream API的解决方案

Java 8 引入的 Stream API 提供了一种声明式、函数式的数据处理方式,非常适合解决上述问题,因为它鼓励通过转换(transformation)而不是修改(mutation)来处理数据。

核心思想是:将每个分批查询的结果视为一个独立的中间集合,然后将所有这些中间集合扁平化并收集到一个最终的不可变集合中。这可以通过 map、flatMap 和 collect 操作组合实现。

以下是重构后的代码示例:

Otter.ai
Otter.ai

一个自动的会议记录和笔记工具,会议内容生成和实时转录

下载
import java.util.Collection;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
import java.util.function.Function; // 用于后续优化

// 假设 Cat, Dog, CatRepo, DogRepo 已经定义
// ... (Cat, Dog 实体类及 CatRepo, DogRepo 接口的定义)

public class BatchDataFetcher {

    // 假设 catRepo 和 dogRepo 已经通过依赖注入或其他方式初始化
    private CatRepo catRepo;
    private DogRepo dogRepo;

    public BatchDataFetcher(CatRepo catRepo, DogRepo dogRepo) {
        this.catRepo = catRepo;
        this.dogRepo = dogRepo;
    }

    public void fetchDataAndProcess() {
        // 用于分批的计数器,确保每个批次编号正确
        AtomicInteger counter = new AtomicInteger(); 

        // 模拟生成5000个数字,并将其分批
        // IntStream.rangeClosed(1, 5000) 生成从1到5000的整数流
        // .boxed() 将 IntStream 转换为 Stream<Integer>
        Collection<List<Integer>> partitionedListOfNumbers = IntStream.rangeClosed(1, 5000)
            .boxed() 
            .collect(Collectors.groupingBy(num -> counter.getAndIncrement() / 500))
            .values(); // 得到 List<List<Integer>> 结构

        // 获取所有猫列表
        List<Cat> catList = partitionedListOfNumbers.stream()
            .map(list -> catRepo.fetchCats(list)) // 对每个分批列表执行查询,得到 Stream<List<Cat>>
            .flatMap(List::stream) // 将 Stream<List<Cat>> 扁平化为 Stream<Cat>
            .collect(Collectors.toList()); // 收集所有 Cat 对象到一个新的 List 中

        // 获取所有狗列表
        List<Dog> dogList = partitionedListOfNumbers.stream()
            .map(list -> dogRepo.fetchDogs(list)) // 对每个分批列表执行查询,得到 Stream<List<Dog>>
            .flatMap(List::stream) // 将 Stream<List<Dog>> 扁平化为 Stream<Dog>
            .collect(Collectors.toList()); // 收集所有 Dog 对象到一个新的 List 中

        // 此时 catList 和 dogList 是通过 Stream 操作“生成”的,而不是“修改”的
        // 它们是不可变的(如果 collect 收集到的是不可变列表,否则是新的可变列表,但不再是共享的)
        System.out.println("Fetched " + catList.size() + " cats and " + dogList.size() + " dogs.");
    }
}

// 模拟 Cat, Dog, CatRepo, DogRepo
class Cat { private int id; private String name; public Cat(int id, String name) { this.id = id; this.name = name; } @Override public String toString() { return "Cat{" + "id=" + id + ", name='" + name + '\'' + '}'; } }
class Dog { private int id; private String name; public Dog(int id, String name) { this.id = id; this.name = name; } @Override public String toString() { return "Dog{" + "id=" + id + ", name='" + name + '\'' + '}'; } }
class CatRepo { public List<Cat> fetchCats(List<Integer> ids) { return ids.stream().map(id -> new Cat(id, "Cat-" + id)).collect(Collectors.toList()); } }
class DogRepo { public List<Dog> fetchDogs(List<Integer> ids) { return ids.stream().map(id -> new Dog(id, "Dog-" + id)).collect(Collectors.toList()); } }

// 示例运行
// public static void main(String[] args) {
//     BatchDataFetcher fetcher = new BatchDataFetcher(new CatRepo(), new DogRepo());
//     fetcher.fetchDataAndProcess();
// }

代码解析:

  1. 数据分批 (groupingBy): IntStream.rangeClosed(1, 5000).boxed().collect(Collectors.groupingBy(num -> counter.getAndIncrement() / 500)).values() 这一步与原始代码类似,负责将连续的数字列表分割成多个子列表,每个子列表包含500个数字。counter 在这里作为分组键的一部分,确保每500个数字被分到同一个组。

  2. map 操作: partitionedListOfNumbers.stream().map(list -> catRepo.fetchCats(list)) 这一步将 Collection<List<Integer>> 转换成 Stream<List<Cat>>。对于 partitionedListOfNumbers 中的每一个 List<Integer>(即一个批次的查询键),catRepo.fetchCats(list) 会被调用,返回一个 List<Cat>。map 操作的输出是一个包含多个 List<Cat> 的流。

  3. flatMap 操作: .flatMap(List::stream) 由于上一步 map 操作的输出是 Stream<List<Cat>>,我们希望得到的是一个单一的 List<Cat>。flatMap 的作用就是将流中的每个元素(这里是 List<Cat>)“扁平化”成一个流(通过 List::stream 方法),然后将所有这些子流连接成一个单一的流。最终,我们得到的是一个 Stream<Cat>。

  4. collect 操作: .collect(Collectors.toList()) 这是流操作的终结操作,它将 Stream<Cat> 中的所有 Cat 对象收集到一个新的 List<Cat> 中。这个 List<Cat> 是一个全新的列表,不与任何外部变量共享,从而避免了共享可变性。

3. 解决方案的优势

通过上述重构,我们获得了以下显著优势:

  • 避免共享可变性:catList 和 dogList 不再在循环内部被修改,而是通过 Stream API 的一系列转换操作“生成”的最终结果。这使得代码更加纯净,更容易理解。
  • 提升代码可读性与函数式风格:Stream API 提供了声明式编程风格,代码清晰地表达了“如何转换数据”而不是“如何一步步修改状态”,符合现代Java的函数式编程范式。
  • 更好的并发安全性:由于没有共享的可变状态,此代码天然支持并发。如果将 partitionedListOfNumbers.stream() 替换为 partitionedListOfNumbers.parallelStream(),查询可以并行执行而无需担心竞态条件,因为每个 map 操作都是独立的,并且最终的 collect 操作会安全地将结果聚合。

4. 进一步优化:消除重复代码

在上述解决方案中,获取 catList 和 dogList 的逻辑非常相似,都遵循 map -> flatMap -> collect 的模式。为了进一步提高代码的复用性并消除重复,我们可以将这部分通用逻辑提取到一个泛型方法中,该方法接受一个函数作为参数,用于执行具体的数据库查询:

public class BatchDataFetcher {
    // ... (构造函数和成员变量不变)

    /**
     * 通用方法:根据分批的键列表和查询函数批量获取数据
     * @param partitionedKeys 分批的查询键列表
     * @param fetchFunction 接受一个 List<K> 并返回 List<T> 的查询函数
     * @param <T> 结果列表中的元素类型
     * @param <K> 查询键列表中的元素类型
     * @return 聚合后的所有 T 类型元素的列表
     */
    public <T, K> List<T> fetchEntitiesInBatches(Collection<List<K>> partitionedKeys, Function<List<K>, List<T>> fetchFunction) {
        return partitionedKeys.stream()
            .map(fetchFunction) // 对每个批次应用查询函数
            .flatMap(List::stream) // 扁平化结果
            .collect(Collectors.toList()); // 收集到新列表
    }

    public void fetchDataAndProcessOptimized() {
        AtomicInteger counter = new AtomicInteger();
        Collection<List<Integer>> partitionedListOfNumbers = IntStream.rangeClosed(1, 5000)
            .boxed()
            .collect(Collectors.groupingBy(num -> counter.getAndIncrement() / 500))
            .values();

        // 使用通用方法获取猫列表
        List<Cat> catList = fetchEntitiesInBatches(partitionedListOfNumbers, catRepo::fetchCats);

        // 使用通用方法获取狗列表
        List<Dog> dogList = fetchEntitiesInBatches(partitionedListOfNumbers, dogRepo::fetchDogs);

        System.out.println("Optimized: Fetched " + catList.size() + " cats and " + dogList.size() + " dogs.");
    }
}

通过引入 fetchEntitiesInBatches 泛型方法,我们将批处理查询的核心逻辑抽象出来,使得 fetchDataAndProcessOptimized 方法更加简洁,并且易于扩展到其他类型的实体查询。

5. 总结与最佳实践

在Java中处理批量数据获取并避免共享可变性是一个常见的需求。通过本教程,我们学习到:

  • 识别共享可变性问题:当代码通过 forEach 循环修改外部集合时,往往存在共享可变性问题,这会降低代码质量和并发安全性。
  • 利用Stream API进行转换:Java Stream API 的 map、flatMap 和 collect 操作是解决这类问题的强大工具。它们允许我们以声明式的方式对数据进行一系列无副作用的转换,最终生成新的集合。
  • 拥抱函数式编程:通过避免副作用和共享可变性,我们能够编写出更纯净、更易于理解、测试和并行化的代码。
  • 代码复用与抽象:对于重复的逻辑模式,应考虑提取为泛型方法或高阶函数,以提高代码的复用性和可维护性。

在设计数据处理逻辑时,始终优先考虑使用不可变数据结构和无副作用的操作,这将显著提升代码的健壮性和可维护性。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

1134

2023.10.12

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

340

2023.10.27

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

381

2024.02.23

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

2194

2024.03.06

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

380

2024.03.06

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

1703

2024.04.07

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

586

2024.04.29

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

440

2024.04.29

TypeScript类型系统进阶与大型前端项目实践
TypeScript类型系统进阶与大型前端项目实践

本专题围绕 TypeScript 在大型前端项目中的应用展开,深入讲解类型系统设计与工程化开发方法。内容包括泛型与高级类型、类型推断机制、声明文件编写、模块化结构设计以及代码规范管理。通过真实项目案例分析,帮助开发者构建类型安全、结构清晰、易维护的前端工程体系,提高团队协作效率与代码质量。

26

2026.03.13

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Kotlin 教程
Kotlin 教程

共23课时 | 4.4万人学习

C# 教程
C# 教程

共94课时 | 11.3万人学习

Java 教程
Java 教程

共578课时 | 82万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号