Examples with BrokerPartitionState io.zeebe.client.clustering.impl.BrokerPartitionState used on opensource projects

Example 1 with BrokerPartitionState

use of io.zeebe.client.clustering.impl.BrokerPartitionState in project zeebe by zeebe-io.

the class TopologyRequestTest method shouldUpdateClientTopologyOnAsyncTopologyRequest.

@Test
public void shouldUpdateClientTopologyOnAsyncTopologyRequest() {
    // given
    final SocketAddress oldLeader = BROKER_1_CLIENT_ADDRESS;
    // when
    final List<Integer> partitions = clusteringRule.getBrokersLeadingPartitions(oldLeader);
    clusteringRule.brokers.remove(oldLeader).close();
    doRepeatedly(this::requestTopologyAsync).until(topologyBrokers -> topologyBrokers != null && topologyBrokers.stream().filter(broker -> !broker.getSocketAddress().equals(oldLeader)).flatMap(broker -> broker.getPartitions().stream()).filter(BrokerPartitionState::isLeader).map(BrokerPartitionState::getPartitionId).collect(Collectors.toSet()).containsAll(partitions));
    // then
    zeebeClient.topics().create("foo", 1).execute();
}

Example 2 with BrokerPartitionState

use of io.zeebe.client.clustering.impl.BrokerPartitionState in project zeebe by zeebe-io.

the class BrokerRestartTest method shouldCreateUniquePartitionIdsAfterRestart.

@Test
public void shouldCreateUniquePartitionIdsAfterRestart() {
    // given
    final ZeebeClient client = clientRule.getClient();
    client.topics().create("foo", 2).execute();
    restartBroker();
    // when
    client.topics().create("bar", 2).execute();
    // then
    final TopologyResponse topology = client.requestTopology().execute();
    final List<TopologyBroker> brokers = topology.getBrokers();
    assertThat(brokers).hasSize(1);
    final TopologyBroker topologyBroker = brokers.get(0);
    final List<BrokerPartitionState> partitions = topologyBroker.getPartitions();
    // default partition + system partition + 4 partitions we create here
    assertThat(partitions).hasSize(6);
    assertThat(partitions).extracting("partitionId").doesNotHaveDuplicates();
}

Example 3 with BrokerPartitionState

use of io.zeebe.client.clustering.impl.BrokerPartitionState in project zeebe by zeebe-io.

the class BrokerRecoveryTest method shouldCreateUniquePartitionIdsAfterRestart.

@Test
public void shouldCreateUniquePartitionIdsAfterRestart() {
    // given
    final ZeebeClient client = clientRule.getClient();
    client.topics().create("foo", 2).execute();
    restartBroker();
    // when
    client.topics().create("bar", 2).execute();
    // then
    final TopologyResponse topology = client.requestTopology().execute();
    final List<TopologyBroker> brokers = topology.getBrokers();
    assertThat(brokers).hasSize(1);
    final TopologyBroker topologyBroker = brokers.get(0);
    final List<BrokerPartitionState> partitions = topologyBroker.getPartitions();
    // default partition + system partition + 4 partitions we create here
    assertThat(partitions).hasSize(6);
    assertThat(partitions).extracting("partitionId").doesNotHaveDuplicates();
}

Example 4 with BrokerPartitionState

use of io.zeebe.client.clustering.impl.BrokerPartitionState in project zeebe by zeebe-io.

the class ClientRule method createDefaultTopic.

private void createDefaultTopic() {
    client.topics().create(DEFAULT_TOPIC, 1).execute();
    final TopologyResponse topology = client.requestTopology().execute();
    defaultPartition = -1;
    final List<TopologyBroker> topologyBrokers = topology.getBrokers();
    for (TopologyBroker leader : topologyBrokers) {
        final List<BrokerPartitionState> partitions = leader.getPartitions();
        for (BrokerPartitionState brokerPartitionState : partitions) {
            if (DEFAULT_TOPIC.equals(brokerPartitionState.getTopicName()) && brokerPartitionState.isLeader()) {
                defaultPartition = brokerPartitionState.getPartitionId();
                break;
            }
        }
    }
    if (defaultPartition < 0) {
        throw new RuntimeException("Could not detect leader for default partition");
    }
}