com.kg.component.utils.BaiduAccessToken Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dashu-frame-component Show documentation

大树快速开发平台::Component::组件库

There is a newer version: 1.1.26

Show newest version

package com.kg.component.utils;

import cn.hutool.json.JSONObject;
import com.baidubce.http.ApiExplorerClient;
import com.baidubce.http.HttpMethodName;
import com.baidubce.model.ApiExplorerRequest;
import com.baidubce.model.ApiExplorerResponse;
import com.kg.component.redis.RedisUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

/**
 * @author yrt
 * @date 2022/7/26 9:08
 */
@Component
public class BaiduAccessToken {

    @Autowired
    private RedisUtils redisUtils;

    private static String CLIENT_ID;
    private static String CLIENT_SECRET;
    private static String GRANT_TYPE;

    /**
     * 百度API Key
     */
    @Value("${com.kg.baidu.client-id}")
    public void setClientId(String str) {
        CLIENT_ID = str;
    }

    /**
     * 百度Secret Key
     */
    @Value("${com.kg.baidu.client-secret}")
    public void setClientSecret(String str) {
        BaiduAccessToken.CLIENT_SECRET = str;
    }

    /**
     * 百度grant_type
     */
    @Value("${com.kg.baidu.grant-type}")
    public void setGrantType(String str) {
        BaiduAccessToken.GRANT_TYPE = str;
    }

    // 获取AccessToken
    public String GetAccessToken() {
        try {
            if (redisUtils.hasKey("accessToken")) {
                return redisUtils.get("accessToken").toString();
            }

            String path = "https://aip.baidubce.com/oauth/2.0/token";
            ApiExplorerRequest request = new ApiExplorerRequest(HttpMethodName.POST, path);


            // 设置header参数
            request.addHeaderParameter("Content-Type", "application/json;charset=UTF-8");

            // 设置query参数
            request.addQueryParameter("client_id", CLIENT_ID);
            request.addQueryParameter("client_secret", CLIENT_SECRET);
            request.addQueryParameter("grant_type", GRANT_TYPE);


            ApiExplorerClient client = new ApiExplorerClient();
            ApiExplorerResponse response = client.sendRequest(request);
            JSONObject jsonObj = new JSONObject(response.getResult());
            String accessToken = jsonObj.get("access_token").toString();
            Long expiresIn =  jsonObj.getLong("expires_in");
            redisUtils.set("accessToken", accessToken, expiresIn - 1000);
            // 返回结果格式为Json字符串
            return accessToken;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
}