56 lines
2.1 KiB
Java
56 lines
2.1 KiB
Java
package com.yj.earth.business.controller;
|
||
|
||
import com.yj.earth.common.util.ApiResponse;
|
||
import com.yj.earth.vo.CsvField;
|
||
import io.swagger.v3.oas.annotations.Operation;
|
||
import io.swagger.v3.oas.annotations.Parameter;
|
||
import io.swagger.v3.oas.annotations.tags.Tag;
|
||
import org.apache.commons.csv.CSVFormat;
|
||
import org.apache.commons.csv.CSVParser;
|
||
import org.apache.commons.csv.CSVRecord;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
import org.springframework.web.bind.annotation.GetMapping;
|
||
import org.springframework.web.bind.annotation.RequestMapping;
|
||
import org.springframework.web.bind.annotation.RequestParam;
|
||
import org.springframework.web.bind.annotation.RestController;
|
||
|
||
import java.io.File;
|
||
import java.io.FileInputStream;
|
||
import java.io.InputStreamReader;
|
||
import java.util.ArrayList;
|
||
import java.util.HashMap;
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
|
||
@Tag(name = "CSV文件解析")
|
||
@RestController
|
||
@RequestMapping("/csv")
|
||
public class CsvController {
|
||
@GetMapping("/parseCsv")
|
||
@Operation(summary = "解析CSV文件")
|
||
public ApiResponse parseCsv(@Parameter(description = "文件路径") @RequestParam String filePath) {
|
||
Map<String, Object> response = new HashMap<>();
|
||
List<CsvField> fieldList = new ArrayList<>();
|
||
|
||
try {
|
||
File file = new File(filePath);
|
||
try (InputStreamReader reader = new InputStreamReader(new FileInputStream(file), "GBK");
|
||
CSVParser parser = new CSVParser(reader, CSVFormat.DEFAULT)) {
|
||
// 遍历所有行（CSVRecord即一行数据）
|
||
for (CSVRecord record : parser) {
|
||
// 确保每行至少有3列（A、B、C列）
|
||
if (record.size() >= 3) {
|
||
String label = record.get(1).trim();
|
||
String key = record.get(2).trim();
|
||
fieldList.add(new CsvField(key, label));
|
||
}
|
||
}
|
||
}
|
||
return ApiResponse.success(fieldList);
|
||
} catch (Exception e) {
|
||
return ApiResponse.failure(e.getMessage());
|
||
}
|
||
}
|
||
}
|