feat 1001121

pcloud-service-book/src/main/java/com/pcloud/book/copyright/biz/impl/BookAuthInfoBizImpl.java

@@ -183,6 +183,10 @@ public class BookAuthInfoBizImpl implements BookAuthInfoBiz {
                     String url = generateExcel(bookAuthCodes, bookDto.getBookName());
                     String zipUrl = CopyrightTools.generateBar4Zip(bookAuthCodes, bookDto.getBookName(), url);
                     noticeUrl = zipUrl;
+                }else if(isHaveBarCode.equals(2)){
+                    String url = generateExcel(bookAuthCodes, bookDto.getBookName());
+                    String zipUrl = CopyrightTools.generateQrcode4Zip(bookAuthCodes, bookDto.getBookName(), url);
+                    noticeUrl = zipUrl;
                 }else {
                     String url = generateExcel(bookAuthCodes, bookDto.getBookName());
                     noticeUrl = url;

pcloud-service-book/src/main/java/com/pcloud/book/copyright/tools/CopyrightTools.java

 package com.pcloud.book.copyright.tools;
 
+import com.google.zxing.BarcodeFormat;
+import com.google.zxing.EncodeHintType;
+import com.google.zxing.MultiFormatWriter;
+import com.google.zxing.common.BitMatrix;
+import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
+
 import com.pcloud.book.book.constant.BookConstant;
 import com.pcloud.book.copyright.dto.DateDTO;
 import com.pcloud.book.copyright.entity.BookAuthCode;
@@ -14,6 +20,7 @@ import com.pcloud.common.utils.LocalDateUtils;
 import com.pcloud.common.utils.UUIDUitl;
 import com.pcloud.common.utils.aliyun.OssUtils;
 import com.pcloud.common.utils.zip.CompressUtils;
+
 import org.jbarcode.JBarcode;
 import org.jbarcode.encode.Code128Encoder;
 import org.jbarcode.paint.BaseLineTextPainter;
@@ -23,7 +30,6 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.stereotype.Component;
 
-import javax.imageio.ImageIO;
 import java.awt.*;
 import java.awt.image.BufferedImage;
 import java.io.ByteArrayOutputStream;
@@ -38,6 +44,8 @@ import java.util.Map;
 import java.util.Set;
 import java.util.UUID;
 
+import javax.imageio.ImageIO;
+
 /**
  * @author lily
  * @date 2018/12/4 16:37
@@ -66,12 +74,13 @@ public class CopyrightTools {
         return shortBuffer.toString().toLowerCase();
     }
 
-   public static void main(String[] args) {
+    public static void main(String[] args) {
         long time = System.currentTimeMillis();
         System.out.println(createShortUuidByCount(100000));
         long time1 = System.currentTimeMillis();
-        System.out.println(time1-time);
+        System.out.println(time1 - time);
     }
+
     /**
      * 生成count位非重复的数
      *
@@ -108,6 +117,13 @@ public class CopyrightTools {
         }
     }
 
+    /**
+     * 生成条形码
+     * @param bookAuthCodes
+     * @param fileName
+     * @param excelUrl
+     * @return
+     */
     public static String generateBar4Zip(List<BookAuthCode> bookAuthCodes, String fileName, String excelUrl) {
         fileName = FileUtils.formatName(fileName);
         String tempZipName = fileName + "_" + LocalDateUtils.getYmdhmss();
@@ -147,6 +163,66 @@ public class CopyrightTools {
         return uploadResultInfo == null ? null : uploadResultInfo.getUrl();
     }
 
+    /**
+     * 生成二维码
+     * @param bookAuthCodes
+     * @param fileName
+     * @param excelUrl
+     * @return
+     */
+    public static String generateQrcode4Zip(List<BookAuthCode> bookAuthCodes, String fileName, String excelUrl) {
+        fileName = FileUtils.formatName(fileName);
+        String tempZipName = fileName + "_" + LocalDateUtils.getYmdhmss();
+        String zipFilePath = ZIP_FILE_PATH + tempZipName + ".zip";
+        String fileFolderPath = FILE_LOCAL_PATH + tempZipName;
+        FileUtils.isDir(fileFolderPath);
+        UploadResultInfo uploadResultInfo;
+        try {
+            for (BookAuthCode bookAuthCode : bookAuthCodes) {
+                LOGGER.info("【二维码】generateQrcode4Zip,<START>.[url]=" + bookAuthCode);
+                Map<EncodeHintType, Object> hints = new HashMap<EncodeHintType, Object>();
+                hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
+                hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
+                hints.put(EncodeHintType.MARGIN, 0); // 设置白边
+                try {
+                    // 生成矩阵
+                    BitMatrix bitMatrix = new MultiFormatWriter().encode(bookAuthCode.getFullCode(), BarcodeFormat.QR_CODE, 55,
+                            55, hints);
+                    int width = bitMatrix.getWidth();
+                    int height = bitMatrix.getHeight();
+                    BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
+
+                    for(int x=0;x<width;x++){
+                        for(int y=0;y<height;y++){
+                            bufferedImage.setRGB(x, y, bitMatrix.get(x, y) ? 0xFF000000 : 0xFFFFFFFF);
+                        }
+                    }
+//                    paintQrcodeText(bufferedImage, bookAuthCode.getFullCode());
+                    File file = new File(fileFolderPath + "/" + bookAuthCode.getFullCode() + ".png");
+                    ImageIO.write(bufferedImage, ImageTypeEnum.PNG.value, file);
+                } catch (Exception e) {
+                    LOGGER.error("【二维码】创建二维码失败,<ERROR>:" + e.getMessage(), e);
+                    throw new FileException(FileException.FILE_UPLOAD_FAILURE, "创建二维码失败");
+                }
+            }
+            String downloadLocalPath = fileFolderPath + "/" + fileName + ".xls";
+            FileUtils.downloadFileFromUrl(excelUrl, downloadLocalPath);
+            // 检查压缩包临时文件夹是否存在，不存在就创建
+            FileUtils.isDir(ZIP_FILE_PATH);
+            CompressUtils.zip(fileFolderPath, zipFilePath);
+            // 上传文件到服务器中
+            uploadResultInfo = OssUtils.uploadLocalFile4CustomName(zipFilePath, fileName);
+        } catch (Exception e) {
+            LOGGER.error("【压缩】压缩失败,<ERROR>:" + e.getMessage(), e);
+            throw new FileException(FileException.ZIP_ERROR, "压缩失败！");
+        } finally {
+            // 删除产生的文件
+            FileUtils.deleteDirectory(fileFolderPath);
+            FileUtils.deleteFile(zipFilePath);
+        }
+        return uploadResultInfo == null ? null : uploadResultInfo.getUrl();
+    }
+
 
     public static String uploadImage(BufferedImage image) throws BizException {
         UploadResultInfo uploadResultInfo = null;
@@ -207,6 +283,27 @@ public class CopyrightTools {
 //        g2d.drawString(text, center, barCodeImage.getHeight() - (height / 10) - 2);
 //    }
 
+    public static void paintQrcodeText(BufferedImage barCodeImage, String text) {
+        //绘图
+        Graphics g2d = barCodeImage.getGraphics();
+        //创建字体
+        Font font = new Font("console", Font.PLAIN, 5);
+        g2d.setFont(font);
+        FontMetrics fm = g2d.getFontMetrics();
+        int height = fm.getHeight();
+//        int center = (barCodeImage.getWidth() - fm.stringWidth(text)) / 2;
+        g2d.setColor(Color.WHITE);
+        g2d.fillRect(0, 0, barCodeImage.getWidth(), barCodeImage.getHeight() / 20);
+        g2d.fillRect(0, barCodeImage.getHeight() - (height * 9 / 10), barCodeImage.getWidth(), (height * 9 / 10));
+        g2d.setColor(Color.BLACK);
+        //g2d.drawString(text, center, barCodeImage.getHeight() - (height / 10) - 2);
+        int start = 17;
+        int length = (barCodeImage.getWidth() - 2 * start) / text.length();
+        for (int i = 0; i < text.length(); i++) {
+            g2d.drawString(text.substring(i, i + 1), start + i * length, barCodeImage.getHeight() - (height / 10) - 2);
+        }
+    }
+
     public static void paintText(BufferedImage barCodeImage, String text) {
         //绘图
         Graphics g2d = barCodeImage.getGraphics();