|
@@ -5,6 +5,8 @@ import cn.reghao.oss.console.account.service.UserContext;
|
|
|
import cn.reghao.oss.console.app.db.repository.UserNodeRepository;
|
|
import cn.reghao.oss.console.app.db.repository.UserNodeRepository;
|
|
|
import cn.reghao.oss.console.app.model.dto.UserNodeDto;
|
|
import cn.reghao.oss.console.app.model.dto.UserNodeDto;
|
|
|
import cn.reghao.oss.console.app.model.po.StoreNode;
|
|
import cn.reghao.oss.console.app.model.po.StoreNode;
|
|
|
|
|
+import cn.reghao.oss.console.app.model.po.UploadChannel;
|
|
|
|
|
+import cn.reghao.oss.console.app.model.po.UserKey;
|
|
|
import cn.reghao.oss.console.app.model.po.UserNode;
|
|
import cn.reghao.oss.console.app.model.po.UserNode;
|
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
|
@@ -20,10 +22,15 @@ import java.util.stream.Collectors;
|
|
|
public class UserNodeService {
|
|
public class UserNodeService {
|
|
|
private final UserNodeRepository userNodeRepository;
|
|
private final UserNodeRepository userNodeRepository;
|
|
|
private final StoreNodeService storeNodeService;
|
|
private final StoreNodeService storeNodeService;
|
|
|
|
|
+ private UserKeyService userKeyService;
|
|
|
|
|
+ private UploadChannelService uploadChannelService;
|
|
|
|
|
|
|
|
- public UserNodeService(UserNodeRepository userNodeRepository, StoreNodeService storeNodeService) {
|
|
|
|
|
|
|
+ public UserNodeService(UserNodeRepository userNodeRepository, StoreNodeService storeNodeService,
|
|
|
|
|
+ UserKeyService userKeyService, UploadChannelService uploadChannelService) {
|
|
|
this.userNodeRepository = userNodeRepository;
|
|
this.userNodeRepository = userNodeRepository;
|
|
|
this.storeNodeService = storeNodeService;
|
|
this.storeNodeService = storeNodeService;
|
|
|
|
|
+ this.userKeyService = userKeyService;
|
|
|
|
|
+ this.uploadChannelService = uploadChannelService;
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
public Result add(UserNodeDto userNodeDto) {
|
|
public Result add(UserNodeDto userNodeDto) {
|
|
@@ -42,16 +49,32 @@ public class UserNodeService {
|
|
|
return Result.fail("node exist");
|
|
return Result.fail("node exist");
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- public void delete(Integer id) {
|
|
|
|
|
|
|
+ public Result delete(Integer id) {
|
|
|
StoreNode storeNode = storeNodeService.getById(id);
|
|
StoreNode storeNode = storeNodeService.getById(id);
|
|
|
if (storeNode != null) {
|
|
if (storeNode != null) {
|
|
|
int nodeId = storeNode.getId();
|
|
int nodeId = storeNode.getId();
|
|
|
int userId = UserContext.getUser().getId();
|
|
int userId = UserContext.getUser().getId();
|
|
|
|
|
+
|
|
|
|
|
+ UserKey userKey = userKeyService.getByNodeIdAndUserId(nodeId, userId);
|
|
|
|
|
+ if (userKey != null) {
|
|
|
|
|
+ return Result.fail("UserKey exists");
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ UploadChannel uploadChannel = uploadChannelService.getByNodeIdAndUserId(nodeId, userId);
|
|
|
|
|
+ if (uploadChannel != null) {
|
|
|
|
|
+ return Result.fail("UploadChannel exists");
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
UserNode userNode = userNodeRepository.findByNodeIdAndUserId(nodeId, userId);
|
|
UserNode userNode = userNodeRepository.findByNodeIdAndUserId(nodeId, userId);
|
|
|
if (userNode != null) {
|
|
if (userNode != null) {
|
|
|
userNodeRepository.delete(userNode);
|
|
userNodeRepository.delete(userNode);
|
|
|
|
|
+ return Result.success();
|
|
|
}
|
|
}
|
|
|
|
|
+
|
|
|
|
|
+ return Result.fail("UserNode not exists");
|
|
|
}
|
|
}
|
|
|
|
|
+
|
|
|
|
|
+ return Result.fail("StoreNode not exists");
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
public List<StoreNode> getUserStoreNodes() {
|
|
public List<StoreNode> getUserStoreNodes() {
|