slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-14 17:53:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(partnerctl-fetch): support fetching raw data via https proxy

Browse Source
This commit is contained in:
slhafzjw
2026-05-13 10:16:18 +08:00
parent 2d45adf8c3
commit 01bfc3ee18
1 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
PartnerCtl/src/main/java/work/slhaf/partner/ctl/support/web_fetch.kt
View File

@@ -1,6 +1,8 @@
package work.slhaf.partner.ctl.support package work.slhaf.partner.ctl.support
import java.io.IOException import java.io.IOException
import java.net.InetSocketAddress
import java.net.ProxySelector
import java.net.URI import java.net.URI
import java.net.http.* import java.net.http.*
import java.time.Duration import java.time.Duration
@@ -8,8 +10,28 @@ import java.time.Duration
private val httpClient: HttpClient = HttpClient.newBuilder() private val httpClient: HttpClient = HttpClient.newBuilder()
.connectTimeout(Duration.ofSeconds(20)) .connectTimeout(Duration.ofSeconds(20))
.followRedirects(HttpClient.Redirect.NEVER) .followRedirects(HttpClient.Redirect.NEVER)
.apply {
proxySelectorFromEnv()?.let(::proxy)
}
.build() .build()
private fun proxySelectorFromEnv(): ProxySelector? {
val proxyText = System.getenv("HTTPS_PROXY")
?: System.getenv("https_proxy")
?: return null
val proxyUri = URI.create(proxyText)
val host = proxyUri.host
?: throw IllegalArgumentException("Invalid HTTPS_PROXY host: $proxyText")
val port = proxyUri.port
if (port == -1) {
throw IllegalArgumentException("HTTPS_PROXY must include port: $proxyText")
}
return ProxySelector.of(InetSocketAddress(host, port))
}
fun fetchText(url: String): String { fun fetchText(url: String): String {
var lastError: Exception? = null var lastError: Exception? = null