package com.thhy.filectrl.service.impl;
|
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import com.alibaba.nacos.api.utils.StringUtils;
|
import com.thhy.filectrl.service.FdfsService;
|
import com.thhy.filectrl.utils.ConvertUtil;
|
import com.thhy.general.common.BasicStatus;
|
import com.thhy.general.exception.BasicException;
|
import org.apache.commons.io.FilenameUtils;
|
import org.csource.common.MyException;
|
import org.csource.common.NameValuePair;
|
import org.csource.fastdfs.FileInfo;
|
import org.csource.fastdfs.StorageClient;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.URLEncoder;
|
import java.text.SimpleDateFormat;
|
import java.util.Date;
|
|
import static com.thhy.general.common.BasicStatus.FILE_NOT_EXISTS_REMOTE;
|
import static com.thhy.general.common.BasicStatus.FILE_UPLOAD_ERROR;
|
|
@Service
|
//@ConditionalOnProperty(prefix = "fastdfs",name = "enbale",havingValue = "true")
|
public class FdfsServiceImpl implements FdfsService {
|
|
private Logger logger = LoggerFactory.getLogger(FdfsServiceImpl.class);
|
|
@Autowired
|
private StorageClient storageClient;
|
|
@Value("${file.workspace}")
|
private String workspace;
|
|
|
public String upload(MultipartFile file) {
|
try {
|
//设置文件信息
|
NameValuePair[] nv = new NameValuePair[2];
|
nv[0] = new NameValuePair("name",file.getOriginalFilename());
|
nv[1] = new NameValuePair("uploadTime",new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()));
|
String[] paths = storageClient.upload_file(file.getBytes(), FilenameUtils.getExtension(file.getOriginalFilename()),nv);
|
return paths[0]+"/"+paths[1];
|
}catch (Exception exception){
|
throw new BasicException(FILE_UPLOAD_ERROR,exception);
|
}
|
}
|
|
@Override
|
public void downfile(String dfsPath,String fileName, HttpServletResponse resp) {
|
//storageClient.download_file()
|
String groupName = dfsPath.substring(0,dfsPath.indexOf("/"));
|
String remoteFile = dfsPath.substring(dfsPath.indexOf("/")+1,dfsPath.length());
|
resp.setCharacterEncoding("utf-8");
|
try {
|
if(StringUtils.isBlank(fileName)){
|
NameValuePair[] nv = storageClient.get_metadata(groupName,remoteFile);
|
for(NameValuePair n : nv){
|
if("name".equals(n.getName())){
|
fileName = n.getValue();
|
break;
|
}
|
}
|
}
|
fileName = URLEncoder.encode(fileName,"UTF-8");
|
resp.setHeader("Content-Disposition","attachment;filename="+fileName);
|
byte[] bytes = storageClient.download_file(groupName,remoteFile);
|
OutputStream os = resp.getOutputStream();
|
os.write(bytes);
|
} catch (IOException e) {
|
throw new BasicException(FILE_NOT_EXISTS_REMOTE);
|
} catch (MyException e) {
|
throw new RuntimeException(e);
|
}
|
|
}
|
|
@Override
|
public void preview(String dfsPath,HttpServletResponse resp) {
|
String groupName = dfsPath.substring(0,dfsPath.indexOf("/"));
|
String remoteFile = dfsPath.substring(dfsPath.indexOf("/")+1,dfsPath.length());
|
resp.setCharacterEncoding("utf-8");
|
|
OutputStream os = null;
|
InputStream inputStream = null;
|
try {
|
String fileName = dfsPath.substring(dfsPath.lastIndexOf("/")+1,dfsPath.lastIndexOf("."));
|
String extendName = dfsPath.substring(dfsPath.lastIndexOf(".")+1,dfsPath.length());
|
if("pdf".equals(extendName)||"PDF".equals(extendName)){
|
downfile(dfsPath,"preview"+extendName,resp);
|
return;
|
}
|
fileName = URLEncoder.encode(fileName,"UTF-8");
|
resp.setHeader("Content-Disposition","attachment;filename="+fileName+".pdf");
|
byte[] bytes = storageClient.download_file(groupName,remoteFile);
|
|
FileOutputStream fos = new FileOutputStream(new File(workspace+"/"+fileName+"."+extendName));
|
fos.write(bytes);
|
fos.close();
|
logger.info("文件保存本地成功:"+workspace+"/"+fileName+"."+extendName);
|
|
ConvertUtil.executeCommand(workspace,fileName+"."+extendName);
|
File file = new File(workspace+"/"+fileName+".pdf");
|
logger.info("命令执行完成,pdf是否存在了"+file.exists());
|
inputStream = new FileInputStream(file);
|
|
byte[] bytes1 = new byte[inputStream.available()];
|
inputStream.read(bytes1);
|
logger.info("读取PDF成功");
|
os = resp.getOutputStream();
|
os.write(bytes1);
|
} catch (IOException e) {
|
throw new BasicException(FILE_NOT_EXISTS_REMOTE);
|
} catch (MyException e) {
|
throw new RuntimeException(e);
|
} finally {
|
try {
|
if(inputStream!=null)inputStream.close();
|
if(os!=null)os.close();
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
|
}
|
}
|
|
@Override
|
public NameValuePair[] fileInfo(String fullPath) {
|
String groupName = fullPath.substring(0,fullPath.indexOf("/"));
|
String remoteFile = fullPath.substring(fullPath.indexOf("/")+1,fullPath.length());
|
try {
|
NameValuePair[] nv = storageClient.get_metadata(groupName,remoteFile);
|
return nv;
|
} catch (IOException e) {
|
throw new BasicException(BasicStatus.FILE_PATH_ERROR);
|
} catch (MyException e) {
|
throw new RuntimeException(e);
|
}
|
}
|
|
public static void main(String[] args) {
|
String dfsPath = "group1/M00/00/00/bx5d1GRGObqAYmLZAAoS1mNy1B8114.png";
|
String groupName = dfsPath.substring(0,dfsPath.indexOf("/"));
|
String remoteFile = dfsPath.substring(dfsPath.indexOf("/")+1,dfsPath.length());
|
String fileName = dfsPath.substring(dfsPath.lastIndexOf("/")+1,dfsPath.lastIndexOf("."));
|
String extendName = dfsPath.substring(dfsPath.lastIndexOf(".")+1,dfsPath.length());
|
System.out.printf(fileName);
|
}
|
|
|
}
|
