Java编写一个文件导入接口,根据校验结果返回错误数据的Excel文件或正确时返回JSON报文。

这段代码展示了如何使用SpringMVC创建一个REST控制器来处理文件上传,对上传文件进行校验,根据校验结果返回JSON响应或包含错误数据的Excel文件。
摘要由CSDN通过智能技术生成
import org.apache.poi.ss.usermodel.*;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

@RestController
public class FileImportController {

    @PostMapping("/import")
    public ResponseEntity<?> importFile(@RequestParam("file") MultipartFile file) {
        // 进行文件校验
        boolean isValid = validateFile(file);

        if (isValid) {
            // 文件校验通过,返回JSON报文
            String jsonResponse = generateJsonResponse();
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_JSON);
            return new ResponseEntity<>(jsonResponse, headers, HttpStatus.OK);
        } else {
            // 文件校验不通过,返回错误数据的Excel文件
            Workbook workbook = generateErrorExcel();
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
            headers.setContentDispositionFormData("attachment", "error_data.xlsx");
            return new ResponseEntity<>(workbookToByteArray(workbook), headers, HttpStatus.OK);
        }
    }

    private boolean validateFile(MultipartFile file) {
        // 实现文件校验的逻辑,根据你的需求进行修改
        // 如果校验通过,返回true;如果校验不通过,返回false
        return true;
    }

    private String generateJsonResponse() {
        // 生成JSON报文的逻辑,根据你的需求进行修改
        return "{\"message\": \"File import successful\"}";
    }

    private Workbook generateErrorExcel() {
        // 生成错误数据的Excel文件的逻辑,根据你的需求进行修改
        Workbook workbook = new XSSFWorkbook();
        Sheet sheet = workbook.createSheet("Error Data");
        Row headerRow = sheet.createRow(0);
        headerRow.createCell(0).setCellValue("Column 1");
        headerRow.createCell(1).setCellValue("Column 2");
        // 添加错误数据行
        Row dataRow = sheet.createRow(1);
        dataRow.createCell(0).setCellValue("Error 1");
        dataRow.createCell(1).setCellValue("Error 2");
        return workbook;
    }

    private byte[] workbookToByteArray(Workbook workbook) {
        // 将Workbook对象转换为字节数组
        try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
            workbook.write(outputStream);
            return outputStream.toByteArray();
        } catch (Exception e) {
            e.printStackTrace();
            return new byte[0];
        }
    }
}
```

在上面的示例代码中,`importFile`方法接收一个`MultipartFile`对象,表示上传的文件。首先,调用`validateFile`方法进行文件校验,根据校验结果决定返回JSON报文还是错误数据的Excel文件。

如果文件校验通过,调用`generateJsonResponse`方法生成JSON报文,并使用`ResponseEntity`返回。设置响应头的`Content-Type`为`application/json`,表示返回的是JSON数据。

如果文件校验不通过,调用`generateErrorExcel`方法生成错误数据的Excel文件,并使用`ResponseEntity`返回。设置响应头的`Content-Type`为`application/octet-stream`,表示返回的是二进制文件。使用`setContentDispositionFormData`方法设置文件名为`error_data.xlsx`,这样浏览器会自动下载该文件。

注意,示例代码中的文件校验、JSON报文生成、错误数据Excel生成等方法都是示例,你需要根据你的实际需求进行修改和扩展。

  • 6
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是一个基于Apache POI的Java代码示例,可以导入Excel文件校验数据。对于正确数据,将它们存入一个List中并返回给前端字符串。对于错误数据,将它们标红并生成一个新的Excel文件供下载。 ```java import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; import java.io.IOException; import java.util.ArrayList; import java.util.List; import org.apache.poi.hssf.usermodel.HSSFCell; import org.apache.poi.hssf.usermodel.HSSFCellStyle; import org.apache.poi.hssf.usermodel.HSSFDataValidation; import org.apache.poi.hssf.usermodel.HSSFDataValidationHelper; import org.apache.poi.hssf.usermodel.HSSFDataValidationList; import org.apache.poi.hssf.usermodel.HSSFRow; import org.apache.poi.hssf.usermodel.HSSFSheet; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.poi.ss.usermodel.Cell; import org.apache.poi.ss.usermodel.CellStyle; import org.apache.poi.ss.usermodel.DataValidation; import org.apache.poi.ss.usermodel.DataValidationConstraint; import org.apache.poi.ss.usermodel.DataValidationHelper; import org.apache.poi.ss.usermodel.DataValidationConstraint.OperatorType; import org.apache.poi.ss.usermodel.DataValidationConstraint.ValidationType; import org.apache.poi.ss.usermodel.DataValidationList; import org.apache.poi.ss.usermodel.DataValidationWorkbook; import org.apache.poi.ss.usermodel.Row; import org.apache.poi.ss.usermodel.Sheet; import org.apache.poi.ss.usermodel.Workbook; import org.apache.poi.ss.util.CellRangeAddressList; import org.apache.poi.ss.util.CellUtil; import org.apache.poi.ss.util.CellRangeAddress; public class ExcelValidator { public static void main(String[] args) throws IOException { // 导入Excel文件 File inputFile = new File("input.xls"); FileInputStream fis = new FileInputStream(inputFile); Workbook workbook = new HSSFWorkbook(fis); Sheet sheet = workbook.getSheetAt(0); // 校验数据 List<Integer> errorRows = new ArrayList<>(); List<String> correctData = new ArrayList<>(); for (int i = 1; i <= sheet.getLastRowNum(); i++) { Row row = sheet.getRow(i); if (row == null) { continue; } Cell idCell = row.getCell(0); Cell nameCell = row.getCell(1); if (idCell == null || nameCell == null || idCell.getCellType() != Cell.CELL_TYPE_NUMERIC || nameCell.getCellType() != Cell.CELL_TYPE_STRING) { errorRows.add(i); } else { correctData.add(idCell.toString() + "," + nameCell.toString()); } } // 标注错误行 CellStyle errorStyle = workbook.createCellStyle(); errorStyle.setFillForegroundColor(IndexedColors.RED.getIndex()); errorStyle.setFillPattern(CellStyle.SOLID_FOREGROUND); for (Integer rowIndex : errorRows) { HSSFRow row = (HSSFRow) sheet.getRow(rowIndex); HSSFCell idCell = row.getCell(0); HSSFCell nameCell = row.getCell(1); CellUtil.setCellStyleProperty(idCell, workbook, CellUtil.FILL_FOREGROUND_COLOR, IndexedColors.RED.getIndex()); CellUtil.setCellStyleProperty(nameCell, workbook, CellUtil.FILL_FOREGROUND_COLOR, IndexedColors.RED.getIndex()); } // 创建新的Excel文件 File outputFile = new File("output.xls"); FileOutputStream fos = new FileOutputStream(outputFile); workbook.write(fos); fos.close(); // 返回正确数据 String correctDataStr = String.join(",", correctData); // TODO: 将正确数据返回给前端 // 下载生成的Excel文件 // TODO: 实现下载功能 } } ``` 在这个示例中,我们假设Excel文件的第一列是ID(数字类型),第二列是名称(字符串类型)。我们使用Apache POI API来读取工作簿和工作表,然后检查单元格类型是否正确。如果存在错误的行,我们使用CellUtil工具类设置单元格样式以标注错误行。对于正确数据,我们将它们存入一个List中,并将List元素拼接成一个字符串返回给前端。最后,我们创建一个新的Excel文件,将修改后的工作簿写入其中,并提供下载功能。 需要注意的是,示例代码中的数据校验仅仅是一种简单的实现方式,你需要根据实际需求进行修改和扩展。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值