blame | history | patch | commit | commitdiff | ignore whitespace
Administrator
2021-09-02 2b5ca6cea92ff6a67582c3331a0704480e63b9b4 
 src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java


@@ -16,40 +16,31 @@


 */


package org.springblade.modules.resource.endpoint;




import com.alibaba.fastjson.JSON;


import io.minio.*;


import io.minio.errors.*;


import io.swagger.annotations.Api;


import javafx.scene.Parent;


import lombok.AllArgsConstructor;


import lombok.SneakyThrows;


import org.apache.http.HttpEntity;


import org.apache.http.client.methods.CloseableHttpResponse;


import org.apache.http.client.methods.HttpPost;


import org.apache.http.entity.ContentType;


import org.apache.http.entity.StringEntity;


import org.apache.http.impl.client.CloseableHttpClient;


import org.apache.http.impl.client.HttpClients;


import org.apache.http.util.EntityUtils;


import org.springblade.common.config.FileConfig;


import org.springblade.common.constant.FtpConstant;


import org.springblade.common.utils.arg;


import org.springblade.common.utils.FileUtil;


import org.springblade.core.launch.constant.AppConstant;


import org.springblade.core.oss.model.BladeFile;


import org.springblade.core.oss.model.OssFile;


import org.springblade.core.secure.BladeUser;


import org.springblade.core.secure.annotation.PreAuth;


import org.springblade.core.secure.utils.AuthUtil;


import org.springblade.core.tenant.annotation.NonDS;


import org.springblade.core.tool.api.R;


import org.springblade.core.tool.constant.RoleConstant;


import org.springblade.core.tool.utils.FileUtil;


import org.springblade.core.tool.utils.Func;


import org.springblade.modules.FTP.FtpUtil;


import org.springblade.modules.resource.builder.oss.OssBuilder;


import org.springblade.modules.resource.entity.Attach;


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


import org.springblade.modules.system.entity.User;


import org.springblade.modules.system.service.IUserService;


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


import org.springframework.web.multipart.MultipartFile;




import java.io.File;


import java.io.FileInputStream;


import java.io.IOException;


@@ -57,10 +48,8 @@


import java.security.InvalidKeyException;


import java.security.NoSuchAlgorithmException;


import java.text.SimpleDateFormat;


import java.util.Date;


import java.util.HashMap;


import java.util.Map;


import java.util.UUID;


import java.util.*;


import java.util.regex.Pattern;




import static org.springblade.common.constant.FtpConstant.*;




@@ -85,6 +74,9 @@


    * 附件表服务


    */


   private final IAttachService attachService;






   private final IUserService userService;




   /**


    * 创建存储桶


@@ -280,6 +272,104 @@






   /**


    * 文件上传,zip


    *


    * @param file 图片对象


    */


   @PostMapping("put-file-zip")


   public R putFileZip(@RequestParam MultipartFile file) throws Exception {


      Map<String, Object> map = new HashMap<>(1);


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


      String url = "http://223.82.109.183: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 namePath = fileName.substring(0,fileName.lastIndexOf("."));


      String fileType = fileName.substring(fileName.lastIndexOf(".") + 1).toLowerCase(Locale.US);


      //判断文件是不是zip类型


      if(!fileType.equals("zip")){


         map.put("data","上传文件类型不符!");


         return R.data(map);


      }


      //FileConfig.localtion是配置文件和config类生产的，测试demo可以直接把FileConfig.localtion替换成D:/test


      String uuid = UUID.randomUUID().toString();


      String desPath = FileConfig.localtion + File.separator + uuid.replaceAll("-", "");


//      String desPath = "D:/test" + File.separator + uuid.replaceAll("-", "");




      //遗漏了这个代码，在本地测试环境不会出问题，在服务器上一定会报没有找到文件的错误


      String savePath = FileConfig.localtion + File.separator;


      FileUtil fileUtil = new FileUtil();


      //解压zip文件


      FileUtil.unZip(file, desPath,savePath);




      List<MultipartFile> fileList = new ArrayList<>();


      //获取图片文件


      fileList = fileUtil.getSubFiles(desPath+File.separator+namePath,fileList);


      for (MultipartFile multipartFile : fileList){


         if(multipartFile.getName().toLowerCase().endsWith(".png") || multipartFile.getName().toLowerCase().endsWith(".jpg")) {


            String newName = "upload/picture/" + UUID.randomUUID().toString().replaceAll("-", "") + multipartFile.getName().substring(multipartFile.getName().lastIndexOf("."));


            InputStream in = multipartFile.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());




            String urls = "http://223.82.109.183:2081/zhba/" + newName;


            //取出身份证号,查询用户信息，更新用户信息


            String pictrueName = multipartFile.getName().substring(0, multipartFile.getName().lastIndexOf("."));


//            String regex ="([1-9]\\d{5}(18|19|20)\\d{2}((0[1-9])|(10|11|12))(([0-2][1-9])|10|20|30|31)\\d{3}[0-9Xx])|([1-9]\\d{5}\\d{2}((0[1-9])|(10|11|12))(([0-2][1-9])|10|20|30|31)\\d{3})";


            String regex ="[\u4e00-\u9fa5]";


            Pattern compile = Pattern.compile(regex);


            String idCardNo = compile.matcher(pictrueName).replaceAll("");


            User user = userService.getUserInfoByIdCardNo(idCardNo);


            //设置用户头像url


            if (null!=user){


               user.setAvatar(urls);


               //更新用户信息


               userService.updateById(user);




               //内网数据推送


               //数据推送


               String s = "update blade_user set avatar = " + "'" + urls + "'" + "where id = " + "'" + user.getId() + "'";


               FtpUtil.sqlFileUpload(s);




               //文件推送


               InputStream inputStream = multipartFile.getInputStream();


               FtpUtil.uploadFile(ftpHost_dev, ftpPort, ftpUserName, ftpPassword, ftpPath, "/", split[2], inputStream);


               in.close();


            }


         }


      }


      //最后要删除文件


      FileUtil.clearFiles(desPath);


      //数据封装


      map.put("data", "上传成功!");


      //返回


      return R.data(map);


   }






   /**


    * 文件上传,自定义上传


    *


    * @param file 图片对象


@@ -387,7 +477,7 @@


    */


   private Long buildAttach(String fileName, Long fileSize, BladeFile bladeFile, String deptid, String type, Long noticeId, String cardid, InputStream in) throws Exception {


      //BladeUser user = AuthUtil.getUser();


      String fileExtension = FileUtil.getFileExtension(fileName);


      String fileExtension = org.springblade.core.tool.utils.FileUtil.getFileExtension(fileName);


      Attach attach = new Attach();


      attach.setDomain(bladeFile.getDomain());


      attach.setLink(bladeFile.getLink());