Merge pull request #4 from kento-kotlin-sandbox/future/Add_file_output_logic

Add csv output logic

Author: Kento75

sample.csv

@@ -0,0 +1 @@
+[email protected],password,Kento75,1993-07-05,25,FALSE,ROLE_ADMIN

src/main/java/com/example/demo/domain/model/repository/jdbc/UserDaoJdbcImpl.java

@@ -151,6 +151,13 @@ public int deleteOne(String userId) throws DataAccessException {
 	// Userテーブルの全データをCSVに出力
 	@Override
 	public void userCsvOut() throws DataAccessException {
+		// m_userテーブルのデータを全件取得
+		String sql = "SELECT * FROM m_user";
+		
+		// ResultSetExtractorの生成
+		UserRowCallbackHandler handler = new UserRowCallbackHandler();
 
+		// SQLの実行＆CSV出力
+		jdbc.query(sql, handler);
 	}
 }

src/main/java/com/example/demo/domain/model/repository/jdbc/UserRowCallbackHandler.java

@@ -0,0 +1,49 @@
+package com.example.demo.domain.model.repository.jdbc;
+
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+import org.springframework.jdbc.core.RowCallbackHandler;
+
+
+public class UserRowCallbackHandler implements RowCallbackHandler {
+
+	@Override
+	public void processRow(ResultSet rs) throws SQLException {
+		try {
+			// ファイル書き込みの準備
+			File file = new File("sample.csv");
+			FileWriter fw = new FileWriter(file.getAbsoluteFile());
+			BufferedWriter bw = new BufferedWriter(fw);
+			
+			// 取得件数分loop
+			do {
+				// ResultSetから値を取得してStringにセット
+				String str = rs.getString("user_id") + ","
+				           + rs.getString("password") + ","
+				           + rs.getString("user_name") + ","
+				           + rs.getString("birthday") + ","
+				           + rs.getString("age") + ","
+				           + rs.getString("marriage") + ","
+				           + rs.getString("role");
+				
+				// ファイルに書き込み＆改行
+				bw.write(str);
+				bw.newLine();
+
+		    } while(rs.next());
+			
+			// 書き込み
+			bw.flush();
+			// ファイルを閉じる
+			bw.close();
+		} catch(IOException e) {
+			e.printStackTrace();
+			throw new SQLException(e);
+		}
+	}
+}

src/main/java/com/example/demo/domain/service/UserService.java

@@ -1,9 +1,15 @@
 package com.example.demo.domain.service;
 
+import java.io.IOException;
+import java.nio.file.FileSystem;
+import java.nio.file.FileSystems;
+import java.nio.file.Files;
+import java.nio.file.Path;
 import java.util.List;
 
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.dao.DataAccessException;
 import org.springframework.stereotype.Service;
 
 import com.example.demo.domain.model.User;
@@ -13,7 +19,7 @@
 @Service
 public class UserService {
     @Autowired
-    @Qualifier("UserDaoJdbcImpl4")
+    @Qualifier("UserDaoJdbcImpl")
     UserDao dao;
 
     // insert用メソッド
@@ -79,4 +85,24 @@ public boolean deleteOne(String userId) {
 
     	return result;
     }
+    
+    // ユーザー一覧をCSV出力
+    public void userCsvOut() throws DataAccessException {
+    	// CSV出力
+    	dao.userCsvOut();
+    }
+    
+    // サーバーに保存されたファイルを取得する（Byte配列に変換）
+    public byte[] getFile(String fileName) throws IOException {
+    	// ファイルシステムの取得
+    	FileSystem fs = FileSystems.getDefault();
+    	
+    	// ファイル取得
+    	Path p = fs.getPath(fileName);
+    	
+    	// ファイルをByte配列に変換
+    	byte[] bytes = Files.readAllBytes(p);
+    	
+    	return bytes;
+    }
 }

src/main/java/com/example/demo/login/controller/HomeController.java

@@ -1,10 +1,14 @@
 package com.example.demo.login.controller;
 
+import java.io.IOException;
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Controller;
 import org.springframework.ui.Model;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -153,8 +157,28 @@ public String postLogout() {
 
     // ユーザー一覧のCSV出力
     @GetMapping("/userList/csv")
-    public String getUserListCsv(Model model) {
-    	// TODO:あとで実装する
-    	return getUserList(model);
+    public ResponseEntity<byte[]> getUserListCsv(Model model) {
+
+        //ユーザーを全件取得して、CSVをサーバーに保存する
+        userService.userCsvOut();
+
+        byte[] bytes = null;
+
+        try {
+
+            //サーバーに保存されているsample.csvファイルをbyteで取得する
+            bytes = userService.getFile("sample.csv");
+
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+
+        //HTTPヘッダーの設定
+        HttpHeaders header = new HttpHeaders();
+        header.add("Content-Type", "text/csv; charset=UTF-8");
+        header.setContentDispositionFormData("filename", "sample.csv");
+
+        //sample.csvを戻す
+        return new ResponseEntity<>(bytes, header, HttpStatus.OK);
     }
 }

src/main/resources/templates/login/userList.html

@@ -34,7 +34,7 @@ <h1>ユーザー一覧</h1>
     <!-- 更新・削除処理の結果表示用 -->
     <label class="text-info" th:text="${result}">結果表示</label><br/>
     <!-- CSV出力用のリンク -->
-    <label class="btn btn-primary" th:href="@{'/userList/csv'}">CSV出力</label>
+    <a class="btn btn-primary" th:href="@{'/userList/csv'}">CSV出力</a>
   </div>
 </body>
 </html>