Skip to content

Commit

Permalink
Polish "Add HealthIndicator for Hazelcast"
Browse files Browse the repository at this point in the history
  • Loading branch information
snicoll committed Jul 18, 2019
1 parent fca5a2b commit be988d7
Show file tree
Hide file tree
Showing 7 changed files with 21 additions and 46 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -64,11 +64,6 @@
<artifactId>hazelcast-spring</artifactId>
<optional>true</optional>
</dependency>
<dependency>
<groupId>com.hazelcast</groupId>
<artifactId>hazelcast-client</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>com.sun.mail</groupId>
<artifactId>jakarta.mail</artifactId>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,6 @@
*/

/**
* Auto-configuration for Hazelcast's actuator.
* Auto-configuration for actuator Hazelcast concerns.
*/
package org.springframework.boot.actuate.autoconfigure.hazelcast;
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,7 @@

package org.springframework.boot.actuate.autoconfigure.hazelcast;

import com.hazelcast.client.config.ClientConfig;
import com.hazelcast.config.Config;
import com.hazelcast.core.Hazelcast;
import com.hazelcast.core.HazelcastInstance;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.Test;

import org.springframework.boot.actuate.autoconfigure.health.HealthIndicatorAutoConfiguration;
Expand All @@ -40,17 +36,10 @@
*/
class HazelcastHealthIndicatorAutoConfigurationIntegrationTests {

private final HazelcastInstance hazelcastServer = Hazelcast.newHazelcastInstance(new Config());

private ApplicationContextRunner contextRunner = new ApplicationContextRunner().withBean(ClientConfig.class)
private ApplicationContextRunner contextRunner = new ApplicationContextRunner()
.withConfiguration(AutoConfigurations.of(HazelcastHealthIndicatorAutoConfiguration.class,
HazelcastAutoConfiguration.class, HealthIndicatorAutoConfiguration.class));

@AfterEach
void shutdown() {
this.hazelcastServer.shutdown();
}

@Test
void hazelcastUp() {
this.contextRunner.run((context) -> {
Expand All @@ -66,7 +55,7 @@ void hazelcastUp() {
@Test
void hazelcastDown() {
this.contextRunner.run((context) -> {
shutdown();
context.getBean(HazelcastInstance.class).shutdown();
assertThat(context).hasSingleBean(HazelcastHealthIndicator.class);
Health health = context.getBean(HazelcastHealthIndicator.class).health();
assertThat(health.getStatus()).isEqualTo(Status.DOWN);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,6 @@ void runShouldCreateIndicator() {
void runWhenDisabledShouldNotCreateIndicator() {
this.contextRunner.withPropertyValues("management.health.hazelcast.enabled:false")
.run((context) -> assertThat(context).doesNotHaveBean(HazelcastHealthIndicator.class)
.doesNotHaveBean(HazelcastHealthIndicator.class)
.hasSingleBean(ApplicationHealthIndicator.class));
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,28 +16,22 @@

package org.springframework.boot.actuate.hazelcast;

import java.util.LinkedHashMap;
import java.util.Map;

import com.hazelcast.core.Endpoint;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.transaction.TransactionalTask;

import org.springframework.boot.actuate.health.AbstractHealthIndicator;
import org.springframework.boot.actuate.health.Health;
import org.springframework.boot.actuate.health.HealthIndicator;
import org.springframework.util.Assert;

/**
* {@link HealthIndicator} for a Hazelcast.
* {@link HealthIndicator} for Hazelcast.
*
* @author Dmytro Nosan
* @author Stephane Nicoll
* @since 2.2.0
*/
public class HazelcastHealthIndicator extends AbstractHealthIndicator {

private static final TransactionalTask<?> TASK = (context) -> null;

private final HazelcastInstance hazelcast;

public HazelcastHealthIndicator(HazelcastInstance hazelcast) {
Expand All @@ -48,16 +42,11 @@ public HazelcastHealthIndicator(HazelcastInstance hazelcast) {

@Override
protected void doHealthCheck(Health.Builder builder) {
this.hazelcast.executeTransaction(TASK);
builder.up().withDetails(getDetails());
}

private Map<String, Object> getDetails() {
Map<String, Object> details = new LinkedHashMap<>();
Endpoint endpoint = this.hazelcast.getLocalEndpoint();
details.put("name", this.hazelcast.getName());
details.put("uuid", endpoint.getUuid());
return details;
this.hazelcast.executeTransaction((context) -> {
builder.up().withDetail("name", this.hazelcast.getName()).withDetail("uuid",
this.hazelcast.getLocalEndpoint().getUuid());
return null;
});
}

}
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
import com.hazelcast.core.Endpoint;
import com.hazelcast.core.HazelcastException;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.transaction.TransactionalTask;
import org.junit.jupiter.api.Test;

import org.springframework.boot.actuate.health.Health;
Expand All @@ -33,22 +34,23 @@
* Tests for {@link HazelcastHealthIndicator}.
*
* @author Dmytro Nosan
* @author Stephane Nicoll
*/
class HazelcastHealthIndicatorTests {

private final HazelcastInstance hazelcast = mock(HazelcastInstance.class);

private final HazelcastHealthIndicator healthIndicator = new HazelcastHealthIndicator(this.hazelcast);

@Test
void hazelcastUp() {
Endpoint endpoint = mock(Endpoint.class);
when(this.hazelcast.getName()).thenReturn("hz0-instance");
when(this.hazelcast.getLocalEndpoint()).thenReturn(endpoint);
when(endpoint.getUuid()).thenReturn("7581bb2f-879f-413f-b574-0071d7519eb0");

Health health = this.healthIndicator.health();

when(this.hazelcast.executeTransaction(any())).thenAnswer((invocation) -> {
TransactionalTask<?> task = invocation.getArgument(0);
return task.execute(null);
});
Health health = new HazelcastHealthIndicator(this.hazelcast).health();
assertThat(health.getStatus()).isEqualTo(Status.UP);
assertThat(health.getDetails()).containsOnlyKeys("name", "uuid").containsEntry("name", "hz0-instance")
.containsEntry("uuid", "7581bb2f-879f-413f-b574-0071d7519eb0");
Expand All @@ -57,9 +59,7 @@ void hazelcastUp() {
@Test
void hazelcastDown() {
when(this.hazelcast.executeTransaction(any())).thenThrow(new HazelcastException());

Health health = this.healthIndicator.health();

Health health = new HazelcastHealthIndicator(this.hazelcast).health();
assertThat(health.getStatus()).isEqualTo(Status.DOWN);
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -764,6 +764,9 @@ The following `HealthIndicators` are auto-configured by Spring Boot when appropr
|{sc-spring-boot-actuator}/elasticsearch/ElasticsearchHealthIndicator.{sc-ext}[`ElasticsearchHealthIndicator`]
|Checks that an Elasticsearch cluster is up.

|{sc-spring-boot-actuator}/hazelcast/HazelcastHealthIndicator.{sc-ext}[`HazelcastHealthIndicator`]
|Checks that an Hazelcast server is up.

|{sc-spring-boot-actuator}/influx/InfluxDbHealthIndicator.{sc-ext}[`InfluxDbHealthIndicator`]
|Checks that an InfluxDB server is up.

Expand Down

0 comments on commit be988d7

Please sign in to comment.