Kaynağa Gözat

update user-service UserServiceImpl

reghao 7 ay önce
ebeveyn
işleme
21a31334a8

+ 1 - 1
user/user-api/src/main/java/cn/reghao/tnb/user/api/iface/UserService.java

@@ -15,5 +15,5 @@ public interface UserService {
     boolean isVip(long userId);
     void sendUserMessage(UserMessageDto userMessageDto);
     int getRecommendMode(long userId);
-    void setRecommendMode(long userId, boolean mode);
+    void setRecommendMode(long userId, int mode);
 }

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

@@ -79,18 +79,13 @@ public class UserServiceImpl implements UserService {
     }
 
     @Override
-    public void setRecommendMode(long userId, boolean mode) {
-        int modeInt = 0;
-        if (mode) {
-            modeInt = 1;
-        }
-
+    public void setRecommendMode(long userId, int mode) {
         Integer mode1 = userRecommendMapper.findByUserId(userId);
         if (mode1 == null) {
-            UserRecommend userRecommend = new UserRecommend(userId, modeInt);
+            UserRecommend userRecommend = new UserRecommend(userId, mode);
             userRecommendMapper.save(userRecommend);
-        } else if (mode1 != modeInt) {
-            userRecommendMapper.updateMode(userId, modeInt);
+        } else if (mode1 != mode) {
+            userRecommendMapper.updateMode(userId, mode);
         }
     }
 }