besscroft/diyfile
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增存储校验

Browse Source
This commit is contained in:
besscroft
2023-03-09 21:55:51 +08:00
parent a1b394ca73
commit 65bab9b215
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
diyfile-system/src/main/java/com/besscroft/diyfile/controller/StorageController.java
View File

@@ -3,8 +3,10 @@ package com.besscroft.diyfile.controller;
import cn.dev33.satoken.annotation.SaCheckRole;
import cn.dev33.satoken.annotation.SaIgnore;
import cn.dev33.satoken.annotation.SaMode;
import cn.hutool.core.util.StrUtil;
import com.besscroft.diyfile.common.constant.RoleConstants;
import com.besscroft.diyfile.common.entity.Storage;
import com.besscroft.diyfile.common.exception.DiyFileException;
import com.besscroft.diyfile.common.param.storage.StorageAddParam;
import com.besscroft.diyfile.common.param.storage.StorageUpdateParam;
import com.besscroft.diyfile.common.param.storage.StorageUpdateStatusParam;
@@ -20,6 +22,7 @@ import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;
import java.util.List;
import java.util.Objects;
/**
* @Description 存储接口
@@ -78,6 +81,10 @@ public class StorageController {
)
@PostMapping("/add")
public AjaxResult add(@RequestBody @Valid StorageAddParam param) {
if (Objects.equals("proxy", param.getStorageKey()))
throw new DiyFileException("存储 key 不能为 proxy");
if (StrUtil.contains(param.getStorageKey(), "/"))
throw new DiyFileException("存储 key 不能包含 /");
storageService.addStorage(param);
return AjaxResult.success("新增成功！");
}