package com.ld.vps.controller;
|
|
import cn.hutool.core.io.FileUtil;
|
import cn.hutool.json.JSONObject;
|
import cn.hutool.json.JSONUtil;
|
import com.ld.vps.bean.BaseResp;
|
import com.ld.vps.service.BitsPressService;
|
import com.ld.vps.service.UploadTaskService;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.repository.query.Param;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
@Controller
|
public class MainController {
|
Logger logger = LoggerFactory.getLogger(MainController.class);
|
@Autowired
|
UploadTaskService taskService;
|
@Autowired
|
BitsPressService bitsPressService;
|
|
@GetMapping("/getUserList")
|
@ResponseBody
|
String getUserList(@RequestParam("count") int count) {
|
return bitsPressService.getTopQueueItem(count);
|
}
|
|
@PostMapping("/uploadFile")
|
@ResponseBody
|
BaseResp<Long> uploadFile(
|
@RequestParam("file") String file,
|
@RequestParam(value = "userId", required = false) String userId) {
|
if (!FileUtil.isFile(file)) {
|
BaseResp<Long> resp = new BaseResp<>(0L);
|
resp.setMsg("文件不存在：" + file);
|
resp.setSuccess(false);
|
return resp;
|
}
|
Long id = taskService.createTask(file, userId == null ? "" : userId);
|
return new BaseResp<>(id);
|
}
|
|
@GetMapping("/refreshToken")
|
@ResponseBody
|
BaseResp<String> refreshToken(@Param("token") String token) {
|
bitsPressService.refreshToken(token);
|
return new BaseResp<>(bitsPressService.getToken());
|
}
|
}
|
