work.gaigeshen.tripartite.qyweixin.openapi.client.AbstractQyWeixinClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tripartite-qyweixin-openapi Show documentation
There is a newer version: 1.22.0
Show newest version
package work.gaigeshen.tripartite.qyweixin.openapi.client;

import lombok.extern.slf4j.Slf4j;
import work.gaigeshen.tripartite.core.client.*;
import work.gaigeshen.tripartite.core.client.accesstoken.AccessToken;
import work.gaigeshen.tripartite.core.client.accesstoken.AccessTokenHelper;
import work.gaigeshen.tripartite.core.client.accesstoken.AccessTokenManager;
import work.gaigeshen.tripartite.core.client.accesstoken.AccessTokenManagerException;
import work.gaigeshen.tripartite.core.client.config.ConfigException;
import work.gaigeshen.tripartite.core.client.parameters.ClientParameters;
import work.gaigeshen.tripartite.core.client.response.ClientResponse;
import work.gaigeshen.tripartite.core.ratelimiter.RateLimiterService;
import work.gaigeshen.tripartite.core.util.ArgumentValidate;
import work.gaigeshen.tripartite.qyweixin.openapi.config.QyWeixinConfig;
import work.gaigeshen.tripartite.qyweixin.openapi.response.QyWeixinResponse;

import java.util.Collection;
import java.util.Objects;

/**
 * 企业微信接口客户端抽象类
 *
 * @author gaigeshen
 */
@Slf4j
public abstract class AbstractQyWeixinClient extends AbstractWebExecutorClient implements BaseQyWeixinClient {

    private final QyWeixinConfig config;

    private final AccessTokenManager accessTokenManager;

    private final RateLimiterService rateLimiterService;

    protected AbstractQyWeixinClient(QyWeixinConfig config, AccessTokenManager accessTokenManager) {
        ArgumentValidate.notNull(config, "config cannot be null");
        ArgumentValidate.notNull(accessTokenManager, "accessTokenManager cannot be null");
        this.config = config;
        this.accessTokenManager = accessTokenManager;
        this.rateLimiterService = RateLimiterService.create(100);
    }

    protected abstract AccessToken getNewAccessToken() throws ClientException;

    @Override
    protected void initInternal() throws ClientException {
        AccessToken accessToken = accessTokenManager.findAccessToken(config);
        if (Objects.nonNull(accessToken) && AccessTokenHelper.isValid(accessToken)) {
            log.info("current valid access token: {}", accessToken);
            return;
        }
        AccessToken newAccessToken = getNewAccessToken();
        try {
            accessTokenManager.addNewAccessToken(config, newAccessToken);
        } catch (AccessTokenManagerException e) {
            throw new ClientException(e.getMessage(), e);
        }
        log.info("new access token added: {}", newAccessToken);
    }

    @Override
    public QyWeixinConfig getConfig() throws ConfigException {
        return config;
    }

    @Override
    public AccessTokenManager getAccessTokenManager() {
        return accessTokenManager;
    }

    @Override
    public RateLimiterService getRateLimiterService() {
        return rateLimiterService;
    }

    @Override
    protected  R validateResponse(R response) throws ClientException {
        R preValidated = super.validateResponse(response);
        if (preValidated instanceof QyWeixinResponse) {
            QyWeixinResponse qyWeixinResponse = (QyWeixinResponse) preValidated;
            if (Objects.nonNull(qyWeixinResponse.errcode) && !Objects.equals(0, qyWeixinResponse.errcode)) {
                throw new ClientException("[ " + qyWeixinResponse.errcode + " ] - [ " + qyWeixinResponse.errmsg + " ]");
            }
        }
        return preValidated;
    }

    @Override
    public ServerHost getServerHost(ClientParameters parameters, Class responseClass) throws ServerHostException {
        return ServerHost.create(config.getServerHost());
    }

    @Override
    public Collection getServerHosts() throws ServerHostException {
        return ServerHosts.create(config.getServerHost());
    }
}