Spring MVC 4.1 No. 023 CSVファイルのダウンロード2
今回もCSVダウンロードについてです。今回はResponseEntityクラスを使用したダウンロードについてです。
ResponseEntityではデータと、ヘッダー、ステータスコードを返せます。
日本語の文字化けをしないようにするために、HttpHeadersクラスのsetContentTypeを使わずに、addメソッドで文字コードと一緒にcontent-typeを指定しています。
CSVデータはただの文字列のデータになります。
package com.example.spring.controller.c023; import org.springframework.http.HttpHeaders; import org.springframework.http.HttpStatus; import org.springframework.http.ResponseEntity; import org.springframework.stereotype.Controller; import org.springframework.util.MimeTypeUtils; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMethod; @Controller public class C023Controller { @RequestMapping("/c023/csvInit") public String csvInit() { return "c023/csvInit"; } @RequestMapping(value = "/c023/csvDown", method = RequestMethod.GET, produces = "application/octet-stream;charset=utf-8") public ResponseEntity<String> csvDown() { HttpHeaders headers = new HttpHeaders(); // headers.setContentType(MediaType.APPLICATION_OCTET_STREAM); headers.add("contet-type", MimeTypeUtils.APPLICATION_OCTET_STREAM_VALUE + ";utf-8"); headers.set("Content-Disposition", "filename=\"test2.csv\""); String csvData = "山田 太郎,33\r\n"; csvData = csvData + "田中 花子,29\r\n"; return new ResponseEntity<String>(csvData, headers, HttpStatus.OK); } }
アンカーを表示する画面です。単純なアンカーのみです。
<%@page contentType="text/html; charset=utf-8" %><%-- --%><!DOCTYPE html> <html> <head> <meta charset="utf-8"> <title>サンプル</title> </head> <body> <a href="csvDown">csvDown</a> </body> </html>
確認用のテストケースは次のとおりです。
package com.example.spring.controller.c023; import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*; import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*; import static org.springframework.test.web.servlet.setup.MockMvcBuilders.*; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; import org.springframework.test.context.web.WebAppConfiguration; import org.springframework.test.web.servlet.MockMvc; import org.springframework.web.context.WebApplicationContext; @RunWith(SpringJUnit4ClassRunner.class) @WebAppConfiguration @ContextConfiguration(locations = { "file:src/main/webapp/WEB-INF/spring/spring-context.xml" }) public class C023ControllerTest { @Autowired private WebApplicationContext wac; private MockMvc mockMvc; @Before public void setup() { mockMvc = webAppContextSetup(wac).build(); } @Test public void csvInitのGET() throws Exception { mockMvc.perform(get("/c023/csvInit")).andExpect(status().isOk()) .andExpect(view().name("c023/csvInit")); } @Test public void csvDownのGET() throws Exception { StringBuilder sb = new StringBuilder(); sb.append("山田 太郎,33\r\n"); sb.append("田中 花子,29\r\n"); mockMvc.perform(get("/c023/csvDown")) .andExpect(status().isOk()) .andExpect( content().contentType( "application/octet-stream;charset=utf-8")) .andExpect(content().string(sb.toString())); } }
ソースは https://github.com/kuwalab/spring-mvc41 にあります。タグ023が今回のサンプルです。
まとめ http://kuwalab.hatenablog.jp/entry/spring
最初 http://kuwalab.hatenablog.jp/entry/spring_mvc41/001
前回 http://kuwalab.hatenablog.jp/entry/spring_mvc41/022
次回 http://kuwalab.hatenablog.jp/entry/spring_mvc41/024