Skip to content

Commit c17580b

Browse files
fix(client): don't close client on withOptions usage when original is gc'd
1 parent 7437b68 commit c17580b

2 files changed

Lines changed: 41 additions & 8 deletions

File tree

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

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,9 @@ private constructor(
9494
apiKey = clientOptions.apiKey
9595
}
9696

97-
fun httpClient(httpClient: HttpClient) = apply { this.httpClient = httpClient }
97+
fun httpClient(httpClient: HttpClient) = apply {
98+
this.httpClient = PhantomReachableClosingHttpClient(httpClient)
99+
}
98100

99101
fun checkJacksonVersionCompatibility(checkJacksonVersionCompatibility: Boolean) = apply {
100102
this.checkJacksonVersionCompatibility = checkJacksonVersionCompatibility
@@ -245,13 +247,11 @@ private constructor(
245247

246248
return ClientOptions(
247249
httpClient,
248-
PhantomReachableClosingHttpClient(
249-
RetryingHttpClient.builder()
250-
.httpClient(httpClient)
251-
.clock(clock)
252-
.maxRetries(maxRetries)
253-
.build()
254-
),
250+
RetryingHttpClient.builder()
251+
.httpClient(httpClient)
252+
.clock(clock)
253+
.maxRetries(maxRetries)
254+
.build(),
255255
checkJacksonVersionCompatibility,
256256
jsonMapper,
257257
streamHandlerExecutor
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.braintrustdata.api.core
4+
5+
import com.braintrustdata.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)