Преглед изворни кода

update UserNode in oss-console

reghao пре 1 година
родитељ
комит
a401197a4d

+ 2 - 2
oss-console/src/main/java/cn/reghao/oss/console/app/db/repository/UserNodeRepository.java

@@ -11,8 +11,8 @@ import java.util.List;
  */
 public interface UserNodeRepository extends JpaRepository<UserNode, Integer> {
     int countByNodeId(int nodeId);
-    UserNode findByUserIdAndNodeId(int nodeId, int userId);
-    List<UserNode> findByUserId(int userId);
+    UserNode findByCreateByAndNodeId(int createBy, int nodeId);
+    List<UserNode> findByCreateBy(int createBy);
     UserNode findByDomain(String domain);
     UserNode findById(int id);
 }

+ 3 - 3
oss-console/src/main/java/cn/reghao/oss/console/app/model/po/UserNode.java

@@ -23,10 +23,10 @@ public class UserNode extends BaseEntity {
     private String domain;
     private String secretKey;
     private String referer;
-    private Integer userId;
+    private Integer createBy;
 
-    public UserNode(UserNodeDto userNodeDto, int userId) {
+    public UserNode(UserNodeDto userNodeDto, int createBy) {
         this.nodeId = userNodeDto.getNodeId();
-        this.userId = userId;
+        this.createBy = createBy;
     }
 }

+ 5 - 5
oss-console/src/main/java/cn/reghao/oss/console/app/service/UserNodeService.java

@@ -36,7 +36,7 @@ public class UserNodeService {
     public Result add(UserNodeDto userNodeDto) {
         int nodeId = userNodeDto.getNodeId();
         Integer userId = UserContext.getUserId();
-        UserNode userNode = userNodeRepository.findByUserIdAndNodeId(userId, nodeId);
+        UserNode userNode = userNodeRepository.findByCreateByAndNodeId(userId, nodeId);
         if (userNode != null) {
             return Result.fail("UserNode exist");
         }
@@ -69,7 +69,7 @@ public class UserNodeService {
     }
 
     public List<UserNode> getByUserId(int userId) {
-        return userNodeRepository.findByUserId(userId);
+        return userNodeRepository.findByCreateBy(userId);
     }
 
     public StoreNode getByDomain(String domain) {
@@ -80,7 +80,7 @@ public class UserNodeService {
 
     public List<StoreNode> getUserStoreNodes() {
         int loginUser = UserContext.getUserId();
-        List<UserNode> list = userNodeRepository.findByUserId(loginUser);
+        List<UserNode> list = userNodeRepository.findByCreateBy(loginUser);
         if (list.isEmpty()) {
             return Collections.emptyList();
         }
@@ -91,7 +91,7 @@ public class UserNodeService {
 
     public List<UserNodeVO> getUserStoreNodes1() {
         int loginUser = UserContext.getUserId();
-        List<UserNode> list = userNodeRepository.findByUserId(loginUser);
+        List<UserNode> list = userNodeRepository.findByCreateBy(loginUser);
         if (list.isEmpty()) {
             return Collections.emptyList();
         }
@@ -108,6 +108,6 @@ public class UserNodeService {
     }
 
     public UserNode getUserNode(int nodeId, int loginUser) {
-        return userNodeRepository.findByUserIdAndNodeId(loginUser, nodeId);
+        return userNodeRepository.findByCreateByAndNodeId(loginUser, nodeId);
     }
 }