在Java Springboot中將CSV轉換為JSON數組

Question

您好，我正在嘗試使用名為 csvReader 的依賴項將 CSV 文件轉換為 JSON 數組，但是當我運行代碼時，它錯誤地打印出 JSON 響應，我很確定為什么有人能夠為我指明正確的方向。

 @GetMapping("/convert") public List<List<String>> convertCSV() throws FileNotFoundException { List<List<String>> records = new ArrayList<List<String>>(); try (CSVReader csvReader = new CSVReader(new FileReader("C:/Download/cities.csv"));) { String[] values = null; while ((values = csvReader.readNext()).= null) { records.add(Arrays;asList(values)). } } catch (IOException e) { e;printStackTrace(); } return values; }

Answer 1

你的案子沒什么大不了的。

您可以閱讀該csv並構建json 。

讀取第一行並確定列。 其余的行是值。

public class Foo{

    public static void main(String[] args) throws Exception{

        List<String> csvRows = null;
        try(var reader = Files.lines(Paths.get("dataFile.csv"))){
            csvRows = reader.collect(Collectors.toList());
        }catch(Exception e){
            e.printStackTrace();
        }

        if(csvRows != null){

            String json = csvToJson(csvRows);
            System.out.println(json);

        }

    }

    public static String csvToJson(List<String> csv){

        //remove empty lines
        //this will affect permanently the list. 
        //be careful if you want to use this list after executing this method
        csv.removeIf(e -> e.trim().isEmpty());

        //csv is empty or have declared only columns
        if(csv.size() <= 1){
            return "[]";
        }

        //get first line = columns names
        String[] columns = csv.get(0).split(",");

        //get all rows
        StringBuilder json = new StringBuilder("[\n");
        csv.subList(1, csv.size()) //substring without first row(columns)
            .stream()
            .map(e -> e.split(","))
            .filter(e -> e.length == columns.length) //values size should match with columns size
            .forEach(row -> {

                json.append("\t{\n");

                    for(int i = 0; i < columns.length; i++){
                        json.append("\t\t\"")
                            .append(columns[i])
                            .append("\" : \"")
                            .append(row[i])
                            .append("\",\n"); //comma-1
                    }

                    //replace comma-1 with \n
                    json.replace(json.lastIndexOf(","), json.length(), "\n");

                json.append("\t},"); //comma-2

            });

        //remove comma-2
        json.replace(json.lastIndexOf(","), json.length(), "");

        json.append("\n]");

        return json.toString();

    }

}

測試：

fname,lname,note
Shaun,Curtis,a
Kirby,Beil,b

-----------------------

[
    {
        "fname" : "Shaun",
        "lname" : "Curtis",
        "note" : "a"
    },  {
        "fname" : "Kirby",
        "lname" : "Beil",
        "note" : "b"
    }
]

此方法適用於csv任何結構。 不需要映射列。

Answer 2

那是因為您讀取字符串中的數據並打印字符串列表。 如果要將 CSV 映射到對象（JSON 對象），則需要將 CSV 讀取為 bean 對象，請找到以下代碼片段以打印為 JSON，將 toString 方法覆蓋為 JSON 格式。

用戶.java

public class User {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @NotNull
    private String name;

    @NotNull
    private String surname;

    //Getter and Setters
}

CsvReaderUtil.java

public static List<User> readCsvFile() throws IOException {
            List<User> list = null;
            CSVReader reader = null;
            InputStream is = null;
            try {
                File initialFile = new File("C:\\Users\\ER\\Desktop\\test.csv");
                is = new FileInputStream(initialFile);
                reader = new CSVReader(new InputStreamReader(is), ',', '"', 1);
                ColumnPositionMappingStrategy strat = new ColumnPositionMappingStrategy();
                strat.setType(User.class);
                String[] columns = new String[]{"id", "name", "surname"};
                strat.setColumnMapping(columns);
                CsvToBean csv = new CsvToBean();
                list = csv.parse(strat, reader);
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                is.close();
                reader.close();
            }
            return list;
        }

現在將此用戶列表打印為 JSON 對象。

Answer 3

這是一個有用的示例，說明如何使用 Java 11+ 將 CSV 轉換為 JSON：

private String fromCsvToJson(String csvFile) {
    String[] lines = file.split("\n");

    if (lines.length <= 1) {
        return List.of();
    }

    var headers = lines[0].split(",");
    var jsonFormat = Arrays.stream(lines)
                           .skip(1)
                           .map(line -> line.split(","))
                           .filter(line -> headers.length == line.length)
                           .map(line -> IntStream.range(0, headers.length).boxed().collect(toMap(i -> headers[i], i -> line[i], (a, b) -> b)))
                           .toList();

    return new ObjectMapper().writeValueAsString(jsonFormat);
}