From 0cc95376f2af6fa5dfbe63ff79cbaf94674531e0 Mon Sep 17 00:00:00 2001
From: zhongrj <123456>
Date: Mon, 24 Oct 2022 13:34:01 +0800
Subject: [PATCH] 新增接口替换二维码地址,数据新增时修改二维码(保存地址)

---
 src/main/java/org/springblade/modules/architecture/controller/ArchitectureController.java |   86 +++++++++++++++++++++++++++++++++++++++----
 1 files changed, 78 insertions(+), 8 deletions(-)

diff --git a/src/main/java/org/springblade/modules/architecture/controller/ArchitectureController.java b/src/main/java/org/springblade/modules/architecture/controller/ArchitectureController.java
index a94bc32..a40f487 100644
--- a/src/main/java/org/springblade/modules/architecture/controller/ArchitectureController.java
+++ b/src/main/java/org/springblade/modules/architecture/controller/ArchitectureController.java
@@ -17,6 +17,8 @@
 package org.springblade.modules.architecture.controller;
 
 import com.google.zxing.WriterException;
+import io.minio.*;
+import io.minio.errors.*;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import io.swagger.annotations.ApiParam;
@@ -26,15 +28,19 @@
 
 import javax.validation.Valid;
 
+import org.springblade.common.utils.Base64Util;
 import org.springblade.common.utils.QRCodeUtil;
 import org.springblade.core.mp.support.Condition;
 import org.springblade.core.mp.support.Query;
+import org.springblade.core.oss.model.BladeFile;
 import org.springblade.core.tool.api.R;
 import org.springblade.core.tool.utils.Func;
 import org.springblade.modules.life.entity.Life;
 import org.springblade.modules.life.service.ILifeService;
 import org.springblade.modules.mechanism.entity.Mechanism;
 import org.springblade.modules.mechanism.service.IMechanismService;
+import org.springblade.modules.resource.builder.oss.OssBuilder;
+import org.springblade.modules.resource.service.IOssService;
 import org.springframework.core.io.support.ResourceRegion;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
@@ -46,10 +52,10 @@
 import springfox.documentation.annotations.ApiIgnore;
 import sun.misc.BASE64Encoder;
 
-import java.io.File;
-import java.io.IOException;
-import java.io.UnsupportedEncodingException;
+import java.io.*;
 import java.net.URLEncoder;
+import java.security.InvalidKeyException;
+import java.security.NoSuchAlgorithmException;
 import java.util.*;
 import java.util.stream.Collectors;
 
@@ -69,6 +75,10 @@
 	private final IArchitectureService architectureService;
 	private final IMechanismService mechanismService;
 	private final ILifeService lifeService;
+	/**
+	 * 对象存储构建类
+	 */
+	private final OssBuilder ossBuilder;
 
 	/**
 	 * 详情
@@ -110,7 +120,7 @@
 	@PostMapping("/save")
 	@ApiOperationSupport(order = 4)
 	@ApiOperation(value = "新增", notes = "传入architecture")
-	public R save(@Valid @RequestBody Architecture architecture) throws IOException, WriterException {
+	public R save(@Valid @RequestBody Architecture architecture) throws IOException, WriterException, ServerException, InsufficientDataException, NoSuchAlgorithmException, InternalException, InvalidResponseException, XmlParserException, InvalidKeyException, ErrorResponseException {
 		architecture.setPitch("-90");
 		architecture.setHeading("0");
 		architecture.setRoll("0");
@@ -172,10 +182,14 @@
 			"&videourl=" + videourl;
 		//生成标注二维码
 		byte[] qrCodeImage = QRCodeUtil.getQRCodeImage(content, 350, 350);
-		String encode = new BASE64Encoder().encode(qrCodeImage);
-		//设置二维码
-		String a = "data:image/png;base64," + encode;
-		architecture.setCodeurl(a);
+//		String encode = new BASE64Encoder().encode(qrCodeImage);
+//		//设置二维码
+//		String a = "data:image/png;base64," + encode;
+		ByteArrayInputStream stream = new ByteArrayInputStream(qrCodeImage);
+		String upload = upload(stream);
+		stream.close();
+		//设置二维码地址
+		architecture.setCodeurl(upload);
 		return R.status(architectureService.save(architecture));
 	}
 
@@ -405,4 +419,60 @@
 //		}
 		return R.data(list2);
 	}
+
+
+	/**
+	 * code url 转换
+	 */
+	@GetMapping("/url")
+	public void url() throws IOException, ServerException, InsufficientDataException, NoSuchAlgorithmException, InternalException, InvalidResponseException, XmlParserException, InvalidKeyException, ErrorResponseException {
+		List<Architecture> list = architectureService.selectArchALL();
+		for (Architecture architecture : list) {
+			int index = architecture.getCodeurl().lastIndexOf(",") +1;
+			String substring = architecture.getCodeurl().substring(index);
+			byte[] qrCodeImage = Base64Util.decode(substring);
+			ByteArrayInputStream inputStream = new ByteArrayInputStream(qrCodeImage);
+			String url = upload(inputStream);
+			architecture.setCodeurl(url);
+			architectureService.updateById(architecture);
+			//关闭流
+			inputStream.close();
+		}
+	}
+
+	private String upload(ByteArrayInputStream in) throws IOException, InvalidKeyException, InvalidResponseException, InsufficientDataException, NoSuchAlgorithmException, ServerException, InternalException, XmlParserException, ErrorResponseException {
+		//填写你文件上传的地址以及相应信息
+		String url = "http://106.225.193.35:2081";
+		String access = "zhbaadmin";
+		String secret = "zhbapassword";
+		String bucket = "zhxy";
+		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 newName = "upload/picture/" + UUID.randomUUID().toString().replaceAll("-", "") + ".png";
+		//创建头部信息
+		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());
+
+		//文件推送
+		String urls = "https://dev.jxpskj.com:8023/zhxy/"+newName;
+		//返回
+		return urls;
+	}
 }

--
Gitblit v1.9.3