ソースを参照

fix: 在获得用户时包含商店名称信息

Gudge-ZL 2 年 前
コミット
2e560b9471

+ 13 - 1
src/main/java/com/seecoder/BlueWhale/serviceImpl/UserServiceImpl.java

@@ -1,11 +1,13 @@
 package com.seecoder.BlueWhale.serviceImpl;
 
 import com.seecoder.BlueWhale.exception.BlueWhaleException;
+import com.seecoder.BlueWhale.repository.StoreRepository;
 import com.seecoder.BlueWhale.repository.UserRepository;
 import com.seecoder.BlueWhale.po.User;
 import com.seecoder.BlueWhale.service.UserService;
 import com.seecoder.BlueWhale.util.SecurityUtil;
 import com.seecoder.BlueWhale.util.TokenUtil;
+import com.seecoder.BlueWhale.vo.OrderVO;
 import com.seecoder.BlueWhale.vo.UserVO;
 
 import java.util.Date;
@@ -31,6 +33,9 @@ public class UserServiceImpl implements UserService {
     @Autowired
     SecurityUtil securityUtil;
 
+    @Autowired
+    StoreRepository storeRepository;
+
     @Override
     public Boolean register(UserVO userVO) {
         User user = userRepository.findByPhone(userVO.getPhone());
@@ -54,7 +59,7 @@ public class UserServiceImpl implements UserService {
 
     @Override
     public UserVO getInformation() {
-        return securityUtil.getCurrentUser().toVO();
+        return wrapWithStoreName(securityUtil.getCurrentUser().toVO());
     }
 
     @Override
@@ -72,4 +77,11 @@ public class UserServiceImpl implements UserService {
         userRepository.save(user);
         return true;
     }
+
+    private UserVO wrapWithStoreName(UserVO userVO){
+        Integer storeId = userVO.getStoreId();
+        String storeName = storeRepository.findById(storeId).get().getName();
+        userVO.setStoreName(storeName);
+        return userVO;
+    }
 }

+ 2 - 0
src/main/java/com/seecoder/BlueWhale/vo/UserVO.java

@@ -29,6 +29,8 @@ public class UserVO {
 
     private Date createTime;
 
+    private String storeName;
+
     public User toPO(){
         User user=new User();
         user.setId(this.id);