Parcourir la source

update user-service UserVipService.java

reghao il y a 3 mois
Parent
commit
c660db5558

+ 3 - 1
user/user-service/src/main/java/cn/reghao/tnb/user/app/controller/UserVipController.java

@@ -1,6 +1,7 @@
 package cn.reghao.tnb.user.app.controller;
 
 import cn.reghao.jutil.jdk.web.result.Result;
+import cn.reghao.tnb.common.auth.UserContext;
 import cn.reghao.tnb.common.web.WebResult;
 import cn.reghao.tnb.common.auth.AuthUser;
 import cn.reghao.tnb.user.app.model.dto.BuyVip;
@@ -32,7 +33,8 @@ public class UserVipController {
     @AuthUser
     @GetMapping("/my")
     public String getMyVip() {
-        MyVip myVip = userVipService.getMyVip();
+        long loginUser = UserContext.getUserId();
+        MyVip myVip = userVipService.getMyVip(loginUser);
         return WebResult.success(myVip);
     }
 

+ 2 - 2
user/user-service/src/main/java/cn/reghao/tnb/user/app/rpc/UserServiceImpl.java

@@ -60,14 +60,14 @@ public class UserServiceImpl implements UserService {
 
     @Override
     public int getRecommendMode(long userId) {
-        MyVip myVip = userVipService.getMyVip();
+        MyVip myVip = userVipService.getMyVip(userId);
         return myVip != null ? myVip.getRcmdMode() : 0;
     }
 
     @Override
     public boolean setRecommendMode(long userId, int mode) {
         boolean changed = true;
-        MyVip myVip = userVipService.getMyVip();
+        MyVip myVip = userVipService.getMyVip(userId);
         if (!myVip.isVip()) {
             return false;
         }

+ 1 - 2
user/user-service/src/main/java/cn/reghao/tnb/user/app/service/UserVipService.java

@@ -31,8 +31,7 @@ public class UserVipService {
         this.walletService = walletService;
     }
 
-    public MyVip getMyVip() {
-        long userId = UserContext.getUserId();
+    public MyVip getMyVip(long userId) {
         UserVip userVip = userVipMapper.findByUserId(userId);
         if (userVip == null) {
             return new MyVip();