package kr.wisestone.owl.service.impl; import com.google.common.collect.Lists; import kr.wisestone.owl.common.ExcelConditionCheck; import kr.wisestone.owl.constant.Constants; import kr.wisestone.owl.constant.MsgConstants; import kr.wisestone.owl.domain.DepartmentManage; import kr.wisestone.owl.exception.OwlRuntimeException; import kr.wisestone.owl.mapper.DepartmentManageMapper; import kr.wisestone.owl.repository.DepartmentManageRepository; import kr.wisestone.owl.service.DepartmentManageService; import kr.wisestone.owl.service.WorkspaceService; import kr.wisestone.owl.util.ConvertUtil; import kr.wisestone.owl.vo.*; import kr.wisestone.owl.web.condition.DepartmentManageCondition; import kr.wisestone.owl.web.form.DepartmentManageForm; import kr.wisestone.owl.web.view.ExcelView; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.domain.Pageable; import org.springframework.data.jpa.repository.JpaRepository; import org.springframework.stereotype.Service; import org.springframework.ui.Model; import org.springframework.web.servlet.ModelAndView; import javax.servlet.http.HttpServletRequest; import java.util.HashMap; import java.util.List; import java.util.Map; @Service public class DepartmentManageServiceImpl extends AbstractServiceImpl> implements DepartmentManageService { @Autowired private DepartmentManageRepository departmentManageRepository; @Autowired private DepartmentManageMapper departmentManageMapper; @Autowired private WorkspaceService workspaceService; @Autowired private ExcelView excelView; @Autowired private ExcelConditionCheck excelConditionCheck; @Override protected JpaRepository getRepository() { return this.departmentManageRepository; } @Override public DepartmentManage addDepartmentManage(DepartmentManageForm departmentManageForm) { DepartmentManage departmentManage = ConvertUtil.copyProperties(departmentManageForm, DepartmentManage.class); departmentManageRepository.saveAndFlush(departmentManage); return departmentManage; } @Override public DepartmentManage getDepartment(Long id) { if (id == null) { throw new OwlRuntimeException( this.messageAccessor.getMessage(MsgConstants.USER_LEVEL_NOT_EXIST)); } DepartmentManage departmentManage = this.findOne(id); if (departmentManage == null) { throw new OwlRuntimeException( this.messageAccessor.getMessage(MsgConstants.USER_LEVEL_NOT_EXIST)); } return departmentManage; } @Override public List findDepartment(Map resJsonData, DepartmentManageCondition condition, Pageable pageable) { condition.setPage(pageable.getPageNumber() * pageable.getPageSize()); condition.setPageSize(pageable.getPageSize()); List> results = this.departmentManageMapper.find(condition); Long totalDepartmentCount = this.departmentManageMapper.count(condition); return this.convertDepartmentManageVoToMap(results, totalDepartmentCount, pageable, resJsonData); } // 검색 결과를 DepartmentManageVo 로 변환한다. private List convertDepartmentManageVoToMap(List> results, Long totalDepartmentCount, Pageable pageable, Map resJsonData) { List departmentManageVos = Lists.newArrayList(); for (Map result : results) { DepartmentManageVo departmentManageVo = ConvertUtil.convertMapToClass(result, DepartmentManageVo.class); // 부서명만 변환 하면 되는건가 ? departmentManageVos.add(departmentManageVo); } int totalPage = (int) Math.ceil((totalDepartmentCount - 1) / pageable.getPageSize()) + 1; resJsonData.put(Constants.RES_KEY_CONTENTS, departmentManageVos); resJsonData.put(Constants.REQ_KEY_PAGE_VO, new ResPage(pageable.getPageNumber(), pageable.getPageSize(), totalPage, totalDepartmentCount)); return departmentManageVos; } // 부서 목록을 엑셀로 다운로드 한다. public ModelAndView downloadExcel(HttpServletRequest request, Model model) { ModelAndView modelAndView = this.workspaceService.checkUseExcelDownload(model); if (modelAndView != null) { return modelAndView; } Map conditions = new HashMap<>(); // 엑셀 다운로드에 필요한 검색 조건 정보를 추출하고 검색 조건 추출에 오류가 발생하면 경고를 표시해준다. modelAndView = this.excelConditionCheck.checkCondition(conditions, request, model); if (modelAndView != null) { return modelAndView; } DepartmentManageCondition departmentManageCondition = DepartmentManageCondition.make(conditions); List> results = this.departmentManageMapper.find(departmentManageCondition); List departmentManageVos = ConvertUtil.convertListToListClass(results, DepartmentManageVo.class); // code_ko_KR 에 code명 설정 ExportExcelVo excelInfo = new ExportExcelVo(); excelInfo.setFileName(this.messageAccessor.message("부서 목록")); excelInfo.addAttrInfos(new ExportExcelAttrVo("departmentName", this.messageAccessor.message("departmentManage.departmentName"), 6, ExportExcelAttrVo.ALIGN_CENTER)); excelInfo.addAttrInfos(new ExportExcelAttrVo("departmentDescription", this.messageAccessor.message("departmentManage.departmentDescription"), 20, ExportExcelAttrVo.ALIGN_CENTER)); excelInfo.setDatas(departmentManageVos); model.addAttribute(Constants.EXCEL, excelInfo); return new ModelAndView(this.excelView); } }