台桌批量生成桌码接口

cash-service/account-service/src/main/java/com/czg/service/account/service/impl/ShopTableServiceImpl.java

@@ -1,21 +1,37 @@
 package com.czg.service.account.service.impl;
 
+import cn.dev33.satoken.context.SaHolder;
 import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.img.ImgUtil;
+import cn.hutool.core.io.FileUtil;
+import cn.hutool.core.io.IoUtil;
+import cn.hutool.core.util.IdUtil;
+import cn.hutool.core.util.RandomUtil;
 import cn.hutool.core.util.StrUtil;
+import cn.hutool.core.util.ZipUtil;
+import cn.hutool.extra.qrcode.QrCodeUtil;
 import com.czg.account.dto.table.ShopTableAddDTO;
 import com.czg.account.entity.ShopArea;
 import com.czg.account.service.ShopAreaService;
 import com.czg.enums.ShopTableStatusEnum;
 import com.czg.exception.ApiNotPrintException;
+import com.mybatisflex.core.paginate.Page;
 import com.mybatisflex.core.query.QueryWrapper;
 import com.mybatisflex.spring.service.impl.ServiceImpl;
 import com.czg.account.entity.ShopTable;
 import com.czg.account.service.ShopTableService;
 import com.czg.service.account.mapper.ShopTableMapper;
 import jakarta.annotation.Resource;
+import jakarta.servlet.http.HttpServletResponse;
 import org.springframework.stereotype.Service;
 
+import java.awt.image.BufferedImage;
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.IOException;
+import java.math.BigDecimal;
 import java.util.ArrayList;
+import java.util.Date;
 import java.util.HashMap;
 import java.util.List;
 
@@ -63,4 +79,57 @@ public class ShopTableServiceImpl extends ServiceImpl<ShopTableMapper, ShopTable
         }
         return true;
     }
+
+    @Override
+    public void createQrCode(Long shopId, Integer num, HttpServletResponse response) throws IOException {
+        // 查询shop下面有多少台桌
+        long count = queryChain().eq(ShopTable::getShopId, shopId).count();
+        if (count == 0) {
+            throw new ApiNotPrintException("未添加台桌，请先添加台桌");
+        }
+
+        // 2. 创建临时目录存储二维码
+        File tempDir = FileUtil.mkdir(FileUtil.getTmpDir() + "/shop_qr_temp");
+        List<File> qrCodeFiles = new ArrayList<>();
+
+        // 获取可用id
+        List<Integer> idList = queryChain().isNull(ShopTable::getTableCode).select("id").orderBy(ShopTable::getId, false).page(new Page<>(1, num)).getRecords().stream().map(ShopTable::getId).toList();
+        long maxId = idList.getLast();
+
+        // 3. 生成唯一二维码
+        for (int i = 0; i < num; i++) {
+            String tableCode = shopId.toString();
+            if (i > idList.size() - 1) {
+                tableCode = tableCode + ++maxId;
+            }else {
+                tableCode = tableCode + idList.get(i);
+            }
+
+            tableCode = tableCode + RandomUtil.randomNumbers(8);
+            // 生成二维码
+            BufferedImage qrImage = QrCodeUtil.generate(tableCode, 300, 300);
+            ByteArrayOutputStream out = new ByteArrayOutputStream();
+            ImgUtil.write(qrImage, ImgUtil.IMAGE_TYPE_PNG, out);
+
+            // 保存二维码到文件
+            File qrFile = new File(tempDir, tableCode + ".png");
+            FileUtil.writeBytes(out.toByteArray(), qrFile);
+            qrCodeFiles.add(qrFile);
+        }
+
+        // 4. 生成 ZIP 文件
+        File zipFile = new File(FileUtil.getTmpDir(), "shop_qrcodes_" + shopId + ".zip");
+        ZipUtil.zip(zipFile, false, qrCodeFiles.toArray(new File[0]));
+
+        // 5. 返回 ZIP 文件
+        response.setContentType("application/zip");
+        response.setHeader("Content-Disposition", "attachment; filename=" + zipFile.getName());
+        response.setContentLength((int) zipFile.length());
+        IoUtil.copy(FileUtil.getInputStream(zipFile), response.getOutputStream());
+
+        // 6. 清理临时文件
+        FileUtil.del(tempDir);
+        FileUtil.del(zipFile);
+
+    }
 }

cash-service/account-service/src/main/resources/mapper/ShopUserMapper.xml

@@ -37,7 +37,7 @@
         where a.user_id = #{userInfoId}
     </select>
     <select id="selectAssetsSummary" resultType="com.czg.account.dto.user.userinfo.UserInfoAssetsSummaryDTO">
-        select sum(IFNULL(b.amount, 0)) amount, sum(IFNULL(b.account_points, 0)) points, sum(IFNULL(c.over_num, 0)) couponNum
+        select sum(IFNULL(b.amount, 0)) as amount, sum(IFNULL(b.account_points, 0)) as points, sum(IFNULL(c.over_num, 0)) as couponNum
         from tb_user_info as a
                  left join tb_shop_user as b on a.id = b.user_id
                  left join tb_shop_activate_in_record as c on c.shop_id = b.shop_id