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