package com.iplatform.tcp;
|
|
import com.walker.infrastructure.utils.JsonUtils;
|
import org.java_websocket.client.WebSocketClient;
|
import org.java_websocket.handshake.ServerHandshake;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
|
import java.net.URI;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
public class DemoWebsocketClient extends WebSocketClient {
|
|
protected final transient Logger logger = LoggerFactory.getLogger(this.getClass());
|
|
public DemoWebsocketClient(URI serverUri) {
|
super(serverUri);
|
}
|
|
@Override
|
public void onOpen(ServerHandshake serverHandshake) {
|
// logger.info("ws 连接成功");
|
String data = null;
|
Map<String, String> map = new HashMap<>(4);
|
map.put("protocol", "login");
|
map.put("uid", this.uid);
|
try {
|
data = JsonUtils.objectToJsonString(map);
|
} catch (Exception e) {
|
throw new RuntimeException(e);
|
}
|
this.send(data);
|
logger.info("客户端发送登录请求，name = {}", this.uid);
|
}
|
|
@Override
|
public void onMessage(String s) {
|
if(s.indexOf("heartbeat") >= 0){
|
return;
|
}
|
logger.info("ws 收到消息:{}", s);
|
try {
|
DemoLoginResponse response = JsonUtils.jsonStringToObject(s, DemoLoginResponse.class);
|
if(response.getProtocol().equals("login")){
|
logger.info("websocket 登录成功：{}", response.getUid());
|
}
|
} catch (Exception e) {
|
throw new RuntimeException(e);
|
}
|
}
|
|
@Override
|
public void onClose(int i, String s, boolean b) {
|
logger.warn("连接被关闭：code={}, reason={}, remote={}", i, s, b);
|
}
|
|
@Override
|
public void onError(Exception e) {
|
logger.error("连接错误：" + e.getMessage(), e);
|
}
|
|
public String getUid() {
|
return uid;
|
}
|
|
public void setUid(String uid) {
|
this.uid = uid;
|
}
|
|
public String getUserName() {
|
return userName;
|
}
|
|
public void setUserName(String userName) {
|
this.userName = userName;
|
}
|
|
private String uid;
|
private String userName;
|
}
|
