|
@@ -33,7 +33,7 @@ import java.util.List;
|
|
|
|
|
|
/**
|
|
/**
|
|
* <p>
|
|
* <p>
|
|
- * 服务实现类
|
|
|
|
|
|
+ * 服务实现类
|
|
* </p>
|
|
* </p>
|
|
*
|
|
*
|
|
* @author 吴涛涛
|
|
* @author 吴涛涛
|
|
@@ -47,81 +47,86 @@ public class PartServiceImpl extends ServiceImpl<PartMapper, Part> implements Pa
|
|
UserMapper userMapper;
|
|
UserMapper userMapper;
|
|
@Resource
|
|
@Resource
|
|
MouldMapper mouldMapper;
|
|
MouldMapper mouldMapper;
|
|
|
|
+
|
|
@Override
|
|
@Override
|
|
public HttpRespMsg importPartExcel(MultipartFile file, UserVO userVO) throws IOException, InvalidFormatException {
|
|
public HttpRespMsg importPartExcel(MultipartFile file, UserVO userVO) throws IOException, InvalidFormatException {
|
|
HttpRespMsg msg = new HttpRespMsg();
|
|
HttpRespMsg msg = new HttpRespMsg();
|
|
User user = userMapper.selectOne(new QueryWrapper<User>().eq("head_imgurl", userVO.getToken()));
|
|
User user = userMapper.selectOne(new QueryWrapper<User>().eq("head_imgurl", userVO.getToken()));
|
|
- if(user != null){
|
|
|
|
- List<Part> parts = partMapper.selectList(new QueryWrapper<Part>().eq("mould_id",userVO.getMouldId()));
|
|
|
|
|
|
+ if (user != null) {
|
|
|
|
+ List<Part> parts = partMapper.selectList(new QueryWrapper<Part>().eq("mould_id", userVO.getMouldId()));
|
|
// try {
|
|
// try {
|
|
- File f = null;
|
|
|
|
- if ("".equals(file) || file.getSize() <= 0) {
|
|
|
|
- file = null;
|
|
|
|
- } else {
|
|
|
|
- //获取输入流
|
|
|
|
- InputStream ins = file.getInputStream();
|
|
|
|
- //新建一个文件
|
|
|
|
- f = new File(file.getOriginalFilename());
|
|
|
|
- //输入流转file
|
|
|
|
- inputStreamToFile(ins, f);
|
|
|
|
- }
|
|
|
|
- Mould mould = mouldMapper.selectById(userVO.getMouldId());
|
|
|
|
- //根据文件创建工作簿
|
|
|
|
- XSSFWorkbook wookbook = new XSSFWorkbook(f);
|
|
|
|
- XSSFSheet sheet = wookbook.getSheetAt(0);
|
|
|
|
- int s = sheet.getLastRowNum();
|
|
|
|
- System.out.println("s==========>"+s);
|
|
|
|
- // 遍历当前sheet中的所有行,第一行是数据对应的字段,不是数据,
|
|
|
|
- // 故从第二行开始遍历拿数据(如果有标题的话,则从第三行开始拿数据)
|
|
|
|
- for (int j = 1; j < sheet.getLastRowNum() + 1; j++) {
|
|
|
|
- XSSFRow row = sheet.getRow(j);
|
|
|
|
- System.out.println("row====>"+row);
|
|
|
|
- //新建零件
|
|
|
|
- Part part = new Part();
|
|
|
|
- part.setMouldId(mould.getId());
|
|
|
|
- // 遍历所有的列,下面的10是excle表格里共有10列即对应了10个字段
|
|
|
|
- for (int y = 0; y < 2; y++) {
|
|
|
|
- XSSFCell cell = row.getCell(y);
|
|
|
|
- System.out.println("cell====>"+row);
|
|
|
|
- cell.setCellType(Cell.CELL_TYPE_STRING);
|
|
|
|
- //取出当前列的值
|
|
|
|
- String value = cell.getStringCellValue();
|
|
|
|
- //判断第几列插入数据,后面就是从列中取数据往对象里放,然后插入到数据库里
|
|
|
|
- if (value == null && "".equals(value)) {
|
|
|
|
- log.error("数据不可为空");
|
|
|
|
- msg.setError("数据不可为空");
|
|
|
|
- return msg;
|
|
|
|
- } else if (y == 0) {
|
|
|
|
- //零件编号
|
|
|
|
- for (Part p : parts) {
|
|
|
|
- if(value.equals(p.getPartNo())){
|
|
|
|
- msg.setError("第"+j+"行的零件编号:"+value+"已被占用,请修改后重新上传");
|
|
|
|
- return msg;
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
- part.setPartNo(value);
|
|
|
|
- } else if (y == 1) {
|
|
|
|
- //零件名称
|
|
|
|
- part.setPartName(value);
|
|
|
|
- } else if (y == 2) {
|
|
|
|
- //零件寿命
|
|
|
|
- part.setPartLife(Integer.parseInt(value));
|
|
|
|
- if(Integer.parseInt(value)<mould.getSettingLife()){
|
|
|
|
- part.setIsVulnerable(1);
|
|
|
|
|
|
+ File f = null;
|
|
|
|
+ if ("".equals(file) || file.getSize() <= 0) {
|
|
|
|
+ file = null;
|
|
|
|
+ } else {
|
|
|
|
+ //获取输入流
|
|
|
|
+ InputStream ins = file.getInputStream();
|
|
|
|
+ //新建一个文件
|
|
|
|
+ f = new File(file.getOriginalFilename());
|
|
|
|
+ //输入流转file
|
|
|
|
+ inputStreamToFile(ins, f);
|
|
|
|
+ }
|
|
|
|
+ Mould mould = mouldMapper.selectById(userVO.getMouldId());
|
|
|
|
+ //根据文件创建工作簿
|
|
|
|
+ XSSFWorkbook wookbook = new XSSFWorkbook(f);
|
|
|
|
+ XSSFSheet sheet = wookbook.getSheetAt(0);
|
|
|
|
+ int s = sheet.getLastRowNum();
|
|
|
|
+ System.out.println("s==========>" + s);
|
|
|
|
+ // 遍历当前sheet中的所有行,第一行是数据对应的字段,不是数据,
|
|
|
|
+ // 故从第二行开始遍历拿数据(如果有标题的话,则从第三行开始拿数据)
|
|
|
|
+ for (int j = 1; j < sheet.getLastRowNum() + 1; j++) {
|
|
|
|
+ XSSFRow row = sheet.getRow(j);
|
|
|
|
+ System.out.println("row====>" + row);
|
|
|
|
+ //新建零件
|
|
|
|
+ Part part = new Part();
|
|
|
|
+ part.setMouldId(mould.getId());
|
|
|
|
+ // 遍历所有的列,下面的10是excle表格里共有10列即对应了10个字段
|
|
|
|
+ for (int y = 0; y < 3; y++) {
|
|
|
|
+ XSSFCell cell = row.getCell(y);
|
|
|
|
+ System.out.println("cell====>" + row);
|
|
|
|
+ cell.setCellType(Cell.CELL_TYPE_STRING);
|
|
|
|
+ //取出当前列的值
|
|
|
|
+ String value = cell.getStringCellValue();
|
|
|
|
+ //判断第几列插入数据,后面就是从列中取数据往对象里放,然后插入到数据库里
|
|
|
|
+ if (value == null && "".equals(value)) {
|
|
|
|
+ log.error("数据不可为空");
|
|
|
|
+ msg.setError("数据不可为空");
|
|
|
|
+ return msg;
|
|
|
|
+ } else if (y == 0) {
|
|
|
|
+ //零件编号
|
|
|
|
+ for (Part p : parts) {
|
|
|
|
+ if (value.equals(p.getPartNo())) {
|
|
|
|
+ msg.setError("第" + j + "行的零件编号:" + value + "已被占用,请修改后重新上传");
|
|
|
|
+ return msg;
|
|
}
|
|
}
|
|
- part.setPartLife(Integer.parseInt(value));
|
|
|
|
}
|
|
}
|
|
|
|
+ part.setPartNo(value);
|
|
|
|
+ } else if (y == 1) {
|
|
|
|
+ //零件名称
|
|
|
|
+ part.setPartName(value);
|
|
|
|
+ } else if (y == 2) {
|
|
|
|
+ //零件寿命
|
|
|
|
+ part.setPartLife(Integer.parseInt(value));
|
|
|
|
+ if (Integer.parseInt(value) < mould.getSettingLife()) {
|
|
|
|
+ part.setIsVulnerable(1);
|
|
|
|
+ }
|
|
|
|
+ part.setPartLife(Integer.parseInt(value));
|
|
}
|
|
}
|
|
- part.setCreatorId(user.getId());
|
|
|
|
- part.setCreator(user.getUsername());
|
|
|
|
- partMapper.insert(part);
|
|
|
|
}
|
|
}
|
|
|
|
+ part.setCreatorId(user.getId());
|
|
|
|
+ part.setCreator(user.getUsername());
|
|
|
|
+ partMapper.insert(part);
|
|
|
|
+ }
|
|
|
|
+ //用完后删除临时文件
|
|
|
|
+ if (!f.isDirectory()) {
|
|
|
|
+ f.delete();
|
|
|
|
+ }
|
|
// } catch (Exception e) {
|
|
// } catch (Exception e) {
|
|
// log.error(e.getMessage(), e);
|
|
// log.error(e.getMessage(), e);
|
|
// msg.setError(e.getMessage());
|
|
// msg.setError(e.getMessage());
|
|
// return msg;
|
|
// return msg;
|
|
// }
|
|
// }
|
|
- }else{
|
|
|
|
|
|
+ } else {
|
|
msg.setError("用户不存在或者未登录");
|
|
msg.setError("用户不存在或者未登录");
|
|
}
|
|
}
|
|
return msg;
|
|
return msg;
|
|
@@ -130,7 +135,7 @@ public class PartServiceImpl extends ServiceImpl<PartMapper, Part> implements Pa
|
|
@Override
|
|
@Override
|
|
public HttpRespMsg add(Part part, UserVO userVO) {
|
|
public HttpRespMsg add(Part part, UserVO userVO) {
|
|
HttpRespMsg msg = new HttpRespMsg();
|
|
HttpRespMsg msg = new HttpRespMsg();
|
|
- if(part.getId() != null){
|
|
|
|
|
|
+ if (part.getId() != null) {
|
|
//修改
|
|
//修改
|
|
Part m = partMapper.selectOne(new QueryWrapper<Part>().eq("part_no", part.getPartNo()));
|
|
Part m = partMapper.selectOne(new QueryWrapper<Part>().eq("part_no", part.getPartNo()));
|
|
if ((m != null && m.getId() == part.getId()) || m == null) {
|
|
if ((m != null && m.getId() == part.getId()) || m == null) {
|
|
@@ -138,11 +143,11 @@ public class PartServiceImpl extends ServiceImpl<PartMapper, Part> implements Pa
|
|
} else {
|
|
} else {
|
|
msg.setError("当前模具编号已存在,请重新输入其他模具编号");
|
|
msg.setError("当前模具编号已存在,请重新输入其他模具编号");
|
|
}
|
|
}
|
|
- }else{
|
|
|
|
|
|
+ } else {
|
|
//添加
|
|
//添加
|
|
Mould mould = mouldMapper.selectById(userVO.getMouldId());
|
|
Mould mould = mouldMapper.selectById(userVO.getMouldId());
|
|
User user = userMapper.selectOne(new QueryWrapper<User>().eq("head_imgurl", userVO.getToken()));
|
|
User user = userMapper.selectOne(new QueryWrapper<User>().eq("head_imgurl", userVO.getToken()));
|
|
- if(part.getPartLife()<mould.getSettingLife()){
|
|
|
|
|
|
+ if (part.getPartLife() < mould.getSettingLife()) {
|
|
part.setIsVulnerable(1);
|
|
part.setIsVulnerable(1);
|
|
}
|
|
}
|
|
part.setCreatorId(user.getId());
|
|
part.setCreatorId(user.getId());
|
|
@@ -156,11 +161,11 @@ public class PartServiceImpl extends ServiceImpl<PartMapper, Part> implements Pa
|
|
public HttpRespMsg getList(UserVO userVO) {
|
|
public HttpRespMsg getList(UserVO userVO) {
|
|
HttpRespMsg msg = new HttpRespMsg();
|
|
HttpRespMsg msg = new HttpRespMsg();
|
|
User user = userMapper.selectOne(new QueryWrapper<User>().eq("head_imgurl", userVO.getToken()));
|
|
User user = userMapper.selectOne(new QueryWrapper<User>().eq("head_imgurl", userVO.getToken()));
|
|
- if(user != null){
|
|
|
|
|
|
+ if (user != null) {
|
|
List<PartVO> list = new ArrayList<>();
|
|
List<PartVO> list = new ArrayList<>();
|
|
list = partMapper.selectPartFileByMouldId(userVO);
|
|
list = partMapper.selectPartFileByMouldId(userVO);
|
|
msg.data = list;
|
|
msg.data = list;
|
|
- }else{
|
|
|
|
|
|
+ } else {
|
|
msg.setError("用户不存在或者未登录");
|
|
msg.setError("用户不存在或者未登录");
|
|
}
|
|
}
|
|
return msg;
|
|
return msg;
|