依赖:
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
</dependency>
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-test</artifactId>
<version>5.1.9.RELEASE</version>
<scope>compile</scope>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-test</artifactId>
</dependency>
测试代码:
package com.jpxx.clsh.util;
import com.alibaba.excel.EasyExcel;
import com.jpxx.clsh.dto.ImportCarDto;
import com.jpxx.clsh.entity.ClshCompanyInfo;
import com.jpxx.clsh.entity.ClshVehicleInfo;
import com.jpxx.clsh.service.CompanyService;
import com.jpxx.clsh.service.VehicleInfoService;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.test.context.junit4.SpringRunner;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
@RunWith(SpringRunner.class)
@SpringBootTest
public class importUtils {
@Autowired
VehicleInfoService vehicleInfoService;
@Autowired
CompanyService companyService;
@Test
public void parsing() {
String srcFile = "C:\\Users\\Administrator\\Desktop\\车辆信息表20200915.xls";
ImportListener importListener = new ImportListener();
EasyExcel.read(new File(srcFile), ImportCarDto.class, importListener).sheet().headRowNumber(2).doRead();
List<ImportCarDto> list = importListener.list;
List<ClshVehicleInfo> clshVehicleInfos = new ArrayList<ClshVehicleInfo>();
list.forEach(importCarDto -> {
ClshCompanyInfo clshCompanyInfo = companyService.getByCompanyName(importCarDto.getCompanyName());
ClshVehicleInfo info = new ClshVehicleInfo();
info.setCompanyId(clshCompanyInfo.getId());
info.setCompanyName(importCarDto.getCompanyName());
info.setPlateNo(importCarDto.getPlateNo());
info.setVehicleType(importCarDto.getType());
info.setEmissionStandard(importCarDto.getEmissionStandards());
info.setFleetNo(importCarDto.getFleetNumber());
info.setBrandAndModel(importCarDto.getBrand());
info.setArea(importCarDto.getArea());
clshVehicleInfos.add(info);
});
vehicleInfoService.saveOrUpdateBatch(clshVehicleInfos);
}
}