diff --git a/ics-admin/src/main/java/com/ics/admin/controller/CustomerStaffController.java b/ics-admin/src/main/java/com/ics/admin/controller/CustomerStaffController.java index b3c2e4d..b0c4e3c 100644 --- a/ics-admin/src/main/java/com/ics/admin/controller/CustomerStaffController.java +++ b/ics-admin/src/main/java/com/ics/admin/controller/CustomerStaffController.java @@ -6,11 +6,15 @@ import com.ics.admin.service.IIcsCustomerStaffService; import com.ics.common.constant.Constants; import com.ics.common.core.controller.BaseController; import com.ics.common.core.domain.R; +import com.ics.system.domain.User; +import com.ics.system.service.IUserService; +import org.checkerframework.checker.units.qual.A; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.wf.jwtp.annotation.RequiresPermissions; import java.util.Date; +import java.util.List; /** * 企业员工 提供者 @@ -26,6 +30,8 @@ public class CustomerStaffController extends BaseController { @Autowired private IIcsCustomerStaffService icsCustomerStaffService; + @Autowired + private IUserService userService; /** * 查询企业员工 @@ -42,8 +48,8 @@ public class CustomerStaffController extends BaseController { @GetMapping("list") public R list(IcsCustomerStaff icsCustomerStaff) { startPage(); - Integer staffId = getLoginStaffId(); - icsCustomerStaff.setIcsCustomerId(staffId.longValue()); + Integer customerId = getLoginCustomerId(); + icsCustomerStaff.setIcsCustomerId(customerId.longValue()); icsCustomerStaff.setDataType(Constants.CUSTOMER_STAFF); return result(icsCustomerStaffService.selectIcsCustomerStaffList(icsCustomerStaff)); } @@ -55,12 +61,18 @@ public class CustomerStaffController extends BaseController { @RequiresPermissions("admin:staff:add") @PostMapping("save") public R addSave(@RequestBody IcsCustomerStaff icsCustomerStaff) { - Integer staffId = getLoginStaffId(); - icsCustomerStaff.setIcsCustomerId(staffId.longValue()); + Integer customerId = getLoginCustomerId(); + icsCustomerStaff.setIcsCustomerId(customerId.longValue()); icsCustomerStaff.setCreateTime(new Date()); icsCustomerStaff.setCreateBy(getLoginName()); icsCustomerStaff.setDataType(Constants.CUSTOMER_STAFF); - return toAjax(icsCustomerStaffService.insertIcsCustomerStaff(icsCustomerStaff)); + User user = userService.selectUserByMobile(icsCustomerStaff.getMobile()); + int i = icsCustomerStaffService.insertIcsCustomerStaff(icsCustomerStaff); + if(i>0){ + user.setStaffId(icsCustomerStaff.getId()); + userService.updateUser(user); + } + return toAjax(i); } /** diff --git a/ics-common/ics-common-core/src/main/java/com/ics/common/core/controller/BaseController.java b/ics-common/ics-common-core/src/main/java/com/ics/common/core/controller/BaseController.java index 1a9639d..b26b652 100644 --- a/ics-common/ics-common-core/src/main/java/com/ics/common/core/controller/BaseController.java +++ b/ics-common/ics-common-core/src/main/java/com/ics/common/core/controller/BaseController.java @@ -113,6 +113,10 @@ public class BaseController { return null; } + /** + * 获取微信id + * @return + */ public Integer getLoginStaffId() { Token token = SubjectUtil.getToken(getRequest()); String value = valueOperations.get(ACCESS_USERID + ":" + token.getUserId()); @@ -123,6 +127,20 @@ public class BaseController { return null; } + /** + * 获取企业id + * @return + */ + public Integer getLoginCustomerId() { + Token token = SubjectUtil.getToken(getRequest()); + String value = valueOperations.get(ACCESS_USERID + ":" + token.getUserId()); + JSONObject jo = StringUtils.isEmpty(value) ? null : JSON.parseObject(value, JSONObject.class); + if (jo != null && jo.containsKey("customerId")) { + return (Integer)jo.get("customerId"); + } + return null; + } + /** * 响应返回结果 diff --git a/ics-system/src/main/java/com/ics/system/domain/User.java b/ics-system/src/main/java/com/ics/system/domain/User.java index 26ba52b..0bf0d29 100644 --- a/ics-system/src/main/java/com/ics/system/domain/User.java +++ b/ics-system/src/main/java/com/ics/system/domain/User.java @@ -81,7 +81,10 @@ public class User extends BaseEntity { @NotBlank(message = "登录名称不能为空") private String username; - private String customerId; + /** + * 企业id + */ + private Long customerId; /** * 用户名称 @@ -167,7 +170,9 @@ public class User extends BaseEntity { */ private Set buttons; - + /** + * 用户id + */ private Long staffId; /** diff --git a/ics-system/src/main/java/com/ics/system/domain/vo/CurrentUserVO.java b/ics-system/src/main/java/com/ics/system/domain/vo/CurrentUserVO.java index 8757e60..ba0ca02 100644 --- a/ics-system/src/main/java/com/ics/system/domain/vo/CurrentUserVO.java +++ b/ics-system/src/main/java/com/ics/system/domain/vo/CurrentUserVO.java @@ -77,8 +77,13 @@ public class CurrentUserVO { private Boolean isAdmin; /** - * 客户id + * 用户id */ private Long staffId; + /** + * 企业id + */ + private Long customerId; + } diff --git a/ics-system/src/main/java/com/ics/system/log/publish/PublishFactory.java b/ics-system/src/main/java/com/ics/system/log/publish/PublishFactory.java index 0d327d5..2c4f1ee 100644 --- a/ics-system/src/main/java/com/ics/system/log/publish/PublishFactory.java +++ b/ics-system/src/main/java/com/ics/system/log/publish/PublishFactory.java @@ -90,6 +90,8 @@ public class PublishFactory { userOnline.setDeptName(user.getDept().getDeptName()); } userOnline.setStaffId(user.getStaffId()); + userOnline.setCustomerId(user.getCustomerId()); + // 发布事件 SpringContextHolder.publishEvent(new UserOnlineEvent(userOnline)); }