提交 110f8cd0 作者: ZhangJingKun

文件上传 zhangjingkun

上级 89af4415
package com.zzsn.knowbase.service; package com.zzsn.knowbase.service;
import com.zzsn.knowbase.entity.Knowledge; import com.zzsn.knowbase.entity.KnowFile;
import com.zzsn.knowbase.vo.Result; import com.zzsn.knowbase.vo.Result;
import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartFile;
...@@ -12,5 +12,5 @@ import org.springframework.web.multipart.MultipartFile; ...@@ -12,5 +12,5 @@ import org.springframework.web.multipart.MultipartFile;
*/ */
public interface ILocalFileService { public interface ILocalFileService {
Result<Knowledge> upload(MultipartFile file, String uid); Result<KnowFile> upload(MultipartFile file, String uid);
} }
package com.zzsn.knowbase.service.impl; package com.zzsn.knowbase.service.impl;
import com.zzsn.knowbase.entity.KnowFile; import com.zzsn.knowbase.entity.KnowFile;
import com.zzsn.knowbase.entity.Knowledge;
import com.zzsn.knowbase.service.ILocalFileService; import com.zzsn.knowbase.service.ILocalFileService;
import com.zzsn.knowbase.util.file.FileUtility; import com.zzsn.knowbase.util.file.FileUtility;
import com.zzsn.knowbase.vo.Result; import com.zzsn.knowbase.vo.Result;
...@@ -36,13 +35,12 @@ public class LocalFileServiceImpl implements ILocalFileService { ...@@ -36,13 +35,12 @@ public class LocalFileServiceImpl implements ILocalFileService {
String filesStorage; String filesStorage;
@Override @Override
public Result<Knowledge> upload(MultipartFile file, String uid) { public Result<KnowFile> upload(MultipartFile file, String uid) {
try { try {
String fullFileName = file.getOriginalFilename(); // get file name String fullFileName = file.getOriginalFilename(); // get file name
String fileExtension = fileUtility.getFileExtension(fullFileName); // get file extension String fileExtension = fileUtility.getFileExtension(fullFileName); // get file extension
long fileSize = file.getSize(); // get file size long fileSize = file.getSize(); // get file size
// check if the file size exceeds the maximum file size or is less than 0 // check if the file size exceeds the maximum file size or is less than 0
if (fileUtility.getMaxFileSize() < fileSize || fileSize <= 0) { if (fileUtility.getMaxFileSize() < fileSize || fileSize <= 0) {
Result result = Result.error("文件大小不正确!"); Result result = Result.error("文件大小不正确!");
...@@ -58,13 +56,15 @@ public class LocalFileServiceImpl implements ILocalFileService { ...@@ -58,13 +56,15 @@ public class LocalFileServiceImpl implements ILocalFileService {
String fileName = file.getOriginalFilename(); String fileName = file.getOriginalFilename();
String fileSuffix = getFileSuffix(fileName); String fileSuffix = getFileSuffix(fileName);
String filePath = getFilePath() + UUID.randomUUID(); uid = UUID.randomUUID().toString();
String filePath = getFilePath() + uid;
byte[] bytes = file.getBytes(); // get file in bytes byte[] bytes = file.getBytes(); // get file in bytes
//Files.write(Paths.get(filePath), bytes); //Files.write(Paths.get(filePath), bytes);
file.transferTo(new File(filePath)); file.transferTo(new File(filePath));
KnowFile knowFile = new KnowFile(); KnowFile knowFile = new KnowFile();
knowFile.setFileId(fileName); knowFile.setFileId(uid);
knowFile.setFileName(fileName); knowFile.setFileName(fileName);
knowFile.setFilePath(filePath); knowFile.setFilePath(filePath);
knowFile.setFileType(fileSuffix); knowFile.setFileType(fileSuffix);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论