private ServiceSocket getConnectionSocket() throws URISyntaxException, Exception {
URI uri = getUri();
String connectionId = getThreadName() + getConnectionId();
ServiceSocket socket;
WebSocketClient webSocketClient;
if (isStreamingConnection()) {
if (connectionList.containsKey(connectionId)) {
socket = connectionList.get(connectionId);
socket.initialize();
return socket;
} else {
socket = new ServiceSocket(this);
connectionList.put(connectionId, socket);
}
} else {
socket = new ServiceSocket(this);
}
SslContextFactory sslContexFactory = new SslContextFactory();
sslContexFactory.setTrustAll(isIgnoreSslErrors());
webSocketClient = new WebSocketClient(sslContexFactory);
webSocketClient.start();
ClientUpgradeRequest request = new ClientUpgradeRequest();
webSocketClient.connect(socket, uri, request);
int connectionTimeout = Integer.parseInt(getConnectionTimeout());
socket.awaitOpen(connectionTimeout, TimeUnit.MILLISECONDS);
return socket;
}
@Override
public SampleResult sample(Entry entry) {
ServiceSocket socket = null;
SampleResult sampleResult = new SampleResult();
sampleResult.setSampleLabel(getName());
sampleResult.setDataEncoding(getContentEncoding());
StringBuilder errorList = new StringBuilder();
errorList.append("\n\n[Problems]\n");
boolean isOK = false;
String payloadMessage = getRequestPayload();
sampleResult.setSamplerData(payloadMessage);
sampleResult.sampleStart();
try {
socket = getConnectionSocket();
if (socket == null) {
sampleResult.setResponseCode("500");
sampleResult.setSuccessful(false);
sampleResult.sampleEnd();
sampleResult.setResponseMessage(errorList.toString());
errorList.append(" - Connection couldn't be opened").append("\n");
return sampleResult;
}
if (!payloadMessage.isEmpty()) {
socket.sendMessage(payloadMessage);
}
int responseTimeout = Integer.parseInt(getResponseTimeout());
socket.awaitClose(responseTimeout, TimeUnit.MILLISECONDS);
if (socket.getResponseMessage() == null || socket.getResponseMessage().isEmpty()) {
sampleResult.setResponseCode("204");
}
if (socket.getError() != 0) {
isOK = false;
sampleResult.setResponseCode(socket.getError().toString());
} else {
sampleResult.setResponseCodeOK();
isOK = true;
}
sampleResult.setResponseData(socket.getResponseMessage(), getContentEncoding());
} catch (URISyntaxException e) {
errorList
.append(" - Invalid URI syntax: ")
.append(e.getMessage())
.append("\n")
.append(StringUtils.join(e.getStackTrace(), "\n"))
.append("\n");
} catch (IOException e) {
errorList
.append(" - IO Exception: ")
.append(e.getMessage())
.append("\n")
.append(StringUtils.join(e.getStackTrace(), "\n"))
.append("\n");
} catch (NumberFormatException e) {
errorList
.append(" - Cannot parse number: ")
.append(e.getMessage())
.append("\n")
.append(StringUtils.join(e.getStackTrace(), "\n"))
.append("\n");
} catch (InterruptedException e) {
errorList
.append(" - Execution interrupted: ")
.append(e.getMessage())
.append("\n")
.append(StringUtils.join(e.getStackTrace(), "\n"))
.append("\n");
} catch (Exception e) {
errorList
.append(" - Unexpected error: ")
.append(e.getMessage())
.append("\n")
.append(StringUtils.join(e.getStackTrace(), "\n"))
.append("\n");
}
sampleResult.sampleEnd();
sampleResult.setSuccessful(isOK);
String logMessage = (socket != null) ? socket.getLogMessage() : "";
sampleResult.setResponseMessage(logMessage + errorList);
return sampleResult;
}
@Override
public void testEnded(String host) {
for (ServiceSocket socket : connectionList.values()) {
socket.close();
}
}