package kr.wisestone.owl.service.impl; import com.google.common.collect.Lists; import kr.wisestone.owl.constant.Constants; import kr.wisestone.owl.constant.MsgConstants; import kr.wisestone.owl.domain.CompanyFieldCategory; import kr.wisestone.owl.domain.enumType.CompanyFieldCategoryType; import kr.wisestone.owl.exception.OwlRuntimeException; import kr.wisestone.owl.mapper.CompanyFieldCategoryMapper; import kr.wisestone.owl.repository.CompanyFieldCategoryRepository; import kr.wisestone.owl.service.*; import kr.wisestone.owl.util.ConvertUtil; import kr.wisestone.owl.vo.*; import kr.wisestone.owl.web.condition.CompanyFieldCategoryCondition; 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 java.util.List; import java.util.Map; @Service public class CompanyFieldCategoryServiceImpl extends AbstractServiceImpl> implements CompanyFieldCategoryService { @Autowired private CompanyFieldCategoryRepository companyFieldCategoryRepository; @Autowired private CompanyFieldCategoryMapper companyFieldCategoryMapper; @Override protected JpaRepository getRepository() { return this.companyFieldCategoryRepository; } @Override public List find(Map resJsonData, CompanyFieldCategoryCondition condition, Pageable pageable) { condition.setPage(pageable.getPageNumber() * pageable.getPageSize()); condition.setPageSize(pageable.getPageSize()); List> results = Lists.newArrayList(); Long totalCount = 0L; if (condition.getTypeCategory() != null && !condition.getTypeCategory().equals("")) { if(condition.getTypeCategory().equals("CHILDSECTOR") && (condition.getParentId() == null || condition.getParentId().equals(""))){ throw new OwlRuntimeException( this.messageAccessor.getMessage(MsgConstants.PARENT_SECTOR_NOT_SELECT)); } results = this.companyFieldCategoryMapper.find(condition); totalCount = this.companyFieldCategoryMapper.count(condition); } return this.convertCompanyFieldCategoryVoToMap(results, totalCount, pageable, resJsonData); } public CompanyFieldCategory find(Long id) { return this.companyFieldCategoryMapper.findById(id); } public List> findByType(CompanyFieldCategoryType categoryType) { CompanyFieldCategoryCondition condition = new CompanyFieldCategoryCondition(); switch (categoryType) { case COMPANYTYPE: condition.setTypeCategory("COMPANYTYPE"); break; case PARENTSECTOR: condition.setTypeCategory("PARENTSECTOR"); break; case CHILDSECTOR: condition.setTypeCategory("CHILDSECTOR"); break; case REGION: condition.setTypeCategory("REGION"); break; case STATUS: condition.setTypeCategory("STATUS"); break; } return this.companyFieldCategoryMapper.find(condition); } // 검색 결과를 Vo 로 변환한다. private List convertCompanyFieldCategoryVoToMap(List> results, Long totalCount, Pageable pageable, Map resJsonData) { List companyFieldCategoryVos = Lists.newArrayList(); for (Map result : results) { CompanyFieldCategoryVo companyFieldCategoryVo = ConvertUtil.convertMapToClass(result, CompanyFieldCategoryVo.class); companyFieldCategoryVos.add(companyFieldCategoryVo); } int totalPage = (int) Math.ceil((totalCount - 1) / pageable.getPageSize()) + 1; resJsonData.put(Constants.RES_KEY_CONTENTS, companyFieldCategoryVos); resJsonData.put(Constants.REQ_KEY_PAGE_VO, new ResPage(pageable.getPageNumber(), pageable.getPageSize(), totalPage, totalCount)); return companyFieldCategoryVos; } }