智慧校园后端
blame | history | patch | commit | commitdiff | ignore whitespace
zrj
2024-08-30 d88ce8f933df6a533eddb06f0f2e340a2c697001 
 src/main/java/org/springblade/modules/mechanism/controller/MechanismController.java


@@ -16,7 +16,10 @@


 */


package org.springblade.modules.mechanism.controller;




import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;


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 +28,16 @@




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.springblade.modules.resource.entity.Oss;


import org.springblade.modules.resource.service.IOssService;


import org.springframework.web.bind.annotation.*;


import com.baomidou.mybatisplus.core.metadata.IPage;


import org.springblade.modules.mechanism.entity.Mechanism;


@@ -37,12 +45,18 @@


import org.springblade.modules.mechanism.service.IMechanismService;


import org.springblade.core.boot.ctrl.BladeController;


import springfox.documentation.annotations.ApiIgnore;


import sun.misc.BASE64Encoder;


//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 +71,13 @@


public class MechanismController extends BladeController {




   private final IMechanismService mechanismService;




   private final IOssService ossService;




   /**


    * 对象存储构建类


    */


   private final OssBuilder ossBuilder;




   /**


    * 详情


@@ -97,7 +118,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");


@@ -119,7 +140,7 @@


      String type = URLEncoder.encode(mechanism.getType(), "UTF-8");


      String mtype = URLEncoder.encode(mechanism.getMtype(), "UTF-8");


      String loutype = URLEncoder.encode(mechanism.getLoutype(), "UTF-8");


      String xtype = URLEncoder.encode(mechanism.getXtype(), "UTF-8");


      String campus = URLEncoder.encode(mechanism.getCampus(), "UTF-8");


      String heading = URLEncoder.encode(mechanism.getHeading(), "UTF-8");


      String pitch = URLEncoder.encode(mechanism.getPitch(), "UTF-8");


      String roll = URLEncoder.encode(mechanism.getRoll(), "UTF-8");


@@ -136,7 +157,7 @@


         "&type=" + type +


         "&mtype=" + mtype +


         "&loutype=" + loutype +


         "&xtype=" + xtype +


         "&campus=" + campus +


         "&heading=" + heading +


         "&pitch=" + pitch +


         "&roll=" + roll +


@@ -144,10 +165,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 +208,67 @@


   }






   /**


    * 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 {


      QueryWrapper<Oss> wrapper = new QueryWrapper<>();


      wrapper.eq("is_deleted",0);


      Oss one = ossService.getOne(wrapper);


      //填写你文件上传的地址以及相应信息


//      String url = "http://106.225.193.35:2081";


//      String access = "zhbaadmin";


//      String secret = "zhbapassword";


//      String bucket = "zhxy";


      String url = one.getEndpoint();


      String access = one.getAccessKey();


      String secret = one.getSecretKey();


      String bucket = one.getBucketName();


      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;


      String urls = url + "/"+ bucket + "/" + newName;


      //返回


      return urls;


   }




}