ゴール

CSVファイルをダウンロードする

f:id:nainaistar:20201206220115p:plain

環境

Java
- 15
org.springframework.boot:spring-boot-starter-web
- 2.4.0
（org.springframework.boot:spring-boot-starter-security）
- 2.4.0（重要ではないですが、検証ソースで使っているので…）
（Lombok）

手順

HTMLでファイルダウンロードボタンを作る

今回は、actionのURLがそのままファイル名となるように作っているので、ログインしている情報からuserIdを取得します。

ログインユーザーは、kirimaruとadminを用意しているので、実際のactionのURLは「/kirimaru.csv」か「admin.csv」になることを想定しています。

   <form method="get" th:action="|/${#authentication.principal.username}.csv|">
        <input type="submit" name="download_file" value="ファイルをダウンロード">
        <input type="hidden" th:name="${_csrf.parameterName}" th:value="${_csrf.token}" />
    </form>>

CSVファイル用のDTOを用意する

@JsonPropertyで出力したいCSVファイルのカラム名を指定します。

@JsonPropertyOrderを付けることで、確実にcsvファイルの出力される順番が保証されるので、書くべきでしょう。

@Value
@AllArgsConstructor
@JsonPropertyOrder({"ID", "UserId", "Action"})
public class TodoCsv {
  @JsonProperty("ID")
  private long id;
  @JsonProperty("UserId")
  private String userId;
  @JsonProperty("Action")
  private String action;

}

DBから取得した値を元にCSVファイルをダウンロードする

@GetMappingのvalueがそのままファイル名となります。producesに文字コード等を設定します。

今回の処理で大事なのは次の処理です。他の項目はアドベントカレンダーとしては必要ですが、この記事だけを見に来た人にとっては蛇足になります。

    CsvMapper mapper = new CsvMapper();
    CsvSchema schema = mapper.schemaFor(TodoCsv.class).withHeader();
    return mapper.writer(schema).writeValueAsString(csvs);

ファイル名：IndexController.java

  @GetMapping(value = "/*.csv", params = "download_file",
      produces = MediaType.APPLICATION_OCTET_STREAM_VALUE + "; charset=UTF-8; Content-Disposition: attachment"
  )
  @ResponseBody
  public Object index() throws JsonProcessingException {
    // User名取得
    SecurityContext securityContext = (SecurityContext) session.getAttribute("SPRING_SECURITY_CONTEXT");
    Authentication authentication = securityContext.getAuthentication();
    AuthTargetUser principal = (AuthTargetUser) authentication.getPrincipal();
    String username = principal.getUsername();
    // DBから取得
    List<TodoDto> todos = todoRepository.findList(username);

    // CSVファイル用のDTOに詰め直す
    List<TodoCsv> csvs = todos.stream().map(
        e -> new TodoCsv(e.getId(), e.getUserId(), e.getAction())
    ).collect(Collectors.toList());
    // ファイルをダウンロードさせる
    CsvMapper mapper = new CsvMapper();
    CsvSchema schema = mapper.schemaFor(TodoCsv.class).withHeader();
    return mapper.writer(schema).writeValueAsString(csvs);
  }}