wg-basic/basic-util/src/main/java/ink/wgink/util/request/AbstractRestTemplate.java

package ink.wgink.util.request;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.*;
import org.springframework.http.client.SimpleClientHttpRequestFactory;
import org.springframework.web.client.HttpClientErrorException;
import org.springframework.web.client.RestTemplate;

/**
 * When you feel like quitting. Think about why you started
 * 当你想要放弃的时候，想想当初你为何开始
 *
 * @ClassName: AbstractRestTemplateUtil
 * @Description:
 * @Author: wanggeng
 * @Date: 2021/3/31 12:18 下午
 * @Version: 1.0
 */
public abstract class AbstractRestTemplate<SuccessResultType, PostBodyType> {
    private static final Logger LOG = LoggerFactory.getLogger(AbstractRestTemplate.class);
    private RestTemplate restTemplate;

    public AbstractRestTemplate() {
        SimpleClientHttpRequestFactory simpleClientHttpRequestFactory = new SimpleClientHttpRequestFactory();
        simpleClientHttpRequestFactory.setConnectTimeout(20 * 1000);
        simpleClientHttpRequestFactory.setReadTimeout(60 * 1000);
        this.restTemplate = new RestTemplate(simpleClientHttpRequestFactory);
    }

    /**
     * 请求头
     *
     * @return
     */
    public HttpHeaders headers() {
        return new HttpHeaders();
    }

    /**
     * json请求头
     *
     * @return
     */
    public HttpHeaders jsonHeaders() {
        HttpHeaders httpHeaders = headers();
        httpHeaders.setContentType(MediaType.APPLICATION_JSON_UTF8);
        return httpHeaders;
    }

    public HttpHeaders formHeaders() {
        HttpHeaders httpHeaders = headers();
        httpHeaders.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
        return httpHeaders;
    }

    /**
     * POST请求主体
     *
     * @return
     */
    public abstract PostBodyType postBody();

    /**
     * 成功返回类型
     *
     * @return
     */
    public abstract Class<SuccessResultType> successResultType();

    /**
     * GET请求
     *
     * @param url
     */
    public void get(String url) {
        Class<SuccessResultType> successResultTypeClass = successResultType();
        if (successResultTypeClass == null) {
            LOG.error("method successResultType can not return null");
            return;
        }
        try {
            ResponseEntity<SuccessResultType> responseEntity = restTemplate.getForEntity(url, successResultTypeClass);
            SuccessResultType result = responseEntity.getBody();
            if (HttpStatus.OK.value() == responseEntity.getStatusCodeValue()) {
                success200(result);
            }
        } catch (Exception e) {
            if (e instanceof HttpClientErrorException) {
                HttpClientErrorException exception = (HttpClientErrorException) e;
                if (exception.getRawStatusCode() == HttpStatus.BAD_REQUEST.value()) {
                    error400(exception);
                    return;
                }
            }
            e.printStackTrace();
        }
    }

    /**
     * POST提交FORM
     *
     * @param url
     */
    public void post(String url) {
        post(url, false);
    }

    /**
     * POST提交JSON
     *
     * @param url
     */
    public void postJson(String url) {
        post(url, true);
    }

    private void post(String url, boolean isJson) {
        PostBodyType postBodyType = postBody();
        Class<SuccessResultType> successResultTypeClass = successResultType();
        if (postBodyType == null) {
            LOG.error("method postBody can not return null");
            return;
        }
        if (successResultTypeClass == null) {
            LOG.error("method successResultType can not return null");
            return;
        }
        try {
            HttpEntity<PostBodyType> httpEntity = new HttpEntity<>(postBodyType, isJson ? jsonHeaders() : formHeaders());
            ResponseEntity<SuccessResultType> responseEntity = restTemplate.postForEntity(url, httpEntity, successResultTypeClass);
            SuccessResultType result = responseEntity.getBody();
            if (HttpStatus.OK.value() == responseEntity.getStatusCodeValue()) {
                success200(result);
            }
        } catch (Exception e) {
            if (e instanceof HttpClientErrorException) {
                HttpClientErrorException exception = (HttpClientErrorException) e;
                if (exception.getRawStatusCode() == HttpStatus.BAD_REQUEST.value()) {
                    error400(exception);
                    return;
                }
            }
            e.printStackTrace();
        }
    }

    /**
     * 成功结果
     *
     * @param result
     */
    public abstract void success200(SuccessResultType result);

    /**
     * @param error
     */
    public abstract void error400(HttpClientErrorException error);

}