213 lines
9.7 KiB
Java
213 lines
9.7 KiB
Java
|
package cn.com.tenlion.systemcard.util;
|
|||
|
|
|
|||
|
|
import com.baidu.aip.util.Base64Util;
|
|||
|
|
import org.json.JSONObject;
|
|||
|
|
|
|||
|
|
import javax.imageio.ImageIO;
|
|||
|
|
import java.awt.image.BufferedImage;
|
|||
|
|
import java.io.*;
|
|||
|
|
import java.net.HttpURLConnection;
|
|||
|
|
import java.net.URL;
|
|||
|
|
import java.net.URLEncoder;
|
|||
|
|
import java.util.HashMap;
|
|||
|
|
import java.util.List;
|
|||
|
|
import java.util.Map;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 统一信用代码证(营业执照)识别工具类
|
|||
|
|
* ClassName: OCRBusinessLicenseUtil
|
|||
|
|
* @Description: TODO
|
|||
|
|
* 创建工具: IDEA
|
|||
|
|
* 运行环境: [Tomcat7以上,MySql5.6以上,JDK7以上]
|
|||
|
|
* @author 崔宝铖
|
|||
|
|
* @date 2021年3月19日
|
|||
|
|
*/
|
|||
|
|
public class OCRBusinessLicenseUtil {
|
|||
|
|
|
|||
|
|
private static String apiKey = "6thm2Soep5COKhhvqGxCID3p";
|
|||
|
|
|
|||
|
|
private static String secretKey = "5f0tjGe6uvYbf72vUrNH5hRFmll4yA1m";
|
|||
|
|
|
|||
|
|
private static String accessToken = "";
|
|||
|
|
|
|||
|
|
// 鉴权token获取Url
|
|||
|
|
private static String authHost = "https://aip.baidubce.com/oauth/2.0/token?";
|
|||
|
|
|
|||
|
|
// 营业执照识别url
|
|||
|
|
private static String businessLicenseUrl = "https://aip.baidubce.com/rest/2.0/ocr/v1/business_license";
|
|||
|
|
|
|||
|
|
public static void main(String[] args) {
|
|||
|
|
BusinessLicenseBean result = businessLicenseFile("D:\\1.jpg");
|
|||
|
|
System.out.println(result);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 传入本地图片路径识别经营许可证
|
|||
|
|
* @param filePath
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
public static BusinessLicenseBean businessLicenseFile(String filePath) {
|
|||
|
|
BusinessLicenseBean bean = new BusinessLicenseBean();
|
|||
|
|
try {
|
|||
|
|
// 本地文件路径
|
|||
|
|
byte[] imgData = getFileBytes(new File(filePath));
|
|||
|
|
String imgStr = Base64Util.encode(imgData);
|
|||
|
|
String imgParam = URLEncoder.encode(imgStr, "UTF-8");
|
|||
|
|
// 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。
|
|||
|
|
Map<String, Object> params = new HashMap<String, Object>();
|
|||
|
|
params.put("image", imgParam);
|
|||
|
|
params.put("access_token", accessToken);
|
|||
|
|
String result = HttpUtil.doPost(businessLicenseUrl, params);
|
|||
|
|
// AccessToken如果过期
|
|||
|
|
if (result.contains("Access token invalid or no longer valid")) {
|
|||
|
|
accessToken = getAuth(apiKey, secretKey);
|
|||
|
|
return businessLicenseFile(filePath);
|
|||
|
|
}
|
|||
|
|
JSONObject jsonObject = new JSONObject(result);
|
|||
|
|
JSONObject wordsResult = jsonObject.getJSONObject("words_result");
|
|||
|
|
String licenseId = wordsResult.getJSONObject("社会信用代码").getString("words");
|
|||
|
|
String licenseNumber = wordsResult.getJSONObject("证件编号").getString("words");
|
|||
|
|
String licenseName = wordsResult.getJSONObject("单位名称").getString("words");
|
|||
|
|
String licenseType = wordsResult.getJSONObject("类型").getString("words");
|
|||
|
|
String licenseResidence = wordsResult.getJSONObject("地址").getString("words");
|
|||
|
|
String licenseLegalPerson = wordsResult.getJSONObject("法人").getString("words");
|
|||
|
|
String licenseCapital = wordsResult.getJSONObject("注册资本").getString("words");
|
|||
|
|
String licenseFoundTime = wordsResult.getJSONObject("成立日期").getString("words");
|
|||
|
|
String licenseTerm = wordsResult.getJSONObject("有效期").getString("words");
|
|||
|
|
String licenseBusiness = wordsResult.getJSONObject("经营范围").getString("words");
|
|||
|
|
String licenseAuthority = wordsResult.getJSONObject("登记机关").getString("words");
|
|||
|
|
String licenseComposition = wordsResult.getJSONObject("组成形式").getString("words");
|
|||
|
|
bean.setLicenseId(licenseId);
|
|||
|
|
bean.setLicenseNumber(licenseNumber);
|
|||
|
|
bean.setLicenseName(licenseName);
|
|||
|
|
bean.setLicenseType(licenseType);
|
|||
|
|
bean.setLicenseResidence(licenseResidence);
|
|||
|
|
bean.setLicenseLegalPerson(licenseLegalPerson);
|
|||
|
|
bean.setLicenseCapital(licenseCapital);
|
|||
|
|
bean.setLicenseFoundTime(licenseFoundTime);
|
|||
|
|
bean.setLicenseTerm(licenseTerm);
|
|||
|
|
bean.setLicenseBusiness(licenseBusiness);
|
|||
|
|
bean.setLicenseAuthority(licenseAuthority);
|
|||
|
|
bean.setLicenseComposition(licenseComposition);
|
|||
|
|
return bean;
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
e.printStackTrace();
|
|||
|
|
}
|
|||
|
|
return bean;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 传入网络图片路径识别经营许可证
|
|||
|
|
* @param url
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
public static BusinessLicenseBean businessLicenseUrl(String url) {
|
|||
|
|
BusinessLicenseBean bean = new BusinessLicenseBean();
|
|||
|
|
try {
|
|||
|
|
// 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。
|
|||
|
|
Map<String, Object> params = new HashMap<String, Object>();
|
|||
|
|
params.put("url", url);
|
|||
|
|
params.put("access_token", accessToken);
|
|||
|
|
String result = HttpUtil.doPost(businessLicenseUrl, params);
|
|||
|
|
// AccessToken如果过期
|
|||
|
|
if (result.contains("Access token invalid or no longer valid")) {
|
|||
|
|
accessToken = getAuth(apiKey, secretKey);
|
|||
|
|
return businessLicenseUrl(url);
|
|||
|
|
}
|
|||
|
|
JSONObject jsonObject = new JSONObject(result);
|
|||
|
|
JSONObject wordsResult = jsonObject.getJSONObject("words_result");
|
|||
|
|
String licenseId = wordsResult.getJSONObject("社会信用代码").getString("words");
|
|||
|
|
String licenseNumber = wordsResult.getJSONObject("证件编号").getString("words");
|
|||
|
|
String licenseName = wordsResult.getJSONObject("单位名称").getString("words");
|
|||
|
|
String licenseType = wordsResult.getJSONObject("类型").getString("words");
|
|||
|
|
String licenseResidence = wordsResult.getJSONObject("地址").getString("words");
|
|||
|
|
String licenseLegalPerson = wordsResult.getJSONObject("法人").getString("words");
|
|||
|
|
String licenseCapital = wordsResult.getJSONObject("注册资本").getString("words");
|
|||
|
|
String licenseFoundTime = wordsResult.getJSONObject("成立日期").getString("words");
|
|||
|
|
String licenseTerm = wordsResult.getJSONObject("有效期").getString("words");
|
|||
|
|
String licenseBusiness = wordsResult.getJSONObject("经营范围").getString("words");
|
|||
|
|
String licenseAuthority = wordsResult.getJSONObject("登记机关").getString("words");
|
|||
|
|
String licenseComposition = wordsResult.getJSONObject("组成形式").getString("words");
|
|||
|
|
bean.setLicenseId(licenseId);
|
|||
|
|
bean.setLicenseNumber(licenseNumber);
|
|||
|
|
bean.setLicenseName(licenseName);
|
|||
|
|
bean.setLicenseType(licenseType);
|
|||
|
|
bean.setLicenseResidence(licenseResidence);
|
|||
|
|
bean.setLicenseLegalPerson(licenseLegalPerson);
|
|||
|
|
bean.setLicenseCapital(licenseCapital);
|
|||
|
|
bean.setLicenseFoundTime(licenseFoundTime);
|
|||
|
|
bean.setLicenseTerm(licenseTerm);
|
|||
|
|
bean.setLicenseBusiness(licenseBusiness);
|
|||
|
|
bean.setLicenseAuthority(licenseAuthority);
|
|||
|
|
bean.setLicenseComposition(licenseComposition);
|
|||
|
|
return bean;
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
e.printStackTrace();
|
|||
|
|
}
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
public static String getAuth(String apiKey, String secretKey) {
|
|||
|
|
// 获取token地址
|
|||
|
|
String getAccessTokenUrl = authHost
|
|||
|
|
// 1. grant_type为固定参数
|
|||
|
|
+ "grant_type=client_credentials"
|
|||
|
|
// 2. 官网获取的 API Key
|
|||
|
|
+ "&client_id=" + apiKey
|
|||
|
|
// 3. 官网获取的 Secret Key
|
|||
|
|
+ "&client_secret=" + secretKey;
|
|||
|
|
try {
|
|||
|
|
URL realUrl = new URL(getAccessTokenUrl);
|
|||
|
|
// 打开和URL之间的连接
|
|||
|
|
HttpURLConnection connection = (HttpURLConnection) realUrl.openConnection();
|
|||
|
|
connection.setRequestMethod("GET");
|
|||
|
|
connection.connect();
|
|||
|
|
// 获取所有响应头字段
|
|||
|
|
Map<String, List<String>> map = connection.getHeaderFields();
|
|||
|
|
// 遍历所有的响应头字段
|
|||
|
|
for (String key : map.keySet()) {
|
|||
|
|
System.err.println(key + "--->" + map.get(key));
|
|||
|
|
}
|
|||
|
|
// 定义 BufferedReader输入流来读取URL的响应
|
|||
|
|
BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
|
|||
|
|
String result = "";
|
|||
|
|
String line;
|
|||
|
|
while ((line = in.readLine()) != null) {
|
|||
|
|
result += line;
|
|||
|
|
}
|
|||
|
|
/**
|
|||
|
|
* 返回结果示例
|
|||
|
|
*/
|
|||
|
|
// System.err.println("result:" + result);
|
|||
|
|
JSONObject jsonObject = new JSONObject(result);
|
|||
|
|
String access_token = jsonObject.getString("access_token");
|
|||
|
|
return access_token;
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
System.err.printf("获取token失败！");
|
|||
|
|
e.printStackTrace(System.err);
|
|||
|
|
}
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static byte[] getFileBytes(File file) throws IOException {
|
|||
|
|
String fileName = file.getName();
|
|||
|
|
String[] names = fileName.split("\\.");
|
|||
|
|
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
|
|||
|
|
if (names.length > 1) {
|
|||
|
|
String fileType = names[names.length - 1];
|
|||
|
|
BufferedImage bi;
|
|||
|
|
try {
|
|||
|
|
bi = ImageIO.read(file);
|
|||
|
|
ImageIO.write(bi, fileType, byteArrayOutputStream);
|
|||
|
|
byte[] bytes = byteArrayOutputStream.toByteArray();
|
|||
|
|
byteArrayOutputStream.close();
|
|||
|
|
return bytes;
|
|||
|
|
} catch (IOException e) {
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|