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 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());