com.luna.baidu.api.BaiduGoodsIdentifyApi Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of baidu-spring-boot-starter Show documentation

Baidu API Toolkit

There is a newer version: 2.1.0-RELEASE

Show newest version

package com.luna.baidu.api;

import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.List;

import com.luna.common.file.FileUtils;
import com.luna.common.net.HttpUtils;
import com.luna.common.net.HttpUtilsConstant;
import com.luna.common.text.Base64Util;
import org.apache.http.HttpResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.alibaba.fastjson.JSON;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Maps;
import com.luna.baidu.dto.goods.GoodsInfoDTO;

/**
 * @author Luna@win10
 * @date 2020/4/30 13:48
 */
public class BaiduGoodsIdentifyApi {

    private static final Logger log = LoggerFactory.getLogger(BaiduGoodsIdentifyApi.class);

    /**
     * 物品人像识别 可联系百度百科
     * 
     * @param key
     * @param image
     * @param baikeNum 联系百度百科条目结果数
     * @return
     * @throws UnsupportedEncodingException
     */
    public static List goodsIdentify(String key, String image, Integer baikeNum)
        throws UnsupportedEncodingException {
        log.info("goodsIdentify start");
        HashMap params = Maps.newHashMap();
        if (Base64Util.isBase64(image)) {
            params.put("image", image);
        } else {
            params.put("image", Base64Util.encodeBase64(FileUtils.readFileToBytes(image)));
        }

        if (baikeNum != null) {
            params.put("baike_num", baikeNum);
        }
        HttpResponse httpResponse = HttpUtils.doPost(BaiduApiConstant.HOST, BaiduApiConstant.GOODS_IDENTIFY,
            ImmutableMap.of("Content-Type", HttpUtilsConstant.X_WWW_FORM_URLENCODED),
            ImmutableMap.of("access_token", key),
            HttpUtils.urlencode(params));
        String s = HttpUtils.checkResponseAndGetResult(httpResponse, true);
        List goodsInfoDTOS = JSON.parseArray(JSON.parseObject(s).getString("result"), GoodsInfoDTO.class);
        log.info("goodsIdentify success goodsInfoDTOS={}", goodsInfoDTOS);
        return goodsInfoDTOS;
    }
}