You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
4.4 KiB
116 lines
4.4 KiB
package com.bupt.note.Controller;
|
|
|
|
import com.bupt.note.Model.PaperNote;
|
|
import com.bupt.note.Repository.FileRepository;
|
|
import com.bupt.note.Repository.PaperNoteRepository;
|
|
import com.bupt.note.ResponseData.ResponseData;
|
|
import com.bupt.note.ResponseData.ResponseDataUtil;
|
|
import com.bupt.note.dto.FileForm;
|
|
import com.bupt.note.dto.QueryContent;
|
|
import com.bupt.note.service.FileService;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.http.MediaType;
|
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.util.UUID;
|
|
|
|
/**
|
|
* 论文文本管理接口
|
|
*/
|
|
@RestController
|
|
@RequestMapping(value = "/v1/api/file")
|
|
public class FileController {
|
|
|
|
|
|
private static Logger logger = LoggerFactory.getLogger(FileController.class);
|
|
|
|
@Autowired
|
|
private FileRepository fileRepository;
|
|
|
|
@Autowired
|
|
private PaperNoteRepository paperNoteRepository;
|
|
|
|
@Autowired
|
|
private FileService fileService;
|
|
|
|
//上传论文文本
|
|
@PostMapping("upload")
|
|
public ResponseData upload(FileForm fileForm) {
|
|
MultipartFile multipartFile = fileForm.getFile();
|
|
if (multipartFile != null && (MediaType.TEXT_PLAIN_VALUE.equals(multipartFile.getContentType()) || MediaType.APPLICATION_PDF_VALUE.equals(multipartFile.getContentType()))) {
|
|
|
|
try {
|
|
String name = multipartFile.getOriginalFilename();
|
|
File file;
|
|
if (name != null && name.contains(".")) {
|
|
file = fileService.newFile(name.replace(multipartFile.getOriginalFilename().split("\\.")[0], UUID.randomUUID().toString()));
|
|
} else {
|
|
file = fileService.newFile();
|
|
}
|
|
logger.debug("论文保存到" + file.getAbsolutePath());
|
|
multipartFile.transferTo(file);
|
|
com.bupt.note.Model.File f = new com.bupt.note.Model.File();
|
|
f.setFilePath(file.getAbsolutePath());
|
|
fileRepository.save(f);
|
|
return ResponseDataUtil.buildSuccess(f.getId());
|
|
} catch (IOException e) {
|
|
logger.error(String.valueOf(e));
|
|
e.printStackTrace();
|
|
return ResponseDataUtil.buildError();
|
|
}
|
|
} else {
|
|
logger.error("上传论文表单校验失败");
|
|
return ResponseDataUtil.buildError();
|
|
}
|
|
}
|
|
|
|
@GetMapping("find")
|
|
public ResponseData get(QueryContent queryContent, @CookieValue("user") String username){
|
|
try {
|
|
if (paperNoteRepository.existsByPaperIdAndUserName(queryContent.getPaperId(), username)) {
|
|
PaperNote paperNote = paperNoteRepository.findByPaperIdAndUserName(queryContent.getPaperId(), username);
|
|
return ResponseDataUtil.buildSuccess(fileService.getContent(paperNote.getFilePath()));
|
|
} else if (fileRepository.existsById(queryContent.getFileId())) {
|
|
com.bupt.note.Model.File file = fileRepository.getOne(queryContent.getFileId());
|
|
return ResponseDataUtil.buildSuccess(fileService.getContent(file.getFilePath()));
|
|
} else {
|
|
logger.error(String.format("论文文件id=%d不存在", queryContent.getFileId()));
|
|
return ResponseDataUtil.buildError();
|
|
}
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
logger.error(String.format("解析文件失败,原因是%s", e));
|
|
return ResponseDataUtil.buildError();
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* 删除论文文本文件
|
|
*
|
|
* @param id 文件id
|
|
* @return
|
|
*/
|
|
@DeleteMapping("remove/{id}")
|
|
public ResponseData remove(@PathVariable Long id) {
|
|
if (fileRepository.existsById(id)) {
|
|
com.bupt.note.Model.File txtFile = fileRepository.getOne(id);
|
|
File f = new File(txtFile.getFilePath());
|
|
if (f.exists() && f.delete()) {
|
|
fileRepository.deleteById(id);
|
|
return ResponseDataUtil.buildSuccess();
|
|
} else {
|
|
logger.error(String.format("删除论文id=%d失败", id));
|
|
return ResponseDataUtil.buildError();
|
|
}
|
|
} else {
|
|
logger.error(String.format("论文id=%d不存在", id));
|
|
return ResponseDataUtil.buildError();
|
|
}
|
|
}
|
|
}
|
|
|