Java读取csv文件,批量写入数据库

功能说明: 该Java类名为TimedTasks,是一个定时任务调度器。它的主要功能是从指定路径的压缩文件中解压出CSV数据,并将解析后的数据使用多线程的方式进行批量插入到数据库中。整个过程包含以下功能:

  1. 获取当前日期和前三天的日期,并构建文件夹路径列表用于查找数据文件。
  2. 遍历文件夹路径列表,查找是否存在名为"test.zip"的压缩文件。
  3. 解压"test.zip"文件中的"test.csv"文件,并读取CSV数据保存在名为"lines"的列表中。
  4. 创建线程池,根据CPU核心数创建对应线程数,用于处理CSV数据的并发插入操作。
  5. 遍历CSV数据列表"lines",从第二行开始(跳过标题行),使用线程池处理每一行的插入操作。
  6. 解析CSV数据并构造对象"DataEntity",将对象添加到并发队列"imQueue"中。
  7. 等待所有线程完成任务后,执行批量插入操作,将并发队列中的数据存储到数据库中。
  8. 定时任务实现数据的定期同步与导入。
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import com.example.service.IDataEntityService; // 假设IDataEntityService接口位于com.example.service包中
import com.example.domain.DataEntity; // 假设DataEntity类位于com.example.domain包中
import com.example.util.AjaxResult; // 假设AjaxResult类位于com.example.util包中

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.zip.ZipFile;

/**
 * 定时任务调度测试
 * 用于解压、读取、导入test.csv数据到数据库
 * 通过多线程处理CSV数据并进行批量插入操作
 * 使用定时任务实现数据的定期同步与导入
 */
@Component("TimedTasks")
public class TimedTasks {

    @Autowired
    private IDataEntityService dataEntityService;

    /**
     * 定时解压、读取、导入test.csv
     * @return 返回导入操作结果
     */
    public AjaxResult exportData() {
        // 获取当前日期,格式为yyyyMMdd
        String currentDate = new SimpleDateFormat("yyyyMMdd").format(new Date());

        // 构建三天前、两天前和一天前的日期
        String threeDaysAgo = LocalDate.now().minusDays(3).format(DateTimeFormatter.BASIC_ISO_DATE);
        String twoDaysAgo = LocalDate.now().minusDays(2).format(DateTimeFormatter.BASIC_ISO_DATE);
        String oneDayAgo = LocalDate.now().minusDays(1).format(DateTimeFormatter.BASIC_ISO_DATE);

        // 定义文件夹路径列表
        List<String> folderPaths = Arrays.asList(currentDate, oneDayAgo, twoDaysAgo);

        // 标志是否找到zip文件
        boolean zipFileFound = false;
        String zipFilePath = "";

        // 遍历文件夹路径列表,查找zip文件
        for (String folderPath : folderPaths) {
            String zipFilePathSftp = "/home/upload/data/" + folderPath + "/test.zip";
            File zipFile = new File(zipFilePathSftp);

            // 如果zip文件存在,设定标志为true并跳出循环
            if (zipFile.exists()) {
                zipFileFound = true;
                zipFilePath = zipFilePathSftp;
                break;
            }
        }

        // 如果没有找到zip文件,返回错误信息
        if (!zipFileFound) {
            return AjaxResult.error("未找到test.zip文件");
        }

        // 解压文件并读取CSV数据
        List<String> lines = readCSVFromZip(zipFilePath, "test.csv");

        if (lines.size() <= 1) {
            return AjaxResult.error();
        }

        // 创建线程池,用于多线程处理
        int numThreads = Runtime.getRuntime().availableProcessors(); // 获取CPU核心数
        ExecutorService executor = Executors.newFixedThreadPool(numThreads);

        // 创建存储数据的并发队列
        ConcurrentLinkedQueue<DataEntity> imQueue = new ConcurrentLinkedQueue<>();

        // 循环处理CSV数据
        for (int i = 1; i < lines.size(); i++) { // 从第二行开始,跳过标题行
            final String line = lines.get(i); // 当前行数据

            // 使用线程池处理每一行的插入操作
            executor.execute(() -> {
                try {
				    // todo 需要替换为csv文件实际使用的分隔符
                    String[] data = line.split("\u0001");
                    for (int j = 0; j < data.length; j++) {
                        data[j] = data[j].replace("\\N", ""); // 去除换行符
                    }

                    DataEntity entity = new DataEntity();
                    entity.setCompanyId(data[0]);
                    entity.setCompanyName(data[1]);
                    entity.setCreditCode(data[2]);
                    entity.setReportYear(data[3]);
                    entity.setReportTermType(data[4]);
                    if (!data[5].isEmpty()) {
                        entity.setBusinessRevenue(Double.valueOf(data[5]));
                    }
                    if (!data[6].isEmpty()) {
                        entity.setTotalProfitAmount(Double.valueOf(data[6]));
                    }

                    String pattern = "yyyy-MM-dd HH:mm:ss";
                    SimpleDateFormat dateFormat = new SimpleDateFormat(pattern);
                    Date date = dateFormat.parse(data[7]);
                    entity.setUpdateTime(date);
                    entity.setBatchNumber(new Date());

                    imQueue.add(entity);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });
        }

        // 等待所有线程完成任务
        executor.shutdown();
        try {
            executor.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        // 执行批量插入操作
        dataEntityService.saveBatch(imQueue);

        return AjaxResult.success("同步成功");
    }

    private List<String> readCSVFromZip(String zipFilePath, String csvFileName) {
        List<String> lines = new ArrayList<>();
        try (ZipFile zipFile = new ZipFile(zipFilePath);
             InputStream inputStream = zipFile.getInputStream(zipFile.getEntry(csvFileName));
             InputStreamReader inputStreamReader = new InputStreamReader(inputStream, StandardCharsets.UTF_8);
             BufferedReader bufferedReader = new BufferedReader(inputStreamReader)) {

            String line;
            while ((line = bufferedReader.readLine()) != null) {
                lines.add(line);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

        return lines;
    }
}
  • 1
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值