From 26b03b6720fe222be3fdad7931a94a4842dca0c2 Mon Sep 17 00:00:00 2001 From: fuliqi <fuliqi@qq.com> Date: 星期三, 21 八月 2024 11:52:49 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java | 196 ++++++++++++++++++++++++++++++++++++++---------- 1 files changed, 153 insertions(+), 43 deletions(-) diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java index 811d071..e41a9c8 100644 --- a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java +++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java @@ -1,30 +1,46 @@ package com.ycl.platform.service.impl; +import com.alibaba.excel.EasyExcel; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper; +import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ycl.dataListener.CurrencyDataListener; +import com.ycl.platform.base.BaseSelect; +import com.ycl.platform.domain.entity.YwPeople; import com.ycl.platform.domain.entity.YwPoint; import com.ycl.platform.domain.entity.YwUnit; -import com.ycl.platform.mapper.YwPointMapper; -import com.ycl.platform.mapper.YwUnitMapper; -import com.ycl.platform.service.YwPointService; -import com.ycl.system.Result; -import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.ycl.platform.domain.form.BatchEditPointForm; import com.ycl.platform.domain.form.YwPointForm; -import com.ycl.platform.domain.vo.YwPointVO; import com.ycl.platform.domain.query.YwPointQuery; -import java.util.List; - -import com.baomidou.mybatisplus.core.metadata.IPage; +import com.ycl.platform.domain.vo.YwPointVO; +import com.ycl.platform.domain.excel.PointExport; +import com.ycl.platform.domain.excel.PointSelectHandler; +import com.ycl.platform.mapper.YwPeopleMapper; +import com.ycl.platform.mapper.YwPointMapper; +import com.ycl.platform.service.YwPointService; +import com.ycl.platform.service.YwUnitService; +import com.ycl.system.Result; +import com.ycl.system.mapper.SysDeptMapper; import com.ycl.system.page.PageUtil; -import org.springframework.stereotype.Service; -import org.springframework.security.core.context.SecurityContextHolder; +import com.ycl.utils.DateUtils; +import com.ycl.utils.SecurityUtils; +import jakarta.servlet.http.HttpServletResponse; +import lombok.RequiredArgsConstructor; import org.springframework.beans.BeanUtils; +import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import org.springframework.util.Assert; -import java.util.ArrayList; -import java.util.stream.Collectors; -import com.baomidou.mybatisplus.extension.plugins.pagination.Page; -import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper; -import lombok.RequiredArgsConstructor; +import org.springframework.util.CollectionUtils; import org.springframework.util.StringUtils; +import org.springframework.web.multipart.MultipartFile; + +import java.io.IOException; +import java.util.Date; +import java.util.List; +import java.util.Objects; +import java.util.function.Consumer; +import java.util.stream.Collectors; /** * 杩愮淮鐐逛綅 鏈嶅姟瀹炵幇绫� @@ -36,8 +52,9 @@ @RequiredArgsConstructor public class YwPointServiceImpl extends ServiceImpl<YwPointMapper, YwPoint> implements YwPointService { - private final YwPointMapper ywPointMapper; - private final YwUnitMapper ywUnitMapper; + private final SysDeptMapper sysDeptMapper; + private final YwUnitService unitService; + private final YwPeopleMapper ywPeopleMapper; /** * 娣诲姞 @@ -47,6 +64,9 @@ @Override public Result add(YwPointForm form) { YwPoint entity = YwPointForm.getEntityByForm(form, null); + Date now = new Date(); + entity.setCreateTime(now); + entity.setUpdateTime(now); if(baseMapper.insert(entity) > 0) { return Result.ok("娣诲姞鎴愬姛"); } @@ -60,9 +80,33 @@ .forEach(form -> { YwPoint entity = YwPointForm.getEntityByForm(form, null); entity.setStatus("鏈紑濮�"); + Date now = new Date(); + entity.setCreateTime(now); + entity.setUpdateTime(now); baseMapper.insert(entity); }); return Result.ok("娣诲姞鎴愬姛"); + } + + @Override + @Transactional(rollbackFor = Exception.class) + public Result batchEdit(BatchEditPointForm form) { + if (Objects.nonNull(form.getUnitId())) { + new LambdaUpdateChainWrapper<>(baseMapper) + .in(YwPoint::getId, form.getIds()) + .set(YwPoint::getUnitId, form.getUnitId()) + .set(YwPoint::getUpdateTime, new Date()) + .update(); + } else { + new LambdaUpdateChainWrapper<>(baseMapper) + .in(YwPoint::getId, form.getIds()) + .set(YwPoint::getUnitId, form.getUnitId()) + .set(YwPoint::getStartTime, null) + .set(YwPoint::getEndTime, null) + .set(YwPoint::getUpdateTime, new Date()) + .update(); + } + return Result.ok("淇敼鎴愬姛"); } /** @@ -72,16 +116,13 @@ */ @Override public Result update(YwPointForm form) { - YwPoint entity = baseMapper.selectById(form.getId()); - - // 涓虹┖鎶汭llegalArgumentException锛屽仛鍏ㄥ眬寮傚父澶勭悊 Assert.notNull(entity, "璁板綍涓嶅瓨鍦�"); BeanUtils.copyProperties(form, entity); - if (baseMapper.updateById(entity) > 0) { - return Result.ok("淇敼鎴愬姛"); - } - return Result.error("淇敼澶辫触"); + Date now = new Date(); + entity.setUpdateTime(now); + baseMapper.updateById(entity); + return Result.ok("淇敼鎴愬姛"); } /** @@ -117,23 +158,9 @@ */ @Override public Result page(YwPointQuery query) { - - IPage<YwPoint> page = new LambdaQueryChainWrapper<>(baseMapper) - .like(StringUtils.hasText(query.getPointName()), YwPoint::getPointName, query.getPointName()) - .orderByDesc(YwPoint::getCreateTime) - .page(PageUtil.getPage(query, YwPoint.class)); - - List<YwPointVO> vos = page.getRecords().stream() - .map( - entity -> { - YwPointVO vo = YwPointVO.getVoByEntity(entity, null); - YwUnit unit = ywUnitMapper.selectById(vo.getUnitId()); - vo.setUnitName(unit.getUnitName()); - return vo; - } - ) - .collect(Collectors.toList()); - return Result.ok().data(vos).total(page.getTotal()); + IPage<YwPointVO> page = PageUtil.getPage(query, YwPointVO.class); + baseMapper.page(page, query); + return Result.ok().data(page.getRecords()).total(page.getTotal()); } /** @@ -143,10 +170,12 @@ */ @Override public Result detail(String id) { - YwPoint entity = baseMapper.selectById(id); Assert.notNull(entity, "璁板綍涓嶅瓨鍦�"); YwPointVO vo = YwPointVO.getVoByEntity(entity, null); + if (Objects.nonNull(entity.getDeptId())) { + vo.setDeptIds(sysDeptMapper.selectParents(entity.getDeptId())); + } return Result.ok().data(vo); } @@ -164,4 +193,85 @@ .collect(Collectors.toList()); return Result.ok().data(vos); } + + @Override + public Result select(String keyword) { + Long userId = SecurityUtils.getUserId(); + // 鑾峰彇褰撳墠杩愮淮浜哄憳鐨勮繍缁村崟浣� + YwPeople people = new LambdaQueryChainWrapper<>(ywPeopleMapper) + .eq(YwPeople::getUserId, userId) + .one(); + Integer unitId = null; + if (Objects.nonNull(people)) { + unitId = people.getBelongUnit(); + } + + List<YwPoint> list = new LambdaQueryChainWrapper<>(baseMapper) + .eq(YwPoint::getUnitId, unitId) + .like(YwPoint::getPointName, keyword) + .list(); + List<BaseSelect> data = list.stream().map(item -> { + BaseSelect baseSelect = new BaseSelect(); + baseSelect.setId(item.getId()); + baseSelect.setValue(item.getPointName()); + return baseSelect; + }).collect(Collectors.toList()); + return Result.ok().data(data); + } + + @Override + public boolean updateRecovery(List<String> serialNumberList, int recovery) { + return new LambdaUpdateChainWrapper<>(baseMapper) + .in(YwPoint::getSerialNumber, serialNumberList) + .set(YwPoint::getRecovery, recovery) + .set(YwPoint::getUpdateTime, new Date()) + .set(recovery == 0, YwPoint::getRecoveryTime, DateUtils.getNowDate()) + .update(); + } + + @Override + public List<YwPoint> home() { + return baseMapper.home(); + } + + @Override + public void export(YwPointQuery query, HttpServletResponse response) throws IOException { + // 瀵煎嚭鏁版嵁 + List<PointExport> exportData = baseMapper.export(query); + + EasyExcel.write(response.getOutputStream(), PointExport.class) + .sheet("鐐逛綅鏇存崲杩愮淮鍗曚綅") + .doWrite(exportData); + } + + @Override + public Result importData(MultipartFile file, Integer unitId, Date startTime, Date endTime, Boolean provinceTag, Boolean importantCommandImageTag) throws IOException { + Consumer<List<PointExport>> consumer = (dataList) -> { + this.updatePoint(dataList, unitId, startTime, endTime, provinceTag, importantCommandImageTag); + }; + EasyExcel.read(file.getInputStream(), PointExport.class , new CurrencyDataListener(consumer)).headRowNumber(1).doReadAll(); + return Result.ok(); + } + + /** + * 淇敼鐐逛綅鐨勮繍缁村崟浣� + * + * @param dataList + * @param unitId + */ + private void updatePoint(List<PointExport> dataList, Integer unitId, Date startTime, Date endTime, Boolean provinceTag, Boolean importantCommandImageTag) { + if (CollectionUtils.isEmpty(dataList)) { + throw new RuntimeException("瀵煎叆鏁版嵁涓嶈兘涓虹┖"); + } + List<String> pointList = dataList.stream().map(PointExport::getSerialNumber).collect(Collectors.toList()); + new LambdaUpdateChainWrapper<>(baseMapper) + .in(YwPoint::getSerialNumber, pointList) + .set(YwPoint::getUnitId, unitId) + .set(YwPoint::getStartTime, startTime) + .set(YwPoint::getEndTime, endTime) + .set(YwPoint::getProvinceTag, provinceTag) + .set(YwPoint::getImportantCommandImageTag, importantCommandImageTag) + .update(); + } + } -- Gitblit v1.8.0