org.apache.pulsar.testclient.PerformanceTransaction Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.pulsar.testclient;
import static java.util.concurrent.TimeUnit.NANOSECONDS;
import static org.apache.commons.lang3.StringUtils.isNotBlank;
import com.beust.jcommander.JCommander;
import com.beust.jcommander.Parameter;
import com.beust.jcommander.ParameterException;
import com.beust.jcommander.Parameters;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.ObjectWriter;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.PrintStream;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Properties;
import java.util.Random;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference;
import java.util.concurrent.atomic.LongAdder;
import org.HdrHistogram.Histogram;
import org.HdrHistogram.HistogramLogWriter;
import org.HdrHistogram.Recorder;
import org.apache.curator.shaded.com.google.common.util.concurrent.RateLimiter;
import org.apache.pulsar.client.admin.PulsarAdmin;
import org.apache.pulsar.client.admin.PulsarAdminBuilder;
import org.apache.pulsar.client.admin.PulsarAdminException;
import org.apache.pulsar.client.api.ClientBuilder;
import org.apache.pulsar.client.api.Consumer;
import org.apache.pulsar.client.api.ConsumerBuilder;
import org.apache.pulsar.client.api.Message;
import org.apache.pulsar.client.api.Producer;
import org.apache.pulsar.client.api.ProducerBuilder;
import org.apache.pulsar.client.api.PulsarClient;
import org.apache.pulsar.client.api.PulsarClientException;
import org.apache.pulsar.client.api.Schema;
import org.apache.pulsar.client.api.SizeUnit;
import org.apache.pulsar.client.api.SubscriptionInitialPosition;
import org.apache.pulsar.client.api.SubscriptionType;
import org.apache.pulsar.client.api.transaction.Transaction;
import org.apache.pulsar.common.partition.PartitionedTopicMetadata;
import org.apache.pulsar.testclient.utils.PaddingDecimalFormat;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class PerformanceTransaction {
private static final LongAdder totalNumEndTxnOpFailed = new LongAdder();
private static final LongAdder totalNumEndTxnOpSuccess = new LongAdder();
private static final LongAdder numTxnOpSuccess = new LongAdder();
private static final LongAdder totalNumTxnOpenTxnFail = new LongAdder();
private static final LongAdder totalNumTxnOpenTxnSuccess = new LongAdder();
private static final LongAdder numMessagesAckFailed = new LongAdder();
private static final LongAdder numMessagesAckSuccess = new LongAdder();
private static final LongAdder numMessagesSendFailed = new LongAdder();
private static final LongAdder numMessagesSendSuccess = new LongAdder();
private static final Recorder messageAckRecorder =
new Recorder(TimeUnit.SECONDS.toMicros(120000), 5);
private static final Recorder messageAckCumulativeRecorder =
new Recorder(TimeUnit.SECONDS.toMicros(120000), 5);
private static final Recorder messageSendRecorder =
new Recorder(TimeUnit.SECONDS.toMicros(120000), 5);
private static final Recorder messageSendRCumulativeRecorder =
new Recorder(TimeUnit.SECONDS.toMicros(120000), 5);
@Parameters(commandDescription = "Test pulsar transaction performance.")
static class Arguments {
@Parameter(names = {"-h", "--help"}, description = "Help message", help = true)
boolean help;
@Parameter(names = { "-cf", "--conf-file" }, description = "Configuration file")
public String confFile;
@Parameter(names = "--topics-c", description = "All topics that need ack for a transaction", required =
true)
public List consumerTopic = Collections.singletonList("test-consume");
@Parameter(names = "--topics-p", description = "All topics that need produce for a transaction",
required = true)
public List producerTopic = Collections.singletonList("test-produce");
@Parameter(names = {"-threads", "--num-test-threads"}, description = "Number of test threads."
+ "This thread is for a new transaction to ack messages from consumer topics and produce message to "
+ "producer topics, and then commit or abort this transaction. "
+ "Increasing the number of threads increases the parallelism of the performance test, "
+ "thereby increasing the intensity of the stress test.")
public int numTestThreads = 1;
@Parameter(names = { "--auth-plugin" }, description = "Authentication plugin class name")
public String authPluginClassName;
@Parameter(
names = { "--auth-params" },
description = "Authentication parameters, whose format is determined by the implementation "
+ "of method `configure` in authentication plugin class, for example \"key1:val1,key2:val2\" "
+ "or \"{\"key1\":\"val1\",\"key2\":\"val2\"}.")
public String authParams;
@Parameter(names = {"-au", "--admin-url"}, description = "Pulsar Admin URL")
public String adminURL;
@Parameter(names = {"-u", "--service-url"}, description = "Pulsar Service URL")
public String serviceURL;
@Parameter(names = {"-np",
"--partitions"}, description = "Create partitioned topics with a given number of partitions, 0 means"
+ "not trying to create a topic")
public Integer partitions = null;
@Parameter(names = {"-c",
"--max-connections"}, description = "Max number of TCP connections to a single broker")
public int maxConnections = 100;
@Parameter(names = {"-time",
"--test-duration"}, description = "Test duration (in second). 0 means keeping publishing")
public long testTime = 0;
@Parameter(names = {"-ioThreads", "--num-io-threads"}, description = "Set the number of threads to be "
+ "used for handling connections to brokers. The default value is 1.")
public int ioThreads = 1;
@Parameter(names = {"-ss",
"--subscriptions"}, description = "A list of subscriptions to consume (for example, sub1,sub2)")
public List subscriptions = Collections.singletonList("sub");
@Parameter(names = {"-ns", "--num-subscriptions"}, description = "Number of subscriptions (per topic)")
public int numSubscriptions = 1;
@Parameter(names = {"-sp", "--subscription-position"}, description = "Subscription position")
private SubscriptionInitialPosition subscriptionInitialPosition = SubscriptionInitialPosition.Earliest;
@Parameter(names = {"-st", "--subscription-type"}, description = "Subscription type")
public SubscriptionType subscriptionType = SubscriptionType.Shared;
@Parameter(names = {"-q", "--receiver-queue-size"}, description = "Size of the receiver queue")
public int receiverQueueSize = 1000;
@Parameter(names = {"-tto", "--txn-timeout"}, description = "Set the time value of transaction timeout,"
+ " and the time unit is second. (After --txn-enable setting to true, --txn-timeout takes effect)")
public long transactionTimeout = 5;
@Parameter(names = {"-ntxn",
"--number-txn"}, description = "Set the number of transaction. 0 means keeping open."
+ "If transaction disabled, it means the number of tasks. The task or transaction produces or "
+ "consumes a specified number of messages.")
public long numTransactions = 0;
@Parameter(names = {"-nmp", "--numMessage-perTransaction-produce"},
description = "Set the number of messages produced in a transaction."
+ "If transaction disabled, it means the number of messages produced in a task.")
public int numMessagesProducedPerTransaction = 1;
@Parameter(names = {"-nmc", "--numMessage-perTransaction-consume"},
description = "Set the number of messages consumed in a transaction."
+ "If transaction disabled, it means the number of messages consumed in a task.")
public int numMessagesReceivedPerTransaction = 1;
@Parameter(names = {"--txn-disable"}, description = "Disable transaction")
public boolean isDisableTransaction = false;
@Parameter(names = {"-abort"}, description = "Abort the transaction. (After --txn-disEnable "
+ "setting to false, -abort takes effect)")
public boolean isAbortTransaction = false;
@Parameter(names = "-txnRate", description = "Set the rate of opened transaction or task. 0 means no limit")
public int openTxnRate = 0;
}
public static void main(String[] args)
throws IOException, PulsarAdminException, ExecutionException, InterruptedException {
final Arguments arguments = new Arguments();
JCommander jc = new JCommander(arguments);
jc.setProgramName("pulsar-perf transaction");
try {
jc.parse(args);
} catch (ParameterException e) {
System.out.println(e.getMessage());
jc.usage();
PerfClientUtils.exit(-1);
}
if (arguments.help) {
jc.usage();
PerfClientUtils.exit(-1);
}
if (arguments.confFile != null) {
Properties prop = new Properties(System.getProperties());
prop.load(new FileInputStream(arguments.confFile));
if (arguments.serviceURL == null) {
arguments.serviceURL = prop.getProperty("brokerServiceUrl");
}
if (arguments.serviceURL == null) {
arguments.serviceURL = prop.getProperty("webServiceUrl");
}
// fallback to previous-version serviceUrl property to maintain backward-compatibility
if (arguments.serviceURL == null) {
arguments.serviceURL = prop.getProperty("serviceUrl", "http://localhost:8080/");
}
if (arguments.adminURL == null) {
arguments.adminURL = prop.getProperty("webServiceUrl");
}
if (arguments.adminURL == null) {
arguments.adminURL = prop.getProperty("adminURL", "http://localhost:8080/");
}
if (arguments.authPluginClassName == null) {
arguments.authPluginClassName = prop.getProperty("authPlugin", null);
}
if (arguments.authParams == null) {
arguments.authParams = prop.getProperty("authParams", null);
}
}
// Dump config variables
PerfClientUtils.printJVMInformation(log);
ObjectMapper m = new ObjectMapper();
ObjectWriter w = m.writerWithDefaultPrettyPrinter();
log.info("Starting Pulsar perf transaction with config: {}", w.writeValueAsString(arguments));
final byte[] payloadBytes = new byte[1024];
Random random = new Random(0);
for (int i = 0; i < payloadBytes.length; ++i) {
payloadBytes[i] = (byte) (random.nextInt(26) + 65);
}
if (arguments.partitions != null) {
PulsarAdminBuilder clientBuilder = PulsarAdmin.builder()
.serviceHttpUrl(arguments.adminURL);
if (isNotBlank(arguments.authPluginClassName)) {
clientBuilder.authentication(arguments.authPluginClassName, arguments.authParams);
}
try (PulsarAdmin client = clientBuilder.build()) {
for (String topic : arguments.producerTopic) {
log.info("Creating produce partitioned topic {} with {} partitions", topic, arguments.partitions);
try {
client.topics().createPartitionedTopic(topic, arguments.partitions);
} catch (PulsarAdminException.ConflictException alreadyExists) {
if (log.isDebugEnabled()) {
log.debug("Topic {} already exists: {}", topic, alreadyExists);
}
PartitionedTopicMetadata partitionedTopicMetadata =
client.topics().getPartitionedTopicMetadata(topic);
if (partitionedTopicMetadata.partitions != arguments.partitions) {
log.error(
"Topic {} already exists but it has a wrong number of partitions: {}, expecting {}",
topic, partitionedTopicMetadata.partitions, arguments.partitions);
PerfClientUtils.exit(-1);
}
}
}
}
}
ClientBuilder clientBuilder =
PulsarClient.builder()
.memoryLimit(0, SizeUnit.BYTES)
.enableTransaction(!arguments.isDisableTransaction)
.serviceUrl(arguments.serviceURL)
.connectionsPerBroker(arguments.maxConnections)
.statsInterval(0, TimeUnit.SECONDS)
.ioThreads(arguments.ioThreads);
if (isNotBlank(arguments.authPluginClassName)) {
clientBuilder.authentication(arguments.authPluginClassName, arguments.authParams);
}
PulsarClient client = clientBuilder.build();
ExecutorService executorService = new ThreadPoolExecutor(arguments.numTestThreads,
arguments.numTestThreads,
0L, TimeUnit.MILLISECONDS,
new LinkedBlockingQueue<>());
long startTime = System.nanoTime();
long testEndTime = startTime + (long) (arguments.testTime * 1e9);
Runtime.getRuntime().addShutdownHook(new Thread(() -> {
if (!arguments.isDisableTransaction) {
printTxnAggregatedThroughput(startTime);
} else {
printAggregatedThroughput(startTime);
}
printAggregatedStats();
}));
// start perf test
AtomicBoolean executing = new AtomicBoolean(true);
RateLimiter rateLimiter = arguments.openTxnRate > 0
? RateLimiter.create(arguments.openTxnRate)
: null;
for (int i = 0; i < arguments.numTestThreads; i++) {
executorService.submit(() -> {
//The producer and consumer clients are built in advance, and then this thread is
//responsible for the production and consumption tasks of the transaction through the loop.
//A thread may perform tasks of multiple transactions in a traversing manner.
List> producers = null;
List>> consumers = null;
AtomicReference atomicReference = null;
try {
producers = buildProducers(client, arguments);
consumers = buildConsumer(client, arguments);
if (!arguments.isDisableTransaction) {
atomicReference = new AtomicReference<>(client.newTransaction()
.withTransactionTimeout(arguments.transactionTimeout, TimeUnit.SECONDS)
.build()
.get());
} else {
atomicReference = new AtomicReference<>(null);
}
} catch (Exception e) {
log.error("Failed to build Producer/Consumer with exception : ", e);
executorService.shutdownNow();
PerfClientUtils.exit(-1);
}
//The while loop has no break, and finally ends the execution through the shutdownNow of
//the executorService
while (true) {
if (arguments.numTransactions > 0) {
if (totalNumTxnOpenTxnFail.sum()
+ totalNumTxnOpenTxnSuccess.sum() >= arguments.numTransactions) {
if (totalNumEndTxnOpFailed.sum()
+ totalNumEndTxnOpSuccess.sum() < arguments.numTransactions) {
continue;
}
log.info("------------------- DONE -----------------------");
executing.compareAndSet(true, false);
executorService.shutdownNow();
PerfClientUtils.exit(0);
break;
}
}
if (arguments.testTime > 0) {
if (System.nanoTime() > testEndTime) {
log.info("------------------- DONE -----------------------");
executing.compareAndSet(true, false);
executorService.shutdownNow();
PerfClientUtils.exit(0);
break;
}
}
Transaction transaction = atomicReference.get();
for (List> subscriptions : consumers) {
for (Consumer consumer : subscriptions) {
for (int j = 0; j < arguments.numMessagesReceivedPerTransaction; j++) {
Message message = null;
try {
message = consumer.receive();
} catch (PulsarClientException e) {
log.error("Receive message failed", e);
executorService.shutdownNow();
PerfClientUtils.exit(-1);
}
long receiveTime = System.nanoTime();
if (!arguments.isDisableTransaction) {
consumer.acknowledgeAsync(message.getMessageId(), transaction)
.thenRun(() -> {
long latencyMicros = NANOSECONDS.toMicros(
System.nanoTime() - receiveTime);
messageAckRecorder.recordValue(latencyMicros);
messageAckCumulativeRecorder.recordValue(latencyMicros);
numMessagesAckSuccess.increment();
}).exceptionally(exception -> {
if (exception instanceof InterruptedException && !executing.get()) {
return null;
}
log.error(
"Ack message failed with transaction {} throw exception",
transaction, exception);
numMessagesAckFailed.increment();
return null;
});
} else {
consumer.acknowledgeAsync(message).thenRun(() -> {
long latencyMicros = NANOSECONDS.toMicros(
System.nanoTime() - receiveTime);
messageAckRecorder.recordValue(latencyMicros);
messageAckCumulativeRecorder.recordValue(latencyMicros);
numMessagesAckSuccess.increment();
}).exceptionally(exception -> {
if (exception instanceof InterruptedException && !executing.get()) {
return null;
}
log.error(
"Ack message failed with transaction {} throw exception",
transaction, exception);
numMessagesAckFailed.increment();
return null;
});
}
}
}
}
for (Producer producer : producers){
for (int j = 0; j < arguments.numMessagesProducedPerTransaction; j++) {
long sendTime = System.nanoTime();
if (!arguments.isDisableTransaction) {
producer.newMessage(transaction).value(payloadBytes)
.sendAsync().thenRun(() -> {
long latencyMicros = NANOSECONDS.toMicros(
System.nanoTime() - sendTime);
messageSendRecorder.recordValue(latencyMicros);
messageSendRCumulativeRecorder.recordValue(latencyMicros);
numMessagesSendSuccess.increment();
}).exceptionally(exception -> {
if (exception instanceof InterruptedException && !executing.get()) {
return null;
}
log.error("Send transaction message failed with exception : ", exception);
numMessagesSendFailed.increment();
return null;
});
} else {
producer.newMessage().value(payloadBytes)
.sendAsync().thenRun(() -> {
long latencyMicros = NANOSECONDS.toMicros(
System.nanoTime() - sendTime);
messageSendRecorder.recordValue(latencyMicros);
messageSendRCumulativeRecorder.recordValue(latencyMicros);
numMessagesSendSuccess.increment();
}).exceptionally(exception -> {
if (exception instanceof InterruptedException && !executing.get()) {
return null;
}
log.error("Send message failed with exception : ", exception);
numMessagesSendFailed.increment();
return null;
});
}
}
}
if (rateLimiter != null){
rateLimiter.tryAcquire();
}
if (!arguments.isDisableTransaction) {
if (!arguments.isAbortTransaction) {
transaction.commit()
.thenRun(() -> {
numTxnOpSuccess.increment();
totalNumEndTxnOpSuccess.increment();
}).exceptionally(exception -> {
if (exception instanceof InterruptedException && !executing.get()) {
return null;
}
log.error("Commit transaction {} failed with exception",
transaction.getTxnID().toString(),
exception);
totalNumEndTxnOpFailed.increment();
return null;
});
} else {
transaction.abort().thenRun(() -> {
numTxnOpSuccess.increment();
totalNumEndTxnOpSuccess.increment();
}).exceptionally(exception -> {
if (exception instanceof InterruptedException && !executing.get()) {
return null;
}
log.error("Commit transaction {} failed with exception",
transaction.getTxnID().toString(),
exception);
totalNumEndTxnOpFailed.increment();
return null;
});
}
while (true) {
try {
Transaction newTransaction = client.newTransaction()
.withTransactionTimeout(arguments.transactionTimeout, TimeUnit.SECONDS)
.build()
.get();
atomicReference.compareAndSet(transaction, newTransaction);
totalNumTxnOpenTxnSuccess.increment();
break;
} catch (Exception throwable){
if (throwable instanceof InterruptedException && !executing.get()) {
break;
}
log.error("Failed to new transaction with exception: ", throwable);
totalNumTxnOpenTxnFail.increment();
}
}
} else {
totalNumTxnOpenTxnSuccess.increment();
totalNumEndTxnOpSuccess.increment();
numTxnOpSuccess.increment();
}
}
});
}
// Print report stats
long oldTime = System.nanoTime();
Histogram reportSendHistogram = null;
Histogram reportAckHistogram = null;
String statsFileName = "perf-transaction-" + System.currentTimeMillis() + ".hgrm";
log.info("Dumping latency stats to {}", statsFileName);
PrintStream histogramLog = new PrintStream(new FileOutputStream(statsFileName), false);
HistogramLogWriter histogramLogWriter = new HistogramLogWriter(histogramLog);
// Some log header bits
histogramLogWriter.outputLogFormatVersion();
histogramLogWriter.outputLegend();
while (executing.get()) {
try {
Thread.sleep(10000);
} catch (InterruptedException e) {
break;
}
long now = System.nanoTime();
double elapsed = (now - oldTime) / 1e9;
long total = totalNumEndTxnOpFailed.sum() + totalNumTxnOpenTxnSuccess.sum();
double rate = numTxnOpSuccess.sumThenReset() / elapsed;
reportSendHistogram = messageSendRecorder.getIntervalHistogram(reportSendHistogram);
reportAckHistogram = messageAckRecorder.getIntervalHistogram(reportAckHistogram);
String txnOrTaskLog = !arguments.isDisableTransaction
? "Throughput transaction: {} transaction executes --- {} transaction/s"
: "Throughput task: {} task executes --- {} task/s";
log.info(
txnOrTaskLog + " --- send Latency: mean: {} ms - med: {} "
+ "- 95pct: {} - 99pct: {} - 99.9pct: {} - 99.99pct: {} - Max: {}" + " --- ack Latency: "
+ "mean: {} ms - med: {} - 95pct: {} - 99pct: {} - 99.9pct: {} - 99.99pct: {} - Max: {}",
INTFORMAT.format(total),
DEC.format(rate),
DEC.format(reportSendHistogram.getMean() / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(50) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(95) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(99) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(99.9) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(99.99) / 1000.0),
DEC.format(reportSendHistogram.getMaxValue() / 1000.0),
DEC.format(reportAckHistogram.getMean() / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(50) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(95) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(99) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(99.9) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(99.99) / 1000.0),
DEC.format(reportAckHistogram.getMaxValue() / 1000.0));
histogramLogWriter.outputIntervalHistogram(reportSendHistogram);
histogramLogWriter.outputIntervalHistogram(reportAckHistogram);
reportSendHistogram.reset();
reportAckHistogram.reset();
oldTime = now;
}
}
private static void printTxnAggregatedThroughput(long start) {
double elapsed = (System.nanoTime() - start) / 1e9;
long numTransactionEndFailed = totalNumEndTxnOpFailed.sum();
long numTransactionEndSuccess = totalNumEndTxnOpSuccess.sum();
long total = numTransactionEndFailed + numTransactionEndSuccess;
double rate = total / elapsed;
long numMessageAckFailed = numMessagesAckFailed.sum();
long numMessageAckSuccess = numMessagesAckSuccess.sum();
long numMessageSendFailed = numMessagesSendFailed.sum();
long numMessageSendSuccess = numMessagesSendSuccess.sum();
long numTransactionOpenFailed = totalNumTxnOpenTxnFail.sum();
long numTransactionOpenSuccess = totalNumTxnOpenTxnSuccess.sum();
log.info(
"Aggregated throughput stats --- {} transaction executed --- {} transaction/s "
+ " --- {} transaction open successfully --- {} transaction open failed"
+ " --- {} transaction end successfully --- {} transaction end failed"
+ " --- {} message ack failed --- {} message send failed"
+ " --- {} message ack success --- {} message send success ",
total,
DEC.format(rate),
numTransactionOpenSuccess,
numTransactionOpenFailed,
numTransactionEndSuccess,
numTransactionEndFailed,
numMessageAckFailed,
numMessageSendFailed,
numMessageAckSuccess,
numMessageSendSuccess);
}
private static void printAggregatedThroughput(long start) {
double elapsed = (System.nanoTime() - start) / 1e9;
long total = totalNumEndTxnOpFailed.sum() + totalNumEndTxnOpSuccess.sum();
double rate = total / elapsed;
long numMessageAckFailed = numMessagesAckFailed.sum();
long numMessageAckSuccess = numMessagesAckSuccess.sum();
long numMessageSendFailed = numMessagesSendFailed.sum();
long numMessageSendSuccess = numMessagesSendSuccess.sum();
log.info(
"Aggregated throughput stats --- {} task executed --- {} task/s"
+ " --- {} message ack failed --- {} message send failed"
+ " --- {} message ack success --- {} message send success",
total,
TOTALFORMAT.format(rate),
numMessageAckFailed,
numMessageSendFailed,
numMessageAckSuccess,
numMessageSendSuccess);
}
private static void printAggregatedStats() {
Histogram reportAckHistogram = messageAckCumulativeRecorder.getIntervalHistogram();
Histogram reportSendHistogram = messageSendRCumulativeRecorder.getIntervalHistogram();
log.info(
"Messages ack aggregated latency stats --- Latency: mean: {} ms - med: {} - 95pct: {} - 99pct: {} - "
+ "99.9pct: {} - "
+ "99.99pct: {} - 99.999pct: {} - Max: {}",
DEC.format(reportAckHistogram.getMean() / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(50) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(95) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(99) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(99.9) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(99.99) / 1000.0),
DEC.format(reportAckHistogram.getValueAtPercentile(99.999) / 1000.0),
DEC.format(reportAckHistogram.getMaxValue() / 1000.0));
log.info(
"Messages send aggregated latency stats --- Latency: mean: {} ms - med: {} - 95pct: {} - 99pct: {} - "
+ "99.9pct: {} - "
+ "99.99pct: {} - 99.999pct: {} - Max: {}",
DEC.format(reportSendHistogram.getMean() / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(50) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(95) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(99) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(99.9) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(99.99) / 1000.0),
DEC.format(reportSendHistogram.getValueAtPercentile(99.999) / 1000.0),
DEC.format(reportSendHistogram.getMaxValue() / 1000.0));
}
static final DecimalFormat DEC = new PaddingDecimalFormat("0.000", 7);
static final DecimalFormat INTFORMAT = new PaddingDecimalFormat("0", 7);
static final DecimalFormat TOTALFORMAT = new DecimalFormat("0.000");
private static final Logger log = LoggerFactory.getLogger(PerformanceProducer.class);
private static List>> buildConsumer(PulsarClient client, Arguments arguments)
throws ExecutionException, InterruptedException {
ConsumerBuilder consumerBuilder = client.newConsumer(Schema.BYTES) //
.subscriptionType(arguments.subscriptionType)
.receiverQueueSize(arguments.receiverQueueSize)
.subscriptionInitialPosition(arguments.subscriptionInitialPosition);
Iterator consumerTopicsIterator = arguments.consumerTopic.iterator();
List>> consumers = new ArrayList<>(arguments.consumerTopic.size());
while (consumerTopicsIterator.hasNext()){
String topic = consumerTopicsIterator.next();
final List> subscriptions = new ArrayList<>(arguments.numSubscriptions);
final List>> subscriptionFutures =
new ArrayList<>(arguments.numSubscriptions);
log.info("Create subscriptions for topic {}", topic);
for (int j = 0; j < arguments.numSubscriptions; j++) {
String subscriberName = arguments.subscriptions.get(j);
subscriptionFutures
.add(consumerBuilder.clone().topic(topic).subscriptionName(subscriberName)
.subscribeAsync());
}
for (Future> future : subscriptionFutures) {
subscriptions.add(future.get());
}
consumers.add(subscriptions);
}
return consumers;
}
private static List> buildProducers(PulsarClient client, Arguments arguments)
throws ExecutionException, InterruptedException {
ProducerBuilder producerBuilder = client.newProducer(Schema.BYTES)
.sendTimeout(0, TimeUnit.SECONDS);
final List>> producerFutures = new ArrayList<>();
for (String topic : arguments.producerTopic) {
log.info("Create producer for topic {}", topic);
producerFutures.add(producerBuilder.clone().topic(topic).createAsync());
}
final List> producers = new ArrayList<>(producerFutures.size());
for (Future> future : producerFutures) {
producers.add(future.get());
}
return producers;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy