Examples with ClusterCapacityBuilder org.bf2.operator.resources.v1alpha1.ClusterCapacityBuilder used on opensource projects

Example 1 with ClusterCapacityBuilder

use of org.bf2.operator.resources.v1alpha1.ClusterCapacityBuilder in project kas-fleetshard by bf2fc6cc711aee1a0c2a.

the class ManagedKafkaAgentController method buildStatus.

/**
 * TODO: this needs to be replaced with actual metrics
 * @return
 */
private ManagedKafkaAgentStatus buildStatus(ManagedKafkaAgent resource) {
    ManagedKafkaAgentStatus status = resource.getStatus();
    ManagedKafkaCondition readyCondition = null;
    if (status != null) {
        readyCondition = ConditionUtils.findManagedKafkaCondition(status.getConditions(), Type.Ready).orElse(null);
    }
    List<StrimziVersionStatus> strimziVersions = this.strimziManager.getStrimziVersions();
    log.debugf("Strimzi versions %s", strimziVersions);
    // consider the fleetshard operator ready when observability is running and a Strimzi bundle is installed (aka at least one available version)
    Status statusValue = this.observabilityManager.isObservabilityRunning() && !strimziVersions.isEmpty() ? ManagedKafkaCondition.Status.True : ManagedKafkaCondition.Status.False;
    if (readyCondition == null) {
        readyCondition = ConditionUtils.buildCondition(ManagedKafkaCondition.Type.Ready, statusValue);
    } else {
        ConditionUtils.updateConditionStatus(readyCondition, statusValue, null, null);
    }
    ClusterCapacity total = new ClusterCapacityBuilder().withConnections(10000).withDataRetentionSize(Quantity.parse("40Gi")).withIngressEgressThroughputPerSec(Quantity.parse("40Gi")).withPartitions(10000).build();
    ClusterCapacity remaining = new ClusterCapacityBuilder().withConnections(10000).withDataRetentionSize(Quantity.parse("40Gi")).withIngressEgressThroughputPerSec(Quantity.parse("40Gi")).withPartitions(10000).build();
    ClusterCapacity delta = new ClusterCapacityBuilder().withConnections(10000).withDataRetentionSize(Quantity.parse("40Gi")).withIngressEgressThroughputPerSec(Quantity.parse("40Gi")).withPartitions(10000).build();
    NodeCounts nodeInfo = new NodeCountsBuilder().withCeiling(0).withCurrent(0).withCurrentWorkLoadMinimum(0).withFloor(0).build();
    ClusterResizeInfo resize = new ClusterResizeInfoBuilder().withDelta(delta).withNodeDelta(3).build();
    return new ManagedKafkaAgentStatusBuilder().withConditions(status == null ? Arrays.asList(readyCondition) : status.getConditions()).withTotal(total).withRemaining(remaining).withNodeInfo(nodeInfo).withResizeInfo(resize).withUpdatedTimestamp(ConditionUtils.iso8601Now()).withStrimzi(strimziVersions).build();
}

Example 2 with ClusterCapacityBuilder

use of org.bf2.operator.resources.v1alpha1.ClusterCapacityBuilder in project kas-fleetshard by bf2fc6cc711aee1a0c2a.

the class AgentPollerTest method testAddDelete.

@Test
public void testAddDelete() {
    assertNull(lookup.getLocalManagedKafkaAgent());
    ManagedKafkaAgent managedKafkaAgent = ManagedKafkaAgentResourceClient.getDummyInstance();
    Mockito.reset(controlPlaneRestClient);
    Mockito.when(controlPlaneRestClient.get(CLUSTER_ID)).thenReturn(managedKafkaAgent);
    // pick up the agent from the control plane
    managedKafkaAgentSync.loop();
    ManagedKafkaAgent local = lookup.getLocalManagedKafkaAgent();
    local.setStatus(new ManagedKafkaAgentStatusBuilder().withRemaining(new ClusterCapacityBuilder().withConnections(1000).build()).build());
    client.replaceStatus(local);
    assertEquals("test-token", local.getSpec().getObservability().getAccessToken());
    assertEquals(ManagedKafkaAgentResourceClient.RESOURCE_NAME, local.getMetadata().getName());
    managedKafkaAgent.getSpec().getObservability().setAccessToken("abc");
    // pick up the update
    managedKafkaAgentSync.loop();
    local = lookup.getLocalManagedKafkaAgent();
    assertEquals("abc", local.getSpec().getObservability().getAccessToken());
    assertEquals(1000, local.getStatus().getRemaining().getConnections());
}