wsy182
/
RuoYi-Cloud
mirror of https://gitee.com/y_project/RuoYi-Cloud.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

新增图片上传接口

pull/371/head
wuyibo 2023-07-10 09:42:00 +08:00
parent 3f377e0fef
commit 21ffc29a9f
4 changed files with 307 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pom.xml
View File

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

4
ruoyi-modules/ruoyi-system/pom.xml
View File

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

207
ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/WxFileController.java Normal file
View File

@ -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);
}
}

91
ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/utils/IdCardTool.java Normal file
View File

@ -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;
}
}