Java后台微信内容安全接口,文字敏感，图片鉴黄

以你之姓@ 2022-12-09 01:57 129阅读 0赞

官方文档:[https://developers.weixin.qq.com/miniprogram/dev/api-backend/open-api/sec-check/security.msgSecCheck.html  
][https_developers.weixin.qq.com_miniprogram_dev_api-backend_open-api_sec-check_security.msgSecCheck.html]

UploadAction类

import org.springframework.web.multipart.MultipartFile;
    
    import java.io.BufferedReader;
    import java.io.DataOutputStream;
    import java.io.InputStreamReader;
    import java.io.OutputStream;
    import java.net.HttpURLConnection;
    import java.net.URL;
    
    public class UploadAction { 
        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;charset=UTF-8");
                // 参数头设置完以后需要两个换行，然后才是参数内容
                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;
        }
    }

返回结果实体类

import com.alibaba.fastjson.JSON;
    import com.shifen.articleservice.utils.UploadAction;
    import org.springframework.http.*;
    import org.springframework.web.client.RestTemplate;
    import org.springframework.web.multipart.MultipartFile;
    
    import java.util.HashMap;
    import java.util.Map;
    import java.util.concurrent.RecursiveTask;
    
    public class CheckTask extends RecursiveTask<AccessTokenWX> { 
        private String url;
        private MultipartFile file;
        private String content;
    
        public CheckTask() { 
        }
    
        public CheckTask(String url, MultipartFile file, String content) { 
            this.url = url;
            this.file = file;
            this.content = content;
        }
    
    
        @Override
        protected AccessTokenWX compute() { 
            AccessTokenWX result = new AccessTokenWX();
            try { 
                if (null == content) { 
                    String json = UploadAction.uploadFile(url, file);
                    result = JSON.parseObject(json, AccessTokenWX.class);
                } else { 
                    RestTemplate rest = new RestTemplate();
                    Map<String, Object> param = new HashMap<String, Object>();
                    param.put("content", content);
                    HttpHeaders headers = new HttpHeaders();
                    headers.setContentType(MediaType.APPLICATION_JSON);
                    HttpEntity requestEntity = new HttpEntity(param, headers);
                    ResponseEntity<byte[]> entity = rest.exchange(url, HttpMethod.POST, requestEntity, byte[].class, new Object[0]);
                    String json = new String(entity.getBody(), "utf-8");
                    result = JSON.parseObject(json, AccessTokenWX.class);
                }
            } catch (Exception e) { 
                result.setErrCode("500");
                result.setErrMsg("system错误");
            }
            return result;
        }
    }

CheckTask类

工具类

import com.shifen.articleservice.entity.AccessTokenWX;
    import com.shifen.articleservice.entity.CheckTask;
    import lombok.extern.slf4j.Slf4j;
    import org.springframework.web.multipart.MultipartFile;
    
    import java.util.HashMap;
    import java.util.Map;
    import java.util.concurrent.ExecutionException;
    import java.util.concurrent.ForkJoinPool;
    import java.util.concurrent.ForkJoinTask;
    
    /** * 微信检测违规图片或文字接口 */
    @Slf4j
    public class WeiXinViolationCheckUtil { 
        /** * 微信图片检测接口 */
        static String IMG_CHECK = "https://api.weixin.qq.com/wxa/img_sec_check?access_token=";
        /** * 微信文字检测接口 */
        static String TXT_CHECK = "https://api.weixin.qq.com/wxa/msg_sec_check?access_token=";
    
        /** * file:图片路径地址 * content:文字内容 * token:微信token */
        public static Map checkImgOrMsg(MultipartFile file, String content, String token) throws ExecutionException, InterruptedException { 
            Map<String, Object> resultMap = new HashMap<>();
            ForkJoinPool pool = new ForkJoinPool();
            String contentUrl = TXT_CHECK + token;
            String imgUrl = IMG_CHECK + token;
            // 返回为0说明检测正常
            String ok = "0";
            ForkJoinTask<AccessTokenWX> contentResult = null;
            ForkJoinTask<AccessTokenWX> imgResult = null;
            resultMap.put("all", "ok");
    
            if (null != content && content.trim().length() > 0) { 
                CheckTask contentTask = new CheckTask(contentUrl, null, content);
                contentResult = pool.submit(contentTask).fork();
                log.info("文字检测结果===" + contentResult.get());
            }
            if (null != file && !file.isEmpty()) { 
                CheckTask contentTask = new CheckTask(imgUrl, file, null);
                imgResult = pool.submit(contentTask).fork();
                log.info("图片检测结果===" + imgResult.get());
            }
            if (null != contentResult) { 
                resultMap.put("content", contentResult.get().getErrCode());
                if (!ok.equals(contentResult.get().getErrCode())) { 
                    resultMap.put("all", "err");
                }
            }
            if (null != imgResult) { 
                resultMap.put("img", imgResult.get().getErrCode());
                if (!ok.equals(imgResult.get().getErrCode())) { 
                    resultMap.put("all", "err");
                }
            }
            return resultMap;
        }
    }

Controller

static Long oldTime = 0L;
        static String tokenCache = "";
        // 过期时间,2h-1m
        static Long outTime = 7200000L - 60000;
    
        public String getToken() { 
            String GET_TOKEN = "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=" + "你的AppId" + "&secret=" + "你的AppSecret";
            System.out.println(GET_TOKEN);
            Long nowTime = System.currentTimeMillis();
            Long cacheTime = nowTime - oldTime;
            if (oldTime <= 0 || cacheTime > outTime) { 
                RestTemplate restTemplate;
                restTemplate = new RestTemplate();
                String json = restTemplate.getForObject(GET_TOKEN, String.class);
                tokenCache = json.split("\"access_token\":\"")[1].split("\",\"")[0].trim();
                oldTime = System.currentTimeMillis();
                return tokenCache;
            } else { 
                return tokenCache;
            }
        }
        //图片检测
         MultipartHttpServletRequest req = (MultipartHttpServletRequest) request;
            MultipartFile file = req.getFile("file");
            String token = getToken();
            Map map = WeiXinViolationCheckUtil.checkImgOrMsg(file, null, token);
            System.out.println(map.get("all"));
            if (!map.get("all").equals("ok")) { 
                return Result.fail("图片包含敏感信息");
            }
           
           
            //文字，学IT找我V:15969390390
            String token = getToken();
                Map map = WeiXinViolationCheckUtil.checkImgOrMsg(null, "检测的文字内容", token);
                if (!map.get("all").equals("ok")) { 
                    return Result.fail("内含有敏感信息");
                }
        }

[https_developers.weixin.qq.com_miniprogram_dev_api-backend_open-api_sec-check_security.msgSecCheck.html]: https://developers.weixin.qq.com/miniprogram/dev/api-backend/open-api/sec-check/security.msgSecCheck.html