package com.qianwen.smartman.modules.smis.service.impl; import java.util.ArrayList; import java.util.List; import org.springframework.stereotype.Service; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.qianwen.core.mp.base.BaseServiceImpl; import com.qianwen.core.tool.utils.Func; import com.qianwen.smartman.modules.smis.entity.EnterpriseInformation; import com.qianwen.smartman.modules.smis.mapper.EnterpriseInformationMapper; import com.qianwen.smartman.modules.smis.service.IEnterpriseInformationService; import com.qianwen.smartman.modules.smis.vo.EnterpriseInVO; import com.qianwen.smartman.modules.smis.vo.EnterpriseInformationVO; import com.qianwen.smartman.modules.smis.vo.EnterprisePictureVO; @Service public class EnterpriseInformationServiceImpl extends BaseServiceImpl implements IEnterpriseInformationService { @Override public Boolean editEnterprise(EnterpriseInVO enterpriseInformation) { return Boolean.valueOf(update(Wrappers.lambdaUpdate() .eq(EnterpriseInformation::getId, enterpriseInformation.getId()) .set(EnterpriseInformation::getObjectKey, enterpriseInformation.getObjectKey()) .set(EnterpriseInformation::getContentLength, enterpriseInformation.getContentLength()) .set(EnterpriseInformation::getLink, enterpriseInformation.getLink()) .set(EnterpriseInformation::getSuffix, enterpriseInformation.getSuffix()) .set(EnterpriseInformation::getFileName, enterpriseInformation.getFileName()))); } @Override public EnterpriseInformationVO getEnterprise() { List list = list(); EnterpriseInformationVO enterpriseInformationVO = new EnterpriseInformationVO(); if (Func.isNotEmpty(list)) { List pictureVOList = new ArrayList<>(); list.forEach(s -> { EnterprisePictureVO pictureVO = EnterprisePictureVO.builder().id(String.valueOf(s.getId())).objectKey(s.getObjectKey()).contentLength(s.getContentLength()).fileName(s.getFileName()).link(s.getLink()).suffix(s.getSuffix()).type(s.getType()).build(); pictureVOList.add(pictureVO); }); enterpriseInformationVO.setList(pictureVOList); } return enterpriseInformationVO; } @Override public Boolean delEnterprise(String id) { return Boolean.valueOf(removeById(Long.valueOf(Func.toLong(id)))); } @Override public Boolean addEnterprise(EnterpriseInVO enterprise) { EnterpriseInformation enterpriseInformation = new EnterpriseInformation(); enterpriseInformation.setLink(enterprise.getLink()).setFileName(enterprise.getFileName()).setSuffix(enterprise.getSuffix()).setObjectKey(enterprise.getObjectKey()).setContentLength(enterprise.getContentLength()).setType(enterprise.getType()); return Boolean.valueOf(save(enterpriseInformation)); } }