com.lehuipay.leona.HttpClient Maven / Gradle / Ivy

Go to download
package com.lehuipay.leona;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.lehuipay.leona.exception.LeonaErrorCodeEnum;
import com.lehuipay.leona.exception.LeonaException;
import com.lehuipay.leona.exception.LeonaRuntimeException;
import com.lehuipay.leona.interceptor.L1Interceptor;
import com.lehuipay.leona.interceptor.L2Interceptor;
import com.lehuipay.leona.interceptor.SignInterceptor;
import com.lehuipay.leona.model.ErrorMessage;
import com.lehuipay.leona.utils.CommonUtil;
import okhttp3.Call;
import okhttp3.MediaType;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;
import org.jetbrains.annotations.NotNull;

import java.io.IOException;
import java.time.Duration;

public class HttpClient {

    private final MediaType mediaTypeJSON = MediaType.parse("application/json; charset=utf-8");

    public HttpClient(Options options) {
        final HMACSigner signer = new HMACSigner(options.getAgentID(), options.getAgentKey());
        final SignInterceptor signInterceptor = new SignInterceptor(signer, options.getAgentID());
        final OkHttpClient.Builder builder = new OkHttpClient.Builder()
                .callTimeout(Duration.ofSeconds(10));

        switch (CommonUtil.NVLL(options.getEncryptionLevel())) {
            case Const.HEADER_ENCRYPTION_LEVEL_L1:
                L1Interceptor l1 = new L1Interceptor(
                        new AESEncryptor(), new RSAEnctryptor(options.getPartnerPriKey(), options.getLhPubKey()), options.getEncryptionAccept()
                );
                builder.addInterceptor(l1);
                break;
            case Const.HEADER_ENCRYPTION_LEVEL_L2:
                L2Interceptor l2 =
                        new L2Interceptor(
                                new AESEncryptor(), options.getSecretKey(), options.getEncryptionAccept());
                builder.addInterceptor(l2);
                break;
        }

        client = builder
                .addInterceptor(signInterceptor)
                .build();
    }

    private final OkHttpClient client;

    /**
     * 同步http请求
     *
     * @param method http方法类型
     * @param url http请求url
     * @param data httpRequestBody
     * @param clazz httpResponseBody marshal类型
     * @param  httpRequestBody的泛型类
     * @param  httpResponseBody的泛型类类
     * @return 指定泛型T的返回值
     * @throws IOException http异常
     * @throws LeonaException 若返回的http code > 300, 则将responseBody中的信息封装为LeonaException
     */
    public  T request(final String method, final String url, R data, Class clazz) throws IOException, LeonaException {
        final Request request = buildRequest(method, url, data);
        final Response response = client.newCall(request).execute();
        return parseResponse(response, clazz);
    }

    /**
     * 异步http请求
     *
     * @param method http方法类型
     * @param url http请求url
     * @param data httpRequestBody
     * @param clazz httpResponseBody marshal类型
     * @param callback 异步请求的callback方法,
     *                 com.lehuipay.leona.Callback接口包含一个方法 void callback(LeonaException e, T data);
     *                 当返回的http code > 300, server返回信息会被包装为LeonaException
     * @param  httpRequestBody的泛型类
     * @param  httpResponseBody的泛型类类
     */
    public  void request(final String method, final String url, R data, Class clazz, Callback callback) {
        final Request request = buildRequest(method, url, data);

        client.newCall(request).enqueue(new okhttp3.Callback() {
            @Override
            public void onFailure(@NotNull Call call, @NotNull IOException e) {
                callback.callback(new LeonaException(new LeonaRuntimeException(LeonaErrorCodeEnum.HTTP_ERROR, e)), null);
            }

            @Override
            public void onResponse(@NotNull Call call, @NotNull Response response) throws IOException {
                try {
                    final T result = parseResponse(response, clazz);
                    callback.callback(null, result);
                } catch (LeonaException e) {
                    callback.callback(e, null);
                }
            }
        });
    }

    private  Request buildRequest(String method, String url, T data) {
        String requestBody = JSONObject.toJSONString(data);
        return new Request.Builder()
                .url(url)
                .method(method, RequestBody.create(requestBody, mediaTypeJSON))
                .build();
    }

    private  T parseResponse(Response response, Class clazz) throws IOException, LeonaException {
        String bodyStr = response.body() == null ? "" : response.body().string();

        if (response.isSuccessful()) {
            return JSON.parseObject(bodyStr, clazz);
        } else {
            final ErrorMessage errorMessage = JSON.parseObject(bodyStr, ErrorMessage.class);
            throw new LeonaException(errorMessage);
        }
    }
}