diff --git a/src/main/java/com/iflytop/sgs/app/service/api/CraftMonitorService.java b/src/main/java/com/iflytop/sgs/app/service/api/CraftMonitorService.java index 4cb459a..1953ef8 100644 --- a/src/main/java/com/iflytop/sgs/app/service/api/CraftMonitorService.java +++ b/src/main/java/com/iflytop/sgs/app/service/api/CraftMonitorService.java @@ -1,5 +1,8 @@ package com.iflytop.sgs.app.service.api; +import cn.hutool.core.bean.BeanUtil; +import cn.hutool.core.util.ObjectUtil; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.iflytop.sgs.app.mapper.CraftMonitorMapper; import com.iflytop.sgs.app.model.entity.CraftMonitor; @@ -15,4 +18,13 @@ import org.springframework.stereotype.Service; @RequiredArgsConstructor public class CraftMonitorService extends ServiceImpl { + public void saveMonitor(CraftMonitor craftMonitor) { + CraftMonitor craftMonitorExist = this.getOne(new LambdaQueryWrapper().eq(CraftMonitor::getHeatId, craftMonitor.getHeatId())); + if (craftMonitorExist != null) { + BeanUtil.copyProperties(craftMonitor, craftMonitorExist,"id"); + this.updateById(craftMonitorExist); + } else { + this.save(craftMonitor); + } + } }