기본 샘플 구조

Author: bnitech

src/main/java/com/bnitech/study/demo/module/ExcelWorkbook.java

@@ -1,33 +1,48 @@
 package com.bnitech.study.demo.module;
 
-import java.io.IOException;
-import java.io.OutputStream;
-
 import org.apache.poi.ss.SpreadsheetVersion;
 import org.apache.poi.xssf.streaming.SXSSFWorkbook;
 
-public class ExcelWorkbook {
-
-private static final SpreadsheetVersion supplyExcelVersion = SpreadsheetVersion.EXCEL2007;
-
-private SXSSFWorkbook wb;
-
-public ExcelWorkbook() {
-wb = new SXSSFWorkbook();
-}
-
-public ExcelSheet createSheet() {
-return new ExcelSheet(wb.createSheet());
-}
+import java.io.IOException;
+import java.io.OutputStream;
 
-public ExcelSheet createSheet(String name) {
-return new ExcelSheet(wb.createSheet(name));
-}
+public class ExcelWorkbook {
 
-public void write(OutputStream stream) throws IOException {
-wb.write(stream);
-wb.close();
-wb.dispose();
-stream.close();
-}
+ private static final SpreadsheetVersion supplyExcelVersion = SpreadsheetVersion.EXCEL2007;
+
+ private SXSSFWorkbook wb;
+ private OutputStream outputStream;
+ private Class<?> aClass;
+
+ public ExcelWorkbook() {
+ wb = new SXSSFWorkbook();
+ }
+
+ public ExcelWorkbook(OutputStream stream, Class<?> aClass) {
+ this();
+ this.outputStream = stream;
+ this.aClass = aClass;
+ }
+
+ public ExcelSheet createSheet() {
+ return new ExcelSheet(wb.createSheet());
+ }
+
+ public ExcelSheet createSheet(String name) {
+ return new ExcelSheet(wb.createSheet(name));
+ }
+
+ public void write(OutputStream stream) throws IOException {
+ wb.write(stream);
+ wb.close();
+ wb.dispose();
+ stream.close();
+ }
+
+ public void write() throws IOException {
+ wb.write(outputStream);
+ wb.close();
+ wb.dispose();
+ outputStream.close();
+ }
 }

src/main/java/com/bnitech/study/demo/sample/controller/PersonController.java

@@ -0,0 +1,42 @@
+package com.bnitech.study.demo.sample.controller;
+
+import com.bnitech.study.demo.module.ExcelSheet;
+import com.bnitech.study.demo.module.ExcelWorkbook;
+import com.bnitech.study.demo.sample.dto.PersonDto;
+import com.bnitech.study.demo.sample.service.IPersonService;
+import lombok.RequiredArgsConstructor;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.util.List;
+
+@RestController
+@RequiredArgsConstructor
+public class PersonController {
+
+ private final IPersonService personService;
+
+ @GetMapping("/person/all")
+ public List<PersonDto> getPersonInfoList() {
+ return personService.getPersonList();
+ }
+
+ @GetMapping("/person/{personId}")
+ public PersonDto getPersonInfo(@PathVariable long personId) {
+ return personService.getPerson(personId);
+ }
+
+ @GetMapping("/person/excel")
+ public void getPersonInfoListByExcel(HttpServletResponse response) throws IOException {
+ response.setHeader("Set-Cookie", "fileDownload=true; path=/");
+ response.setHeader("Content-Disposition", "attachment; filename=\"Car List.xlsx\"");
+
+ ExcelWorkbook workbook = new ExcelWorkbook(response.getOutputStream(), PersonDto.class);
+ ExcelSheet sheet = workbook.createSheet();
+
+ workbook.write();
+ }
+}

src/main/java/com/bnitech/study/demo/sample/dao/Person.java

@@ -0,0 +1,22 @@
+package com.bnitech.study.demo.sample.dao;
+
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+import lombok.ToString;
+
+import javax.persistence.Entity;
+import javax.persistence.Id;
+
+@Entity
+@Getter
+@Setter
+@NoArgsConstructor
+@ToString
+public class Person {
+ @Id
+ private long id;
+ private String name;
+ private String company;
+ private int salary;
+}

src/main/java/com/bnitech/study/demo/sample/dto/PersonDto.java

@@ -0,0 +1,17 @@
+package com.bnitech.study.demo.sample.dto;
+
+import com.bnitech.study.demo.module.ExcelHeader;
+import com.bnitech.study.demo.module.dto.ExcelDto;
+
+public class PersonDto extends ExcelDto {
+
+ @Override
+ protected void setExcelHeader(ExcelHeader excelHeader) {
+ super.excelHeader = excelHeader;
+ }
+
+ @Override
+ protected int getExcelValue() {
+ return 0;
+ }
+}

src/main/java/com/bnitech/study/demo/sample/repository/PersonRepository.java

@@ -0,0 +1,9 @@
+package com.bnitech.study.demo.sample.repository;
+
+import com.bnitech.study.demo.sample.dao.Person;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface PersonRepository extends JpaRepository<Person, Long> {
+}

src/main/java/com/bnitech/study/demo/sample/service/IPersonService.java

@@ -0,0 +1,11 @@
+package com.bnitech.study.demo.sample.service;
+
+import com.bnitech.study.demo.sample.dto.PersonDto;
+
+import java.util.List;
+
+public interface IPersonService {
+ List<PersonDto> getPersonList();
+
+ PersonDto getPerson(long personId);
+}

src/main/java/com/bnitech/study/demo/sample/service/impl/PersonService.java

@@ -0,0 +1,20 @@
+package com.bnitech.study.demo.sample.service.impl;
+
+import com.bnitech.study.demo.sample.dto.PersonDto;
+import com.bnitech.study.demo.sample.service.IPersonService;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+
+@Service
+public class PersonService implements IPersonService {
+ @Override
+ public List<PersonDto> getPersonList() {
+ return null;
+ }
+
+ @Override
+ public PersonDto getPerson(long personId) {
+ return null;
+ }
+}