package com.doumee.core.oss;
|
|
import com.doumee.core.utils.ID;
|
import lombok.Cleanup;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.mock.web.MockMultipartFile;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.BufferedOutputStream;
|
import java.io.File;
|
import java.io.FileOutputStream;
|
import java.io.InputStream;
|
import java.time.LocalDate;
|
import java.time.format.DateTimeFormatter;
|
import java.util.Map;
|
|
/**
|
* 附件上传
|
*/
|
@Service
|
public class UploadFileService {
|
|
|
@Autowired
|
private AliOssService aliOssService;
|
|
public FileModel uploadWxCode(InputStream inputStream,String id) throws Exception{
|
MultipartFile multipartFile = new MockMultipartFile(id+".png",id+".png","", inputStream);
|
return this.uploadAliOssFile(multipartFile,"wxCode",true);
|
}
|
|
/**
|
* 2、上传阿里云OSS
|
*
|
* @param file: 表单文件
|
* @param model:是否存在目录
|
* @param ifCreateFileName:是否创建新名字 true/false
|
*/
|
public FileModel uploadAliOssFile(MultipartFile file, String model, Boolean ifCreateFileName) {
|
FileModel res = new FileModel();
|
//OSS
|
String origin=FileContent.fileOrigin.get(1);
|
//原文件名(如:xxxx.doc)
|
String originalFilename = file.getOriginalFilename();
|
//文件后缀(如:doc)
|
String suffix = originalFilename.substring(file.getOriginalFilename().lastIndexOf(".") + 1);
|
//文件名
|
String fileName = ifCreateFileName ? ID.nextGUID() + originalFilename.substring(file.getOriginalFilename().lastIndexOf(".")) : originalFilename;
|
//--------OOS文件存放根路径---------
|
String fullFileName = this.getRootPath(model,fileName);
|
res.setOrigin(origin);
|
res.setSuffix(suffix);
|
//新文件名
|
res.setNewFilename(fileName);
|
//原文件名
|
res.setFileName(originalFilename);
|
res.setFilePath(fullFileName);
|
try {
|
Map map = aliOssService.uploadFile(fullFileName, file.getInputStream());
|
String code = map.get("code").toString();
|
String url = map.get("url").toString();
|
String msg = map.get("msg").toString();
|
if (code.equals("200")) {
|
res.setCode(0);
|
res.setFileUrl(url);
|
res.setMsg(msg);
|
} else {
|
res.setCode(-1);
|
res.setMsg(msg);
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
res.setMsg(e.getMessage());
|
res.setCode(-1);
|
}
|
return res;
|
}
|
|
|
/**
|
* 获取根目录及下载URL
|
*
|
* @param model
|
* @param fileName
|
* @return
|
*/
|
public String getRootPath(String model, String fileName) {
|
String time = LocalDate.now().format(DateTimeFormatter.BASIC_ISO_DATE);
|
//根目录
|
String catalog = "file";
|
String oss_doc= time+ "/" + model + "/" + fileName;
|
if(StringUtils.isNotBlank(catalog)){
|
oss_doc=catalog.trim()+"/"+oss_doc;
|
}
|
return oss_doc;
|
}
|
}
|
