Merge branch '2.1.x' into 2.2.x

Closes gh-20717
spring-projects · Mar 29, 2020 · 05f1081 · 05f1081
2 parents a9a6df2 + c9e32aa
commit 05f1081
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/...ain/java/org/springframework/boot/actuate/cassandra/CassandraReactiveHealthIndicator.java b/...ain/java/org/springframework/boot/actuate/cassandra/CassandraReactiveHealthIndicator.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2019 the original author or authors.
+ * Copyright 2012-2020 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -15,8 +15,9 @@
  */
 package org.springframework.boot.actuate.cassandra;
 
-import com.datastax.driver.core.querybuilder.QueryBuilder;
-import com.datastax.driver.core.querybuilder.Select;
+import com.datastax.driver.core.ConsistencyLevel;
+import com.datastax.driver.core.SimpleStatement;
+import com.datastax.driver.core.Statement;
 import reactor.core.publisher.Mono;
 
 import org.springframework.boot.actuate.health.AbstractReactiveHealthIndicator;
@@ -33,6 +34,9 @@
  */
 public class CassandraReactiveHealthIndicator extends AbstractReactiveHealthIndicator {
 
+	private static final Statement SELECT = new SimpleStatement("SELECT release_version FROM system.local")
+			.setConsistencyLevel(ConsistencyLevel.LOCAL_ONE);
+
 	private final ReactiveCassandraOperations reactiveCassandraOperations;
 
 	/**
@@ -47,8 +51,7 @@ public CassandraReactiveHealthIndicator(ReactiveCassandraOperations reactiveCass
 
 	@Override
 	protected Mono<Health> doHealthCheck(Health.Builder builder) {
-		Select select = QueryBuilder.select("release_version").from("system", "local");
-		return this.reactiveCassandraOperations.getReactiveCqlOperations().queryForObject(select, String.class)
+		return this.reactiveCassandraOperations.getReactiveCqlOperations().queryForObject(SELECT, String.class)
 				.map((version) -> builder.up().withDetail("version", version).build()).single();
 	}
 

diff --git a/...ava/org/springframework/boot/actuate/cassandra/CassandraReactiveHealthIndicatorTests.java b/...ava/org/springframework/boot/actuate/cassandra/CassandraReactiveHealthIndicatorTests.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2019 the original author or authors.
+ * Copyright 2012-2020 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -15,7 +15,7 @@
  */
 package org.springframework.boot.actuate.cassandra;
 
-import com.datastax.driver.core.querybuilder.Select;
+import com.datastax.driver.core.Statement;
 import org.junit.jupiter.api.Test;
 import reactor.core.publisher.Mono;
 import reactor.test.StepVerifier;
@@ -42,7 +42,8 @@ class CassandraReactiveHealthIndicatorTests {
 	@Test
 	void testCassandraIsUp() {
 		ReactiveCqlOperations reactiveCqlOperations = mock(ReactiveCqlOperations.class);
-		given(reactiveCqlOperations.queryForObject(any(Select.class), eq(String.class))).willReturn(Mono.just("6.0.0"));
+		given(reactiveCqlOperations.queryForObject(any(Statement.class), eq(String.class)))
+				.willReturn(Mono.just("6.0.0"));
 		ReactiveCassandraOperations reactiveCassandraOperations = mock(ReactiveCassandraOperations.class);
 		given(reactiveCassandraOperations.getReactiveCqlOperations()).willReturn(reactiveCqlOperations);