字节图片内容安全接入

字节图片内容安全接入

AccessToken

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.yqxc.aiface.common.http.HttpClientHelper;
import com.yqxc.aiface.common.redis.helper.RedisHelper;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 曳戈泰尔
 * @date 2023/4/7 4:52 PM
 */
@Component
@Slf4j
public class AccessToken {

    /** 注入配置类 */
    private final DouConfig douConfig;

    @Autowired
    public AccessToken(DouConfig douConfig) {
        this.douConfig = douConfig;
    }

    /** 注入Redis */
    private RedisHelper redisHelper;

    @Autowired
    public void setRedisHelper(RedisHelper redisHelper) {
        this.redisHelper = redisHelper;
    }

    /** 初始化的时候执行一次 */
    @PostConstruct
    public void init() {
        // 获取access_token
        getAccessToken();
    }

    /** 每 100 分钟执行一次 */
    @Scheduled(fixedRate = 100 * 60 * 1000)
    public void schedule() {
        // 获取access_token
        getAccessToken();
    }

    /** 获取access_token */
    public void getAccessToken() {

        // 封装请求参数
        JSONObject jsonObject = new JSONObject();
        jsonObject.put("appid", douConfig.getAppId());
        jsonObject.put("secret", douConfig.getSecret());
        jsonObject.put("grant_type", "client_credential");

        // 发送请求
        String result =
                HttpClientHelper.doPostJson(
                        douConfig.getAccessTokenEndPoint(), JSON.toJSONString(jsonObject));
        // 判断返回结果是否为空
        if (StringUtils.isBlank(result)) {
            log.error("获取access_token失败");

            return;
        }

        // 解析返回结果
        JSONObject resultJson = JSON.parseObject(result);
        // 判断返回结果是否成功
        if (resultJson.getInteger("err_no") != 0) {
            log.error(
                    "获取access_token失败，错误码：{}，错误信息：{}",
                    resultJson.getInteger("err_no"),
                    resultJson.getString("err_tips"));

            return;
        }

        // 获取access_token
        String accessToken =
                JSON.parseObject(resultJson.getString("data")).getString("access_token");
        // 判断access_token是否为空
        if (StringUtils.isBlank(accessToken)) {
            log.error("获取access_token失败");

            return;
        }

        // 设置access_token到Redis
        // 超时时间设置为110分钟
        String key = "douyin_access_token:";
        redisHelper.set(key, accessToken, 110 * 60L);
    }
}

ImageDouyinModeration

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.yqxc.aiface.common.http.HttpClientHelper;
import com.yqxc.aiface.common.mergeFace.config.FaceConfig;
import com.yqxc.aiface.common.redis.helper.RedisHelper;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 抖音 内容安全检测
 * @date 2023/3/25 3:40 PM
 */
@Component
@Slf4j
public class ImageDouyinModeration {

    /** Face配置类 */
    private FaceConfig faceConfig;

    @Autowired
    public ImageDouyinModeration(FaceConfig faceConfig) {
        this.faceConfig = faceConfig;
    }

    /** 注入Redis */
    private RedisHelper redisHelper;

    @Autowired
    public void setRedisHelper(RedisHelper redisHelper) {
        this.redisHelper = redisHelper;
    }

    public Boolean execImageModeration(String fileUrl) {

        // 获取access_token
        String key = "douyin_access_token:";
        String accessToken = redisHelper.get(key);
        if (StringUtils.isBlank(accessToken)) {
            log.error("获取抖音access_token失败");

            return false;
        }

        // 构造请求参数
        JSONObject jsonObject = new JSONObject();
        jsonObject.put("app_id", faceConfig.getDouyinAppId());
        jsonObject.put("access_token", accessToken);
        jsonObject.put("image", fileUrl);

        // 发送请求
        String result =
                HttpClientHelper.doPostJson(
                        faceConfig.getDouyinScanImageEndPoint(), jsonObject.toJSONString());
        // 检查返回结果
        if (StringUtils.isBlank(result)) {
            log.error("抖音内容安全检测失败");

            return false;
        }

        // 解析返回结果
        JSONObject resultJson = JSON.parseObject(result);
        if (resultJson.getInteger("error") != 0) {
            log.error("抖音内容安全检测失败, 错误信息: {}", resultJson.getString("message"));

            return false;
        }

        // 获取置信度列表
        JSONArray jsonArray = resultJson.getJSONArray("predicts");
        if (jsonArray == null || jsonArray.isEmpty()) {
            log.error("抖音内容安全检测失败, 错误信息: {}", resultJson.getString("message"));

            return false;
        }

        // 遍历获取检测结果
        boolean whetherPass = true;
        for (Object o : jsonArray) {
            JSONObject jsonObj = (JSONObject) o;

            // 存在非法标签
            if (Boolean.TRUE.equals(jsonObj.getBoolean("hit"))) {
                whetherPass = false;

                break;
            }
        }

        return whetherPass;
    }
}

使用

Boolean scanImageWhether = imageDouyinModeration.execImageModeration(ossUrl);
if (Boolean.FALSE.equals(scanImageWhether)) {

    // 删除OSS资源
    String fileName = ossUrl.substring("https://face.folkspace.cn/aiFace/".length());
    ossHelper.deleteFile(fileName);

    throw new AiException("图片中含有敏感或违规信息");
}