腾讯图片审核Demo

腾讯图片审核Demo

application.yml

tencent:
  image:
    moderation:
      end-point: ims.tencentcloudapi.com
      secret-id: AKID96nEPJwUpvlMhmbAcolS0MUN1PYR5TbU
      secret-key: e3C6lAJDsISM0OXQ3PWNyJODn7C1atwk

ModerationConfig

import lombok.Data;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 曳戈泰尔
 * @date 2023/3/14 11:50 AM
 */
@Configuration
@Data
public class ModerationConfig {

    @Value("${tencent.image.moderation.end-point}")
    private String endPoint;

    @Value("${tencent.image.moderation.secret-id}")
    private String tencentSecretId;

    @Value("${tencent.image.moderation.secret-key}")
    private String tencentSecretKey;
}

ImageModeration

import com.alibaba.fastjson.JSONObject;
import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.ims.v20201229.ImsClient;
import com.tencentcloudapi.ims.v20201229.models.ImageModerationRequest;
import com.tencentcloudapi.ims.v20201229.models.ImageModerationResponse;
import lombok.extern.slf4j.Slf4j;
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 ImageModeration {

    /** Moderation配置类 */
    private ModerationConfig moderationConfig;

    @Autowired
    public ImageModeration(ModerationConfig moderationConfig) {
        this.moderationConfig = moderationConfig;
    }

    public Boolean execImageModeration(String fileUrl) {
        try {
            // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
            // 代码泄露可能会导致 SecretId 和 SecretKey
            // 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
            // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
            Credential cred =
                    new Credential(
                            moderationConfig.getTencentSecretId(),
                            moderationConfig.getTencentSecretKey());

            // 实例化一个http选项，可选的，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            httpProfile.setEndpoint(moderationConfig.getEndPoint());

            // 实例化一个client选项，可选的，没有特殊需求可以跳过
            ClientProfile clientProfile = new ClientProfile();
            clientProfile.setHttpProfile(httpProfile);

            // 实例化要请求产品的client对象,clientProfile是可选的
            ImsClient client = new ImsClient(cred, "ap-beijing", clientProfile);

            // 实例化一个请求对象,每个接口都会对应一个request对象
            ImageModerationRequest req = new ImageModerationRequest();
            req.setFileUrl(fileUrl);

            // 返回的resp是一个ImageModerationResponse的实例，与请求对象对应
            ImageModerationResponse resp = client.ImageModeration(req);

            // 输出json格式的字符串回包
            log.info("腾讯内容安全检测结果：{}", ImageModerationResponse.toJsonString(resp));

            // 检测返回结果是否Normal
            return "Normal"
                    .equals(
                            JSONObject.parseObject(ImageModerationResponse.toJsonString(resp))
                                    .get("Label")
                                    .toString());
        } catch (Exception e) {
            log.error("腾讯内容安全检测失败，原因：{}", e.getMessage());

            return false;
        }
    }
}