129 lines
4.9 KiB
Groovy
129 lines
4.9 KiB
Groovy
@GrabResolver(name='Paho', root='https://repo.eclipse.org/content/repositories/paho-releases/')
|
|
//@Grab(group='org.eclipse.paho', module='mqtt-client', version='0.4.0')
|
|
@Grab(group='org.eclipse.paho', module='org.eclipse.paho.client.mqttv3', version='1.1.0')
|
|
@Grab(group='org.eclipse.jetty.toolchain', module='jetty-perf-helper', version='1.0.5')
|
|
//@Grab(group='org.slf4j', module='slf4j-api', version='1.7.21')
|
|
//@Grab(group='org.slf4j', module='slf4j-jdk14', version='1.7.21')
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttClient
|
|
import org.eclipse.paho.client.mqttv3.MqttMessage
|
|
import org.eclipse.paho.client.mqttv3.MqttException
|
|
import org.eclipse.paho.client.mqttv3.MqttCallback
|
|
import org.eclipse.paho.client.mqttv3.IMqttDeliveryToken
|
|
import org.eclipse.paho.client.mqttv3.persist.MqttDefaultFilePersistence
|
|
import org.HdrHistogram.Histogram
|
|
import org.eclipse.jetty.toolchain.perf.PlatformTimer
|
|
|
|
import java.util.concurrent.Semaphore
|
|
//import groovy.util.logging.Slf4j
|
|
|
|
|
|
/**
|
|
* Performance metric test for issue 204.
|
|
* One topic per client
|
|
* Flow:
|
|
* [MEASURE START]
|
|
* - connect
|
|
* - subscribe to topic "test"+i
|
|
* - publish something on "test"+i
|
|
* - unsubscribe from "test"+i
|
|
* - disconnect
|
|
* [MEASURE STOP]
|
|
* */
|
|
//@Slf4j
|
|
class WorkflowTask implements MqttCallback {
|
|
private MqttClient client
|
|
private int clientCounter
|
|
private long startTime
|
|
private long numOfIterations
|
|
Histogram histogram
|
|
private int pauseMicroseconds
|
|
private PlatformTimer timer = PlatformTimer.detect()
|
|
private Semaphore disconnectReceived = new Semaphore(0)
|
|
|
|
private static String printDate() {
|
|
return new Date().format("HH:mm:ss,SSS") + " " + Thread.currentThread().getName()
|
|
}
|
|
|
|
void execute() {
|
|
client.callback = this
|
|
|
|
(1..numOfIterations).each { it ->
|
|
startTime = System.nanoTime()
|
|
println "\nAsk connect " + printDate()
|
|
client.connect()//.waitForCompletion()
|
|
println "connected ${it}, " + printDate()
|
|
client.subscribe("/topic${clientCounter}", 0)
|
|
println "subscribed to /topic${clientCounter} " + printDate()
|
|
byte[] bytes = "TestPayload ${it}".bytes
|
|
client.publish("/topic${clientCounter}", bytes, 0, false)
|
|
println "published ${it}, " + printDate()
|
|
// println "published to /topic${clientCounter}..wait for receive of message"
|
|
|
|
disconnectReceived.acquire()
|
|
println "Acquired semaphore, value ${disconnectReceived.availablePermits()} " + printDate()
|
|
println "Ask disconnect " + printDate()
|
|
client.disconnect()//.waitForCompletion()
|
|
println "Disconnected ${it}" + printDate() + "\n"
|
|
client.debug.dumpClientDebug()
|
|
// client.close()
|
|
long stopTime = System.nanoTime()
|
|
long spentNanoSeconds = stopTime - startTime
|
|
histogram.recordValue(spentNanoSeconds)
|
|
|
|
// long waitMicroseconds = pauseMicroseconds - (spentNanoSeconds / 1000)
|
|
// println "${spentNanoSeconds} spent nanos, pauseMicroseconds: ${pauseMicroseconds}"
|
|
// timer.sleep(waitMicroseconds)
|
|
}
|
|
println "Latency diagram (micro secs):\n"
|
|
histogram.outputPercentileDistribution(System.out, 1000.0)
|
|
}
|
|
|
|
public void messageArrived(String topic, MqttMessage message) throws Exception {
|
|
println "Received published " + printDate()
|
|
if (topic != "/topic${clientCounter}".toString()) {
|
|
println "Error received a publish on not matching topic ${topic}"
|
|
return
|
|
}
|
|
// long stopTime = System.nanoTime()
|
|
// long spentNanoSeconds = stopTime - startTime
|
|
|
|
disconnectReceived.release()
|
|
println "Released semaphore, value ${disconnectReceived.availablePermits()} " + printDate()
|
|
}
|
|
|
|
public void connectionLost(Throwable cause) {
|
|
println "Client ${clientCounter} lost connection, caused by ${cause}"
|
|
}
|
|
|
|
public void deliveryComplete(IMqttDeliveryToken token) {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
MqttClient createClient(int i) {
|
|
String tmpDir = System.getProperty("java.io.tmpdir")
|
|
MqttDefaultFilePersistence dataStore = new MqttDefaultFilePersistence(tmpDir)
|
|
new MqttClient("tcp://localhost:1883", "TestClient${i}", dataStore)
|
|
}
|
|
|
|
Histogram histogram = new Histogram(5)
|
|
if (args.size() != 2) {
|
|
println "Usage:"
|
|
println "\n groovy conSubPubUnsub_issue204.groovy <num of iterations per second> <seconds>"
|
|
return 1
|
|
}
|
|
int numOfIterationsPerSecond = args[0] as int
|
|
int duration = args[1] as int
|
|
long pauseMicroseconds = (1 / numOfIterationsPerSecond) * 1000 * 1000
|
|
long numOfIterations = numOfIterationsPerSecond * duration
|
|
|
|
MqttClient client = createClient(0)
|
|
def task = new WorkflowTask(client: client, clientCounter: 0, histogram: histogram,
|
|
pauseMicroseconds: pauseMicroseconds, numOfIterations: numOfIterations)
|
|
|
|
task.execute()
|
|
|
|
//println "Latency diagram:\n"
|
|
//histogram.outputPercentileDistribution(System.out, 1000.0) |