WL/laws_weilai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cos工具类 + 上传下载接口cos切换

Browse Source
This commit is contained in:
liyawei
2022-06-08 18:59:54 +08:00
parent d5497cf5bc
commit 1ff997103a
10 changed files with 501 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jero-boot/jero-boot-base/jero-boot-base-core/pom.xml
+12 -1
View File
@@ -241,7 +241,18 @@
<groupId>com.squareup.okhttp3</groupId>
<artifactId>okhttp</artifactId>
</dependency>
<!-- 腾讯 cos -->
<dependency>
<groupId>com.qcloud</groupId>
<artifactId>cos_api</artifactId>
<version>5.2.4</version>
<exclusions>
<exclusion>
<artifactId>slf4j-log4j12</artifactId>
<groupId>org.slf4j</groupId>
</exclusion>
</exclusions>
</dependency>
</dependencies>
</project>
jero-boot/jero-boot-base/jero-boot-base-core/src/main/java/com/jero/common/util/oss/CosBootUtil.java
+260
View File
@@ -0,0 +1,260 @@
package com.jero.common.util.oss;
import com.jero.config.oss.CosConfiguration;
import com.qcloud.cos.COSClient;
import com.qcloud.cos.ClientConfig;
import com.qcloud.cos.auth.BasicCOSCredentials;
import com.qcloud.cos.auth.COSCredentials;
import com.qcloud.cos.exception.CosClientException;
import com.qcloud.cos.exception.CosServiceException;
import com.qcloud.cos.http.HttpMethodName;
import com.qcloud.cos.http.HttpProtocol;
import com.qcloud.cos.region.Region;
import com.qcloud.cos.utils.IOUtils;
import lombok.extern.slf4j.Slf4j;
import org.jeecgframework.core.util.ApplicationContextUtil;
import org.jeecgframework.dict.service.AutoPoiDictServiceI;
import org.springframework.web.multipart.MultipartFile;
import java.io.*;
import java.net.*;
import java.util.Date;
/**
* @Author: liyawei
* @Description:
* @Date: Created in 9:12 2022/6/8
*/
@Slf4j
public class CosBootUtil {
private static String secretId;
private static String secretKey;
private static String region;
private static String bucketName;
private static String appId;
private static String endPoint;
public static void setSecretId(String secretId) {
CosBootUtil.secretId = secretId;
}
public static void setSecretKey(String secretKey) {
CosBootUtil.secretKey = secretKey;
}
public static void setRegion(String region) {
CosBootUtil.region = region;
}
public static void setBucketName(String bucketName) {
CosBootUtil.bucketName = bucketName;
}
public static void setAppId(String appId) {
CosBootUtil.appId = appId;
}
public static void setEndPoint(String endPoint) {
CosBootUtil.endPoint = endPoint;
}
public static String getSecretId() {
return secretId;
}
public static String getSecretKey() {
return secretKey;
}
public static String getRegion() {
return region;
}
public static String getBucketName() {
return bucketName;
}
public static String getAppId() {
return appId;
}
public static String getEndPoint() {
return endPoint;
}
/**
* cos 工具客户端
*/
private static COSClient cosClient = null;
private static COSCredentials cred = null;
/**
* 初始化
*/
private static void init() {
CosConfiguration ccf = ApplicationContextUtil.getContext().getBean(CosConfiguration.class);
ccf.initCosBootConfiguration();
// 1 初始化用户身份信息(secretId, secretKey)
cred = new BasicCOSCredentials(secretId, secretKey);
// 2 设置bucket的区域,
ClientConfig clientConfig = new ClientConfig(new Region(region));
clientConfig.setHttpProtocol(HttpProtocol.https);
// 以下的设置，是可选的：
// 设置 socket 读取超时，默认 30s
// clientConfig.setSocketTimeout(30*1000);
// 设置建立连接超时，默认 30s
// clientConfig.setConnectionTimeout(30*1000);
// 3 生成cos客户端
cosClient = new COSClient(cred, clientConfig);
}
/**
* cos 上传
* 相同key 文件会覆盖
* @param file
* @param key
*/
public static void upload(MultipartFile file, String key){
init();
Date expirationTime = new Date(System.currentTimeMillis() + 30 * 60 * 1000);
// 生成预签名上传 URL
URL url = cosClient.generatePresignedUrl(bucketName, key, expirationTime, HttpMethodName.PUT);
log.info("生成预签名上传URL:" + url);
int responseCode = upLoadFile(url.toString(), file);
if (responseCode == 200) {
log.info("上传成功！");
} else {
log.info("上传失败！");
}
cosClient.shutdown();
}
private static int upLoadFile(String singnedurl,MultipartFile file){
InputStream inputStream= null;
DataOutputStream out = null;
int responseCode = 0;
try {
URL url= null;
url = new URL(singnedurl);
inputStream= file.getInputStream();
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
connection.setDoOutput(true);
connection.setRequestMethod("PUT");
out = new DataOutputStream(connection.getOutputStream());
// 写入要上传的数据
IOUtils.copy(inputStream, out);
responseCode = connection.getResponseCode();
log.info("Service returned response code " + responseCode);
} catch (ProtocolException e) {
log.error(e.getMessage(), e);
} catch (IOException e) {
log.error(e.getMessage(), e);
} finally {
try {
inputStream.close();
} catch (IOException e) {
log.error(e.getMessage(), e);
}
try {
out.close();
} catch (IOException e) {
log.error(e.getMessage(), e);
}
return responseCode;
}
}
/**
* cos 下载
* @param key
* @return
*/
public static InputStream download(String key) {
Date expirationTime = new Date(System.currentTimeMillis() + 30 * 60 * 1000);
// 生成预签名上传 URL
URL url = cosClient.generatePresignedUrl(bucketName, key, expirationTime, HttpMethodName.GET);
log.info("生成预签名上传URL:" + url);
InputStream in = downFile(url.toString());
if (in != null) {
log.info("下载成功！");
} else {
log.info("下载失败！");
}
// System.out.println("下载成功");
cosClient.shutdown();
return in;
}
private static InputStream downFile(String singnedurl) {
InputStream in = null;
try {
URL url=new URL(singnedurl);
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
connection.setDoInput(true);
connection.setRequestMethod("GET");
in = connection.getInputStream();
int responseCode = connection.getResponseCode();
log.info("Service returned response code " + responseCode);
// 写入要上传的数据
// IOUtils.copy(in, outputStream);
return in;
} catch (IOException e) {
e.printStackTrace();
}/* finally {
try {
in.close();
} catch (IOException e) {
e.printStackTrace();
}
// fileOutputStream.close();
}*/
return in;
}
/**
* 判断文件是否存在
* 上传，下载，删除时 必需调用此方法
* @param key
* @return
*/
public static boolean doesObjectExist(String key) {
init();
boolean objectExists = false;
try {
objectExists = cosClient.doesObjectExist(bucketName, key);
return objectExists;
} catch (CosServiceException e) {
e.printStackTrace();
} catch (CosClientException e) {
e.printStackTrace();
} finally {
// 确认本进程不再使用 cosClient 实例之后，关闭之
cosClient.shutdown();
}
return objectExists;
}
/**
* 删除
*/
public static void delete(String key) {
new Thread(new Runnable() {
public void run() {
// 指定要删除的 bucket 和路径
try {
cosClient.deleteObject(bucketName, key);
log.info("删除成功");
} catch (Throwable tb) {
log.info("删除文件失败");
tb.printStackTrace();
}
}
}).start();
}
}
jero-boot/jero-boot-base/jero-boot-base-core/src/main/java/com/jero/config/oss/CosConfiguration.java
+39
View File
@@ -0,0 +1,39 @@
package com.jero.config.oss;
import com.jero.common.util.oss.CosBootUtil;
import com.jero.common.util.oss.OssBootUtil;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
/**
* @Author: liyawei
* @Description:
* @Date: Created in 9:11 2022/6/8
*/
@Configuration
public class CosConfiguration {
@Value("${jero.cos.secretId}")
private String secretId;
@Value("${jero.cos.secretKey}")
private String secretKey;
@Value("${jero.cos.region}")
private String region;
@Value("${jero.cos.bucketName}")
private String bucketName;
@Value("${jero.cos.appId}")
private String appId;
@Value("${jero.cos.endPoint}")
private String endPoint;
@Bean
public void initCosBootConfiguration() {
CosBootUtil.setSecretId(secretId);
CosBootUtil.setSecretKey(secretKey);
CosBootUtil.setRegion(region);
CosBootUtil.setBucketName(bucketName);
CosBootUtil.setAppId(appId);
CosBootUtil.setEndPoint(endPoint);
}
}
jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/oss/service/IOSSFileService.java
+2
View File
@@ -19,6 +19,8 @@ public interface IOSSFileService extends IService<OSSFile> {
*/
public OSSFile uploadLocal(MultipartFile mf, String bizPath,String state,String cut);
public OSSFile uploadLocalOfCos(MultipartFile mf, String bizPath,String state,String cut);
public OSSFile uploadLocalForSplit(MultipartFile mf, String bizPath,String state,String cut);
jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/oss/service/impl/OSSFileServiceImpl.java
+98
View File
@@ -7,6 +7,7 @@ import com.jero.common.exception.JeroBootException;
import com.jero.common.system.vo.LoginUser;
import com.jero.common.util.CommonUtils;
import com.jero.common.util.oConvertUtils;
import com.jero.common.util.oss.CosBootUtil;
import com.jero.common.util.oss.OssBootUtil;
import com.jero.modules.oss.entity.OSSFile;
import com.jero.modules.oss.mapper.OSSFileMapper;
@@ -164,6 +165,103 @@ public class OSSFileServiceImpl extends ServiceImpl<OSSFileMapper, OSSFile> impl
return oSSFile;
}
/**
* 本地文件上传 cos
*
* @param mf 文件
* @param bizPath 自定义路径
* @return
*/
@Override
public OSSFile uploadLocalOfCos(MultipartFile mf, String bizPath,String state,String cut) {
//处理文件大小,大于100M抛出异常
long fileSize = mf.getSize() / 1024 / 1024;
String originalFilename = mf.getOriginalFilename();
if (fileSize >= 100) {
if(cut.equals(CutEnum.CN.getValue())){
throw new JeroBootException(originalFilename + "文件大小超出100MB, 请压缩或降低文件质量!");
}else{
throw new JeroBootException(originalFilename + "File size out 100MB, Please compress or reduce file quality!");
}
}
OSSFile oSSFile = new OSSFile();
try {
String ctxPath = uploadpath;
String fileName = null;
String fileType = null;
String orgName = mf.getOriginalFilename();// 获取文件名
orgName = CommonUtils.getFileName(orgName);
if (orgName.indexOf(".") != -1) {
fileName = orgName.substring(0, orgName.lastIndexOf(".")) + "_" + System.currentTimeMillis() + orgName.substring(orgName.lastIndexOf("."));
} else {
fileName = orgName + "_" + System.currentTimeMillis();
}
String filePath = ctxPath + "/" + fileName;
fileType = orgName.substring(orgName.lastIndexOf("."));
String fileTypeStr = ".doc,.DOC,.docx,.DOCX,.xls, .XLS,.xlsx,.XLSX,.pdf,.PDF";
//判断文件类型
if("1".equals(state)){
//固定文件
if(!fileTypeStr.contains(fileType)){
if(cut.equals(CutEnum.CN.getValue())){
throw new JeroBootException("只能够上传pdf,word,excel类型的文件。请重新选择文件！");
}else{
throw new JeroBootException("Only upload pdf,word,excel type of file Please select the file again！");
}
}
}else{
if (CommonUtils.limitFileSuffix(orgName, fileSuffixLimits)) {
if(cut.equals(CutEnum.CN.getValue())){
throw new JeroBootException("不能上传"+StringUtils.join(fileSuffixLimits, ",")+"类型的文件。请重新选择文件！");
}else{
throw new JeroBootException("Can't upload"+StringUtils.join(fileSuffixLimits, ",")+"type of file Please select the file again！");
}
}
}
if (StringUtils.isNotEmpty(fileType)) {
String[] fileTypeArr = {".doc",".DOC",".txt",".TXT", ".docx",".DOCX",
".xls", ".XLS",".xlsx",".XLSX", ".pdf",".PDF", ".png",".PNG",
".jfif",".JFIF", ".pjpeg",".PJPEG", ".jpeg",".JPEG", ".pjp",".PJP",
".jpg",".JPG", ".swf",".SWF", ".bmp",".BMP",".rar",".RAR",".zip",".ZIP",".ppt",".PPT",".pptx",".PPTX",".csv",".CSV"};
boolean flag = true;
for (String fileTypeTemp : fileTypeArr) {
if (fileTypeTemp.equalsIgnoreCase(fileType)) {
flag = false;
}
}
if (flag) {
if(cut.equals(CutEnum.CN.getValue())){
throw new JeroBootException("只能够上传pdf/word/excel/csv/ppt/txt/zip/rar/静态图片类型的文件。请重新选择文件！");
}else{
throw new JeroBootException("Only upload pdf/word/excel/csv/ppt/txt/zip/rar/静态图片类型的文件。请重新选择文件！");
}
}
}
// 将文件上传至腾讯云 cos
CosBootUtil.upload(mf, filePath);
LoginUser loginUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
//存入文件表
oSSFile.setFileName(orgName);
oSSFile.setId(UuidUtils.getUUID());
oSSFile.setUrl(filePath);
oSSFile.setCreateBy(loginUser.getUsername());
oSSFile.setCreateTime(new Date());
this.save(oSSFile);
oSSFile.setUrl(null);
return oSSFile;
} catch (Exception e) {
log.error(e.getMessage(), e);
}
return oSSFile;
}
/**
* 本地文件上传-文档拆分专用
*
jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/system/controller/CommonController.java
+75 -7
View File
@@ -11,6 +11,7 @@ import com.jero.common.system.api.ISysBaseAPI;
import com.jero.common.util.RestUtil;
import com.jero.common.util.TokenUtils;
import com.jero.common.util.oConvertUtils;
import com.jero.common.util.oss.CosBootUtil;
import com.jero.modules.oss.entity.OSSFile;
import com.jero.modules.oss.service.IOSSFileService;
import com.jero.modules.system.util.PDFUtils;
@@ -158,14 +159,16 @@ public class CommonController {
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
MultipartFile file = multipartRequest.getFile("file");// 获取上传文件对象
if (oConvertUtils.isEmpty(bizPath)) {
if (CommonConstant.UPLOAD_TYPE_OSS.equals(uploadType)) {
//未指定目录，则用阿里云默认目录 upload
bizPath = "upload";
} else {
bizPath = "";
}
bizPath = "";
// if (CommonConstant.UPLOAD_TYPE_OSS.equals(uploadType)) {
// //未指定目录，则用阿里云默认目录 upload
// bizPath = "upload";
// } else {
// bizPath = "";
// }
}
OSSFile oSSFile = ossFileService.uploadLocal(file, bizPath,state,cut);
// OSSFile oSSFile = ossFileService.uploadLocalOfCos(file, bizPath,state,cut); // 切换cos
if (oConvertUtils.isNotEmpty(oSSFile)) {
result.setResult(oSSFile);
result.setSuccess(true);
@@ -331,8 +334,73 @@ public class CommonController {
}
}
}
}
/**
* word,excel预览 cos
*
* @param id 传入文件id
* @param request
* @param response
*/
// @GetMapping(value = "/downLoadFile")
public void downLoadFromCos(String id,HttpServletRequest request, HttpServletResponse response) {
if(StringUtils.isBlank(id)){
throw new JeroBootException("参数信息不全");
}
id = id.split("\\.")[0];
// 查询数据表数据是否存在
LambdaQueryWrapper<OSSFile> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(OSSFile::getId,id);
OSSFile ossFile = ossFileService.getOne(queryWrapper);
if( null == ossFile){
throw new JeroBootException("文件不存在");
}
InputStream inputStream = null;
OutputStream outputStream = null;
try {
String fileName = "";
//本地下载
String filePath = ossFile.getUrl();
if(!CosBootUtil.doesObjectExist(filePath)){
response.setStatus(404);
throw new RuntimeException("文件不存在..");
}
// 文件名称
fileName = ossFile.getFileName();
response.setContentType("application/force-download");// 设置强制下载不打开
response.addHeader("Content-Disposition", "attachment;fileName=" + new String(fileName.getBytes("UTF-8"),"iso-8859-1"));
outputStream = response.getOutputStream();
inputStream = CosBootUtil.download(filePath);
byte[] buf = new byte[1024];
int len;
while ((len = inputStream.read(buf)) > 0) {
outputStream.write(buf, 0, len);
}
response.flushBuffer();
} catch (IOException e) {
log.error("文件下载失败" + e.getMessage());
response.setStatus(404);
e.printStackTrace();
} finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException e) {
log.error(e.getMessage(), e);
}
}
if (outputStream != null) {
try {
outputStream.close();
} catch (IOException e) {
log.error(e.getMessage(), e);
}
}
}
}
/**
* 文件下载
*
jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/system/controller/SysPermissionController.java
+3
View File
@@ -7,6 +7,7 @@ import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.jero.common.constant.enums.CutEnum;
import lombok.extern.slf4j.Slf4j;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.apache.shiro.authz.annotation.RequiresRoles;
import com.jero.common.api.vo.Result;
import com.jero.common.constant.CommonConstant;
@@ -387,6 +388,7 @@ public class SysPermissionController {
*
* @return
*/
@RequiresPermissions("sys:role:auth")
@RequestMapping(value = "/queryRolePermission", method = RequestMethod.GET)
public Result<List<String>> queryRolePermission(@RequestParam(name = "roleId", required = true) String roleId) {
Result<List<String>> result = new Result<>();
@@ -405,6 +407,7 @@ public class SysPermissionController {
*
* @return
*/
@RequiresPermissions("sys:role:auth")
@RequestMapping(value = "/saveRolePermission", method = RequestMethod.POST)
//@RequiresRoles({ "admin" })
public Result<String> saveRolePermission(@RequestBody JSONObject json) {
jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/system/controller/SysRoleController.java
+1 -1
View File
@@ -368,7 +368,7 @@ public class SysRoleController {
* @param request
* @return
*/
@RequiresPermissions("sys:role:list")
@RequiresPermissions("sys:role:auth")
@RequestMapping(value = "/queryTreeList", method = RequestMethod.GET)
public Result<Map<String,Object>> queryTreeList(HttpServletRequest request) {
Result<Map<String,Object>> result = new Result<>();
jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/system/controller/SysUserController.java
+2 -2
View File
@@ -271,7 +271,7 @@ public class SysUserController {
* @param userId
* @return
*/
@RequiresPermissions("sys:user:edit")
@RequiresPermissions("sys:user:detail")
@ApiOperation(value = "用户管理-查询用户部门")
@RequestMapping(value = "/userDepartList", method = RequestMethod.GET)
public Result<List<DepartIdModel>> getUserDepartsList(@RequestParam(name = "userId", required = true) String userId) {
@@ -318,7 +318,7 @@ public class SysUserController {
* @param userid
* @return
*/
@RequiresPermissions("sys:user:edit")
@RequiresPermissions("sys:user:detail")
@ApiOperation(value = "用户管理-查询用户角色")
@RequestMapping(value = "/queryUserRole", method = RequestMethod.GET)
public Result<List<String>> queryUserRole(@RequestParam(name = "userid", required = true) String userid) {
jero-boot/jero-boot-single-startup/src/main/resources/application-dev.yml
+9 -1
View File
@@ -188,7 +188,7 @@ mybatis-plus:
call-setters-on-nulls: true
#jero专用配置
jero :
# 本地：local\Minio：minio\阿里云：alioss
# 本地：local\Minio：minio\阿里云：alioss\腾讯云 cos
uploadType: local
backUrl: http://localhost:3000
#拆分图片展示地址
@@ -212,6 +212,14 @@ jero :
secretKey: ??
bucketName: jeroos
staticDomain: ??
#腾讯云cos存储配置 测试桶
cos:
secretId: AKIDoTt3xbz82g4C4f05tRjlX2HGEwOZU8Xf
secretKey: uYv3vXsxnNgjW4MN9j36tPfeNp1xe5GB
region: ap-beijing
bucketName: dd-grp-laws-test-1253431691
appId: 1253431691
endPoint: https://dd-grp-laws-test-1253431691.cos.ap-beijing.myqcloud.com
# ElasticSearch 6设置
elasticsearch:
cluster-name: elasticsearch