|
|
@@ -1,18 +1,16 @@
|
|
|
package cn.reghao.tnb.file.app.service;
|
|
|
|
|
|
import cn.reghao.jutil.jdk.result.Result;
|
|
|
-import cn.reghao.tnb.common.auth.UserContext;
|
|
|
import cn.reghao.tnb.common.util.ConstantId;
|
|
|
import cn.reghao.tnb.file.app.db.mapper.StoreNodeMapper;
|
|
|
import cn.reghao.tnb.file.app.db.mapper.UploadChannelMapper;
|
|
|
import cn.reghao.tnb.file.app.db.mapper.UserNodeMapper;
|
|
|
-import cn.reghao.tnb.file.app.model.dto.UserNodeDto;
|
|
|
-import cn.reghao.tnb.file.app.model.dto.UserNodeUpdateDto;
|
|
|
+import cn.reghao.tnb.file.app.model.dto.NodeAddDto;
|
|
|
+import cn.reghao.tnb.file.app.model.dto.NodeUpdateDto;
|
|
|
import cn.reghao.tnb.file.app.model.po.StoreNode;
|
|
|
import cn.reghao.tnb.file.app.model.po.UploadChannel;
|
|
|
import cn.reghao.tnb.file.app.model.po.UserNode;
|
|
|
import cn.reghao.tnb.common.db.SelectOption;
|
|
|
-import cn.reghao.tnb.file.app.model.vo.UserNodeInfo;
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
@@ -39,8 +37,8 @@ public class UserNodeService {
|
|
|
this.storeConfigService = storeConfigService;
|
|
|
}
|
|
|
|
|
|
- public Result add(UserNodeDto userNodeDto) {
|
|
|
- int storeNodeId = userNodeDto.getStoreNodeId();
|
|
|
+ public Result add(NodeAddDto nodeAddDto) {
|
|
|
+ int storeNodeId = nodeAddDto.getStoreNodeId();
|
|
|
int ossUser = storeConfigService.getLocalOssUser();
|
|
|
UserNode userNode = userNodeMapper.findByCreateByAndStoreNodeId(ossUser, storeNodeId);
|
|
|
if (userNode != null) {
|
|
|
@@ -52,7 +50,7 @@ public class UserNodeService {
|
|
|
return Result.success("StoreNode not exist");
|
|
|
}
|
|
|
|
|
|
- userNode = new UserNode(userNodeDto, ossUser);
|
|
|
+ userNode = new UserNode(nodeAddDto, ossUser);
|
|
|
userNodeMapper.save(userNode);
|
|
|
return Result.success("node added");
|
|
|
}
|
|
|
@@ -78,8 +76,8 @@ public class UserNodeService {
|
|
|
userNodeMapper.save(userNode);
|
|
|
}
|
|
|
|
|
|
- public void updateUserNode(UserNodeUpdateDto userNodeUpdateDto) {
|
|
|
- userNodeMapper.updateUserNode(userNodeUpdateDto);
|
|
|
+ public void updateUserNode(NodeUpdateDto nodeUpdateDto) {
|
|
|
+ userNodeMapper.updateUserNode(nodeUpdateDto);
|
|
|
}
|
|
|
|
|
|
public Result delete(int id) {
|
|
|
@@ -99,33 +97,22 @@ public class UserNodeService {
|
|
|
return Result.fail("UserNode not exists");
|
|
|
}
|
|
|
|
|
|
- public List<UserNodeInfo> getUserNodes(int loginUser) {
|
|
|
+ public List<UserNode> getUserNodes(int loginUser) {
|
|
|
List<UserNode> list = userNodeMapper.findByCreateBy(loginUser);
|
|
|
- return list.stream()
|
|
|
- .map(this::getUserNodeInfo)
|
|
|
- .collect(Collectors.toList());
|
|
|
- }
|
|
|
-
|
|
|
- private UserNodeInfo getUserNodeInfo(UserNode userNode) {
|
|
|
- int userNodeId = userNode.getId();
|
|
|
- int storeNodeId = userNode.getStoreNodeId();
|
|
|
- StoreNode storeNode = storeNodeMapper.findById(storeNodeId);
|
|
|
- String nodeAddr = storeNode.getNodeAddr();
|
|
|
- String domain = userNode.getDomain();
|
|
|
- String referer = userNode.getReferer();
|
|
|
- String secretKey = userNode.getSecretKey();
|
|
|
- return new UserNodeInfo(storeNodeId, nodeAddr, userNodeId, domain, referer, secretKey);
|
|
|
+ list.forEach(userNode -> {
|
|
|
+ int storeNodeId = userNode.getStoreNodeId();
|
|
|
+ StoreNode storeNode = storeNodeMapper.findById(storeNodeId);
|
|
|
+ String nodeAddr = storeNode.getNodeAddr();
|
|
|
+ userNode.setStoreNodeAddr(nodeAddr);
|
|
|
+ });
|
|
|
+
|
|
|
+ return list;
|
|
|
}
|
|
|
|
|
|
public UserNode getUserNode(int userNodeId) {
|
|
|
return userNodeMapper.findById(userNodeId);
|
|
|
}
|
|
|
|
|
|
- public UserNodeInfo getUserNodeInfo(int userNodeId) {
|
|
|
- UserNode userNode = userNodeMapper.findById(userNodeId);
|
|
|
- return getUserNodeInfo(userNode);
|
|
|
- }
|
|
|
-
|
|
|
public UserNode getUserNodeByDomain(String domain) {
|
|
|
return userNodeMapper.findByDomain(domain);
|
|
|
}
|