|
|
@@ -2,6 +2,7 @@ package com.ruoyi.asset.service.impl;
|
|
|
|
|
|
import java.io.File;
|
|
|
import java.io.FileInputStream;
|
|
|
+import java.io.IOException;
|
|
|
import java.util.ArrayList;
|
|
|
import java.util.HashSet;
|
|
|
import java.util.List;
|
|
|
@@ -12,6 +13,7 @@ import com.ruoyi.common.code.QrDTO;
|
|
|
import com.ruoyi.common.core.domain.TreeSelect;
|
|
|
import com.ruoyi.common.utils.DateUtils;
|
|
|
import com.ruoyi.common.utils.SecurityUtils;
|
|
|
+import org.apache.commons.io.FileUtils;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
@@ -229,13 +231,15 @@ public class TbLocationServiceImpl implements ITbLocationService
|
|
|
|
|
|
private final String dataName = "/location/qrCodes";
|
|
|
|
|
|
+ private final String cacheName = "/location/cache";
|
|
|
+
|
|
|
@Override
|
|
|
public void createQrCodeDataByLocation(TbLocation tbLocation) {
|
|
|
QrDTO qrDTO = new QrDTO();
|
|
|
qrDTO.setName(tbLocation.getName());
|
|
|
qrDTO.setValue(tbLocation.getNumber());
|
|
|
|
|
|
- File file = new File(savePath + dataName + "/" + qrDTO.getValue() + ".png");
|
|
|
+ File file = new File(getFileUrl(qrDTO.getValue()));
|
|
|
if (file.exists()) {
|
|
|
return;
|
|
|
}
|
|
|
@@ -248,7 +252,7 @@ public class TbLocationServiceImpl implements ITbLocationService
|
|
|
if (location == null) {
|
|
|
return null;
|
|
|
}
|
|
|
- String fileURL = savePath + dataName + "/" + locationNumber + ".png";
|
|
|
+ String fileURL = getFileUrl(locationNumber);
|
|
|
File file = new File(fileURL);
|
|
|
if (!file.exists()) {
|
|
|
createQrCodeDataByLocation(location);
|
|
|
@@ -263,4 +267,70 @@ public class TbLocationServiceImpl implements ITbLocationService
|
|
|
}
|
|
|
return bytes;
|
|
|
}
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String downloadQrCode(List<Long> locationIds) {
|
|
|
+ ArrayList<String> fileUrlList = new ArrayList<>();
|
|
|
+ for (Long locationId : locationIds) {
|
|
|
+ TbLocation location = tbLocationMapper.selectTbLocationById(locationId);
|
|
|
+ if (location == null) {
|
|
|
+ continue;
|
|
|
+ }
|
|
|
+ String fileUrl = getFileUrl(location.getNumber());
|
|
|
+ fileUrlList.add(fileUrl);
|
|
|
+ }
|
|
|
+
|
|
|
+ // 新建一个目录
|
|
|
+ String path = savePath + cacheName;
|
|
|
+ File file = new File(path);
|
|
|
+ if (file.exists()) {
|
|
|
+ // 清空文件夹
|
|
|
+ delete(path);
|
|
|
+ } else {
|
|
|
+ file.mkdir();
|
|
|
+ }
|
|
|
+
|
|
|
+ for (String fileUrl : fileUrlList) {
|
|
|
+ try {
|
|
|
+ FileUtils.copyFileToDirectory(new File(fileUrl), new File(path));
|
|
|
+ } catch (IOException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return path;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void delete(String path) {
|
|
|
+ // 为传进来的路径参数创建一个文件对象
|
|
|
+ File file = new File(path);
|
|
|
+ // 如果目标路径是一个文件，那么直接调用delete方法删除即可
|
|
|
+ // file.delete();
|
|
|
+ // 如果是一个目录，那么必须把该目录下的所有文件和子目录全部删除，才能删除该目标目录，这里要用到递归函数
|
|
|
+ // 创建一个files数组，用来存放目标目录下所有的文件和目录的file对象
|
|
|
+ File[] files;
|
|
|
+ // 将目标目录下所有的file对象存入files数组中
|
|
|
+ files = file.listFiles();
|
|
|
+ if (files == null) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ // 循环遍历files数组
|
|
|
+ for(File temp : files){
|
|
|
+ // 判断该temp对象是否为文件对象
|
|
|
+ if (temp.isFile()) {
|
|
|
+ temp.delete();
|
|
|
+ }
|
|
|
+ // 判断该temp对象是否为目录对象
|
|
|
+ if (temp.isDirectory()) {
|
|
|
+ // 将该temp目录的路径给delete方法（自己），达到递归的目的
|
|
|
+ delete(temp.getAbsolutePath());
|
|
|
+ // 确保该temp目录下已被清空后，删除该temp目录
|
|
|
+ temp.delete();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private String getFileUrl(String number) {
|
|
|
+ return savePath + dataName + "/" + number + ".png";
|
|
|
+ }
|
|
|
}
|
