From ebd2f4397a92c6a5096de1b86d59154363344720 Mon Sep 17 00:00:00 2001
From: vincentlu <t1341870251@gmail.com>
Date: 星期二, 13 五月 2025 08:48:15 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/AgvModelServiceImpl.java |   54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 54 insertions(+), 0 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/AgvModelServiceImpl.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/AgvModelServiceImpl.java
index cae99f7..bce4b2d 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/AgvModelServiceImpl.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/manager/service/impl/AgvModelServiceImpl.java
@@ -1,12 +1,66 @@
 package com.zy.acs.manager.manager.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.zy.acs.manager.manager.entity.Agv;
 import com.zy.acs.manager.manager.entity.AgvModel;
 import com.zy.acs.manager.manager.mapper.AgvModelMapper;
 import com.zy.acs.manager.manager.service.AgvModelService;
+import com.zy.acs.manager.manager.service.AgvService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import javax.annotation.PostConstruct;
+import java.util.HashMap;
+import java.util.Map;
 
 @Service("agvModelService")
 public class AgvModelServiceImpl extends ServiceImpl<AgvModelMapper, AgvModel> implements AgvModelService {
 
+    public static final Map<Long, AgvModel> AGV_ID_MODEL_CACHE = new HashMap<>();
+    public static final Map<String, AgvModel> AGV_NO_MODEL_CACHE = new HashMap<>();
+
+    @Autowired
+    private AgvService agvService;
+
+    @PostConstruct
+    public void init() {
+        for (Agv agv : agvService.list()) {
+            AgvModel agvModel = this.getById(agv.getAgvModel());
+            AGV_ID_MODEL_CACHE.put(agv.getId(), agvModel);
+            AGV_NO_MODEL_CACHE.put(agv.getUuid(), agvModel);
+        }
+    }
+
+    @Override
+    public AgvModel selectByType(String type) {
+        return this.getOne(new LambdaQueryWrapper<AgvModel>().eq(AgvModel::getType, type));
+    }
+
+    @Override
+    public AgvModel getByAgvId(Long agvId) {
+        AgvModel agvModel = AGV_ID_MODEL_CACHE.get(agvId);
+        if (agvModel == null) {
+            Agv agv = agvService.getById(agvId);
+            agvModel = this.getById(agv.getAgvModel());
+            if(agvModel != null) {
+                AGV_ID_MODEL_CACHE.put(agv.getId(), agvModel);
+            }
+        }
+        return agvModel;
+    }
+
+    @Override
+    public AgvModel getByAgvNo(String agvNo) {
+        AgvModel agvModel = AGV_NO_MODEL_CACHE.get(agvNo);
+        if (agvModel == null) {
+            Agv agv = agvService.selectByUuid(agvNo);
+            agvModel = this.getById(agv.getAgvModel());
+            if(agvModel != null) {
+                AGV_NO_MODEL_CACHE.put(agvNo, agvModel);
+            }
+        }
+        return agvModel;
+    }
+
 }

--
Gitblit v1.9.1