Use non-blocking API in CouchbaseReactiveHealthIndicator

See gh-32505
spring-projects · Sep 27, 2022 · f51c4c3 · f51c4c3
1 parent 8a27f0b
commit f51c4c3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/...ain/java/org/springframework/boot/actuate/couchbase/CouchbaseReactiveHealthIndicator.java b/...ain/java/org/springframework/boot/actuate/couchbase/CouchbaseReactiveHealthIndicator.java
@@ -46,9 +46,10 @@ public CouchbaseReactiveHealthIndicator(Cluster cluster) {
 
 	@Override
 	protected Mono<Health> doHealthCheck(Health.Builder builder) {
-		DiagnosticsResult diagnostics = this.cluster.diagnostics();
-		new CouchbaseHealth(diagnostics).applyTo(builder);
-		return Mono.just(builder.build());
+		return this.cluster.reactive().diagnostics().map(diagnostics -> {
+			new CouchbaseHealth(diagnostics).applyTo(builder);
+			return builder.build();
+		});
 	}
 
 }
diff --git a/...ava/org/springframework/boot/actuate/couchbase/CouchbaseReactiveHealthIndicatorTests.java b/...ava/org/springframework/boot/actuate/couchbase/CouchbaseReactiveHealthIndicatorTests.java
@@ -28,10 +28,12 @@
 import com.couchbase.client.core.endpoint.EndpointState;
 import com.couchbase.client.core.service.ServiceType;
 import com.couchbase.client.java.Cluster;
+import com.couchbase.client.java.ReactiveCluster;
 import org.junit.jupiter.api.Test;
 
 import org.springframework.boot.actuate.health.Health;
 import org.springframework.boot.actuate.health.Status;
+import reactor.core.publisher.Mono;
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.mockito.BDDMockito.given;
@@ -52,13 +54,15 @@ void couchbaseClusterIsUp() {
 				Collections.singletonList(new EndpointDiagnostics(ServiceType.KV, EndpointState.CONNECTED, "127.0.0.1",
 						"127.0.0.1", Optional.empty(), Optional.of(1234L), Optional.of("endpoint-1"))));
 		DiagnosticsResult diagnostics = new DiagnosticsResult(endpoints, "test-sdk", "test-id");
-		given(cluster.diagnostics()).willReturn(diagnostics);
+		ReactiveCluster reactiveCluster = mock(ReactiveCluster.class);
+		given(reactiveCluster.diagnostics()).willReturn(Mono.just(diagnostics));
+		given(cluster.reactive()).willReturn(reactiveCluster);
 		Health health = healthIndicator.health().block(Duration.ofSeconds(30));
 		assertThat(health.getStatus()).isEqualTo(Status.UP);
 		assertThat(health.getDetails()).containsEntry("sdk", "test-sdk");
 		assertThat(health.getDetails()).containsKey("endpoints");
 		assertThat((List<Map<String, Object>>) health.getDetails().get("endpoints")).hasSize(1);
-		then(cluster).should().diagnostics();
+		then(reactiveCluster).should().diagnostics();
 	}
 
 	@Test
@@ -73,13 +77,15 @@ void couchbaseClusterIsDown() {
 						new EndpointDiagnostics(ServiceType.KV, EndpointState.CONNECTING, "127.0.0.1", "127.0.0.1",
 								Optional.empty(), Optional.of(1234L), Optional.of("endpoint-2"))));
 		DiagnosticsResult diagnostics = new DiagnosticsResult(endpoints, "test-sdk", "test-id");
-		given(cluster.diagnostics()).willReturn(diagnostics);
+		ReactiveCluster reactiveCluster = mock(ReactiveCluster.class);
+		given(reactiveCluster.diagnostics()).willReturn(Mono.just(diagnostics));
+		given(cluster.reactive()).willReturn(reactiveCluster);
 		Health health = healthIndicator.health().block(Duration.ofSeconds(30));
 		assertThat(health.getStatus()).isEqualTo(Status.DOWN);
 		assertThat(health.getDetails()).containsEntry("sdk", "test-sdk");
 		assertThat(health.getDetails()).containsKey("endpoints");
 		assertThat((List<Map<String, Object>>) health.getDetails().get("endpoints")).hasSize(2);
-		then(cluster).should().diagnostics();
+		then(reactiveCluster).should().diagnostics();
 	}
 
 }