From 79f4827cbd2fa4a3a31cbf9263ca23df0b633413 Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Thu, 12 May 2022 18:38:43 +0800
Subject: [PATCH] 农产品记录新增,图片上传接口新增

---
 src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java |   59 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 59 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java b/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
index f1b57b4..e048470 100644
--- a/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
+++ b/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
@@ -16,6 +16,8 @@
  */
 package org.springblade.modules.resource.endpoint;
 
+import io.minio.*;
+import io.minio.errors.*;
 import io.swagger.annotations.Api;
 import lombok.AllArgsConstructor;
 import lombok.SneakyThrows;
@@ -33,6 +35,14 @@
 import org.springblade.modules.resource.service.IAttachService;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.security.InvalidKeyException;
+import java.security.NoSuchAlgorithmException;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.UUID;
 
 /**
  * 对象存储端点
@@ -150,6 +160,55 @@
 	}
 
 	/**
+	 * 文件上传,自定义上传
+	 *
+	 * @param file 图片对象
+	 */
+	@PostMapping("put-files")
+	public R putFiles(@RequestParam MultipartFile file) throws IOException, ServerException, InsufficientDataException, InternalException, InvalidResponseException, InvalidKeyException, NoSuchAlgorithmException, XmlParserException, ErrorResponseException {
+		//填写你文件上传的地址以及相应信息
+		String url = "http://61.131.136.25:2081";
+		String access = "zhbaadmin";
+		String secret = "zhbapassword";
+		String bucket = "zhba";
+		MinioClient minioClient =
+			MinioClient.builder()
+				.endpoint(url)
+				.credentials(access, secret)
+				.build();
+		// 检查存储桶是否已经存在
+		boolean isExist = minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucket).build());
+		if (!isExist) {
+			// 创建一个名为zip的存储桶,用于zip文件。
+			minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucket).build());
+			minioClient.setBucketPolicy(SetBucketPolicyArgs.builder().bucket(bucket).build());
+		}
+		String fileName = file.getOriginalFilename();
+		String newName = "upload/picture/" + UUID.randomUUID().toString().replaceAll("-", "")
+			+ fileName.substring(fileName.lastIndexOf("."));
+		InputStream in = file.getInputStream();
+		String[] split = newName.split("/");
+		//创建头部信息
+		Map<String, String> headers = new HashMap<>(1 << 2);
+		//添加自定义内容类型
+		headers.put("Content-Type", "application/octet-stream");
+		//上传
+		minioClient.putObject(
+			PutObjectArgs.builder().bucket(bucket).object(newName).stream(
+				in, in.available(), -1)
+				.headers(headers)
+				.build());
+		in.close();
+		String urls = "http://61.131.136.25:2081/zhba/" + newName;
+		//数据封装
+		Map<String, Object> map = new HashMap<>(2);
+		map.put("name", newName);
+		map.put("url", urls);
+		//返回
+		return R.data(map);
+	}
+
+	/**
 	 * 上传文件
 	 *
 	 * @param fileName 存储桶对象名称

--
Gitblit v1.9.3