From 34150a16ef56b933c59838da129d5b35b9dc09d8 Mon Sep 17 00:00:00 2001
From: skyouc
Date: 星期三, 26 三月 2025 13:29:21 +0800
Subject: [PATCH] 添加仓库，库区树状接口
---
rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java | 32 ++++++++++++++++++++++++++++++++
1 files changed, 32 insertions(+), 0 deletions(-)
diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java b/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java
index 784116a..fc56f98 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/manager/service/impl/WarehouseServiceImpl.java
@@ -1,12 +1,44 @@
package com.vincent.rsf.server.manager.service.impl;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.vincent.rsf.framework.exception.CoolException;
+import com.vincent.rsf.server.manager.entity.WarehouseAreas;
import com.vincent.rsf.server.manager.mapper.WarehouseMapper;
import com.vincent.rsf.server.manager.entity.Warehouse;
+import com.vincent.rsf.server.manager.service.WarehouseAreasService;
import com.vincent.rsf.server.manager.service.WarehouseService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.List;
@Service("warehouseService")
public class WarehouseServiceImpl extends ServiceImpl<WarehouseMapper, Warehouse> implements WarehouseService {
+ @Autowired
+ private WarehouseAreasService warehouseAreasService;
+
+ @Override
+ public List<Warehouse> getAllWarehouseAreas() {
+ List<Warehouse> warehouses = this.list(new LambdaQueryWrapper<>());
+ if (warehouses.isEmpty()) {
+ throw new CoolException("浠撳簱涓虹┖锛岃娣诲姞浠撳簱鍚庡啀鎿嶄綔锛侊紒");
+ }
+ List<WarehouseAreas> warehouseAreas = warehouseAreasService.list(new LambdaQueryWrapper<>());
+ if (warehouseAreas.isEmpty()) {
+ return warehouses;
+ }
+ warehouses.forEach(warehouse -> {
+ List<WarehouseAreas> areas = new ArrayList<>();
+ warehouseAreas.forEach(warehouseAreas1 -> {
+ if (warehouse.getId().equals(warehouseAreas1.getWarehouseId())) {
+ areas.add(warehouseAreas1);
+ }
+ });
+ warehouse.setChildren(areas);
+ });
+ return warehouses;
+ }
}
--
Gitblit v1.9.1