|
|
@@ -1,5 +1,6 @@
|
|
|
package cn.reghao.dfs.store.task;
|
|
|
|
|
|
+import cn.reghao.dfs.store.db.repository.ImageRepository;
|
|
|
import cn.reghao.dfs.store.model.vo.ObjectResult;
|
|
|
import cn.reghao.dfs.store.task.processor.AudioFileProcessor;
|
|
|
import cn.reghao.dfs.store.task.processor.DiskFileProcessor;
|
|
|
@@ -26,13 +27,16 @@ public class FileProcessor {
|
|
|
private final VideoFileProcessor videoFileProcessor;
|
|
|
private final AudioFileProcessor audioFileProcessor;
|
|
|
private final DiskFileProcessor diskFileProcessor;
|
|
|
+ private final ImageRepository imageRepository;
|
|
|
|
|
|
public FileProcessor(ImageFileProcessor imageFileProcessor, VideoFileProcessor videoFileProcessor,
|
|
|
- AudioFileProcessor audioFileProcessor, DiskFileProcessor diskFileProcessor) {
|
|
|
+ AudioFileProcessor audioFileProcessor, DiskFileProcessor diskFileProcessor,
|
|
|
+ ImageRepository imageRepository) {
|
|
|
this.imageFileProcessor = imageFileProcessor;
|
|
|
this.videoFileProcessor = videoFileProcessor;
|
|
|
this.audioFileProcessor = audioFileProcessor;
|
|
|
this.diskFileProcessor = diskFileProcessor;
|
|
|
+ this.imageRepository = imageRepository;
|
|
|
}
|
|
|
|
|
|
public UploadFileRet process(ObjectResult objectResult, int channelId) {
|
|
|
@@ -56,6 +60,12 @@ public class FileProcessor {
|
|
|
case image:
|
|
|
uploadFileRet = imageFileProcessor.processImage(objectResult);
|
|
|
break;
|
|
|
+ case img:
|
|
|
+ uploadFileRet = imageFileProcessor.processImage(objectResult);
|
|
|
+ String uploadId = uploadFileRet.getUploadId();
|
|
|
+ String url = imageRepository.getImageUrl(uploadId).getOriginalUrl();
|
|
|
+ uploadFileRet.setUrl(url);
|
|
|
+ break;
|
|
|
case photo:
|
|
|
uploadFileRet = imageFileProcessor.processPhoto(objectResult);
|
|
|
break;
|