com.tencent.devops.common.web.GlobalProxyConfiguration.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-web Show documentation
Tencent blueking ci project
There is a newer version: 3.1.0-rc.4
package com.tencent.devops.common.web

import com.tencent.devops.common.web.proxy.CustomProxySelector
import org.slf4j.LoggerFactory
import org.springframework.beans.factory.annotation.Value
import org.springframework.context.annotation.Configuration
import org.springframework.core.Ordered
import org.springframework.core.annotation.Order
import java.net.InetSocketAddress
import java.net.Proxy
import java.net.ProxySelector

@Order(Ordered.HIGHEST_PRECEDENCE)
@Configuration("globalProxyConfiguration")
class GlobalProxyConfiguration(
    /**
     * 是否开启，设置为 true 才生效
     */
    @Value("\${net.proxy.enable:false}")
    private val enableProxy: String,
    /**
     *  需要代理的hosts，多个使用","隔开，支持正则表达式
     */
    @Value("\${net.proxy.hosts:}")
    private val proxyHosts: String,
    /**
     *  代理服务器类型，可 HTTP, SOCKS
     */
    @Value("\${net.proxy.server.type:}")
    private val proxyServerType: String,
    /**
     *  代理服务器主机，host 或者 ip
     */
    @Value("\${net.proxy.server.host:}")
    private val proxyServerHost: String,
    /**
     *  代理服务器端口
     */
    @Value("\${net.proxy.server.port:}")
    private val proxyServerPort: String
) {

    companion object {
        private val spliterator = Regex(",")
        private val LOG = LoggerFactory.getLogger(GlobalProxyConfiguration::class.java.name)
    }

    init {
        proxyConfiguration()
    }

    private fun proxyConfiguration() {
        LOG.info("proxy configuration enable: $enableProxy")
        LOG.info("proxy configuration proxyHosts: $proxyHosts")
        LOG.info("proxy configuration proxyServerType: $proxyServerType")
        LOG.info("proxy configuration proxyServerHost: $proxyServerHost")
        LOG.info("proxy configuration proxyServerPort: $proxyServerPort")
        val port = proxyServerPort.toIntOrNull()
        val enableFlag = enableProxy.toBoolean() && proxyHosts.isNotBlank() &&
            proxyServerType.isNotBlank() && proxyServerHost.isNotBlank()
        if (enableFlag && port != null) {
            Proxy.Type.values().firstOrNull { it.name == proxyServerType }?.also { type ->
                val proxyHostsRegex = spliterator.split(proxyHosts).toSet().map { Regex(it.trim()) }
                val socketAddress = InetSocketAddress(proxyServerHost, port)
                val proxy = Proxy(type, socketAddress)
                val proxySelector = CustomProxySelector(proxy, proxyHostsRegex)
                LOG.info("$proxySelector will use $proxy to proxy request")
                ProxySelector.setDefault(proxySelector)
            }
        }
        LOG.info("current default proxy selector: ${ProxySelector.getDefault()}")
    }
}