From 35a1c88cc80ca675f6173648968ab5394b49e31e Mon Sep 17 00:00:00 2001 From: whycxzp <glperry@163.com> Date: 星期三, 28 五月 2025 17:14:50 +0800 Subject: [PATCH] 消防机器人websocket逻辑更新 --- src/main/java/com/whyc/service/RobotStatusService.java | 15 +++++++++++++++ 1 files changed, 15 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/whyc/service/RobotStatusService.java b/src/main/java/com/whyc/service/RobotStatusService.java index fdaadcf..03f43fa 100644 --- a/src/main/java/com/whyc/service/RobotStatusService.java +++ b/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); + } + } } -- Gitblit v1.9.1