Examples with Status io.pravega.shared.health.Status used on opensource projects

Example 1 with Status

use of io.pravega.shared.health.Status in project pravega by pravega.

the class RetentionServiceHealthContributorTest method testHealthCheck.

@Test
public void testHealthCheck() throws Exception {
    retentionService.startAsync();
    Status status = contributor.doHealthCheck(builder);
    Assert.assertEquals(Status.UP, status);
    retentionService.stopAsync();
    status = contributor.doHealthCheck(builder);
    Assert.assertEquals(Status.DOWN, status);
}

Example 2 with Status

use of io.pravega.shared.health.Status in project pravega by pravega.

the class WatermarkingServiceHealthContibutorTest method testHealthCheck.

@Test
public void testHealthCheck() throws Exception {
    watermarkingService.startAsync();
    Assert.assertTrue(watermarkingService.isRunning());
    Status status = contributor.doHealthCheck(builder);
    Assert.assertEquals(Status.UP, status);
    watermarkingService.stopAsync();
    status = contributor.doHealthCheck(builder);
    Assert.assertEquals(Status.DOWN, status);
}

Example 3 with Status

use of io.pravega.shared.health.Status in project pravega by pravega.

the class SegmentContainerHealthContributor method doHealthCheck.

@Override
public Status doHealthCheck(Health.HealthBuilder builder) {
    Status status = Status.DOWN;
    if (segmentContainer.state() == Service.State.NEW) {
        status = Status.NEW;
    }
    if (segmentContainer.state() == Service.State.STARTING) {
        status = Status.STARTING;
    }
    if (segmentContainer.state() == Service.State.RUNNING) {
        status = Status.UP;
    }
    builder.details(ImmutableMap.of("Id", segmentContainer.getId(), "ActiveSegments", segmentContainer.getActiveSegments()));
    return status;
}

Example 4 with Status

use of io.pravega.shared.health.Status in project pravega by pravega.

the class ServiceStarterTest method testCuratorClientCreation.

/**
 * Check that the client created by ServiceStarter can correctly connect to a Zookeeper server using the custom
 * Zookeeper client factory.
 *
 * @throws Exception
 */
@Test
public void testCuratorClientCreation() throws Exception {
    @Cleanup CuratorFramework zkClient = serviceStarter.createZKClient();
    zkClient.blockUntilConnected();
    @Cleanup ZKHealthContributor zkHealthContributor = new ZKHealthContributor(zkClient);
    Health.HealthBuilder builder = Health.builder().name(zkHealthContributor.getName());
    Status zkStatus = zkHealthContributor.doHealthCheck(builder);
    Assert.assertTrue(zkClient.getZookeeperClient().isConnected());
    Assert.assertEquals("HealthContributor should report an 'UP' Status.", Status.UP, zkStatus);
    zkClient.close();
    zkStatus = zkHealthContributor.doHealthCheck(builder);
    Assert.assertEquals("HealthContributor should report an 'DOWN' Status.", Status.DOWN, zkStatus);
}

Example 5 with Status

use of io.pravega.shared.health.Status in project pravega by pravega.

the class ZKHealthContributor method doHealthCheck.

@Override
public Status doHealthCheck(Health.HealthBuilder builder) {
    Status status = Status.DOWN;
    boolean running = this.zk.getState() == CuratorFrameworkState.STARTED;
    if (running) {
        status = Status.NEW;
    }
    boolean ready = this.zk.getZookeeperClient().isConnected();
    if (ready) {
        status = Status.UP;
    }
    builder.details(ImmutableMap.of("zk-connection-url", this.zk.getZookeeperClient().getCurrentConnectionString()));
    return status;
}