forked from kk-open/kkFileView
57 lines
2.0 KiB
Java
57 lines
2.0 KiB
Java
package cn.keking.service.impl;
|
||
|
||
import cn.keking.model.FileAttribute;
|
||
import cn.keking.model.ReturnResponse;
|
||
import cn.keking.service.FilePreview;
|
||
import cn.keking.utils.DownloadUtils;
|
||
import cn.keking.service.FileHandlerService;
|
||
import org.springframework.stereotype.Service;
|
||
import org.springframework.ui.Model;
|
||
import org.springframework.util.CollectionUtils;
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
|
||
/**
|
||
* Created by kl on 2018/1/17.
|
||
* Content :图片文件处理
|
||
*/
|
||
@Service
|
||
public class PictureFilePreviewImpl implements FilePreview {
|
||
|
||
private final FileHandlerService fileHandlerService;
|
||
|
||
public PictureFilePreviewImpl(FileHandlerService fileHandlerService) {
|
||
this.fileHandlerService = fileHandlerService;
|
||
}
|
||
|
||
@Override
|
||
public String filePreviewHandle(String url, Model model, FileAttribute fileAttribute) {
|
||
List<String> imgUrls = new ArrayList<>();
|
||
imgUrls.add(url);
|
||
String fileKey = fileAttribute.getFileKey();
|
||
List<String> zipImgUrls = fileHandlerService.getImgCache(fileKey);
|
||
if (!CollectionUtils.isEmpty(zipImgUrls)) {
|
||
imgUrls.addAll(zipImgUrls);
|
||
}
|
||
// 不是http开头，浏览器不能直接访问，需下载到本地
|
||
if (url != null && !url.toLowerCase().startsWith("http")) {
|
||
ReturnResponse<String> response = DownloadUtils.downLoad(fileAttribute, null);
|
||
if (response.isFailure()) {
|
||
model.addAttribute("fileType", fileAttribute.getSuffix());
|
||
model.addAttribute("msg", response.getMsg());
|
||
return "fileNotSupported";
|
||
} else {
|
||
String file = fileHandlerService.getRelativePath(response.getContent());
|
||
imgUrls.clear();
|
||
imgUrls.add(file);
|
||
model.addAttribute("imgurls", imgUrls);
|
||
model.addAttribute("currentUrl", file);
|
||
}
|
||
} else {
|
||
model.addAttribute("imgurls", imgUrls);
|
||
model.addAttribute("currentUrl", url);
|
||
}
|
||
return "picture";
|
||
}
|
||
}
|