com.tencent.devops.common.client.ClientAutoConfiguration.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-client Show documentation
Tencent blueking ci project
There is a newer version: 3.1.0-rc.4
/*
 * Tencent is pleased to support the open source community by making BK-CI 蓝鲸持续集成平台 available.
 *
 * Copyright (C) 2019 THL A29 Limited, a Tencent company.  All rights reserved.
 *
 * BK-CI 蓝鲸持续集成平台 is licensed under the MIT license.
 *
 * A copy of the MIT License is included in this file.
 *
 *
 * Terms of the MIT License:
 * ---------------------------------------------------
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
 * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of
 * the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
 * LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
 * NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

package com.tencent.devops.common.client

import com.fasterxml.jackson.databind.ObjectMapper
import com.tencent.devops.common.client.consul.ConsulFilter
import com.tencent.devops.common.redis.RedisOperation
import com.tencent.devops.common.service.BkTag
import com.tencent.devops.common.service.ServiceAutoConfiguration
import com.tencent.devops.common.service.config.CommonConfig
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.beans.factory.annotation.Value
import org.springframework.boot.autoconfigure.AutoConfigureAfter
import org.springframework.boot.autoconfigure.AutoConfigureOrder
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean
import org.springframework.cloud.client.discovery.composite.CompositeDiscoveryClient
import org.springframework.cloud.client.loadbalancer.LoadBalancerAutoConfiguration
import org.springframework.context.annotation.Bean
import org.springframework.context.annotation.Configuration
import org.springframework.context.annotation.PropertySource
import org.springframework.core.Ordered

/**
 *
 * Powered By Tencent
 */
@Configuration
@PropertySource("classpath:/common-client.properties")
@AutoConfigureOrder(Ordered.HIGHEST_PRECEDENCE)
@AutoConfigureAfter(ServiceAutoConfiguration::class, LoadBalancerAutoConfiguration::class)
class ClientAutoConfiguration {

    @Value("\${spring.cloud.consul.discovery.tags:prod}")
    private val consulTag: String = "prod"

    @Bean
    @ConditionalOnMissingBean(CommonConfig::class)
    fun commonConfig() = CommonConfig()

    @Bean
    fun bkTag(): BkTag {
        val finalConsulTag = consulTag.split(',')[0].trim()
        return BkTag(finalConsulTag)
    }

    @Bean
    fun clientErrorDecoder(objectMapper: ObjectMapper) = ClientErrorDecoder(objectMapper)

    @Bean
    @ConditionalOnMissingBean(Client::class)
    fun client(
        clientErrorDecoder: ClientErrorDecoder,
        commonConfig: CommonConfig,
        bkTag: BkTag,
        objectMapper: ObjectMapper,
        @Autowired(required = false) compositeDiscoveryClient: CompositeDiscoveryClient?
    ) = Client(compositeDiscoveryClient, clientErrorDecoder, commonConfig, bkTag, objectMapper)

    @Bean
    fun consulFilter(bkTag: BkTag) = ConsulFilter(bkTag)

    @Bean
    fun clientTokenService(redisOperation: RedisOperation, bkTag: BkTag) = ClientTokenService(redisOperation, bkTag)
}