机器和商品部门权限

yudao-module-mall/yudao-module-product-biz/src/main/java/cn/iocoder/yudao/module/product/dal/dataobject/device/DeviceDO.java

@@ -58,4 +58,9 @@ public class DeviceDO extends BaseDO {
      */
     private String cabinetChannel;
 
+    /**
+     * 部门 ID
+     */
+    private Long deptId;
+
 }

yudao-module-mall/yudao-module-product-biz/src/main/java/cn/iocoder/yudao/module/product/dal/dataobject/spu/ProductSpuDO.java

@@ -181,4 +181,8 @@ public class ProductSpuDO extends BaseDO {
      * 浏览量
      */
     private Integer browseCount;
+    /**
+     * 部门 ID
+     */
+    private Long deptId;
 }

yudao-module-mall/yudao-module-product-biz/src/main/java/cn/iocoder/yudao/module/product/framework/web/config/ProductWebConfiguration.java

@@ -30,7 +30,8 @@ public class ProductWebConfiguration {
     public DeptDataPermissionRuleCustomizer productDeptDataPermissionRuleCustomizer() {
         return rule -> {
             // dept
-//            rule.addDeptColumn(AdminUserDO.class);
+            rule.addDeptColumn(ProductSpuDO.class);
+            rule.addDeptColumn(DeviceDO.class);
 //            rule.addDeptColumn(DeptDO.class, "id");
             // user
             rule.addUserColumn(ProductSpuDO.class, "creator");

yudao-module-mall/yudao-module-product-biz/src/main/java/cn/iocoder/yudao/module/product/service/device/DeviceServiceImpl.java

@@ -37,6 +37,7 @@ import cn.iocoder.yudao.module.product.dal.mysql.device.DeviceMapper;
 import org.springframework.web.client.RestTemplate;
 
 import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
+import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserDeptId;
 import static cn.iocoder.yudao.module.product.enums.ErrorCodeConstants.*;
 
 /**
@@ -76,6 +77,8 @@ public class DeviceServiceImpl implements DeviceService {
         }
         // 插入
         DeviceDO device = BeanUtils.toBean(createReqVO, DeviceDO.class);
+        // 传入部门
+        device.setDeptId(getLoginUserDeptId());
         deviceMapper.insert(device);
         DispenseRequestBo request = new DispenseRequestBo();
         request.setDeviceId(createReqVO.getDeviceId());

yudao-module-mall/yudao-module-product-biz/src/main/java/cn/iocoder/yudao/module/product/service/spu/ProductSpuServiceImpl.java

@@ -35,6 +35,7 @@ import java.util.*;
 
 import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
 import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.*;
+import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserDeptId;
 import static cn.iocoder.yudao.module.product.dal.dataobject.category.ProductCategoryDO.CATEGORY_LEVEL;
 import static cn.iocoder.yudao.module.product.enums.ErrorCodeConstants.*;
 
@@ -80,6 +81,8 @@ public class ProductSpuServiceImpl implements ProductSpuService {
         ProductSpuDO spu = BeanUtils.toBean(createReqVO, ProductSpuDO.class);
         // 初始化 SPU 中 SKU 相关属性
         initSpuFromSkus(spu, skuSaveReqList);
+        // 传入部门
+        spu.setDeptId(getLoginUserDeptId());
         // 插入 SPU
         productSpuMapper.insert(spu);
         // 插入 SKU