新增图片上传接口

pom.xml

@@ -190,6 +190,11 @@
                 <artifactId>mybatis-plus-boot-starter</artifactId>
                 <version>3.5.3.1</version>
             </dependency>-->
+            <dependency>
+                <groupId>com.baidu.aip</groupId>
+                <artifactId>java-sdk</artifactId>
+                <version>4.15.1</version>
+            </dependency>
             <dependency>
                 <groupId>com.alibaba</groupId>
                 <artifactId>easyexcel</artifactId>

ruoyi-modules/ruoyi-system/pom.xml

@@ -88,6 +88,10 @@
             <groupId>com.ruoyi</groupId>
             <artifactId>ruoyi-common-swagger</artifactId>
         </dependency>
+        <dependency>
+            <groupId>com.baidu.aip</groupId>
+            <artifactId>java-sdk</artifactId>
+        </dependency>
         <dependency>
             <groupId>org.projectlombok</groupId>
             <artifactId>lombok</artifactId>

ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/WxFileController.java

@@ -0,0 +1,207 @@
+package com.ruoyi.system.controller;
+
+import com.alibaba.fastjson.JSON;
+import com.ruoyi.common.core.exception.CheckedException;
+import com.ruoyi.common.core.exception.ServiceException;
+import com.ruoyi.common.core.web.controller.BaseController;
+import com.ruoyi.common.core.web.domain.AjaxResult;
+import com.ruoyi.common.core.web.page.TableDataInfo;
+import com.ruoyi.system.utils.IdCardTool;
+import io.swagger.annotations.*;
+import org.json.JSONObject;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.File;
+import java.text.SimpleDateFormat;
+import java.util.*;
+
+/**
+ * @author 吴一博
+ * @date 2023年07月10日 8:50
+ * @Description
+ */
+@RestController
+@RequestMapping(value="/file")
+@Api(description = "文件上传和下载")
+public class WxFileController extends BaseController {
+
+    @Value("${image.location.linux}")
+    private String linuxLocation;
+
+    @Value("${image.location.windows}")
+    private String winLocation;
+
+    @ApiOperation("文件上传")
+    @PostMapping(value = "/uploadMore")
+    @ResponseBody
+    public TableDataInfo handleFileUpload(@RequestParam("files") MultipartFile[] fileList) {
+        if(fileList==null||fileList.length==0){
+            throw new CheckedException("请选择需要上传的文件！");
+        }
+        List<String> fileUrls=new ArrayList<>();
+        for (int i = 0; i < fileList.length; i++) {
+            MultipartFile file = fileList[i];
+            //String filePath = winLocation;
+            Date date = new Date();
+            String time =  new SimpleDateFormat("yyyy-MM-dd").format(date);
+            System.out.println("获取到精确到日的时间格式为"+time);
+            String[] str = time.split("-");//根据‘-’进行拆分字符串 拆分出来的日期有，年，日，月，根据年日月创建文件夹
+            String datePath="/"+str[0]+"/"+str[1]+"/"+str[2]+"/";
+            //创建文件夹
+            String filePath = linuxLocation+datePath;
+            if (!file.isEmpty()) {
+                try {
+                    String originalFilename=file.getOriginalFilename();//原文件名称
+                    String newFileName = UUID.randomUUID().toString()+"."+originalFilename.substring(originalFilename.lastIndexOf(".")+1).toLowerCase();
+                    String dest=filePath + newFileName;
+                    File destFile = new File(filePath+newFileName);
+                    //文件目录不存在需要先创建
+                    if(!destFile.getParentFile().exists()){
+                        destFile.getParentFile().mkdirs();
+                    }
+                    file.transferTo(destFile);
+                    fileUrls.add(datePath+newFileName);
+                } catch (Exception e) {
+                    throw new CheckedException("第 " + i + " 个文件上传失败 ==> " + e.getMessage());
+                }
+            } else {
+                throw new CheckedException("第 " + i + " 个文件上传失败因为文件为空 ");
+            }
+        }
+        return getDataTable(fileUrls);
+    }
+    @ApiOperation("PC端专用多文件上传")
+    @PostMapping(value = "/uploadMoreFiles")
+    @ResponseBody
+    @ApiImplicitParams(@ApiImplicitParam(name = "files",paramType = "formData",value = "图片",required = true,dataType = "file",allowMultiple=true))
+    public TableDataInfo uploadMoreFiles(@RequestParam("files") MultipartFile[] fileList) {
+        if(fileList==null||fileList.length==0){
+            throw new CheckedException("请选择需要上传的文件！");
+        }
+        List<String> fileUrls=new ArrayList<>();
+        for (int i = 0; i < fileList.length; i++) {
+            MultipartFile file = fileList[i];
+            //String filePath = winLocation;
+            Date date = new Date();
+            String time =  new SimpleDateFormat("yyyy-MM-dd").format(date);
+            System.out.println("获取到精确到日的时间格式为"+time);
+            String[] str = time.split("-");//根据‘-’进行拆分字符串 拆分出来的日期有，年，日，月，根据年日月创建文件夹
+            String datePath="/"+str[0]+"/"+str[1]+"/"+str[2]+"/";
+            //创建文件夹
+            String filePath = linuxLocation+datePath;
+            if (!file.isEmpty()) {
+                try {
+                    String originalFilename=file.getOriginalFilename();//原文件名称
+                    String newFileName = UUID.randomUUID().toString()+"."+originalFilename.substring(originalFilename.lastIndexOf(".")+1).toLowerCase();
+                    String dest=filePath + newFileName;
+                    File destFile = new File(filePath+newFileName);
+                    //文件目录不存在需要先创建
+                    if(!destFile.getParentFile().exists()){
+                        destFile.getParentFile().mkdirs();
+                    }
+                    file.transferTo(destFile);
+                    fileUrls.add(datePath+newFileName);
+                } catch (Exception e) {
+                    throw new CheckedException("第 " + i + " 个文件上传失败 ==> " + e.getMessage());
+                }
+            } else {
+                throw new CheckedException("第 " + i + " 个文件上传失败因为文件为空 ");
+            }
+        }
+        return getDataTable(fileUrls);
+    }
+
+    @ApiOperation(value = "身份证上传自动识别身份证数据", notes = "身份证上传自动识别身份证数据")
+    @PostMapping(value = "/uploadIdCardImg")
+    @ResponseBody
+    public AjaxResult uploadIdCardImg(@RequestParam(value = "front：身份证含照片的一面；back：身份证带国徽的一面", required = true) String idCardSide,
+                                           @RequestParam("files") MultipartFile[] fileList) {
+        if(fileList==null||fileList.length==0){
+            throw new CheckedException("请选择需要上传的文件！");
+        }if(fileList.length!=1){
+            throw new CheckedException("只能上传一张图片哦！");
+        }
+        Map returnMap = new HashMap();
+        for (int i = 0; i < fileList.length; i++) {
+            MultipartFile file = fileList[i];
+            //String filePath = winLocation;
+            Date date = new Date();
+            String time =  new SimpleDateFormat("yyyy-MM-dd").format(date);
+            System.out.println("获取到精确到日的时间格式为"+time);
+            String[] str = time.split("-");//根据‘-’进行拆分字符串 拆分出来的日期有，年，日，月，根据年日月创建文件夹
+            String datePath=File.separator+"idCard"+File.separator+str[0]+File.separator+str[1]+File.separator+str[2]+File.separator;
+            //创建文件夹
+            String filePath = linuxLocation+datePath;
+            if (!file.isEmpty()) {
+                try {
+                    String originalFilename=file.getOriginalFilename();//原文件名称
+                    String newFileName = UUID.randomUUID().toString()+"."+originalFilename.substring(originalFilename.lastIndexOf(".")+1).toLowerCase();
+                    String dest=filePath + newFileName;
+                    File destFile = new File(dest);
+                    //文件目录不存在需要先创建
+                    if(!destFile.getParentFile().exists()){
+                        destFile.getParentFile().mkdirs();
+                    }
+                    file.transferTo(destFile);
+                    returnMap.put("relativePath",datePath+newFileName);
+                    //自动识别
+                    //上传成功后获取图片数据-自动识别
+                    org.json.JSONObject idCardInfo = IdCardTool.automaticRecognition(idCardSide,dest);
+                    returnMap.put("idCardInfo",idCardInfo);
+                } catch (Exception e) {
+                    throw new ServiceException("上传失败"+ JSON.toJSONString(returnMap));
+                }
+            } else {
+                throw new ServiceException("第 " + i + " 个文件上传失败因为文件为空 ");
+            }
+        }
+        return AjaxResult.success(returnMap);
+    }
+
+    @PostMapping(value = "/uploadIdCardImg1", consumes = "multipart/*", headers = "content-type=multipart/form-data")
+    @ApiOperation(value = "身份证上传自动识别身份证数据", notes = "身份证上传自动识别身份证数据", httpMethod = "POST")
+    @ApiResponses(value = {
+            @ApiResponse(code = 200, message = "上传成功！"),
+            @ApiResponse(code = 500, message = "上传失败！")
+    })
+    public AjaxResult uploadIdCardImg1(@RequestParam(value = "front：身份证含照片的一面；back：身份证带国徽的一面", required = true) String idCardSide,
+                                        @ApiParam(value = "身份证照片", required = true) MultipartFile file) {
+        Map returnMap = new HashMap();
+        if (!file.isEmpty()) {
+            if (file.getContentType().contains("image")) {
+                try {
+                    Date date = new Date();
+                    String time =  new SimpleDateFormat("yyyy-MM-dd").format(date);
+                    System.out.println("获取到精确到日的时间格式为"+time);
+                    String[] str = time.split("-");//根据‘-’进行拆分字符串 拆分出来的日期有，年，日，月，根据年日月创建文件夹
+                    String datePath=File.separator+"idCard"+File.separator+str[0]+File.separator+str[1]+File.separator+str[2]+File.separator;
+                    // 获取图片的文件名
+                    String originalFilename = file.getOriginalFilename();
+                    // 获取图片的扩展名
+                    String extensionName = originalFilename.substring(originalFilename.indexOf("."));
+                    // 新的图片文件名 = 获取时间戳+"."图片扩展名
+                    String newFileName = UUID.randomUUID().toString()+extensionName;
+                    // 文件目录
+                    String filePath = linuxLocation+datePath;
+                    //文件路径
+                    String distfilePath = filePath + newFileName;
+                    File destFile = new File(distfilePath);
+                    //文件目录不存在需要先创建
+                    if(!destFile.getParentFile().exists()){
+                        destFile.getParentFile().mkdirs();
+                    }
+                    file.transferTo(destFile);
+                    returnMap.put("relativePath",datePath+newFileName);
+                    //上传成功后获取图片数据-自动识别
+                    JSONObject idCardInfo = IdCardTool.automaticRecognition(idCardSide,distfilePath);
+                    returnMap.put("idCardInfo",idCardInfo);
+                }catch (Exception e){
+                    throw new CheckedException("上传失败"+JSON.toJSONString(returnMap));
+                }
+            }
+        }
+        return AjaxResult.success(returnMap);
+    }
+}

ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/utils/IdCardTool.java

@@ -0,0 +1,91 @@
+package com.ruoyi.system.utils;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import com.baidu.aip.ocr.AipOcr;
+import com.baidu.aip.util.Util;
+import org.springframework.util.StringUtils;
+
+import java.io.IOException;
+import java.util.HashMap;
+
+/**
+ * 身份证识别
+ */
+public class IdCardTool {
+    //设置APPID/AK/SK
+    public static final String APP_ID = "22812439";
+    public static final String API_KEY = "orXDtC8LLQGBDmW3EKiE81Fv";
+    public static final String SECRET_KEY = "1eTVMN545qvXe1hNryUa1ctTZqkCqMRf";
+
+    public static void main(String[] args) throws IOException {
+        // 初始化一个AipOcr
+        AipOcr client = new AipOcr(APP_ID, API_KEY, SECRET_KEY);
+
+        // 可选：设置网络连接参数
+        client.setConnectionTimeoutInMillis(2000);
+        client.setSocketTimeoutInMillis(60000);
+
+        // 可选：设置代理服务器地址, http和socket二选一，或者均不设置
+        //client.setHttpProxy("proxy_host", proxy_port);  // 设置http代理
+        //client.setSocketProxy("proxy_host", proxy_port);  // 设置socket代理
+
+        // 可选：设置log4j日志输出格式，若不设置，则使用默认配置
+        // 也可以直接通过jvm启动参数设置此环境变量
+        //System.setProperty("aip.log4j.conf", "path/to/your/log4j.properties");
+
+        // 传入可选参数调用接口
+        HashMap<String, String> options = new HashMap<String, String>();
+        options.put("detect_direction", "true");
+        options.put("detect_risk", "false");
+
+        String idCardSide = "front";
+
+        // 参数为本地图片路径
+        String image = "H://wyb.png";
+        org.json.JSONObject res = client.idcard(image, idCardSide, options);
+        System.out.println(res.toString(2));
+
+        // 参数为本地图片二进制数组
+        byte[] file = Util.readFileByBytes(image);
+        res = client.idcard(file, idCardSide, options);
+        System.out.println(JSON.toJSONString(res));
+    }
+
+    /**
+     * 身份证正反面的自动识别
+     * @param idCardSide 不传默认正面
+     * @return
+     * @throws IOException
+     */
+    public static org.json.JSONObject automaticRecognition(String idCardSide,String localPath) throws IOException {
+        // 初始化一个AipOcr
+        AipOcr client = new AipOcr(APP_ID, API_KEY, SECRET_KEY);
+        // 可选：设置网络连接参数
+        client.setConnectionTimeoutInMillis(2000);
+        client.setSocketTimeoutInMillis(60000);
+        // 可选：设置代理服务器地址, http和socket二选一，或者均不设置
+        //client.setHttpProxy("proxy_host", proxy_port);  // 设置http代理
+        //client.setSocketProxy("proxy_host", proxy_port);  // 设置socket代理
+        // 可选：设置log4j日志输出格式，若不设置，则使用默认配置
+        // 也可以直接通过jvm启动参数设置此环境变量
+        //System.setProperty("aip.log4j.conf", "path/to/your/log4j.properties");
+        // 传入可选参数调用接口
+        HashMap<String, String> options = new HashMap<String, String>();
+        //是否检测图像朝向，默认不检测，即：false。朝向是指输入图像是正常方向、逆时针旋转90/180/270度。可选值包括:
+        //- true：检测朝向；
+        //- false：不检测朝向。
+        options.put("detect_direction", "true");
+        //是否开启身份证风险类型(身份证复印件、临时身份证、身份证翻拍、修改过的身份证)功能，默认不开启，即：false。可选值:true-开启；false-不开启
+        options.put("detect_risk", "false");
+        if(StringUtils.isEmpty(idCardSide)){
+            idCardSide="front";
+        }
+        // localPath 参数为本地图片路径
+        org.json.JSONObject res = client.idcard(localPath, idCardSide, options);
+        // 参数为本地图片二进制数组
+/*        byte[] file = Util.readFileByBytes(image);
+        res = client.idcard(file, idCardSide, options);*/
+        System.out.println(JSON.toJSONString(res));
+        return res;
+    }
+}