From 8c037dd262b9f30397dba96eb8899ad68af5ec56 Mon Sep 17 00:00:00 2001
From: sxu <602087911@qq.com>
Date: Mon, 16 Dec 2024 11:01:41 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B6=E8=B4=A7=E5=9C=B0=E5=9D=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../material/basic/controller/BmCompanyAddressController.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/basic/controller/BmCompanyAddressController.java b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/basic/controller/BmCompanyAddressController.java
index 14a3f41..d764c5a 100644
--- a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/basic/controller/BmCompanyAddressController.java
+++ b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/basic/controller/BmCompanyAddressController.java
@@ -3,6 +3,7 @@ package com.bonus.material.basic.controller;
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
 import com.bonus.common.log.enums.OperaType;
+import com.bonus.common.security.utils.SecurityUtils;
 import com.bonus.material.common.annotation.PreventRepeatSubmit;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -45,6 +46,7 @@ public class BmCompanyAddressController extends BaseController {
     @GetMapping("/list")
     public TableDataInfo list(BmCompanyAddress bmCompanyAddress) {
         startPage();
+        bmCompanyAddress.setCompanyId(SecurityUtils.getLoginUser().getSysUser().getCompanyId());
         List<BmCompanyAddress> list = bmCompanyAddressService.selectBmCompanyAddressList(bmCompanyAddress);
         return getDataTable(list);
     }
@@ -83,6 +85,7 @@ public class BmCompanyAddressController extends BaseController {
     @PostMapping
     public AjaxResult add(@RequestBody BmCompanyAddress bmCompanyAddress) {
         try {
+            bmCompanyAddress.setCompanyId(SecurityUtils.getLoginUser().getSysUser().getCompanyId());
             return toAjax(bmCompanyAddressService.insertBmCompanyAddress(bmCompanyAddress));
         } catch (Exception e) {
             return error("系统错误, " + e.getMessage());
@@ -99,6 +102,7 @@ public class BmCompanyAddressController extends BaseController {
     @PostMapping("/edit")
     public AjaxResult edit(@RequestBody BmCompanyAddress bmCompanyAddress) {
         try {
+            bmCompanyAddress.setCompanyId(SecurityUtils.getLoginUser().getSysUser().getCompanyId());
             return toAjax(bmCompanyAddressService.updateBmCompanyAddress(bmCompanyAddress));
         } catch (Exception e) {
             return error("系统错误, " + e.getMessage());