java Minio使用(参考一些文章,自己使用的工具类)

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;

import org.springframework.web.multipart.MultipartFile;

import com.cetcnav.lbs.base.util.CommonUtils;
import com.cetcnav.lbs.base.util.DateUtil;
import com.cetcnav.lbs.pis.ws.RestResponse;
import com.cetcnav.lbs.pis.ws.WsResponse.ResultCode;

import io.minio.BucketExistsArgs;
import io.minio.GetObjectArgs;
import io.minio.MakeBucketArgs;
import io.minio.MinioClient;
import io.minio.PutObjectArgs;
import io.minio.StatObjectArgs;
import io.minio.StatObjectResponse;
import io.minio.errors.MinioException;

public class MinioUtils {

	/**
	 * 上传一个文件
	 * 
	 * @param bucketName
	 *            桶名称
	 * @param sourceName
	 *            上传文件路径名
	 * @param targetName
	 *            保存名称
	 * @throws IOException
	 * @throws NoSuchAlgorithmException
	 * @throws InvalidKeyException
	 */
	public static RestResponse uploadFile(InputStream source, String targetName)
			throws IOException, NoSuchAlgorithmException, InvalidKeyException {
		RestResponse response = new RestResponse();
		try {

			// 获取endpoint等信息
			String endpoint = MinIOPropertiesUtils.getString("minio.endpoint");
			String accessKey = MinIOPropertiesUtils.getString("minio.accessKey");
			String secretKey = MinIOPropertiesUtils.getString("minio.secretKey");
			String bucketName = MinIOPropertiesUtils.getString("bucket.name");

			// Create a minioClient with the MinIO server playground, its access
			// key and secret key.
			MinioClient minioClient = MinioClient.builder().endpoint(endpoint).credentials(accessKey, secretKey)
					.build();
			// Make bucketName bucket if not exist.
			boolean found = minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build());
			if (!found) {
				// Make a new bucket called
				minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
			} else {
				System.out.println("Bucket " + bucketName + " already exists.");
			}
			// 将source上传到bucketName桶中,重命名为tatgetName,可以包含多层路径
			minioClient.putObject(PutObjectArgs.builder().bucket(bucketName).object(targetName)
					.stream(source, source.available(), -1).build());
			response.setCode(ResultCode.SUCCESS);
			response.setMessage("上传成功");
			// 返回路径包含桶名
			// response.setData("/"+bucketName+"/"+targetName);
			// 返回路径不包含桶名
			response.setData("/" + bucketName + "/" + targetName);
		} catch (MinioException e) {
			System.out.println("Error occurred: " + e.toString());
			// System.out.println("HTTP trace: " + e.httpTrace());
			// CommonUtils.printStackTrack(e);
			response.setCode(ResultCode.ERROR);
			response.setMessage("上传失败");
		}
		return response;
	}
	
	/**
	 * 上传多个文件
	 * 
	 */
	public static RestResponse uploadFiles(String departmentCode, String identityCard, MultipartFile[] files) {

		RestResponse response = new RestResponse();
		StringBuffer targetUrl = new StringBuffer();

		try {
			StringBuffer targetName = new StringBuffer(MinIOPropertiesUtils.getString("bucket.visitorMaterial"));
			targetName.append(departmentCode + "/");
			targetName.append(DateUtil.date2Str(DateUtil.getCurrentTime(), "yyyy") + "/");
			targetName.append(DateUtil.date2Str(DateUtil.getCurrentTime(), "MM") + "/");
			targetName.append(DateUtil.date2Str(DateUtil.getCurrentTime(), "dd") + "/");
			targetName.append(identityCard);
			// 循环上传
			for (int i = 0; i < files.length; i++) {
				MultipartFile file = files[i];
				InputStream sourceStream = file.getInputStream();
				StringBuffer targetFile = new StringBuffer();
				// 生成目标路径
				targetFile.append(targetName);
				targetFile.append(CommonUtils.UNDERLINE + (i + 1));
				targetFile.append(file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")));
				// 上传
				RestResponse res = MinioUtils.uploadFile(sourceStream, targetFile.toString());
				if (res.getCode() == ResultCode.SUCCESS) {
					targetUrl.append(res.getData());
					// 每个返回路径分号分隔
					if (i < files.length - 1) {
						targetUrl.append(";");
					}

				}
				if (res.getCode() == ResultCode.ERROR) {
					response.setCode(ResultCode.ERROR);
					response.setMessage("上传失败");
					return response;
				}
			}
			response.setCode(ResultCode.SUCCESS);
			response.setMessage("上传成功");
			response.setData(targetUrl);

		} catch (Exception e) {

			response.setCode(ResultCode.ERROR);
			response.setMessage("上传失败");
		}
		return response;
	}

	/**
	 * 
	 * @param fileName
	 *            minio存储路径,文件第一个/前是桶名,需要先截取出去
	 * @return 文件流
	 */
	public static InputStream downloadFile(String fileUrl) {
		InputStream stream = null;
		RestResponse response = new RestResponse();
		try {
			// 获取endpoint等信息
			String endpoint = MinIOPropertiesUtils.getString("minio.endpoint");
			String accessKey = MinIOPropertiesUtils.getString("minio.accessKey");
			String secretKey = MinIOPropertiesUtils.getString("minio.secretKey");
			String bucketName = MinIOPropertiesUtils.getString("bucket.name");
			// 从fileUrl中截取相对路径
			String fileName = (fileUrl.split(bucketName))[1];

			// Create a minioClient with the MinIO server playground, its access
			// key and secret key.
			MinioClient minioClient = MinioClient.builder().endpoint(endpoint).credentials(accessKey, secretKey)
					.build();

			StatObjectResponse objectStat = minioClient
					.statObject(StatObjectArgs.builder().bucket(bucketName).object(fileName).build());
			if (objectStat == null) {
				response.setCode(ResultCode.ERROR);
				response.setMessage("文件不存在");
				return null;
			}

			stream = minioClient.getObject(GetObjectArgs.builder().bucket(bucketName).object(fileName).build());

		} catch (Exception e) {
			response.setCode(ResultCode.ERROR);
			response.setMessage("文件不存在");
		}
		return stream;
	}

	/**
	 * 
	 * @param fileName
	 *            minio存储路径,文件第一个/前是桶名,需要先截取出去
	 * @return 文件流
	 */
	public static byte[] downloadFileToByte(String fileUrl) {
		InputStream stream = null;
		RestResponse response = new RestResponse();
		try {
			// 获取endpoint等信息
			String endpoint = MinIOPropertiesUtils.getString("minio.endpoint");
			String accessKey = MinIOPropertiesUtils.getString("minio.accessKey");
			String secretKey = MinIOPropertiesUtils.getString("minio.secretKey");
			String bucketName = MinIOPropertiesUtils.getString("bucket.name");
			// 从fileUrl中截取相对路径
			String fileName = (fileUrl.split(bucketName))[1];

			// Create a minioClient with the MinIO server playground, its access
			// key and secret key.
			MinioClient minioClient = MinioClient.builder().endpoint(endpoint).credentials(accessKey, secretKey)
					.build();

			StatObjectResponse objectStat = minioClient
					.statObject(StatObjectArgs.builder().bucket(bucketName).object(fileName).build());
			if (objectStat == null) {
				response.setCode(ResultCode.ERROR);
				response.setMessage("文件不存在");
				return null;
			}

			stream = minioClient.getObject(GetObjectArgs.builder().bucket(bucketName).object(fileName).build());

			ByteArrayOutputStream os = new ByteArrayOutputStream();
			byte[] buffer = new byte[1024];
			int n = 0;
			while (-1 != (n = stream.read(buffer))) {
				os.write(buffer, 0, n);
			}
			stream.close();
			return os.toByteArray();
		} catch (Exception e){
			response.setCode(ResultCode.ERROR);
			response.setMessage("文件不存在");
			return null;
		}
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值