heimoshuiyu/msw-open-music
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

check limit

Browse Source
This commit is contained in:
黑墨水鱼
2021-05-22 21:16:43 +08:00
parent 60c110eff3
commit dfd1f136bc
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/pkg/api/api.go
View File

@@ -236,8 +236,7 @@ func (api *API) HandleSearchFiles(w http.ResponseWriter, r *http.Request) {
api.HandleErrorString(w, r, `"filename" can't be empty`)
return
}
if searchFilesRequest.Limit == 0 {
api.HandleErrorString(w, r, `"limit" can't be zero`)
if api.CheckLimit(w, r, searchFilesRequest.Limit) != nil {
return
}
@@ -254,6 +253,16 @@ func (api *API) HandleSearchFiles(w http.ResponseWriter, r *http.Request) {
json.NewEncoder(w).Encode(searchFilesResponse)
}
func (api *API) CheckLimit(w http.ResponseWriter, r *http.Request, limit int64) (error) {
if limit <= 0 || limit > 10 {
log.Println("[api] [Warning] Limit error", limit)
err := errors.New(`"limit" can't be zero or more than 10`)
api.HandleError(w, r, err)
return err
}
return nil
}
func (api *API) HandleSearchFolders(w http.ResponseWriter, r *http.Request) {
searchFoldersRequest := &SearchFoldersRequest{}
err := json.NewDecoder(r.Body).Decode(searchFoldersRequest)
@@ -267,8 +276,7 @@ func (api *API) HandleSearchFolders(w http.ResponseWriter, r *http.Request) {
api.HandleErrorString(w, r, `"foldername" can't be empty`)
return
}
if searchFoldersRequest.Limit == 0 {
api.HandleErrorString(w, r, `"limit" can't be zero`)
if api.CheckLimit(w, r, searchFoldersRequest.Limit) != nil {
return
}