package com.whyc.service; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.whyc.mapper.EnvironmentMapper; import com.whyc.mapper.RobotAlarmMapper; import com.whyc.pojo.db_fire_robot.RobotAlarm; import org.springframework.stereotype.Service; import javax.annotation.Resource; @Service public class RobotAlarmService { @Resource private RobotAlarmMapper mapper; public void add(RobotAlarm alarm) { mapper.insert(alarm); } public void addOrUpdate(RobotAlarm alarm) { QueryWrapper query = Wrappers.query(); query.last(" limit 1"); RobotAlarm alarmInDB = mapper.selectOne(query); if (alarmInDB ==null) { mapper.insert(alarm); }else{ alarm.setId(alarmInDB.getId()); mapper.updateById(alarm); } } }