From df054296327ebca617a9c19bb01139b9c40bf232 Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Tue, 07 Jun 2022 11:25:44 +0800
Subject: [PATCH] 培训报名,保安员头像导入新增修改

---
 src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java |  141 +++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 141 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 d866716..97b996a 100644
--- a/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
+++ b/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
@@ -462,6 +462,147 @@
 		return R.data(map);
 	}
 
+	/**
+	 * 文件上传,zip(保安员头像上传)
+	 *
+	 * @param file 图片对象
+	 */
+	@PostMapping("put-file-zips")
+	public R putFileZips(@RequestParam MultipartFile file) throws Exception {
+		Map<String, Object> map = new HashMap<>(1);
+		//填写你文件上传的地址以及相应信息
+		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 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);
+			throw new ServiceException("上传文件类型不符!必须是 zip 压缩文件格式!");
+		}
+		//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);
+		fileList = fileUtil.getSubFiles(desPath,fileList);
+		//将不能导入的保安员账号存起来
+		List<String> errorList = new ArrayList<>();
+		List<String> picList = new ArrayList<>();
+		//导入状态,默认为true ,如果有一个出现问题则为 false
+		AtomicBoolean pic = new AtomicBoolean(true);
+		AtomicBoolean status = new AtomicBoolean(true);
+		//遍历
+		for (MultipartFile multipartFile : fileList){
+			long size = multipartFile.getSize();
+			if (size<30*1024 || size>500*1024){
+				status.set(false);
+				//取出身份证号,查询用户信息,更新用户信息
+				String pictrueName = multipartFile.getName().substring(0, multipartFile.getName().lastIndexOf("."));
+				String regex ="[\u4e00-\u9fa5]";
+				Pattern compile = Pattern.compile(regex);
+				String idCardNo = compile.matcher(pictrueName).replaceAll("");
+				//加入集合
+				errorList.add(idCardNo);
+
+			}
+			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://61.131.136.25:2081/zhba/" + newName;
+				//内网
+				String inUrl = FtpConfig.ip + "/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.getById(idCardNo);
+				//设置用户头像url
+				if (null!=user){
+//					user.setAvatar(urls);
+//					user.setAvatar("");
+//					//更新用户信息
+//					userService.updateById(user);
+
+					//内网数据推送
+					//数据推送
+					String s = "update blade_user set avatar = " + "'" + inUrl + "'" + "where id = " + "'" + user.getId() + "'";
+					//FtpUtil.sqlFileUpload(s);
+					myAsyncService.FTP(s);
+
+					//文件推送
+					InputStream inputStream = multipartFile.getInputStream();
+					FtpUtil.uploadFile(FtpConfig.ftpHost, ftpPort, FtpConfig.ftpUserName, ftpPassword, ftpPath, "/", "f"+split[2], inputStream);
+					in.close();
+				}else {
+					pic.set(false);
+					picList.add(idCardNo);
+				}
+			}
+		}
+		//最后要删除文件
+		FileUtil.clearFiles(FileConfig.localtion + File.separator + file.getOriginalFilename());
+		FileUtil.clearFiles(desPath);
+		//数据封装
+		map.put("data", "上传成功!");
+		//如果所有数据导入有一个异常
+		StringBuilder errorBuilder = new StringBuilder();
+		if (!status.get()) {
+			String errorAccount = StringUtils.join(errorList, "\\\n");
+			errorBuilder.append("用户:[" + errorAccount + "]照片导入失败!照片不能小于30KB,且不能大于500KB!!");
+		}
+
+		if (!pic.get()){
+			String errorAccount = StringUtils.join(picList, "\\\n");
+			errorBuilder.append("用户:[" + errorAccount + "]导入失败!编号不存在,请检查核对后再上传!");
+		}
+		//抛出异常
+		if (errorBuilder.length()>0){
+			throw new ServiceException(errorBuilder.toString());
+		}
+		//返回
+		return R.data(map);
+	}
+
 
 	/**
 	 * 文件上传,zip , 缴费凭证

--
Gitblit v1.9.3