Add: support select ramdom files by tag

This commit is contained in:
2021-12-13 07:07:43 +08:00
parent 7a10922ec4
commit d4718ac120
5 changed files with 178 additions and 47 deletions

View File

@@ -72,6 +72,7 @@ func NewAPI(config Config) (*API, error) {
apiMux.HandleFunc("/search_folders", api.HandleSearchFolders)
apiMux.HandleFunc("/get_files_in_folder", api.HandleGetFilesInFolder)
apiMux.HandleFunc("/get_random_files", api.HandleGetRandomFiles)
apiMux.HandleFunc("/get_random_files_with_tag", api.HandleGetRandomFilesWithTag)
apiMux.HandleFunc("/get_file_stream", api.HandleGetFileStream)
apiMux.HandleFunc("/get_ffmpeg_config_list", api.HandleGetFfmpegConfigs)
apiMux.HandleFunc("/feedback", api.HandleFeedback)

View File

@@ -23,3 +23,29 @@ func (api *API) HandleGetRandomFiles(w http.ResponseWriter, r *http.Request) {
log.Println("[api] Get random files")
json.NewEncoder(w).Encode(getRandomFilesResponse)
}
type GetRandomFilesWithTagRequest struct {
ID int64 `json:"id"`
}
func (api *API) HandleGetRandomFilesWithTag(w http.ResponseWriter, r *http.Request) {
req := &GetRandomFilesWithTagRequest{}
err := json.NewDecoder(r.Body).Decode(req)
if err != nil {
api.HandleError(w, r, err)
return
}
files, err := api.Db.GetRandomFilesWithTag(req.ID, 10)
if err != nil {
api.HandleError(w, r, err)
return
}
getRandomFilesResponse := &GetRandomFilesResponse{
Files: &files,
}
log.Println("[api] Get random files with tag", req.ID)
json.NewEncoder(w).Encode(getRandomFilesResponse)
}