diff --git a/src/main/java/com/gxwebsoft/common/system/controller/MainController.java b/src/main/java/com/gxwebsoft/common/system/controller/MainController.java index 09a3daa..6ff719d 100644 --- a/src/main/java/com/gxwebsoft/common/system/controller/MainController.java +++ b/src/main/java/com/gxwebsoft/common/system/controller/MainController.java @@ -88,6 +88,8 @@ public class MainController extends BaseController { private UserRoleService userRoleService; @Resource private CompanyService companyService; + @Resource + private UserRefereeService userRefereeService; @ApiOperation("用户登录") @@ -225,7 +227,10 @@ public class MainController extends BaseController { public ApiResult userInfo() { final Integer loginUserId = getLoginUserId(); if (loginUserId != null) { - return success(userService.getByIdRel(getLoginUserId())); + User user = userService.getByIdRel(getLoginUserId()); + UserReferee userReferee = userRefereeService.getByUserId(loginUserId); + user.setHasParent(userReferee != null); + return success(user); } return fail("loginUserId不存在", null); } diff --git a/src/main/java/com/gxwebsoft/common/system/entity/User.java b/src/main/java/com/gxwebsoft/common/system/entity/User.java index a500a04..07f6144 100644 --- a/src/main/java/com/gxwebsoft/common/system/entity/User.java +++ b/src/main/java/com/gxwebsoft/common/system/entity/User.java @@ -319,6 +319,10 @@ public class User implements UserDetails { @TableField(exist = false) private String authCode; + @ApiModelProperty("是否有上级") + @TableField(exist = false) + private Boolean hasParent; + // @ApiModelProperty("企业信息") // @TableField(exist = false) // private Company companyInfo;