1、概述
文件上传是我们比较常见的功能,主要代码如下。
2、代码
@RestController
public class FileUploadController {
@PostMapping("/upload")
public ResponseEntity<String> uploadFile(@RequestPart("file") MultipartFile file) {
// 处理文件上传逻辑
if (!file.isEmpty()) {
try {
// 获取文件字节数据
byte[] bytes = file.getBytes();
// 设置文件保存路径,这里假设保存在项目根目录下的upload文件夹中
String savePath = "upload" + File.separator + file.getOriginalFilename();
// 创建保存文件的File对象
File saveFile = new File(savePath);
if (!saveFile.getParentFile().exists()) {
saveFile.getParentFile().mkdirs(); // 如果父目录不存在,则创建父目录
}
// 将文件字节数据写入到保存的File对象中
FileOutputStream fos = new FileOutputStream(saveFile);
fos.write(bytes);
fos.close();
return new ResponseEntity<>("File uploaded successfully.", HttpStatus.OK);
} catch (Exception e) {
return new ResponseEntity<>("File upload failed: " + e.getMessage(), HttpStatus.INTERNAL_SERVER_ERROR);
}
} else {
return new ResponseEntity<>("File upload failed: Empty file.", HttpStatus.BAD_REQUEST);
}
}
}