背景：websocket在Html5里面有着举足轻重的地位，最近工做中须要对websocket接口进行压测，使用jmeter的websocket

sample进行，可是通常现网的websocket都是采用的wss://协议，直接使用ws://协议的较少，毕竟安全仍是很是重要的，这里就涉及到ssl协议，模拟websocket客户端，使用证书和公钥，摸索了半天终于摸索出来了。html

java websocket client

maven项目中引入git

org.java-websocket

Java-WebSocket

1.4.0

什么是wss

wss 是 Web Socket Secure 的简称， wss是websocket 在TLS之上的加密版本。web

如何获取证书，并使用keytool工具生成本地store秘钥文件

keytool -import -alias "my.store" -file wss.cer -keystore my.store

websocket ssl client 代码示例

首先继承抽象类WebSocketClient，并实现websocket的onMessage、onOpen、onClose、onError，其实就是websocket的事件浏览器

import java.net.URI;

import org.java_websocket.client.WebSocketClient;

import org.java_websocket.handshake.ServerHandshake;

class WebSocketClientInst extends WebSocketClient {

public WebSocketClientInst(URI serverUri ) {

super( serverUri );

}

@Override

public void onOpen( ServerHandshake handshakedata ) {

System.out.println( "Connected" );

}

@Override

public void onMessage( String message ) {

System.out.println( "got: " + message );

}

@Override

public void onClose( int code, String reason, boolean remote ) {

System.out.println( "Disconnected" );

}

@Override

public void onError( Exception ex ) {

ex.printStackTrace();

}

}

而后就是wss的重头戏了，这里面的核心实际上是SSLSocketFactory 和SSLSocket，感兴趣能够深刻百度了解下：安全

import org.java_websocket.enums.ReadyState;

import javax.net.ssl.KeyManagerFactory;

import javax.net.ssl.SSLContext;

import javax.net.ssl.SSLSocketFactory;

import javax.net.ssl.TrustManagerFactory;

import java.io.*;

import java.net.URI;

import java.security.KeyStore;

class SSLClientExample {

public static void main( String[] args ) throws Exception {

WebSocketClientInst chatclient = new WebSocketClientInst( new URI( "wss://ip:port/test/path" ) );

// load up the key store

String KEYSTORE = "xx/xxxx/my.store"; //基于证书生成的store秘钥文件的路径

String STOREPASSWORD = "123456"; //使用keytool工具时，输入的密码

String KEYPASSWORD = "123456";

KeyStore ks = KeyStore.getInstance( KeyStore.getDefaultType() );

File kf = new File( KEYSTORE );

ks.load( new FileInputStream( kf ), STOREPASSWORD.toCharArray() );

KeyManagerFactory kmf = KeyManagerFactory.getInstance( "SunX509" );

kmf.init( ks, KEYPASSWORD.toCharArray() );

TrustManagerFactory tmf = TrustManagerFactory.getInstance( "SunX509" );

tmf.init( ks );

SSLContext sslContext = null;

sslContext = SSLContext.getInstance( "TLS" );

sslContext.init( kmf.getKeyManagers(), tmf.getTrustManagers(), null );

SSLSocketFactory factory = sslContext.getSocketFactory();// (SSLSocketFactory) SSLSocketFactory.getDefault();

chatclient.setSocketFactory( factory );

chatclient.connectBlocking();

boolean loop = true;

int times = 0;

while (loop){

times++;

if (ReadyState.OPEN.equals(chatclient.getReadyState())) {

chatclient.send(getBytes("a.wav")); //发送二进制文件

}else {

System.out.println("还没ready, 继续进行中");

if (times<=10) {

Thread.sleep(1000);

}else{

System.out.println("超时");

break;

}

}

}

}

public static byte[] getBytes(String filePath){

byte[] buffer = null;

try {

File file = new File(filePath);

FileInputStream fis = new FileInputStream(file);

ByteArrayOutputStream bos = new ByteArrayOutputStream(1000);

byte[] b = new byte[1000];

int n;

while ((n = fis.read(b)) != -1) {

bos.write(b, 0, n);

}

fis.close();

bos.close();

buffer = bos.toByteArray();

} catch (FileNotFoundException e) {

e.printStackTrace();

} catch (IOException e) {

e.printStackTrace();

}

return buffer;

}

}

至此完成代码，你能够使用这个代码，而后实现jmeter的java sample，用来做压测了websocket