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/mechanism/controller/MechanismController.java | 85 ++++++++++++++++++++++++++++++++++++++++--
1 files changed, 80 insertions(+), 5 deletions(-)
diff --git a/src/main/java/org/springblade/modules/mechanism/controller/MechanismController.java b/src/main/java/org/springblade/modules/mechanism/controller/MechanismController.java
index 2a8743f..1a7caf9 100644
--- a/src/main/java/org/springblade/modules/mechanism/controller/MechanismController.java
+++ b/src/main/java/org/springblade/modules/mechanism/controller/MechanismController.java
@@ -17,6 +17,8 @@
package org.springblade.modules.mechanism.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;
@@ -25,11 +27,14 @@
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.resource.builder.oss.OssBuilder;
import org.springframework.web.bind.annotation.*;
import com.baomidou.mybatisplus.core.metadata.IPage;
import org.springblade.modules.mechanism.entity.Mechanism;
@@ -39,10 +44,16 @@
import springfox.documentation.annotations.ApiIgnore;
import sun.misc.BASE64Encoder;
+import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
+import java.security.InvalidKeyException;
+import java.security.NoSuchAlgorithmException;
+import java.util.HashMap;
+import java.util.List;
import java.util.Map;
+import java.util.UUID;
/**
* 控制器
@@ -57,6 +68,11 @@
public class MechanismController extends BladeController {
private final IMechanismService mechanismService;
+
+ /**
+ * 对象存储构建类
+ */
+ private final OssBuilder ossBuilder;
/**
* 详情
@@ -97,7 +113,7 @@
@PostMapping("/save")
@ApiOperationSupport(order = 4)
@ApiOperation(value = "新增", notes = "传入mechanism")
- public R save(@Valid @RequestBody Mechanism mechanism) throws IOException, WriterException {
+ public R save(@Valid @RequestBody Mechanism mechanism) throws IOException, WriterException, ServerException, InsufficientDataException, NoSuchAlgorithmException, InternalException, InvalidResponseException, XmlParserException, InvalidKeyException, ErrorResponseException {
mechanism.setPitch("-90");
mechanism.setHeading("0");
mechanism.setRoll("0");
@@ -144,10 +160,14 @@
"&videourl=" + videourl;
//生成标注二维码
byte[] qrCodeImage = QRCodeUtil.getQRCodeImage(content, 350, 350);
- String encode = new BASE64Encoder().encode(qrCodeImage);
- //设置二维码
- String a = "data:image/png;base64," + encode;
- mechanism.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();
+ //设置二维码地址
+ mechanism.setCodeurl(upload);
return R.status(mechanismService.save(mechanism));
}
@@ -183,4 +203,59 @@
}
+ /**
+ * code url 转换
+ */
+ @GetMapping("/url")
+ public void url() throws IOException, ServerException, InsufficientDataException, NoSuchAlgorithmException, InternalException, InvalidResponseException, XmlParserException, InvalidKeyException, ErrorResponseException {
+ List<Mechanism> list = mechanismService.selectAllList();
+ for (Mechanism mechanism : list) {
+ int index = mechanism.getCodeurl().lastIndexOf(",") +1;
+ String substring = mechanism.getCodeurl().substring(index);
+ byte[] qrCodeImage = Base64Util.decode(substring);
+ ByteArrayInputStream inputStream = new ByteArrayInputStream(qrCodeImage);
+ String url = upload(inputStream);
+ mechanism.setCodeurl(url);
+ mechanismService.updateById(mechanism);
+ //关闭流
+ 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