From da2c594ade5d69621dd11a13bb758477e2dc079e Mon Sep 17 00:00:00 2001
From: “djh” <“3298565835@qq.com”>
Date: 星期三, 07 五月 2025 15:57:51 +0800
Subject: [PATCH] 危化品新版修改

---
 hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzHazmatServiceImpl.java |   28 ++++++++++++++++++++++------
 1 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzHazmatServiceImpl.java b/hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzHazmatServiceImpl.java
index ebb6530..8bec69e 100644
--- a/hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzHazmatServiceImpl.java
+++ b/hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzHazmatServiceImpl.java
@@ -1,7 +1,10 @@
 package com.gkhy.hazmat.system.service.impl;
 
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.github.pagehelper.PageHelper;
 import com.gkhy.hazmat.common.api.CommonPage;
 import com.gkhy.hazmat.common.config.IdTableNameHandler;
 import com.gkhy.hazmat.common.domain.entity.SysUser;
@@ -21,6 +24,7 @@
 import org.springframework.transaction.annotation.Transactional;
 
 import java.math.BigDecimal;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
@@ -54,9 +58,13 @@
         }
         SysUser currentUser = SecurityUtils.getLoginUser().getUser();
         checkUserAllowed(null,currentUser);
-        //设置分表id
-        IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
-        hzHazmat.setCompanyId(currentUser.getCompanyId());
+        if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){
+            IdTableNameHandler.setCurrentId(hzHazmat.getCompanyId());
+        }else {
+            //设置分表id
+            IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
+            hzHazmat.setCompanyId(currentUser.getCompanyId());
+        }
         PageUtils.startPage();
         List<HzHazmat> hazmatList = baseMapper.selectHazmatList(hzHazmat);
         IdTableNameHandler.removeCurrentId();
@@ -67,9 +75,14 @@
     public CommonPage selectHazmatGroupWarehouse(HzHazmat hzHazmat) {
         SysUser currentUser = SecurityUtils.getLoginUser().getUser();
         checkUserAllowed(null,currentUser);
-        hzHazmat.setCompanyId(currentUser.getCompanyId());
-        //设置分表id
-        IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
+        if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){
+            IdTableNameHandler.setCurrentId(hzHazmat.getCompanyId());
+        }else {
+            hzHazmat.setCompanyId(currentUser.getCompanyId());
+            //设置分表id
+            //todo
+            IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
+        }
         PageUtils.startPage();
         List<HzHazmatWarehouseVO> hazmatList = baseMapper.selectHazmatGroupWareHouse(hzHazmat);
         if(!hazmatList.isEmpty()) {
@@ -195,6 +208,7 @@
         HzWarehouseRecord warehouseRecord = new HzWarehouseRecord()
                 .setWarehouseId(hazmat.getWarehouseId())
                 .setBasicId(hazmat.getBasicId())
+                .setCupboardId(hazmat.getCupboardId())
                 .setCreateId(currentUser.getId())
                 .setNum(-1)
                 .setState(OperateStatusEnum.USING.getCode())
@@ -249,6 +263,7 @@
         HzWarehouseRecord warehouseRecord = new HzWarehouseRecord()
                 .setWarehouseId(dbhazmat.getWarehouseId())
                 .setBasicId(dbhazmat.getBasicId())
+                .setCupboardId(hazmat.getCupboardId())
                 .setCreateId(currentUser.getId())
                 .setNum(1)
                 .setState(OperateStatusEnum.RETURN.getCode())
@@ -318,6 +333,7 @@
         HzWarehouseRecord warehouseRecord = new HzWarehouseRecord()
                 .setWarehouseId(hazmat.getWarehouseId())
                 .setBasicId(hazmat.getBasicId())
+                .setCupboardId(hazmat.getCupboardId())
                 .setCreateId(currentUser.getId())
                 .setNum(-1)
                 .setState(OperateStatusEnum.DISCARD.getCode())

--
Gitblit v1.9.2