private String printEventHubRuntimeInfo(EventHubRuntimeInformation ehInfo) { if (ehInfo == null) { return "[EventHubRuntimeInformation: null]"; } return String.format("[EventHubRuntimeInformation: createAt=%s, partitionCount=%d, path=%s]", ehInfo.getCreatedAt(), ehInfo.getPartitionCount(), ehInfo.getPath()); }
@Override public CompletableFuture<EventHubRuntimeInformation> apply(Map<String, Object> rawdata) { CompletableFuture<EventHubRuntimeInformation> future2 = new CompletableFuture<EventHubRuntimeInformation>(); future2.complete(new EventHubRuntimeInformation( (String) rawdata.get(ClientConstants.MANAGEMENT_ENTITY_NAME_KEY), ((Date) rawdata.get(ClientConstants.MANAGEMENT_RESULT_CREATED_AT)).toInstant(), (int) rawdata.get(ClientConstants.MANAGEMENT_RESULT_PARTITION_COUNT), (String[]) rawdata.get(ClientConstants.MANAGEMENT_RESULT_PARTITION_IDS))); return future2; } }, this.executor);
this.partitionIds = ehInfo.getPartitionIds(); TRACE_LOGGER.info(this.hostContext.withHost("Eventhub " + this.hostContext.getEventHubPath() + " count of partitions: " + ehInfo.getPartitionCount())); for (String id : this.partitionIds) { TRACE_LOGGER.info(this.hostContext.withHost("Found partition with id: " + id));
long timeoutMs = config.getRuntimeInfoWaitTimeMS(systemName); Integer numPartitions = ehClient.getRuntimeInformation().get(timeoutMs, TimeUnit.MILLISECONDS).getPartitionCount();
EventHubRuntimeInformation ehInfo = runtimeInfo.get(timeoutMs, TimeUnit.MILLISECONDS); LOG.info(String.format("Adding partition ids=%s for stream=%s. EHRuntimetInfo=%s", Arrays.toString(ehInfo.getPartitionIds()), streamName, printEventHubRuntimeInfo(ehInfo))); streamPartitions.put(streamName, ehInfo.getPartitionIds());
PowerMockito.when(mockRuntimeInfo.getPartitionCount()).thenReturn(2);