SpringBoot 2.X Kotlin系列之RestTemplate配置及使用

image

自從RESTFul API興起後,Spring就給開發者提供了一個訪問Rest的客服端,RestTemplate不僅可以很方便的調用http接口,還可以調用同一註冊中心下的微服務,同時還有負載均衡和熔斷機制。當然我也聽說OKhttp,HTTPClient之類的網絡框架,但是我們這裏重點講的是RestTemplate(其中一種實現就是基於HTTPClient實現的),因爲這是SpringBoot自帶,而且基本使用場景都覆蓋了。

什麼是RestTemplate?

RestTemplate是Spring提供的用於訪問Rest服務的客戶端,

RestTemplate提供了多種便捷訪問遠程Http服務的方法,能夠大大提高客戶端的編寫效率。

調用RestTemplate的默認構造函數,RestTemplate對象在底層通過使用java.net包下的實現創建HTTP 請求,

可以通過使用ClientHttpRequestFactory指定不同的HTTP請求方式。

ClientHttpRequestFactory接口主要提供了兩種實現方式

1、一種是SimpleClientHttpRequestFactory,使用J2SE提供的方式(既java.net包提供的方式)創建底層的Http請求連接。

2、一種方式是使用HttpComponentsClientHttpRequestFactory方式,底層使用HttpClient訪問遠程的Http服務,使用HttpClient可以配置連接池和證書等信息。

簡單配置RestTemplate

@Configuration
class RestTemplateConfig {

    /**
     * 構建一個使用默認配置RestTemplate Bean
     */
    @Bean
    fun restTemplate():RestTemplate {
        return RestTemplateBuilder().build()
    }
}

發送POST請求,通過Form表單提交參數

@Component
class RestTemplateClient {

    @Autowired lateinit var restTemplate: RestTemplate

    val logger:Logger = LoggerFactory.getLogger(this.javaClass)

    /**
     * 發送POST請求,通過Form表單提交參數
     */
    fun <T> postForm(params: Map<String, String>, url: String, t:T) : ResultDTO<T>? {
        logger.info("請求URL===========:{}", url)
        logger.info("請求參數===========:{}", params)
        val httpHeaders = HttpHeaders()
        httpHeaders.contentType = MediaType.APPLICATION_FORM_URLENCODED
        val requestParams = LinkedMultiValueMap<String, String>()
        params.forEach(requestParams::add)
        val httpEntity = HttpEntity<MultiValueMap<String, String>>(requestParams, httpHeaders)
        return restTemplate.postForObject(url, httpEntity, ResultDTO<T>().javaClass)
    }
}

編寫Controller及測試

@RestController
@RequestMapping("/api")
class LotteryController {

    @Autowired lateinit var restTemplateClient: RestTemplateClient

    /**
     * 彩票類型查詢
     */
    @PostMapping("/lottery/types")
    fun lotteryTypes(@RequestBody params:Map<String, String>) : ResultDTO<LotteryDTO>? {
        return restTemplateClient.postForm(params, UrlEnums.DEMO1.url, LotteryDTO())
    }
}

image

發送POST請求,通JSON傳參

/**
 * 發送POST請求,通過Form表單提交參數
 */
fun <T> postBody(params: Map<String, String>, url: String, t:T) : ResultDTO<T>? {
    logger.info("請求URL===========:{}", url)
    logger.info("請求參數===========:{}", params)
    val httpHeaders = HttpHeaders()
    httpHeaders.contentType = MediaType.APPLICATION_JSON_UTF8
    val httpEntity = HttpEntity(params, httpHeaders)
    return restTemplate.postForObject(url, httpEntity, ResultDTO<T>().javaClass)
}

發送GET請求

/**
 * 發送POST請求,通過Form表單提交參數
 */
fun <T> get(params: Map<String, String>, url: String, t:T) : ResultDTO<T>? {
    logger.info("請求URL===========:{}", url)
    logger.info("請求參數===========:{}", params)
    val builder = StringBuilder()
    params.forEach{k, v -> builder.append(k).append("=").append(v).append("&") }
    val param : String = builder.toString().substring(0, builder.length - 1)
    return restTemplate.getForObject(url.plus(param), ResultDTO<T>().javaClass)
}

GET下載資源

/**
 * 下載資源
 */
fun downGet(url: String, headerParams: Map<String, String>) : ByteArray? {
    logger.info("資源URL==========>:{}", url)
    val httpHeaders = HttpHeaders()
    if (!CollectionUtils.isEmpty(headerParams)) {
        headerParams.forEach(httpHeaders::add)
    }
    val httpEntity = HttpEntity(LinkedMultiValueMap<String, String>().putAll(httpHeaders))
    return restTemplate.exchange(url, HttpMethod.GET, httpEntity, ByteArray::class.java).body
}

調用微服務接口

我們在開發微服務應用的時候,有時候一個應用可能會部署很多臺,而且還是不在不通的機器上,這個時候我們可以用Nginx做一個負載均衡,但是我們這裏推薦使用的是SpringCloud提供的負載均衡,只要在Bean上面加一個註解即可@LoadBalanced,如下所示:

@Bean
@LoadBalanced
fun restTemplate():RestTemplate {
    return RestTemplateBuilder().build()
}

和調用普通的http請求有點不同的地方是,調用微服務是http://application.name/xxx,我們訪問的路徑是微服務應用名稱+業務路由,因爲服務會註冊到註冊中心,註冊中心會維護這個應用名稱,當我們通過應用去訪問時,就可以知道這個應用在那臺機器上了。

基礎使用講完了,但是這裏還存在一個問題,因爲我們使用的是最簡單的配置,在這裏我們沒有設置默認超時時間,官方默認的是60S,但是我們業務中通常對響應速度是有要求,一分鐘的等待時間對於用戶來說太長了,這個時候我們就需要有一個超時機制,具體的設置如下:

package io.intodream.kotlin05.config

import org.apache.http.impl.client.HttpClientBuilder
import org.springframework.beans.factory.annotation.Value
import org.springframework.boot.context.properties.ConfigurationProperties
import org.springframework.boot.web.client.RestTemplateBuilder
import org.springframework.context.annotation.Bean
import org.springframework.context.annotation.Configuration
import org.springframework.http.client.ClientHttpRequestFactory
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory
import org.springframework.http.client.SimpleClientHttpRequestFactory
import org.springframework.http.converter.HttpMessageConverter
import org.springframework.http.converter.StringHttpMessageConverter
import org.springframework.stereotype.Component
import org.springframework.web.client.RestTemplate
import java.nio.charset.StandardCharsets

/**
 * @description
 * RestTemplate 配置
 * @author Jwenk
 * @copyright intoDream.io 築夢科技
 * @email [email protected]
 * @date 2019-04-05,22:42
 */
@Configuration
class RestTemplateConfig {

    @Value("\${remote.maxTotalConnect}")
    val maxTotalConnect: Int = 0
    @Value("\${remote.maxConnectPerRoute}")
    val maxConnectPerRoute: Int = 200
    @Value("\${remote.connectTimeout}")
    val connectTimeout: Int = 2000
    @Value("\${remote.readTimeout}")
    val readTimeout: Int = 3000

    /**
     * 創建HTTP客戶端工廠
     */
    private fun createFactory() : ClientHttpRequestFactory {
        if (this.maxTotalConnect <= 0) {
            val factory = SimpleClientHttpRequestFactory()
            factory.setReadTimeout(this.readTimeout)
            factory.setConnectTimeout(this.connectTimeout)
            return factory
        }
        val httpClient = HttpClientBuilder.create()
                .setMaxConnTotal(this.maxTotalConnect)
                .setMaxConnPerRoute(this.maxConnectPerRoute).build()
        val factory = HttpComponentsClientHttpRequestFactory(httpClient)
        factory.setConnectTimeout(this.connectTimeout)
        factory.setReadTimeout(this.readTimeout)
        return factory
    }

    /**
     * 構建一個使用默認配置RestTemplate Bean
     */
    @Bean
    fun restTemplate():RestTemplate {
        val restTemplate = RestTemplate(this.createFactory())
        val converterList : MutableList<HttpMessageConverter<*>> = restTemplate.messageConverters
        var converterTarget : HttpMessageConverter<*>? = null
        /**
         * 重新設置StringHttpMessageConverter字符集爲UTF-8,解決中文亂碼問題
         */
        for (item: HttpMessageConverter<*> in converterList) {
            if (StringHttpMessageConverter::class.java == item.javaClass) {
                converterTarget = item
                break
            }
        }
        if (null != converterTarget) {
            converterList.remove(converterTarget)
        }
        converterList.add(1, StringHttpMessageConverter(StandardCharsets.UTF_8))
        return restTemplate
    }
}

記得在pom.xml裏面引入HttpClient的jar,不然啓動會報錯的

<dependency>
    <groupId>org.apache.httpcomponents</groupId>
    <artifactId>httpclient</artifactId>
    <version>4.5.6</version>
</dependency>

關於RestTemplate的一些使用及配置優化到此就結束了,喜歡的朋友可以關注本人的博客https://www.tisnz.com,如果有不對的地方歡迎指正。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章