20220606

xiaoyong931011

2022-07-07 f17a3980d0dfec11c1db69503e930fa20b6684f8

20220606

 src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java

@@ -401,9 +401,23 @@

    @Override
    public Result viewHomePage() {
        UserEntity loginUser = LoginUserUtil.getLoginUser();
        Long companyId = LoginUserUtil.getCompanyId();

        AdminHomePageInfoVo adminHomePageInfoVo = new AdminHomePageInfoVo();

        QueryWrapper<UserRoleEntity> userRoleQueryWrapper = new QueryWrapper<>();
        userRoleQueryWrapper.eq("user_id",loginUser.getId());
        List<UserRoleEntity> userRoleEntities = userRoleMapper.selectList(userRoleQueryWrapper);
        if(CollUtil.isNotEmpty(userRoleEntities)){
            List<String> roleNames = new ArrayList<>();
            for(UserRoleEntity userRoleEntity : userRoleEntities){
                RoleEntity roleEntity = roleMapper.selectById(userRoleEntity.getRoleId());
                roleNames.add(roleEntity.getRoleName());
            }
            adminHomePageInfoVo.setRoleNames(roleNames);
        }

        QueryWrapper<ColumnEntity> columnQueryWrapper = new QueryWrapper<>();
        columnQueryWrapper.eq("company_id",companyId);
        adminHomePageInfoVo.setColumnNum(columnMapper.selectCount(columnQueryWrapper));

 src/main/java/com/xcong/farmer/cms/modules/system/vo/AdminHomePageInfoVo.java

@@ -4,6 +4,8 @@
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.List;

@Data
@ApiModel(value = "AdminHomePageInfoVo", description = "首页信息返回")
public class AdminHomePageInfoVo {
@@ -17,4 +19,7 @@
    @ApiModelProperty(value = "留言数量")
    private Integer messageNum;

    @ApiModelProperty(value = "角色")
    private List<String> roleNames;

}