|
@@ -3,6 +3,7 @@ package com.dayou.service.impl;
|
|
|
import cn.dev33.satoken.stp.StpUtil;
|
|
|
import com.alibaba.excel.EasyExcel;
|
|
|
import com.alibaba.excel.read.listener.PageReadListener;
|
|
|
+import com.dayou.dto.calculate.equipment.EquipmentBaseInfoDTO;
|
|
|
import com.dayou.entity.AssetsCalculate;
|
|
|
import com.dayou.entity.AssetsCalculateEqptData;
|
|
|
import com.dayou.mapper.AssetsCalculateMapper;
|
|
@@ -14,6 +15,7 @@ import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
+import org.springframework.transaction.annotation.Transactional;
|
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
|
|
import java.io.IOException;
|
|
@@ -82,6 +84,7 @@ public class AssetsCalculateServiceImpl extends ServiceImpl<AssetsCalculateMappe
|
|
|
* @return Boolean
|
|
|
*/
|
|
|
@Override
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
public void importEqptExcelData(MultipartFile multipartFile, Long calculateId) throws IOException {
|
|
|
// 读取excel内容并批量保存到数据库
|
|
|
InputStream inputStream = multipartFile.getInputStream();
|
|
@@ -91,5 +94,21 @@ public class AssetsCalculateServiceImpl extends ServiceImpl<AssetsCalculateMappe
|
|
|
// 批量保存
|
|
|
assetsCalculateEqptDataService.saveBatch(dataList);
|
|
|
}, 1000)).sheet().doRead();
|
|
|
+ // 更新测算表进度
|
|
|
+ assetsCalculateMapper.updateCalculateProgress(calculateId,"GENERATE");
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 更新测算表基础测算信息
|
|
|
+ * @param calculateId id
|
|
|
+ * @param baseInfo 基础信息json字符串
|
|
|
+ * @return Boolean
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
+ public Boolean updateCalculateBaseInfo(Long calculateId, String baseInfo) {
|
|
|
+ // 更新测算表进度
|
|
|
+ assetsCalculateMapper.updateCalculateProgress(calculateId,"GENERATE");
|
|
|
+ return assetsCalculateMapper.updateCalculateBaseInfo(calculateId, baseInfo);
|
|
|
}
|
|
|
}
|