~chenjiahe/hx_common.git

chenjiahe

2022-07-12 3f40721b45fda889f1e48598ef1aa833d09a271e

提交 \| 用户 \| age
7b319c	1	package com.hx.util;
C	2
cd5e48	3	import com.hx.exception.TipsException;
7b319c	4	import com.qcloud.cos.COSClient;
C	5	import com.qcloud.cos.ClientConfig;
	6	import com.qcloud.cos.auth.BasicCOSCredentials;
	7	import com.qcloud.cos.auth.COSCredentials;
cd5e48	8	import com.qcloud.cos.exception.CosClientException;
C	9	import com.qcloud.cos.exception.CosServiceException;
	10	import com.qcloud.cos.http.HttpProtocol;
	11	import com.qcloud.cos.model.GetObjectRequest;
ccd95b	12	import com.qcloud.cos.model.ObjectMetadata;
7b319c	13	import com.qcloud.cos.model.PutObjectRequest;
C	14	import com.qcloud.cos.model.PutObjectResult;
	15	import com.qcloud.cos.region.Region;
cd5e48	16	import com.qcloud.cos.transfer.Download;
C	17	import com.qcloud.cos.transfer.TransferManager;
7b319c	18	import org.springframework.web.multipart.MultipartFile;
C	19
	20	import java.io.File;
	21	import java.io.IOException;
cd5e48	22	import java.util.concurrent.ExecutorService;
C	23	import java.util.concurrent.Executors;
7b319c	24
C	25	/**腾讯云 COS
	26	* @author ChenJiaHe
	27	* @date 2020-12-03
	28	*/
	29	public class COSUtil {
	30
	31	/** 后端调用上传图片
	32	* @param key 上传路径（包括图片名称和和后缀），指定要上传到 COS 上对象键
	33	* @param localFile
	34	* @param secretId 用户id
	35	* @param secretKey 用户秘钥
	36	* @param regionName 存在域，参考腾讯云
	37	* @param bucketName 指定要上传到的存储桶
	38	* @return
	39	* @throws IOException
	40	*/
ccd95b	41	public static String uploadImg(String key,MultipartFile localFile,String secretId, String secretKey,String regionName,String bucketName) throws IOException {
7b319c	42
C	43	// 1 初始化用户身份信息（secretId, secretKey）。
	44	COSCredentials cred = new BasicCOSCredentials(secretId, secretKey);
	45	// 2 设置 bucket 的区域, COS 地域的简称请参照 https://cloud.tencent.com/document/product/436/6224
	46	// clientConfig 中包含了设置 region, https(默认 http), 超时, 代理等 set 方法, 使用可参见源码或者常见问题 Java SDK 部分。
	47	Region region = new Region(regionName);
	48	ClientConfig clientConfig = new ClientConfig(region);
	49	// 3 生成 cos 客户端。
	50	COSClient cosClient = new COSClient(cred, clientConfig);
	51
	52	//开始上传
ccd95b	53	ObjectMetadata objectMetadata = new ObjectMetadata();
C	54	// 设置输入流长度为500
ee9728	55	objectMetadata.setContentLength(localFile.getSize());
ccd95b	56	// 设置 Content type, 默认是 application/octet-stream,对于本地文件上传，默认根据本地文件的后缀进行映射
C	57	// ，例如 jpg 文件映射为image/jpeg对于流式上传默认是 application/octet-stream
	58	//objectMetadata.setContentType("application/pdf");
	59	PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, key,localFile.getInputStream(),objectMetadata);
7b319c	60	PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
C	61
	62	//拼接路径
	63	StringBuilder imgUrl = new StringBuilder();
486aaa	64	imgUrl.append("https://"+bucketName+".cos."+regionName+".myqcloud.com");
7b319c	65	if(key.startsWith("/")){
C	66	imgUrl.append(key);
	67	}else{
	68	imgUrl.append("/"+key);
	69	}
a99c1c	70	cosClient.shutdown();
7b319c	71	return imgUrl.toString();
C	72
	73	}
	74
49f28b	75
C	76	/** 后端调用上传图片
	77	* @param key 上传路径（包括图片名称和和后缀），指定要上传到 COS 上对象键
	78	* @param localFile
	79	* @param secretId 用户id
	80	* @param secretKey 用户秘钥
	81	* @param regionName 存在域，参考腾讯云
	82	* @param bucketName 指定要上传到的存储桶
	83	* @return
	84	* @throws IOException
	85	*/
	86	public static String uploadImg(String key,File localFile,String secretId, String secretKey,String regionName,String bucketName) throws IOException {
	87
	88	// 1 初始化用户身份信息（secretId, secretKey）。
	89	COSCredentials cred = new BasicCOSCredentials(secretId, secretKey);
	90	// 2 设置 bucket 的区域, COS 地域的简称请参照 https://cloud.tencent.com/document/product/436/6224
	91	// clientConfig 中包含了设置 region, https(默认 http), 超时, 代理等 set 方法, 使用可参见源码或者常见问题 Java SDK 部分。
	92	Region region = new Region(regionName);
	93	ClientConfig clientConfig = new ClientConfig(region);
	94	// 3 生成 cos 客户端。
	95	COSClient cosClient = new COSClient(cred, clientConfig);
	96
	97	// 设置 Content type, 默认是 application/octet-stream,对于本地文件上传，默认根据本地文件的后缀进行映射
	98	// ，例如 jpg 文件映射为image/jpeg对于流式上传默认是 application/octet-stream
	99	//objectMetadata.setContentType("application/pdf");
	100	PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, key, localFile);
	101	PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
	102
	103	//拼接路径
	104	StringBuilder imgUrl = new StringBuilder();
	105	imgUrl.append("https://"+bucketName+".cos."+regionName+".myqcloud.com");
	106	if(key.startsWith("/")){
	107	imgUrl.append(key);
	108	}else{
	109	imgUrl.append("/"+key);
	110	}
a99c1c	111	cosClient.shutdown();
49f28b	112	return imgUrl.toString();
C	113
	114	}
	115
cd5e48	116	/**下载文件
C	117	* @param key 上传路径（包括图片名称和和后缀），指定要上传到 COS 上对象键
	118	* @param secretId 用户id
	119	* @param secretKey 用户秘钥
	120	* @param regionName 存在域，参考腾讯云
	121	* @param bucketName 指定要上传到的存储桶
	122	*/
	123	public static File download(String key,String secretId, String secretKey,String regionName,String bucketName){
	124
	125	// 使用高级接口必须先保证本进程存在一个 TransferManager 实例，如果没有则创建
	126	// 详细代码参见本页：高级接口 -> 创建 TransferManager
	127	TransferManager transferManager = createTransferManager( secretId, secretKey,regionName);
	128	// 本地文件路径
	129	File downloadFile = null;
	130	try {
	131	GetObjectRequest getObjectRequest;
	132	System.out.println("key:"+key);
	133	System.out.println("bucketName:"+bucketName);
	134	System.out.println("regionName:"+regionName);
	135	System.out.println("regionName:"+regionName+".myqcloud.com");
	136	//截取文件名称
	137	String[] datas = key.split(regionName+".myqcloud.com/");
	138	if(datas.length ==1){
	139	getObjectRequest = new GetObjectRequest(bucketName, datas[0]);
	140	}else if(datas.length ==2){
	141	System.out.println("datas[1]:"+datas[1]);
	142	getObjectRequest = new GetObjectRequest(bucketName, datas[1]);
	143	}else{
	144	throw new TipsException("文件路径错误【key】");
	145	}
	146
	147	////生成临时文件
	148	//获取后缀名称
	149	String suffix = key.substring(key.lastIndexOf("."));
	150	downloadFile = File.createTempFile("temp", suffix);
	151
	152	// 返回一个异步结果 Donload, 可同步的调用 waitForCompletion 等待下载结束, 成功返回 void, 失败抛出异常
	153	Download download = transferManager.download(getObjectRequest, downloadFile);
	154	download.waitForCompletion();
	155	} catch (CosServiceException e) {
	156	e.printStackTrace();
	157	} catch (CosClientException e) {
	158	e.printStackTrace();
	159	} catch (InterruptedException e) {
	160	e.printStackTrace();
	161	}catch (IOException e){
	162	e.printStackTrace();
	163	}finally {
	164	// 确定本进程不再使用 transferManager 实例之后，关闭之
	165	// 详细代码参见本页：高级接口 -> 关闭 TransferManager
	166	shutdownTransferManager(transferManager);
	167	}
	168
	169	return downloadFile;
	170	}
	171
	172	/**
	173	* 创建 TransferManager 实例，这个实例用来后续调用高级接口
	174	* @param secretId
	175	* @param secretKey
	176	* @return
	177	*/
	178	public static TransferManager createTransferManager(String secretId, String secretKey,String regionName) {
	179	// 创建一个 COSClient 实例，这是访问 COS 服务的基础实例。
	180	// 详细代码参见本页: 简单操作 -> 创建 COSClient
	181	COSClient cosClient = createCOSClient( secretId, secretKey,regionName);
	182
	183	// 自定义线程池大小，建议在客户端与 COS 网络充足（例如使用腾讯云的 CVM，同地域上传 COS）的情况下，设置成16或32即可，可较充分的利用网络资源
	184	// 对于使用公网传输且网络带宽质量不高的情况，建议减小该值，避免因网速过慢，造成请求超时。
	185	ExecutorService threadPool = Executors.newFixedThreadPool(32);
	186
	187	// 传入一个 threadpool, 若不传入线程池，默认 TransferManager 中会生成一个单线程的线程池。
	188	TransferManager transferManager = new TransferManager(cosClient, threadPool);
	189
	190	return transferManager;
	191	}
	192
	193	/**
	194	* 关闭管理
	195	* @param transferManager
	196	*/
	197	public static void shutdownTransferManager(TransferManager transferManager) {
	198	// 指定参数为 true, 则同时会关闭 transferManager 内部的 COSClient 实例。
	199	// 指定参数为 false, 则不会关闭 transferManager 内部的 COSClient 实例。
	200	transferManager.shutdownNow(true);
	201	}
	202
	203	/**
	204	* 创建 COSClient 实例，这个实例用来后续调用请求
	205	* @param secretId
	206	* @param secretKey
	207	* @return
	208	*/
	209	public static COSClient createCOSClient(String secretId, String secretKey,String regionName) {
	210	// 设置用户身份信息。
	211	// SECRETID 和 SECRETKEY 请登录访问管理控制台 https://console.cloud.tencent.com/cam/capi 进行查看和管理
	212	COSCredentials cred = new BasicCOSCredentials(secretId, secretKey);
	213
	214	// ClientConfig 中包含了后续请求 COS 的客户端设置：
	215	ClientConfig clientConfig = new ClientConfig();
	216
	217	// 设置 bucket 的地域
	218	// COS_REGION 请参照 https://cloud.tencent.com/document/product/436/6224
	219	clientConfig.setRegion(new Region(regionName));
	220
	221	// 设置请求协议, http 或者 https
	222	// 5.6.53 及更低的版本，建议设置使用 https 协议
	223	// 5.6.54 及更高版本，默认使用了 https
	224	clientConfig.setHttpProtocol(HttpProtocol.https);
	225
	226	// 以下的设置，是可选的：
	227
	228	// 设置 socket 读取超时，默认 30s
	229	clientConfig.setSocketTimeout(300*1000);
	230	// 设置建立连接超时，默认 30s
	231	clientConfig.setConnectionTimeout(30*1000);
	232
	233	// 如果需要的话，设置 http 代理，ip 以及 port
	234	//clientConfig.setHttpProxyIp("httpProxyIp");
	235	//clientConfig.setHttpProxyPort(80);
	236
	237	// 生成 cos 客户端。
	238	return new COSClient(cred, clientConfig);
	239	}
	240
	241
7b319c	242	}