Skip to content

Commit 2595481

Browse files
fix(client): don't close client on withOptions usage when original is gc'd
1 parent 69737c8 commit 2595481

File tree

2 files changed

+41
-8
lines changed

2 files changed

+41
-8
lines changed

openlayer-java-core/src/main/kotlin/com/openlayer/api/core/ClientOptions.kt

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,9 @@ private constructor(
8787
apiKey = clientOptions.apiKey
8888
}
8989

90-
fun httpClient(httpClient: HttpClient) = apply { this.httpClient = httpClient }
90+
fun httpClient(httpClient: HttpClient) = apply {
91+
this.httpClient = PhantomReachableClosingHttpClient(httpClient)
92+
}
9193

9294
fun checkJacksonVersionCompatibility(checkJacksonVersionCompatibility: Boolean) = apply {
9395
this.checkJacksonVersionCompatibility = checkJacksonVersionCompatibility
@@ -234,13 +236,11 @@ private constructor(
234236

235237
return ClientOptions(
236238
httpClient,
237-
PhantomReachableClosingHttpClient(
238-
RetryingHttpClient.builder()
239-
.httpClient(httpClient)
240-
.clock(clock)
241-
.maxRetries(maxRetries)
242-
.build()
243-
),
239+
RetryingHttpClient.builder()
240+
.httpClient(httpClient)
241+
.clock(clock)
242+
.maxRetries(maxRetries)
243+
.build(),
244244
checkJacksonVersionCompatibility,
245245
jsonMapper,
246246
clock,
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
// File generated from our OpenAPI spec by Stainless.
2+
3+
package com.openlayer.api.core
4+
5+
import com.openlayer.api.core.http.HttpClient
6+
import org.assertj.core.api.Assertions.assertThat
7+
import org.junit.jupiter.api.Test
8+
import org.junit.jupiter.api.extension.ExtendWith
9+
import org.mockito.junit.jupiter.MockitoExtension
10+
import org.mockito.kotlin.mock
11+
import org.mockito.kotlin.never
12+
import org.mockito.kotlin.verify
13+
14+
@ExtendWith(MockitoExtension::class)
15+
internal class ClientOptionsTest {
16+
17+
@Test
18+
fun toBuilder_whenOriginalClientOptionsGarbageCollected_doesNotCloseOriginalClient() {
19+
val httpClient = mock<HttpClient>()
20+
var clientOptions =
21+
ClientOptions.builder().httpClient(httpClient).apiKey("My API Key").build()
22+
verify(httpClient, never()).close()
23+
24+
// Overwrite the `clientOptions` variable so that the original `ClientOptions` is GC'd.
25+
clientOptions = clientOptions.toBuilder().build()
26+
System.gc()
27+
Thread.sleep(100)
28+
29+
verify(httpClient, never()).close()
30+
// This exists so that `clientOptions` is still reachable.
31+
assertThat(clientOptions).isEqualTo(clientOptions)
32+
}
33+
}

0 commit comments

Comments
 (0)