package org.springblade.common.utils;
|
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import org.apache.http.HttpEntity;
|
import org.apache.http.HttpResponse;
|
import org.apache.http.client.HttpClient;
|
import org.apache.http.client.methods.HttpPost;
|
import org.apache.http.entity.ByteArrayEntity;
|
import org.apache.http.entity.ContentType;
|
import org.apache.http.impl.client.HttpClients;
|
import org.apache.http.util.EntityUtils;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.*;
|
import java.net.HttpURLConnection;
|
import java.net.URL;
|
import java.util.HashMap;
|
import java.util.Map;
|
import org.apache.commons.io.FileUtils;
|
import java.awt.image.BufferedImage;
|
import javax.imageio.ImageIO;
|
import net.coobird.thumbnailator.Thumbnails;
|
|
/**
|
* 微信安全检验工具类
|
*/
|
public class WeiXinSecurityUtil {
|
private static Integer IMG_WIDTH = 750;
|
private static Integer IMG_HEIGHT = 1334;
|
/**
|
* 图片检测接口
|
*/
|
private static String IMG_SEC_URL = "https://api.weixin.qq.com/wxa/img_sec_check?access_token=";
|
|
//获取wxAccessToken
|
private static String WX_GET_ACCESS_TOKEN = "https://api.weixin.qq.com/cgi-bin/token";
|
|
private static String WX_APP_ID = "wx41aa8a5d2e565a05";
|
private static String WX_SECRET = "e0e9218ac368cd4ca620ff4c7030f468";
|
|
/**
|
* 恶意图片过滤-校验图片是否含有违法违规内容
|
* MultipartFile multipartFile, String accessToken
|
* @param multipartFile
|
* @return
|
*/
|
public static String checkImg(MultipartFile multipartFile){
|
String result = "";
|
//压缩图片
|
InputStream inputStream = compressImage(multipartFile);
|
// 校验api url
|
String url = IMG_SEC_URL + getWxAccessToken();
|
// 获得Http客户端(可以理解为:你得先有一个浏览器;注意:实际上HttpClient与浏览器是不一样的
|
HttpClient httpclient = HttpClients.createDefault();
|
//创建一个post请求
|
HttpPost request = new HttpPost(url);
|
//设置响应头 （ application/octet-stream：二进制流，不知道下载文件类型）
|
request.addHeader("Content-Type", "application/octet-stream");
|
try {
|
//创建一个byte数组，和输入的文件的大小一样
|
byte[] byt = new byte[inputStream.available()];
|
//从输入流中读取全部，并将其存储在缓冲区数组byt 中。
|
inputStream.read(byt);
|
//定制提交内容
|
request.setEntity(new ByteArrayEntity(byt, ContentType.create("image/jpg")));
|
//由客户端执行(发送)请求,执行校验
|
HttpResponse response = httpclient.execute(request);
|
// 从响应模型中获取响应实体
|
HttpEntity entity = response.getEntity();
|
result = EntityUtils.toString(entity, "UTF-8");// 转成string
|
//打印校验结果
|
System.out.println("result：" + result);
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
if(inputStream != null){
|
try {
|
inputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
// 响应返回结果
|
return result;
|
}
|
|
|
|
/**
|
* 图片过滤检测
|
* @param file 图片文件
|
* @return
|
*/
|
public String checkPic(MultipartFile file) {
|
String access_token = "";
|
String result = "";
|
//自己写一个定时任务或其他方式 获取AccessToken
|
try {
|
access_token = getWxAccessToken();
|
String url = "https://api.weixin.qq.com/wxa/img_sec_check?access_token=" + access_token;
|
result = uploadFile(url, file);
|
System.out.println("图片检测结果 = " + result);
|
} catch (Exception e) {
|
System.out.println("----------------调用腾讯内容过滤系统出错------------------" + e.getMessage());
|
}
|
// 返回
|
return result;
|
}
|
|
/**
|
* 上传二进制文件
|
* @param graphurl 接口地址
|
* @param file 图片文件
|
* @return
|
*/
|
public static String uploadFile(String graphurl,MultipartFile file) {
|
String line = null;//接口返回的结果
|
try {
|
// 换行符
|
final String newLine = "\r\n";
|
final String boundaryPrefix = "--";
|
// 定义数据分隔线
|
String BOUNDARY = "========7d4a6d158c9";
|
// 服务器的域名
|
URL url = new URL(graphurl);
|
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
|
// 设置为POST情
|
conn.setRequestMethod("POST");
|
// 发送POST请求必须设置如下两行
|
conn.setDoOutput(true);
|
conn.setDoInput(true);
|
conn.setUseCaches(false);
|
// 设置请求头参数
|
conn.setRequestProperty("connection", "Keep-Alive");
|
conn.setRequestProperty("Charsert", "UTF-8");
|
conn.setRequestProperty("Content-Type","multipart/form-data; boundary=" + BOUNDARY);
|
conn.setRequestProperty("User-Agent","Mozilla/5.0 (iPhone; CPU iPhone OS 11_0 like Mac OS X) AppleWebKit/604.1.38 (KHTML, like Gecko) Version/11.0 Mobile/15A372 Safari/604.1");
|
OutputStream out = new DataOutputStream(conn.getOutputStream());
|
|
// 上传文件
|
StringBuilder sb = new StringBuilder();
|
sb.append(boundaryPrefix);
|
sb.append(BOUNDARY);
|
sb.append(newLine);
|
// 文件参数,photo参数名可以随意修改
|
sb.append("Content-Disposition: form-data;name=\"image\";filename=\""
|
+ "https://api.weixin.qq.com" + "\"" + newLine);
|
sb.append("Content-Type:application/octet-stream");
|
// 参数头设置完以后需要两个换行，然后才是参数内容
|
sb.append(newLine);
|
sb.append(newLine);
|
|
// 将参数头的数据写入到输出流中
|
out.write(sb.toString().getBytes());
|
|
// 读取文件数据
|
out.write(file.getBytes());
|
// 最后添加换行
|
out.write(newLine.getBytes());
|
|
// 定义最后数据分隔线，即--加上BOUNDARY再加上--。
|
byte[] end_data = (newLine + boundaryPrefix + BOUNDARY
|
+ boundaryPrefix + newLine).getBytes();
|
// 写上结尾标识
|
out.write(end_data);
|
out.flush();
|
out.close();
|
// 定义BufferedReader输入流来读取URL的响应
|
BufferedReader reader = new BufferedReader(new InputStreamReader(
|
conn.getInputStream()));
|
while ((line = reader.readLine()) != null) {
|
return line;
|
}
|
} catch (Exception e) {
|
System.out.println("发送POST请求出现异常！" + e);
|
}
|
return line;
|
}
|
|
/**
|
* 上传二进制文件
|
* @param apiurl 接口地址
|
* @param file 图片文件
|
* @return
|
*/
|
public static String uploadFile(String apiurl, byte[] file) {
|
//接口返回的结果
|
String line = null;
|
try {
|
// 换行符
|
final String newLine = "\r\n";
|
final String boundaryPrefix = "--";
|
// 定义数据分隔线
|
String BOUNDARY = "========7d4a6d158c9";
|
// 服务器的域名
|
URL url = new URL(apiurl);
|
HttpURLConnection conn = (HttpURLConnection) url.openConnection();
|
// 设置为POST情
|
conn.setRequestMethod("POST");
|
// 发送POST请求必须设置如下两行
|
conn.setDoOutput(true);
|
conn.setDoInput(true);
|
conn.setUseCaches(false);
|
// 设置请求头参数
|
conn.setRequestProperty("connection", "Keep-Alive");
|
conn.setRequestProperty("Charsert", "UTF-8");
|
conn.setRequestProperty("Content-Type","multipart/form-data; boundary=" + BOUNDARY);
|
conn.setRequestProperty("User-Agent","Mozilla/5.0 (iPhone; CPU iPhone OS 14_0_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148 MicroMessenger/7.0.15(0x17000f31) NetType/WIFI Language/zh_CN");
|
OutputStream out = new DataOutputStream(conn.getOutputStream());
|
|
// 上传文件
|
StringBuilder sb = new StringBuilder();
|
sb.append(boundaryPrefix);
|
sb.append(BOUNDARY);
|
sb.append(newLine);
|
// 文件参数,photo参数名可以随意修改
|
sb.append("Content-Disposition: form-data;name=\"image\";filename=\""
|
+ "https://api.weixin.qq.com" + "\"" + newLine);
|
sb.append("Content-Type:application/octet-stream");
|
// 参数头设置完以后需要两个换行，然后才是参数内容
|
sb.append(newLine);
|
sb.append(newLine);
|
|
// 将参数头的数据写入到输出流中
|
out.write(sb.toString().getBytes());
|
|
// 读取文件数据
|
out.write(file);
|
// 最后添加换行
|
out.write(newLine.getBytes());
|
|
// 定义最后数据分隔线，即--加上BOUNDARY再加上--。
|
byte[] end_data = (newLine + boundaryPrefix + BOUNDARY
|
+ boundaryPrefix + newLine).getBytes();
|
// 写上结尾标识
|
out.write(end_data);
|
out.flush();
|
out.close();
|
// 定义BufferedReader输入流来读取URL的响应
|
BufferedReader reader = new BufferedReader(new InputStreamReader(
|
conn.getInputStream()));
|
while ((line = reader.readLine()) != null) {
|
return line;
|
}
|
} catch (Exception e) {
|
System.out.println("发送POST请求出现异常！" + e);
|
}
|
return line;
|
}
|
|
/**
|
* 获取 WxAccessToken
|
* @return
|
*/
|
public static String getWxAccessToken() {
|
//参数
|
Map<String, String> params = new HashMap<>();
|
params.put("grant_type", "client_credential");
|
params.put("appid", WX_APP_ID);
|
params.put("secret", WX_SECRET);
|
String result = HttpClientUtils.doGet(WX_GET_ACCESS_TOKEN, params);
|
JSONObject jsonObject = JSON.parseObject(result);
|
String accessToken = jsonObject.getString("access_token");
|
return accessToken;
|
}
|
|
|
|
/**
|
* 压缩图片
|
* @param multipartFile
|
*/
|
public static InputStream compressImage(MultipartFile multipartFile){
|
//*************对不是jpg格式的图片转换成jpg格式***************
|
//获取文件名后缀，判断其格式
|
int begin = multipartFile.getOriginalFilename().lastIndexOf(".");
|
int last = multipartFile.getOriginalFilename().length();
|
//获得文件后缀名
|
String houzuiFileName = multipartFile.getOriginalFilename().substring(begin, last);
|
//创建临时文件
|
File tempFile = new File(multipartFile.getOriginalFilename());
|
//写入临时File文件 tempFile，将multipartFile转换成File
|
try {
|
//import org.apache.commons.io.FileUtils;
|
FileUtils.copyInputStreamToFile(multipartFile.getInputStream(), tempFile);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
//如果文件不是jpg格式,转换其格式
|
if (!"jpg".equalsIgnoreCase(houzuiFileName)){
|
//ImageUtils是一个工具类，下面给出
|
//将png格式转换成jpg，输出到tempFile
|
//测试OK
|
ImageUtils.convert(multipartFile.getOriginalFilename(), "jpg", tempFile.getAbsolutePath());
|
//*************对不是jpg格式的图片转换成jpg格式***************
|
}
|
try {
|
//压缩图片
|
BufferedImage bufferedImage = Thumbnails.of(tempFile)
|
.size(740, 1330)//指定压缩之后的图片尺寸
|
.outputQuality(0.8f)//图片压缩质量
|
.asBufferedImage();
|
ByteArrayOutputStream os = new ByteArrayOutputStream();
|
ImageIO.write(bufferedImage, "jpg", os);
|
InputStream inputStream = new ByteArrayInputStream(os.toByteArray());
|
// System.out.println(inputStream);
|
return inputStream;
|
} catch (IOException e) {
|
e.printStackTrace();
|
}finally {
|
// 会在本地产生临时文件，用完后需要删除
|
if (tempFile.exists()) {
|
tempFile.delete();
|
}
|
}
|
return null;
|
}
|
|
}
|
