From 17efddc6a667670dca682bf36b51a43e99615e6d Mon Sep 17 00:00:00 2001 From: k94314517 <8417338+k94314517@user.noreply.gitee.com> Date: 星期三, 16 十月 2024 15:59:38 +0800 Subject: [PATCH] 代码初始化 --- server/system_service/src/main/java/com/doumee/core/model/LoginUserInfo.java | 21 ++++++++++++++++----- 1 files changed, 16 insertions(+), 5 deletions(-) diff --git a/server/system_service/src/main/java/com/doumee/core/model/LoginUserInfo.java b/server/system_service/src/main/java/com/doumee/core/model/LoginUserInfo.java index f3b08e5..907be16 100644 --- a/server/system_service/src/main/java/com/doumee/core/model/LoginUserInfo.java +++ b/server/system_service/src/main/java/com/doumee/core/model/LoginUserInfo.java @@ -1,8 +1,11 @@ package com.doumee.core.model; +import com.doumee.dao.business.model.Company; import com.doumee.dao.system.model.SystemPermission; import com.doumee.dao.system.model.SystemRole; import com.doumee.dao.system.model.SystemUser; +import com.github.xiaoymin.knife4j.core.util.CollectionUtils; +import io.swagger.annotations.ApiModelProperty; import lombok.Data; import org.springframework.beans.BeanUtils; @@ -37,31 +40,39 @@ private String sessionId; private Date loginDate; + @ApiModelProperty(value = "璐︽埛鏉ユ簮锛�0=鍚庡彴娣诲姞 锛� 2=鍙告満娉ㄥ唽") + private Integer source; private List<String> roles; private List<String> permissions; private List<Integer> companyIdList; + private Company company; //jwt鐧诲綍璁剧疆 - public static LoginUserInfo from(SystemUser user, List<SystemRole> roles, List<SystemPermission> permissions,String sessionId) { + public static LoginUserInfo from(SystemUser user, List<SystemRole> roles, List<SystemPermission> permissions,Company company,String sessionId) { if (user == null) { return null; } // 鎷疯礉鐢ㄦ埛淇℃伅 LoginUserInfo loginUserInfo = new LoginUserInfo(); BeanUtils.copyProperties(user, loginUserInfo); + loginUserInfo.setCompany(company); // 璁剧疆瑙掕壊淇℃伅 List<String> rs = new ArrayList<>(); - for (SystemRole role : roles) { - rs.add(role.getCode()); + if(CollectionUtils.isNotEmpty(roles)){ + for (SystemRole role : roles) { + rs.add(role.getCode()); + } } loginUserInfo.setSessionId(sessionId); loginUserInfo.setRoles(rs); // 璁剧疆鏉冮檺淇℃伅 List<String> pms = new ArrayList<>(); - for (SystemPermission permission : permissions) { - pms.add(permission.getCode()); + if(CollectionUtils.isNotEmpty(permissions)){ + for (SystemPermission permission : permissions) { + pms.add(permission.getCode()); + } } loginUserInfo.setPermissions(pms); return loginUserInfo; -- Gitblit v1.9.3