新增二维码/zip压缩工具类，新增查看位置二维码接口

pom.xml

@@ -30,6 +30,7 @@
         <poi.version>4.1.2</poi.version>
         <velocity.version>2.3</velocity.version>
         <jwt.version>0.9.1</jwt.version>
+        <google.core.version>3.3.3</google.core.version>
     </properties>
 	
     <!-- 依赖声明 -->
@@ -128,6 +129,19 @@
                 <version>${jwt.version}</version>
             </dependency>
 
+            <!-- zxing生成二维码 -->
+            <dependency>
+                <groupId>com.google.zxing</groupId>
+                <artifactId>core</artifactId>
+                <version>${google.core.version}</version>
+            </dependency>
+
+            <dependency>
+                <groupId>com.google.zxing</groupId>
+                <artifactId>javase</artifactId>
+                <version>${google.core.version}</version>
+            </dependency>
+
             <!-- 验证码 -->
             <dependency>
                 <groupId>pro.fessional</groupId>

ruoyi-admin/src/main/java/com/ruoyi/asset/controller/TbLocationController.java

@@ -5,6 +5,7 @@ import javax.servlet.http.HttpServletResponse;
 
 import com.ruoyi.common.core.domain.TreeSelect;
 import io.swagger.v3.oas.annotations.Parameter;
+import org.springframework.http.MediaType;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -119,4 +120,12 @@ public class TbLocationController extends BaseController
     {
         return toAjax(tbLocationService.deleteTbLocationByIds(ids));
     }
+
+    /**
+     * 查看位置二维码
+     */
+    @GetMapping(value = "/code/{number}", produces = MediaType.IMAGE_PNG_VALUE)
+    public byte[] selectQrCode(@PathVariable("number") String locationNumber) {
+        return tbLocationService.selectQrCodeByLocationNumber(locationNumber);
+    }
 }

ruoyi-admin/src/main/java/com/ruoyi/asset/mapper/TbLocationMapper.java

@@ -19,6 +19,8 @@ public interface TbLocationMapper
      */
     public TbLocation selectTbLocationById(Long id);
 
+    TbLocation selectTbLocationByNumber(String number);
+
     /**
      * 查询所属位置列表
      * 

ruoyi-admin/src/main/java/com/ruoyi/asset/service/ITbLocationService.java

@@ -66,4 +66,16 @@ public interface ITbLocationService
      * @return 结果
      */
     public int deleteTbLocationById(Long id);
+
+    /**
+     * 生成位置二维码库
+     */
+    void createQrCodeDataByLocation(TbLocation tbLocation);
+
+    /**
+     * 查看位置二维码
+     *
+     * @param locationNumber 位置编码
+     */
+    byte[] selectQrCodeByLocationNumber(String locationNumber);
 }

ruoyi-admin/src/main/java/com/ruoyi/asset/service/impl/TbLocationServiceImpl.java

@@ -1,14 +1,19 @@
 package com.ruoyi.asset.service.impl;
 
+import java.io.File;
+import java.io.FileInputStream;
 import java.util.ArrayList;
 import java.util.HashSet;
 import java.util.List;
 
 import cn.hutool.core.util.StrUtil;
+import com.ruoyi.common.code.QRCodeUtils;
+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.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import com.ruoyi.asset.mapper.TbLocationMapper;
 import com.ruoyi.asset.domain.TbLocation;
@@ -152,6 +157,7 @@ public class TbLocationServiceImpl implements ITbLocationService
         if (flag < 1) {
             return flag;
         }
+        createQrCodeDataByLocation(tbLocation);
 
         String sequence = getSequence(parentId, tbLocation.getId());
         tbLocation.setSequence(sequence);
@@ -217,4 +223,44 @@ public class TbLocationServiceImpl implements ITbLocationService
     {
         return tbLocationMapper.deleteTbLocationById(id);
     }
+
+    @Value("${ruoyi.profile}")
+    private String savePath;
+
+    private final String dataName = "/location/qrCodes";
+
+    @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");
+        if (file.exists()) {
+            return;
+        }
+        QRCodeUtils.createCodeToFile(qrDTO, new File(savePath + dataName), qrDTO.getValue() + ".png");
+    }
+
+    @Override
+    public byte[] selectQrCodeByLocationNumber(String locationNumber) {
+        TbLocation location = tbLocationMapper.selectTbLocationByNumber(locationNumber);
+        if (location == null) {
+            return null;
+        }
+        String fileURL = savePath + dataName + "/" + locationNumber + ".png";
+        File file = new File(fileURL);
+        if (!file.exists()) {
+            createQrCodeDataByLocation(location);
+        }
+        byte[] bytes = new byte[0];
+        try {
+            FileInputStream fileInputStream = new FileInputStream(file);
+            bytes = new byte[fileInputStream.available()];
+            fileInputStream.read(bytes, 0, fileInputStream.available());
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return bytes;
+    }
 }

ruoyi-admin/src/main/resources/mapper/asset/TbLocationMapper.xml

@@ -36,6 +36,12 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
         <include refid="selectTbLocationVo"/>
         where id = #{id}
     </select>
+
+    <!--selectTbLocationByNumber-->
+    <select id="selectTbLocationByNumber" parameterType="String" resultMap="TbLocationResult">
+        <include refid="selectTbLocationVo"/>
+        where number = #{number}
+    </select>
         
     <insert id="insertTbLocation" parameterType="TbLocation" useGeneratedKeys="true" keyProperty="id">
         insert into tb_location

ruoyi-admin/src/main/resources/mapper/change/TbAssetAllocationDTOMapper.xml

@@ -22,8 +22,7 @@
                a.new_location_number newLocationNumber,
                b.asset_number assetNumber,
                b.old_responsible_person oldResponsiblePerson,
-               b.old_location_number oldLocationNumber,
-
+               b.old_location_number oldLocationNumber
     </select>
 
 </mapper>

ruoyi-common/pom.xml

@@ -14,6 +14,18 @@
     <description>
         common通用工具
     </description>
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <source>10</source>
+                    <target>10</target>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
 
     <dependencies>
 
@@ -126,6 +138,17 @@
             <artifactId>javax.servlet-api</artifactId>
         </dependency>
 
+        <!-- zxing生成二维码 -->
+        <dependency>
+            <groupId>com.google.zxing</groupId>
+            <artifactId>core</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>com.google.zxing</groupId>
+            <artifactId>javase</artifactId>
+        </dependency>
+
         <!--huTool-->
         <dependency>
             <groupId>cn.hutool</groupId>
@@ -133,6 +156,12 @@
             <version>5.7.17</version>
         </dependency>
 
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <optional>true</optional>
+        </dependency>
+
     </dependencies>
 
 </project>

ruoyi-common/src/main/java/com/ruoyi/common/code/QRCodeUtils.java

@@ -0,0 +1,175 @@
+package com.ruoyi.common.code;
+
+
+import cn.hutool.core.util.StrUtil;
+import com.google.zxing.BarcodeFormat;
+import com.google.zxing.EncodeHintType;
+import com.google.zxing.MultiFormatWriter;
+import com.google.zxing.WriterException;
+import com.google.zxing.common.BitMatrix;
+import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Component;
+
+import javax.imageio.ImageIO;
+import javax.swing.filechooser.FileSystemView;
+import java.awt.*;
+import java.awt.image.BufferedImage;
+import java.io.File;
+import java.io.OutputStream;
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * 二维码工具
+ **/
+@Component
+public class QRCodeUtils {
+    private static final Logger log= LoggerFactory.getLogger(QRCodeUtils.class);
+
+    //CODE_WIDTH：二维码宽度，单位像素
+    private static final int CODE_WIDTH = 400;
+    //CODE_HEIGHT：二维码高度，单位像素
+    private static final int CODE_HEIGHT = 400;
+    //FRONT_COLOR：二维码前景色，0x000000 表示黑色
+    private static final int FRONT_COLOR = 0x000000;
+    //BACKGROUND_COLOR：二维码背景色，0xFFFFFF 表示白色
+    //演示用 16 进制表示，和前端页面 CSS 的取色是一样的，注意前后景颜色应该对比明显，如常见的黑白
+    private static final int BACKGROUND_COLOR = 0xFFFFFF;
+
+    public static void createCodeToFile(QrDTO content, File codeImgFileSaveDir, String fileName) {
+        try {
+            if (content == null || StrUtil.isBlank(content.getValue()) || StrUtil.isBlank(fileName)) {
+                return;
+            }
+            if (codeImgFileSaveDir==null || codeImgFileSaveDir.isFile()) {
+                //二维码图片存在目录为空，默认放在桌面...
+                codeImgFileSaveDir = FileSystemView.getFileSystemView().getHomeDirectory();
+            }
+            if (!codeImgFileSaveDir.exists()) {
+                //二维码图片存在目录不存在，开始创建...
+                codeImgFileSaveDir.mkdirs();
+            }
+
+            //核心代码-生成二维码
+            BufferedImage bufferedImage = getBufferedImage(content);
+
+            File codeImgFile = new File(codeImgFileSaveDir, fileName);
+            ImageIO.write(bufferedImage, "png", codeImgFile);
+
+            log.info("二维码图片生成成功：" + codeImgFile.getPath());
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
+    /**
+     * 生成二维码并输出到输出流, 通常用于输出到网页上进行显示，输出到网页与输出到磁盘上的文件中，区别在于最后一句 ImageIO.write
+     * write(RenderedImage im,String formatName,File output)：写到文件中
+     * write(RenderedImage im,String formatName,OutputStream output)：输出到输出流中
+     * @param content  ：二维码内容
+     * @param outputStream ：输出流，比如 HttpServletResponse 的 getOutputStream
+     */
+    public static void createCodeToOutputStream(QrDTO content, OutputStream outputStream) {
+        try {
+            if (content == null || StrUtil.isBlank(content.getValue())) {
+                return;
+            }
+            //核心代码-生成二维码
+            BufferedImage bufferedImage = getBufferedImage(content);
+
+            //区别就是这一句，输出到输出流中，如果第三个参数是 File，则输出到文件中
+            ImageIO.write(bufferedImage, "png", outputStream);
+
+            log.info("二维码图片生成到输出流成功...");
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
+    //核心代码-生成二维码
+    private static BufferedImage getBufferedImage(QrDTO content) throws WriterException {
+
+        String name = content.getName();
+        String value = content.getValue().trim();
+
+        //com.google.zxing.EncodeHintType：编码提示类型,枚举类型
+        Map<EncodeHintType, Object> hints = new HashMap();
+
+        //EncodeHintType.CHARACTER_SET：设置字符编码类型
+        hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
+
+        //EncodeHintType.ERROR_CORRECTION：设置误差校正
+        //ErrorCorrectionLevel：误差校正等级，L = ~7% correction、M = ~15% correction、Q = ~25% correction、H = ~30% correction
+        //不设置时，默认为 L 等级，等级不一样，生成的图案不同，但扫描的结果是一样的
+        hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);
+
+        //EncodeHintType.MARGIN：设置二维码边距，单位像素，值越小，二维码距离四周越近
+        hints.put(EncodeHintType.MARGIN, 1);
+
+        /*
+          MultiFormatWriter:多格式写入，这是一个工厂类，里面重载了两个 encode 方法，用于写入条形码或二维码
+               encode(String contents,BarcodeFormat format,int width, int height,Map<EncodeHintType,?> hints)
+               contents:条形码/二维码内容
+               format：编码类型，如 条形码，二维码 等
+               width：码的宽度
+               height：码的高度
+               hints：码内容的编码类型
+          BarcodeFormat：枚举该程序包已知的条形码格式，即创建何种码，如 1 维的条形码，2 维的二维码 等
+          BitMatrix：位(比特)矩阵或叫2D矩阵，也就是需要的二维码
+         */
+        // 生成二维码
+        MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
+        /*参数顺序分别为：编码内容，编码类型，生成图片宽度，生成图片高度，设置参数
+          BitMatrix 的 get(int x, int y) 获取比特矩阵内容，指定位置有值，则返回true，将其设置为前景色，否则设置为背景色
+          BufferedImage 的 setRGB(int x, int y, int rgb) 方法设置图像像素
+               x：像素位置的横坐标，即列
+               y：像素位置的纵坐标，即行
+               rgb：像素的值，采用 16 进制,如 0xFFFFFF 白色
+         */
+        BitMatrix bitMatrix = multiFormatWriter.encode(value, BarcodeFormat.QR_CODE, CODE_WIDTH, CODE_HEIGHT, hints);
+        // 将二维码放入缓冲流
+        BufferedImage bufferedImage = new BufferedImage(CODE_WIDTH, CODE_HEIGHT, BufferedImage.TYPE_INT_BGR);
+        for (int x = 0; x < CODE_WIDTH; x++) {
+            for (int y = 0; y < CODE_HEIGHT; y++) {
+                // 循环将二维码内容写入图片
+                bufferedImage.setRGB(x, y, bitMatrix.get(x, y) ? FRONT_COLOR : BACKGROUND_COLOR);
+            }
+        }
+
+        // ------------------------------------------自定义文本描述-------------------------------------------------
+        // 1、在内存创建图片缓冲区 这里设置画板的宽高和类型
+        BufferedImage outImage = new BufferedImage(CODE_WIDTH, CODE_HEIGHT, BufferedImage.TYPE_4BYTE_ABGR);
+
+        // 2、创建画布，获取图像对象
+        Graphics2D graphics2D = outImage.createGraphics();
+
+        // 3、抗锯齿，防止模糊
+        RenderingHints rh = new RenderingHints(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
+        rh.put(RenderingHints.KEY_COLOR_RENDERING, RenderingHints.VALUE_COLOR_RENDER_QUALITY);
+        rh.put(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_LCD_HRGB);
+        graphics2D.setRenderingHints(rh);
+
+        // 4、在画布上画上二维码 X轴Y轴，宽度高度
+        graphics2D.drawImage(bufferedImage, 0, 0, bufferedImage.getWidth(), bufferedImage.getHeight(), null);
+
+        // 设置文字颜色为黑色
+        graphics2D.setColor(Color.BLACK);
+        // 字体、字型、字号
+        graphics2D.setFont(new Font("黑体", Font.PLAIN, 18));
+
+        // 获取字体宽度
+        int contentWidth = graphics2D.getFontMetrics().stringWidth(name);
+
+        // drawString(文字信息、x轴、y轴)方法根据参数设置文字的坐标轴 ，根据需要来进行调整
+        graphics2D.drawString(name, (CODE_WIDTH - contentWidth) / 2, CODE_HEIGHT - 5);
+
+        graphics2D.dispose();
+        outImage.flush();
+        bufferedImage = outImage;
+
+        return bufferedImage;
+    }
+}
+

ruoyi-common/src/main/java/com/ruoyi/common/code/QrDTO.java

@@ -0,0 +1,14 @@
+package com.ruoyi.common.code;
+
+import lombok.Data;
+
+import javax.validation.constraints.NotBlank;
+
+@Data
+public class QrDTO {
+    /** 标签名字 */
+    private String name;
+    /** 标签携带值 */
+    @NotBlank(message = "携带值不能为空")
+    private String value;
+}

ruoyi-common/src/main/java/com/ruoyi/common/filter/Folder2ZipUtils.java

@@ -0,0 +1,119 @@
+package com.ruoyi.common.filter;
+
+import javax.servlet.http.HttpServletResponse;
+import java.io.*;
+import java.net.URLEncoder;
+import java.nio.charset.StandardCharsets;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
+
+/**
+ * 文件夹压缩成zip文件
+ */
+public class Folder2ZipUtils {
+    /**
+     * zip打包
+     *
+     * @param sourceFileName 资源文件名称
+     * @param response 响应头
+     */
+    public static void zip(String sourceFileName, HttpServletResponse response) {
+        ZipOutputStream zipOut = null;
+        BufferedOutputStream bOs = null;
+
+        try {
+            SimpleDateFormat format = new SimpleDateFormat("yyyyMMdd");
+            String time = format.format(new Date());
+            String downloadFileName = time + "_my.zip";
+            // 转换为中文否则可能乱码
+            downloadFileName = URLEncoder.encode(downloadFileName, StandardCharsets.UTF_8);
+
+            // 将zip以流的形式输出到前台 指明response的返回对象是文件流
+            response.setHeader("content-type", "application/octet-stream");
+            response.setCharacterEncoding("utf-8");
+            // 设置浏览器响应头对呀的Content-disposition 设置在下载框默认显示的文件名
+            response.setHeader("Content-disposition", "attachment;filename=" + downloadFileName);
+            // 创建ZIP输出流
+            zipOut = new ZipOutputStream(response.getOutputStream());
+            // 创建缓冲输出流
+            bOs = new BufferedOutputStream(zipOut);
+            File file = new File(sourceFileName);
+            // 调用压缩函数
+            compress(zipOut, bOs, file, file.getName());
+            zipOut.flush();
+        } catch (Exception e) {
+            e.printStackTrace();
+        } finally {
+            try {
+                if (zipOut != null) {
+                    zipOut.close();
+                }
+                if (bOs != null) {
+                    bOs.close();
+                }
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+
+    /**
+     * 压缩函数
+     *
+     * @param zipOut ZIP流
+     * @param bOs 缓冲流
+     * @param sourceFile 资源文件
+     * @param base 根路径
+     */
+    public static void compress(ZipOutputStream zipOut, BufferedOutputStream bOs, File sourceFile, String base) {
+        FileInputStream fOs = null;
+        BufferedInputStream bIs = null;
+        try {
+            // 判断路径是否为目录
+            if (sourceFile.isDirectory()) {
+                // 取出文件夹中的文件(或子文件夹)
+                File[] files = sourceFile.listFiles();
+                if (files == null) {
+                    throw new RuntimeException("【ZIP工具类】文件夹不存在");
+                }
+                if (files.length == 0) {
+                    // 如果文件夹为空, 则只需在目的地zip文件中写入一个目录进入点
+                    zipOut.putNextEntry(new ZipEntry(base + "/"));
+                } else {
+                    // 如果文件夹不为空，则递归调用compress对文件夹中每一个文件(或文件夹)进行压缩
+                    for (File file : files) {
+                        compress(zipOut, bOs, file, base + "/" + file.getName());
+                    }
+                }
+            } else {
+                // 如果不是目录，即为文件，则先写入目录进入点，再将文件写入zip文件中
+                zipOut.putNextEntry(new ZipEntry(base));
+                fOs = new FileInputStream(sourceFile);
+                bIs = new BufferedInputStream(fOs);
+
+                int tag;
+                // 将源文件写入到zip文件中
+                while ((tag = bIs.read()) != -1) {
+                    zipOut.write(tag);
+                }
+                bIs.close();
+                fOs.close();
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        } finally {
+            try {
+                if (bIs != null) {
+                    bIs.close();
+                }
+                if (fOs != null) {
+                    fOs.close();
+                }
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+}

ruoyi-framework/src/main/java/com/ruoyi/framework/config/SecurityConfig.java

@@ -113,7 +113,7 @@ public class SecurityConfig extends WebSecurityConfigurerAdapter
                 // 对于登录login 注册register 验证码captchaImage 允许匿名访问
                 .antMatchers("/login", "/register", "/captchaImage","/inventory/inventory/takeStock").permitAll()
                 // 静态资源，可匿名访问
-                .antMatchers(HttpMethod.GET, "/", "/*.html", "/**/*.html", "/**/*.css", "/**/*.js", "/profile/**", "/asset/location/tree", "/asset/location/treeSelect").permitAll()
+                .antMatchers(HttpMethod.GET, "/", "/*.html", "/**/*.html", "/**/*.css", "/**/*.js", "/profile/**", "/asset/location/tree", "/asset/location/code/**").permitAll()
                 .antMatchers("/swagger-ui.html", "/swagger-resources/**", "/webjars/**", "/*/api-docs", "/druid/**").permitAll()
                 // 除上面外的所有请求全部需要鉴权认证
                 .anyRequest().authenticated()

ruoyi-ui/src/api/asset/location.js

@@ -64,3 +64,7 @@ export function delLocation(id) {
     method: 'delete'
   })
 }
+
+export function selectQrCode(number) {
+  return '/asset/location/code/' + number
+}

ruoyi-ui/src/views/asset/location/index.vue

@@ -46,6 +46,11 @@
       <el-table-column type="selection" width="55" align="center" />
       <!-- <el-table-column label="位置编号" align="center" prop="number" /> -->
       <el-table-column label="名称" align="center" prop="name" />
+      <el-table-column label="二维码" align="center" prop="number">
+        <template slot-scope="scope">
+          <image-preview :src="getCode(scope.row.number)" :width="80" :height="80"/>
+        </template>
+      </el-table-column>
       <el-table-column label="编号" align="center" prop="id" />
       <el-table-column label="上级位置" align="center" prop="parentId" />
       <!-- <el-table-column label="祖级列表" align="center" prop="sequence" /> -->
@@ -99,7 +104,7 @@
 </template>
 
 <script>
-import { getLocation, delLocation, addLocation, updateLocation, tree, treeSelect } from '@/api/asset/location'
+import { getLocation, delLocation, addLocation, updateLocation, tree, treeSelect, selectQrCode } from '@/api/asset/location'
 import Treeselect from "@riophae/vue-treeselect";
 import "@riophae/vue-treeselect/dist/vue-treeselect.css";
 
@@ -176,6 +181,9 @@ export default {
     getLevel(sequence) {
       return sequence.split(",").length
     },
+    getCode(number) {
+      return selectQrCode(number)
+    },
     // 取消按钮
     cancel() {
       this.open = false