feats

pom.xml

@@ -84,6 +84,14 @@
 		</dependency>
 		<!--excel end-->
 
+		<!--zip start-->
+		<dependency>
+			<groupId>net.lingala.zip4j</groupId>
+			<artifactId>zip4j</artifactId>
+			<version>2.10.0</version>
+		</dependency>
+		<!--zip end-->
+
 		<!--weixin-miniapp start-->
 		<dependency>
 			<groupId>com.github.binarywang</groupId>

src/main/java/com/njyunzhi/pet_identity/common/HttpUtils.java

@@ -4,12 +4,26 @@ import org.springframework.http.*;
 import org.springframework.stereotype.Component;
 import org.springframework.web.client.RestTemplate;
 
+import java.io.InputStream;
 import java.io.UnsupportedEncodingException;
 import java.net.URLDecoder;
 import java.util.*;
 
 @Component
 public class HttpUtils {
+
+    public InputStream download(String url) {
+        RestTemplate restTemplate = new RestTemplate();
+        HttpHeaders headers = new HttpHeaders();
+        HttpEntity<String> entity = new HttpEntity<String>(headers);
+        Map<String, Object> urlAndParams = parseURL(url);
+        String formatURL = (String) urlAndParams.get("url");
+        Map<String, String> params = (Map<String, String>) urlAndParams.get("params");
+
+        ResponseEntity<InputStream> resp = restTemplate.exchange(formatURL, HttpMethod.GET, entity, InputStream.class, params);
+        return resp.getBody();
+    }
+
     public String get(String url) throws Exception {
         RestTemplate restTemplate = new RestTemplate();
         HttpHeaders headers = new HttpHeaders();

src/main/java/com/njyunzhi/pet_identity/common/ZipUtil.java

@@ -0,0 +1,100 @@
+package com.njyunzhi.pet_identity.common;
+
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.write.builder.ExcelWriterBuilder;
+import com.alibaba.fastjson.JSON;
+import lombok.Data;
+import net.lingala.zip4j.ZipFile;
+import net.lingala.zip4j.exception.ZipException;
+import net.lingala.zip4j.model.ZipParameters;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import javax.servlet.http.HttpServletResponse;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.nio.ByteBuffer;
+import java.nio.channels.Channels;
+import java.nio.channels.ReadableByteChannel;
+import java.nio.channels.WritableByteChannel;
+import java.time.LocalDateTime;
+import java.time.ZoneId;
+import java.time.format.DateTimeFormatter;
+import java.util.List;
+
+@Component
+public class ZipUtil {
+
+    @Autowired
+    HttpUtils httpUtils;
+
+    private String today() {
+        String formatterStr = "yyyy-MM-dd";
+        DateTimeFormatter formatter = DateTimeFormatter.ofPattern(formatterStr);
+        return LocalDateTime.now(ZoneId.of("Asia/Shanghai")).format(formatter);
+    }
+
+    public ZipFile zipLink(List<RFile> files) throws IOException {
+        ZipFile zipFile = new ZipFile(today() + ".zip");
+
+        if (files == null || files.size() < 1) {
+            return zipFile;
+        }
+
+        for (RFile f : files) {
+            ZipParameters zipParameters = new ZipParameters();
+            zipParameters.setFileNameInZip(f.getName());
+            InputStream inputStream = httpUtils.download(f.getUrl());
+            zipFile.addStream(inputStream, zipParameters);
+            inputStream.close();
+        }
+
+        return zipFile;
+    }
+
+    private long copyStream(InputStream input, OutputStream output) throws IOException {
+        ReadableByteChannel inputChannel = Channels.newChannel(input);
+        WritableByteChannel outputChannel = Channels.newChannel(output);
+        ByteBuffer buffer = ByteBuffer.allocateDirect(10240);
+        long size = 0;
+
+        while (inputChannel.read(buffer) != -1) {
+            buffer.flip();
+            size += outputChannel.write(buffer);
+            buffer.clear();
+        }
+
+        return size;
+    }
+
+    public void flush(HttpServletResponse response, ZipFile zipFile) throws IOException {
+        try {
+            String fileName = zipFile.getFile().getName();
+            InputStream inputStream = new FileInputStream(zipFile.getFile());
+
+            response.setContentType("application/zip");
+            response.setCharacterEncoding("utf-8");
+            response.setHeader("Content-disposition", "attachment;filename="+StringUtils.urlEncode(fileName));
+
+            response.setContentLength(zipFile.getBufferSize());
+            copyStream(inputStream, response.getOutputStream());
+        } catch (Exception e) {
+            e.printStackTrace();
+
+            // 重置response
+            response.reset();
+            response.setContentType("application/json");
+            response.setCharacterEncoding("utf-8");
+            ResponseBean res = ResponseBean.error(e.getMessage(), ResponseBean.ERROR_UNAVAILABLE);
+            response.getWriter().println(JSON.toJSONString(res));
+        }
+    }
+
+    @Data
+    public static class RFile {
+        String url;
+        String name;
+    }
+}

src/main/java/com/njyunzhi/pet_identity/controller/TaCardNoController.java

@@ -6,12 +6,14 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.njyunzhi.pet_identity.common.BaseController;
 import com.njyunzhi.pet_identity.common.ResponseBean;
 import com.njyunzhi.pet_identity.common.StringUtils;
+import com.njyunzhi.pet_identity.common.ZipUtil;
 import com.njyunzhi.pet_identity.entity.TaSequence;
 import com.njyunzhi.pet_identity.service.ITaSequenceService;
 import com.njyunzhi.pet_identity.vo.BatchCardParams;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import io.swagger.annotations.ApiParam;
+import net.lingala.zip4j.ZipFile;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -24,6 +26,10 @@ import com.njyunzhi.pet_identity.service.ITaCardNoService;
 import com.njyunzhi.pet_identity.entity.TaCardNo;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.servlet.http.HttpServletResponse;
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * <p>
     * 卡号库 前端控制器
@@ -46,6 +52,9 @@ public class TaCardNoController extends BaseController {
     @Autowired
     public ITaSequenceService iTaSequenceService;
 
+    @Autowired
+    public ZipUtil zipUtil;
+
     /**
      * 分页查询列表
      * @param pageNum
@@ -55,10 +64,19 @@ public class TaCardNoController extends BaseController {
     @RequestMapping(value="/cardNo",method= RequestMethod.GET)
     @ApiOperation(value="列表", notes = "列表", httpMethod = "GET", response = ResponseBean.class)
     public ResponseBean taCardNoList(@ApiParam("页码") @RequestParam(value ="pageNum",defaultValue = "1") Integer pageNum,
-									 @ApiParam("单页数据量") @RequestParam(value ="pageSize",defaultValue = "10") Integer pageSize) throws Exception{
+									 @ApiParam("单页数据量") @RequestParam(value ="pageSize",defaultValue = "10") Integer pageSize,
+                                     @ApiParam("前缀") @RequestParam(value ="prefix", required = false) String prefix,
+                                     @ApiParam("起始号码") @RequestParam(value ="start", required = false) Integer start,
+                                     @ApiParam("终止号码") @RequestParam(value ="end", required = false) Integer end) throws Exception{
+
+        if (StringUtils.isEmpty(prefix) && (null != start || null != end)) {
+            return ResponseBean.error("卡号前缀不能为空");
+        }
 
         IPage<TaCardNo> pg = new Page<>(pageNum, pageSize);
         QueryWrapper<TaCardNo> queryWrapper = new QueryWrapper<>();
+        queryWrapper.ge(null != start, "card_no", iTaCardNoService.getFormatNo(prefix, start));
+        queryWrapper.le(null != end, "card_no", iTaCardNoService.getFormatNo(prefix, end));
         queryWrapper.orderByDesc("create_date");
 
         IPage<TaCardNo> result = iTaCardNoService.page(pg, queryWrapper);
@@ -66,6 +84,42 @@ public class TaCardNoController extends BaseController {
     }
 
 
+    /**
+     * 分页查询列表
+     * @return
+     */
+    @RequestMapping(value="/cardNo/export",method= RequestMethod.GET)
+    @ApiOperation(value="导出", notes = "导出", httpMethod = "GET", response = ResponseBean.class)
+    public ResponseBean export(@ApiParam("前缀") @RequestParam(value ="prefix") String prefix,
+                               @ApiParam("起始号码") @RequestParam(value ="start", required = false) Integer start,
+                               @ApiParam("终止号码") @RequestParam(value ="end", required = false) Integer end,
+                               HttpServletResponse response) throws Exception{
+
+        if (StringUtils.isEmpty(prefix)) {
+            return ResponseBean.error("卡号前缀不能为空");
+        }
+
+        QueryWrapper<TaCardNo> queryWrapper = new QueryWrapper<>();
+        queryWrapper.ge(null != start, "card_no", iTaCardNoService.getFormatNo(prefix, start));
+        queryWrapper.le(null != end, "card_no", iTaCardNoService.getFormatNo(prefix, end));
+        queryWrapper.orderByDesc("create_date");
+
+        List<TaCardNo> result = iTaCardNoService.list(queryWrapper);
+
+        List<ZipUtil.RFile> files = new ArrayList<>();
+        if (null != result && result.size() > 0) {
+            for (TaCardNo item : result) {
+                ZipUtil.RFile f = new ZipUtil.RFile();
+                f.setName(item.getCardNo() + ".png");
+                f.setUrl(item.getQrImage());
+            }
+        }
+
+        ZipFile data = zipUtil.zipLink(files);
+        zipUtil.flush(response, data);
+        return null;
+    }
+
     /**
      * 分页查询列表
      * @param id

src/main/java/com/njyunzhi/pet_identity/service/ITaCardNoService.java

@@ -17,4 +17,6 @@ public interface ITaCardNoService extends IService<TaCardNo> {
     TaCardNo getUnuseCard(String cardNo);
 
     void createBatchCardNo(BatchCardParams batchCardParams) throws Exception;
+
+    String getFormatNo(String prefix, Integer num);
 }

src/main/java/com/njyunzhi/pet_identity/service/impl/TaCardNoServiceImpl.java

@@ -62,7 +62,7 @@ public class TaCardNoServiceImpl extends ServiceImpl<TaCardNoMapper, TaCardNo> i
 
         for (int i = 0; i < batchCardParams.getNum(); i += 1) {
             Integer nextVal = taSequenceMapper.getNextVal("card-" + batchCardParams.getPrefix());
-            String cardNo = batchCardParams.getPrefix() + StringUtils.lpad(nextVal.toString(), "0", 5);
+            String cardNo = getFormatNo(batchCardParams.getPrefix(), nextVal);
             MiniAppScene scene = new MiniAppScene().setPage(cardpath).setQueryStr("cardno=" + StringUtils.urlEncode(cardNo));
 
             // 生成小程序码
@@ -82,4 +82,10 @@ public class TaCardNoServiceImpl extends ServiceImpl<TaCardNoMapper, TaCardNo> i
             }
         }
     }
+
+    @Override
+    public String getFormatNo(String prefix, Integer num) {
+        int maxLen = 5;
+        return String.format("%s%s", prefix, StringUtils.lpad(num.toString(), "0", maxLen));
+    }
 }