java client

This commit is contained in:
heavyrain lee
2019-04-29 23:06:46 +08:00
parent c1bb45fe14
commit 9aa29685ef
@@ -15,11 +15,16 @@ import org.fusesource.hawtbuf.Buffer;
import org.fusesource.hawtbuf.UTF8Buffer; import org.fusesource.hawtbuf.UTF8Buffer;
import org.fusesource.mqtt.client.*; import org.fusesource.mqtt.client.*;
import java.io.IOException; import java.io.*;
import java.nio.ByteBuffer;
import java.nio.charset.Charset; import java.nio.charset.Charset;
import java.util.Base64; import java.util.Base64;
import java.util.List; import java.util.List;
import static cn.wildfirechat.client.IMClient.ConnectionStatus.ConnectionStatus_Connected;
import static cn.wildfirechat.client.IMClient.ConnectionStatus.ConnectionStatus_Connecting;
import static cn.wildfirechat.client.IMClient.ConnectionStatus.ConnectionStatus_Unconnected;
public class IMClient implements Listener { public class IMClient implements Listener {
private final String userId; private final String userId;
@@ -38,19 +43,25 @@ public class IMClient implements Listener {
private long messageHead; private long messageHead;
interface ReceiveMessageCallback { interface ReceiveMessageCallback {
void onReceiveMessages(List<WFCMessage.Message> messageList, boolean hasMore); void onReceiveMessages(IMClient client, List<WFCMessage.Message> messageList, boolean hasMore);
void onRecallMessage(long messageUid); void onRecallMessage(IMClient client, long messageUid);
} }
interface ConnectionStatusCallback { interface ConnectionStatusCallback {
void onConnectionStatusChanged(int newStatus); void onConnectionStatusChanged(IMClient client, ConnectionStatus newStatus);
} }
interface SendMessageCallback { interface SendMessageCallback {
void onSuccess(long messageUid); void onSuccess(long messageUid, long timestamp);
void onFailure(int errorCode); void onFailure(int errorCode);
} }
public enum ConnectionStatus {
ConnectionStatus_Unconnected,
ConnectionStatus_Connecting,
ConnectionStatus_Connected,
}
public IMClient(String userId, String token, String clientId, String host, int port, ReceiveMessageCallback messageCallback, ConnectionStatusCallback connectionStatusCallback) { public IMClient(String userId, String token, String clientId, String host, int port, ReceiveMessageCallback messageCallback, ConnectionStatusCallback connectionStatusCallback) {
this.userId = userId; this.userId = userId;
@@ -76,7 +87,7 @@ public class IMClient implements Listener {
if(route(userId, token)) { if(route(userId, token)) {
try { try {
mqtt.setHost("tcp://" + mqttServerIp + ":" + mqttServerPort); mqtt.setHost("tcp://" + mqttServerIp + ":" + mqttServerPort);
mqtt.setVersion("1.1"); mqtt.setVersion("3.1.1");
mqtt.setKeepAlive((short)180); mqtt.setKeepAlive((short)180);
mqtt.setClientId(clientId); mqtt.setClientId(clientId);
@@ -90,6 +101,9 @@ public class IMClient implements Listener {
connection = mqtt.callbackConnection(); connection = mqtt.callbackConnection();
connection.listener(this); connection.listener(this);
//connecting
connectionStatusCallback.onConnectionStatusChanged(this, ConnectionStatus_Connecting);
connection.connect(new Callback<byte[]>() { connection.connect(new Callback<byte[]>() {
@Override @Override
public void onSuccess(byte[] value) { public void onSuccess(byte[] value) {
@@ -103,12 +117,16 @@ public class IMClient implements Listener {
} }
} }
System.out.println("on connect success"); System.out.println("on connect success");
//connected
connectionStatusCallback.onConnectionStatusChanged(IMClient.this, ConnectionStatus_Connected);
} }
@Override @Override
public void onFailure(Throwable value) { public void onFailure(Throwable value) {
System.out.println("on connect failure"); System.out.println("on connect failure");
connectionStatusCallback.onConnectionStatusChanged(IMClient.this, ConnectionStatus_Unconnected);
} }
}); });
@@ -127,7 +145,17 @@ public class IMClient implements Listener {
@Override @Override
public void onSuccess(byte[] value) { public void onSuccess(byte[] value) {
if (value[0] == 0) { if (value[0] == 0) {
callback.onSuccess(0L); byte[] data = new byte[value.length-1];
for (int i = 0; i < data.length; i++) {
data[i] = value[i+1];
}
data = AES.AESDecrypt(data, privateSecret, true);
ByteBuffer buffer = ByteBuffer.wrap(data, 0,16);
long messageUid = buffer.getLong();
long timestamp = buffer.getLong();
callback.onSuccess(messageUid, timestamp);
} else { } else {
callback.onFailure(value[0]); callback.onFailure(value[0]);
} }
@@ -150,7 +178,9 @@ public class IMClient implements Listener {
inputRoute.setClientId(clientId); inputRoute.setClientId(clientId);
inputRoute.setToken(token); inputRoute.setToken(token);
WFCMessage.IMHttpWrapper request = WFCMessage.IMHttpWrapper.newBuilder().setClientId(clientId).setToken(token).setRequest("ROUTE").setData(ByteString.copyFrom(token.getBytes())).build(); WFCMessage.RouteRequest routeRequest = WFCMessage.RouteRequest.newBuilder().setPlatform(0).build();
WFCMessage.IMHttpWrapper request = WFCMessage.IMHttpWrapper.newBuilder().setClientId(clientId).setToken(token).setRequest("ROUTE").setData(ByteString.copyFrom(routeRequest.toByteArray())).build();
byte[] data = AES.AESEncrypt(request.toByteArray(), privateSecret); byte[] data = AES.AESEncrypt(request.toByteArray(), privateSecret);
data = Base64.getEncoder().encode(data); data = Base64.getEncoder().encode(data);
@@ -186,7 +216,7 @@ public class IMClient implements Listener {
for (int i = 0; i < bytes1.length; i++) { for (int i = 0; i < bytes1.length; i++) {
bytes1[i] = bytes[i+1]; bytes1[i] = bytes[i+1];
} }
byte[] rawData = AES.AESDecrypt(bytes1, "", true); byte[] rawData = AES.AESDecrypt(bytes1, privateSecret, true);
WFCMessage.RouteResponse routeResponse = WFCMessage.RouteResponse.parseFrom(rawData); WFCMessage.RouteResponse routeResponse = WFCMessage.RouteResponse.parseFrom(rawData);
mqttServerIp = routeResponse.getHost(); mqttServerIp = routeResponse.getHost();
mqttServerPort = routeResponse.getLongPort(); mqttServerPort = routeResponse.getLongPort();
@@ -237,8 +267,9 @@ public class IMClient implements Listener {
try { try {
WFCMessage.PullMessageResult result = WFCMessage.PullMessageResult.parseFrom(data); WFCMessage.PullMessageResult result = WFCMessage.PullMessageResult.parseFrom(data);
if (receiveMessageCallback != null) { if (receiveMessageCallback != null) {
receiveMessageCallback.onReceiveMessages(result.getMessageList(), false); receiveMessageCallback.onReceiveMessages(IMClient.this, result.getMessageList(), false);
} }
messageHead = result.getHead();
} catch (InvalidProtocolBufferException e) { } catch (InvalidProtocolBufferException e) {
e.printStackTrace(); e.printStackTrace();
} }
@@ -267,20 +298,33 @@ public class IMClient implements Listener {
} }
public static void main(String[] args) { public static void main(String[] args) {
IMClient client = new IMClient("cgc8c8VV", "oOVX4iTM/JSDtWPTqRYVDSDdSwLSjSgtU99gTYd73PLMRZQAxce1IEyEvYVHt05ar7A8M3cTlxQHh1TywGcb8WmXn3zu5BZAAZIE4QpDwhaXf1p6OiF5IDL6Ksa1KgAyMQf4Q7UHK3COkg8XBGf2TRo8AIpligitH2loeyh0Ol4=", "DD72C212-26C7-4B38-A5FC-88550896B170", "wildfirechat.cn", 80, new ReceiveMessageCallback() { //token与userid和clientid是绑定的,使用时一定要传入正确的userid和clientid,不然会认为token非法
IMClient client = new IMClient("yzyOyOKK", "7SJk13q+YdHHe6EwDzry9BKogxTNf3UgtYj50cBTZgWNkNuxEkiqg2koKg0lXViONIX1LmwCR1jN0Mw8hvk6KGpiSKFi+IRaRkIb3mNzgIfrq4afhyIHaQfa2HOfsi6Ws+9YobkdDgdq7W70bEdVfiCSU9+JOIY449nxZzfg2Zw=", "DD72C212-26C7-4B38-A5FC-88550896B170", "192.168.1.101", 80, new ReceiveMessageCallback() {
@Override @Override
public void onReceiveMessages(List<WFCMessage.Message> messageList, boolean hasMore) { public void onReceiveMessages(IMClient client, List<WFCMessage.Message> messageList, boolean hasMore) {
} }
@Override @Override
public void onRecallMessage(long messageUid) { public void onRecallMessage(IMClient client, long messageUid) {
} }
}, new ConnectionStatusCallback() { }, new ConnectionStatusCallback() {
@Override @Override
public void onConnectionStatusChanged(int newStatus) { public void onConnectionStatusChanged(IMClient client, ConnectionStatus newStatus) {
if (newStatus == ConnectionStatus_Connected) {
client.sendMessage(WFCMessage.Conversation.newBuilder().setType(0).setTarget("yzyOyOKK").setLine(0).build(), WFCMessage.MessageContent.newBuilder().setContent("helloworld").setType(1).build(), new SendMessageCallback() {
@Override
public void onSuccess(long messageUid, long timestamp) {
System.out.println("send success");
}
@Override
public void onFailure(int errorCode) {
System.out.println("send failure");
}
});
}
} }
}); });