きり丸アドベントカレンダー2020の19記事目です。

ToDoアプリケーションでは、CSVファイルをアップロードして一括で登録するような機能があると便利です。ですので、ファイルをアップロードする仕組みを作りましょう。

ゴール

HTMLからJavaにファイルをアップロードする
アップロードしたCSVファイルを読み込んでDBに登録する

環境

Java
- 15
org.springframework.boot:spring-boot-starter-web
- 2.4.0

手順

HTMLでファイルアップロードボタンを作る

HTMLのFormタグのenctypeに「multipart/form-data」を指定するとファイルをアップロードできます。

また、Javaで受け取れるようにinputタグのnameを適当に指定します。

   <form method="post" action="/" enctype="multipart/form-data">
        <input type="file" name="file">
        <input type="submit" name="upload_file" value="ファイルをアップロード">
        <input type="hidden" th:name="${_csrf.parameterName}" th:value="${_csrf.token}" />
    </form>

CSVファイルを準備する

事前にCSVファイルを作成します。今回はid、userIdとactionの3項目をもつCSVファイルを作成します。

4,admin,味噌
5,admin,オイスターソース
6,admin,家に帰る

JavaでCSVファイルを受け取ってDBに登録する

@RequestParamにHTMLのinputタグで指定していたnameを指定します。アップロードしたファイルはMultipartFileクラスで受け取ります。

文字コード UTF-8で読み込んだ後に、BufferdReaderに変換します。

あとは、CSVファイルなのでカンマで分解してDBに渡せるように加工します。

  @PostMapping(value = "/", params = "upload_file")
  public String uploadFile(@RequestParam("file") MultipartFile uploadFile) {
    try (BufferedReader br = new BufferedReader(new InputStreamReader(uploadFile.getInputStream(), StandardCharsets.UTF_8))){
      String line;
      while ((line = br.readLine()) != null) {
        final String[] split = line.split(",");
        final TodoDto todo = TodoDto.builder().id(Integer.parseInt(split[0])).userId(split[1]).action(split[2]).build();
        todoRepository.insert(todo);
      }
    } catch (IOException e) {
      throw new RuntimeException("ファイルが読み込めません", e);
    }

    return "redirect:/";
  }