diff --git a/src/main/java/com/peanut/modules/common/controller/UserController.java b/src/main/java/com/peanut/modules/common/controller/UserController.java index 8562ca51..7bb80e73 100644 --- a/src/main/java/com/peanut/modules/common/controller/UserController.java +++ b/src/main/java/com/peanut/modules/common/controller/UserController.java @@ -5,8 +5,11 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.peanut.common.utils.MD5Utils; import com.peanut.common.utils.MailUtil; import com.peanut.common.utils.R; +import com.peanut.common.utils.ShiroUtils; import com.peanut.modules.common.entity.MyUserEntity; +import com.peanut.modules.common.entity.UserVip; import com.peanut.modules.common.service.MyUserService; +import com.peanut.modules.common.service.UserVipService; import com.peanut.modules.sys.service.SysUserTokenService; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; @@ -15,6 +18,8 @@ import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RestController; + +import java.util.List; import java.util.Map; import java.util.Random; import java.util.concurrent.TimeUnit; @@ -33,6 +38,8 @@ public class UserController { private StringRedisTemplate redisTemplate; @Autowired private SysUserTokenService sysUserTokenService; + @Autowired + private UserVipService userVipService; /** * 常规注册 发短信验证码 @@ -94,6 +101,22 @@ public class UserController { return MailUtil.sendMail("疯子读书邮箱验证码",code,email); } + @RequestMapping("/getUserInfo") + public R getUserInfo(){ + int uid = ShiroUtils.getUId(); + MyUserEntity userEntity = userService.getById(uid); + if(!userEntity.getVip().equals("0")){ + List userVips = userVipService.getBaseMapper().selectList(new LambdaQueryWrapper() + .eq(UserVip::getUserId, uid).eq(UserVip::getState,0)); + if (userVips.size()==1){ + userEntity.setUserVip(userVips.get(0)); + } + } + return R.ok().put("result",userEntity); + + + } + /** * 绑定电话号 * @return diff --git a/src/main/java/com/peanut/modules/common/entity/MyUserEntity.java b/src/main/java/com/peanut/modules/common/entity/MyUserEntity.java index 12021930..4cbc89df 100644 --- a/src/main/java/com/peanut/modules/common/entity/MyUserEntity.java +++ b/src/main/java/com/peanut/modules/common/entity/MyUserEntity.java @@ -143,4 +143,7 @@ public class MyUserEntity implements Serializable { private String remark; + @TableField(exist = false) + private UserVip userVip; + }