消防机器人websocket逻辑更新

whycxzp

2025-05-28 35a1c88cc80ca675f6173648968ab5394b49e31e

 src/main/java/com/whyc/service/RobotStatusService.java

@@ -1,6 +1,9 @@
package com.whyc.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.mapper.RobotStatusMapper;
import com.whyc.pojo.db_fire_robot.Environment;
import com.whyc.pojo.db_fire_robot.RobotStatus;
import org.springframework.stereotype.Service;

@@ -15,4 +18,16 @@
    public void add(RobotStatus status) {
        mapper.insert(status);
    }

    public void addOrUpdate(RobotStatus status) {
        QueryWrapper<RobotStatus> query = Wrappers.query();
        query.last(" limit 1");
        RobotStatus statusInDB = mapper.selectOne(query);
        if (statusInDB ==null) {
            mapper.insert(status);
        }else{
            status.setId(statusInDB.getId());
            mapper.updateById(status);
        }
    }
}