0

0

解决JVM堆内存溢出:Java应用高效处理海量数据批次迁移实践

心靈之曲

心靈之曲

发布时间:2025-07-10 19:36:31

|

768人浏览过

|

来源于php中文网

原创

解决jvm堆内存溢出:java应用高效处理海量数据批次迁移实践

本文旨在解决Java微服务在处理海量数据(如百万级记录迁移)时遇到的JVM堆内存耗尽问题。通过深入探讨内存溢出的根源,并提出一种高效的批处理策略,利用数据库的LIMIT和OFFSET机制分批次查询和处理数据,从而显著降低单次操作的内存消耗,避免服务崩溃,确保数据迁移的稳定性和效率。

1. 问题背景与内存溢出分析

在Java微服务架构中,当需要处理或迁移大量数据(例如百万级数据库记录)时,常见的做法是查询所有符合条件的数据并加载到内存中进行后续处理。然而,这种一次性加载海量数据的策略极易导致JVM堆内存耗尽(java.lang.OutOfMemoryError: Java heap space),表现为服务崩溃或响应缓慢。

错误日志通常会显示类似以下信息:

Resource exhaustion event: the JVM was unable to allocate memory from the heap.
ResourceExhausted! (1/0)

这表明应用程序试图分配的内存超出了JVM堆的可用容量。在数据迁移场景中,如将一个表中的大量记录复制到另一个表,如果一次性查询并缓存所有源数据(例如使用 JdbcTemplate.queryForList()),即使后续使用了JDBC批处理进行写入,内存压力也可能在数据查询阶段就达到极限。

原始代码片段中的 List> sourceRecords = sourceDbTemplate.queryForList(...) 是导致内存问题的核心。当 sourceRecords 包含数百万条记录时,每条记录作为一个 Map 对象,连同其内部的键值对,将迅速消耗大量堆内存。

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

2. 解决方案:基于数据库分页的批处理策略

为了解决一次性加载海量数据导致的内存溢出问题,核心思想是将大批量数据处理分解为多个小批次处理。这可以通过结合数据库的分页查询能力和应用程序的迭代处理逻辑来实现。

2.1 数据库层面的批次查询 (LIMIT 和 OFFSET)

数据库提供了 LIMIT(或 TOP)和 OFFSET(或 SKIP)子句,允许我们指定查询结果的数量以及从哪个位置开始返回。这是实现数据分批查询的基础。

ONLYOFFICE
ONLYOFFICE

用ONLYOFFICE管理你的网络私人办公室

下载

SQL查询示例:

SELECT *
FROM your_table
WHERE your_condition
ORDER BY unique_id_column -- 必须指定一个排序字段,确保每次查询的顺序稳定
LIMIT batch_size
OFFSET current_offset;
  • LIMIT batch_size: 定义每个批次要查询的记录数量。
  • OFFSET current_offset: 定义从结果集的哪个位置开始返回记录。current_offset 会随着已处理记录的数量递增。
  • ORDER BY unique_id_column: 至关重要! 必须根据一个稳定、唯一且通常是索引的列(如主键ID、创建时间戳等)进行排序。这确保了:
    • 每次分页查询的结果顺序是确定的。
    • 不会遗漏或重复获取数据。
    • 对于大型数据集和高 OFFSET 值,数据库可能需要扫描大量跳过的行,因此 OFFSET 的性能会随之下降。在极端情况下,可以考虑使用基于游标(Keyset Pagination)的方法(即 WHERE id > last_id ORDER BY id LIMIT batch_size),但这超出了本教程的范围。

2.2 应用程序层面的迭代处理

在应用程序中,我们需要构建一个循环,在每次迭代中:

  1. 计算当前的 OFFSET 值。
  2. 使用 LIMIT 和计算出的 OFFSET 从源数据库中查询一个批次的数据。
  3. 处理这个批次的数据(例如,复制到目标表)。
  4. 更新 OFFSET 值,为下一次迭代做准备。
  5. 当查询结果为空时,表示所有数据已处理完毕,退出循环。

3. 示例代码实现

下面我们将基于原有的代码结构,展示如何修改 archiveTableRecords 和 buildSQLQueryToFetchSourceRecords 方法以实现批处理。

3.1 修改SQL查询构建方法

为了支持 LIMIT 和 OFFSET,我们需要修改 ArchiveSQLQueries.buildSQLQueryToFetchSourceRecords 方法,使其能够接收批次大小、偏移量以及用于排序的列名。

import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.ParameterizedPreparedStatementSetter;

// 假设 ArchiveConfigDTO 和 getCSTDateNew 方法已定义
// 假设 ArchiveSQLQueries 是一个包含静态方法的类

public class DataArchiverService {

    private static final Logger logger = LoggerFactory.getLogger(DataArchiverService.class);

    @Value("${batch-size}")
    private int batchSize; // 配置的批处理大小,例如1000或5000

    // 辅助方法,假设存在
    private String getCSTDateNew(String archiveMonths) {
        // 实现日期计算逻辑
        return "2022-09-01"; // 示例值
    }

    // 辅助方法,假设存在
    private int getSumOfArray(int[][] array) {
        int sum = 0;
        for (int[] innerArray : array) {
            for (int value : innerArray) {
                sum += value;
            }
        }
        return sum;
    }

    // --- 修改后的 ArchiveSQLQueries 类中的方法 ---
    public static class ArchiveSQLQueries {
        public static String buildSQLQueryToFetchSourceRecords(String sourceTable, String orderByColumn, int limit, int offset) {
            StringBuilder sb = new StringBuilder("SELECT * FROM " + sourceTable + " where update_dts <= ?");
            // 确保 orderByColumn 是安全的,避免SQL注入风险
            if (orderByColumn != null && !orderByColumn.isEmpty()) {
                sb.append(" ORDER BY ").append(orderByColumn);
            }
            sb.append(" LIMIT ").append(limit);
            sb.append(" OFFSET ").append(offset);
            return sb.toString();
        }

        public static String buildSQLTargetRecordInsertionQuery(String targetTable, Map record, String primaryKeyColumn) {
            // 假设此方法已正确实现,根据Map构建INSERT语句
            // 示例:INSERT INTO target_table (col1, col2) VALUES (?, ?)
            StringBuilder sb = new StringBuilder("INSERT INTO ").append(targetTable).append(" (");
            StringBuilder values = new StringBuilder(" VALUES (");

            boolean firstColumn = true;
            for (String key : record.keySet()) {
                if (!key.equals(primaryKeyColumn)) { // 假设主键在插入时是自增的,不包含在VALUES中
                    if (!firstColumn) {
                        sb.append(", ");
                        values.append(", ");
                    }
                    sb.append(key);
                    values.append("?");
                    firstColumn = false;
                }
            }
            sb.append(")").append(values).append(")");
            return sb.toString();
        }
    }

    // --- 原始的 copySourceRecords 方法,其内部已使用批处理写入 ---
    public int copySourceRecords(JdbcTemplate targetDbTemplate, String targetTable, String primaryKeyColumn,
                                 List> sourceRecords, List primaryKeyValueList) {
        int result = 0;
        logger.info("Copying {} records to {}", sourceRecords.size(), targetTable);

        if (sourceRecords.isEmpty()) {
            return 0;
        }

        // 构建插入语句,基于第一个记录的结构
        String insertSql = ArchiveSQLQueries.buildSQLTargetRecordInsertionQuery(targetTable, sourceRecords.get(0), primaryKeyColumn);

        int[][] insertResult = targetDbTemplate.batchUpdate(
                insertSql,
                sourceRecords,
                batchSize, // 这里使用了配置的batchSize进行JDBC批处理写入
                new ParameterizedPreparedStatementSetter>() {
                    @Override
                    public void setValues(PreparedStatement ps, Map argument) throws SQLException {
                        int index = 1;
                        for (Entry obj : argument.entrySet()) {
                            // 假设主键列在目标表是自增的,或者不作为插入参数
                            if (obj.getKey().equals(primaryKeyColumn)) {
                                primaryKeyValueList.add(obj.getValue()); // 收集主键值
                            } else {
                                ps.setObject(index++, obj.getValue());
                            }
                        }
                    }
                });

        result = getSumOfArray(insertResult);
        logger.info("Inserted {} record(s) in {}", result, targetTable);
        return result;
    }

    // --- 修改后的 archiveTableRecords 方法,实现批次循环 ---
    public void archiveTableRecords(JdbcTemplate sourceDbTemplate, JdbcTemplate targetDbTemplate,
                                    ArchiveConfigDTO archiveObj) {
        try {
            String sourceTable = archiveObj.getSourceTable();
            String targetTable = archiveObj.getTargetTable();
            String primaryKeyColumn = archiveObj.getPrimaryKeyColumn(); // 假设这是用于排序的列
            String archive_months = archiveObj.getArchiveCriteriaMonths();
            String compareDate1 = getCSTDateNew(archive_months);
            logger.info("Archive criteria date: {}", compareDate1);

            int processedRecords = 0;
            List allPrimaryKeyValueList = new ArrayList<>(); // 用于收集所有已处理记录的主键,以便后续删除

            while (true) {
                // 1. 批次查询源数据
                List> sourceRecordsBatch = sourceDbTemplate.queryForList(
                        ArchiveSQLQueries.buildSQLQueryToFetchSourceRecords(sourceTable, primaryKeyColumn, batchSize, processedRecords),
                        compareDate1
                );

                if (sourceRecordsBatch.isEmpty()) {
                    logger.info("No more records to fetch for table {}. Total processed: {}", sourceTable, processedRecords);
                    break; // 没有更多记录,退出循环
                }

                logger.info("Fetched batch of {} records from {} (offset: {})", sourceRecordsBatch.size(), sourceTable, processedRecords);

                // 2. 准备当前批次的主键列表
                List currentBatchPrimaryKeys = new ArrayList<>();

                // 3. 复制当前批次数据到目标表
                int recordsInsertedInBatch = copySourceRecords(targetDbTemplate, targetTable, primaryKeyColumn, sourceRecordsBatch, currentBatchPrimaryKeys);

                if (recordsInsertedInBatch > 0) {
                    // 将当前批次的主键添加到总列表中
                    allPrimaryKeyValueList.addAll(currentBatchPrimaryKeys);
                    logger.info("Copied {} record(s) to {}. Total copied: {}", recordsInsertedInBatch, targetTable, allPrimaryKeyValueList.size());
                }

                // 4. 更新已处理记录数,作为下一次查询的偏移量
                processedRecords += sourceRecordsBatch.size();
            }

            // 5. 所有批次处理完毕后,根据收集到的主键删除源数据
            if (!allPrimaryKeyValueList.isEmpty()) {
                logger.info("Initiating deletion of {} records from source table {}.", allPrimaryKeyValueList.size(), sourceTable);
                // 注意:如果 allPrimaryKeyValueList 极其庞大,deleteSourceRecords 方法也可能需要内部批处理
                deleteSourceRecords(sourceDbTemplate, sourceTable, primaryKeyColumn, allPrimaryKeyValueList);
                logger.info("Deletion completed for {} records from source table {}.", allPrimaryKeyValueList.size(), sourceTable);
            }

        } catch (Exception e) {
            logger.error("Exception in archiveTableRecords: {} {}", e.getMessage(), e);
            // 在实际应用中,这里需要更健壮的异常处理,例如记录失败的批次信息,以便后续重试
        }
    }

    // 假设 deleteSourceRecords 方法已存在
    public void deleteSourceRecords(JdbcTemplate sourceDbTemplate, String sourceTable, String primaryKeyColumn, List primaryKeyValueList) {
        // 示例:DELETE FROM source_table WHERE primaryKeyColumn IN (?, ?, ...)
        // 对于非常大的 primaryKeyValueList,需要考虑分批次执行DELETE语句,或使用其他高效的删除策略
        String deleteSql = "DELETE FROM " + sourceTable + " WHERE " + primaryKeyColumn + " IN (" +
                           String.join(",", java.util.Collections.nCopies(primaryKeyValueList.size(), "?")) + ")";

        try {
            sourceDbTemplate.batchUpdate(deleteSql, new ArrayList<>(primaryKeyValueList)); // 假设可以一次性处理
        } catch (Exception e) {					
					

					

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

阿里巴巴推出的全能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,提供了直观易用的用户界面等等。

728

2023.10.12

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

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

328

2023.10.27

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

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

350

2024.02.23

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

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

1263

2024.03.06

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

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

360

2024.03.06

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

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

841

2024.04.07

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

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

581

2024.04.29

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

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

423

2024.04.29

java入门学习合集
java入门学习合集

本专题整合了java入门学习指南、初学者项目实战、入门到精通等等内容,阅读专题下面的文章了解更多详细学习方法。

1

2026.01.29

热门下载

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

精品课程

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

共17课时 | 2.4万人学习

微信小程序开发之API篇
微信小程序开发之API篇

共15课时 | 1.2万人学习

进程与SOCKET
进程与SOCKET

共6课时 | 0.4万人学习

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

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