221 lines
10 KiB
Java
221 lines
10 KiB
Java
package de.bitsharesmunich.graphenej.api;
|
|
|
|
import com.google.gson.Gson;
|
|
import com.google.gson.GsonBuilder;
|
|
import com.google.gson.reflect.TypeToken;
|
|
import com.neovisionaries.ws.client.WebSocket;
|
|
import com.neovisionaries.ws.client.WebSocketFrame;
|
|
|
|
import java.io.Serializable;
|
|
import java.lang.reflect.Type;
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import de.bitsharesmunich.graphenej.AssetAmount;
|
|
import de.bitsharesmunich.graphenej.RPC;
|
|
import de.bitsharesmunich.graphenej.Transaction;
|
|
import de.bitsharesmunich.graphenej.UserAccount;
|
|
import de.bitsharesmunich.graphenej.errors.RepeatedRequestIdException;
|
|
import de.bitsharesmunich.graphenej.interfaces.SubscriptionHub;
|
|
import de.bitsharesmunich.graphenej.interfaces.SubscriptionListener;
|
|
import de.bitsharesmunich.graphenej.interfaces.WitnessResponseListener;
|
|
import de.bitsharesmunich.graphenej.models.ApiCall;
|
|
import de.bitsharesmunich.graphenej.models.DynamicGlobalProperties;
|
|
import de.bitsharesmunich.graphenej.models.SubscriptionResponse;
|
|
import de.bitsharesmunich.graphenej.models.WitnessResponse;
|
|
import de.bitsharesmunich.graphenej.operations.LimitOrderCreateOperation;
|
|
import de.bitsharesmunich.graphenej.operations.TransferOperation;
|
|
|
|
/**
|
|
* A websocket adapter prepared to be used as a basic dispatch hub for subscription messages.
|
|
*
|
|
* Created by nelson on 1/26/17.
|
|
*/
|
|
public class SubscriptionMessagesHub extends BaseGrapheneHandler implements SubscriptionHub {
|
|
|
|
private WebSocket mWebsocket;
|
|
|
|
// Sequence of message ids
|
|
public final static int LOGIN_ID = 1;
|
|
public final static int GET_DATABASE_ID = 2;
|
|
public final static int SUBCRIPTION_REQUEST = 3;
|
|
|
|
// ID of subscription notifications
|
|
public final static int SUBCRIPTION_NOTIFICATION = 4;
|
|
|
|
private SubscriptionResponse.SubscriptionResponseDeserializer mSubscriptionDeserializer;
|
|
private Gson gson;
|
|
private String user;
|
|
private String password;
|
|
private boolean clearFilter;
|
|
private int currentId;
|
|
private int databaseApiId = -1;
|
|
private int subscriptionCounter = 0;
|
|
private HashMap<Long, BaseGrapheneHandler> mHandlerMap = new HashMap<>();
|
|
|
|
/**
|
|
* Id used to separate requests regarding the subscriptions
|
|
*/
|
|
private final int SUBSCRIPTION_ID = 10;
|
|
|
|
/**
|
|
* Constructor used to create a subscription message hub that will call the set_subscribe_callback
|
|
* API with the clear_filter parameter set to false, meaning that it will only receive automatic updates
|
|
* from objects we register.
|
|
*
|
|
* A list of ObjectTypes must be provided, otherwise we won't get any update.
|
|
*
|
|
* @param user: User name, in case the node to which we're going to connect to requires authentication
|
|
* @param password: Password, same as above
|
|
* @param clearFilter: Whether to automatically subscribe of not to the notification feed.
|
|
* @param errorListener: Callback that will be fired in case there is an error.
|
|
*/
|
|
public SubscriptionMessagesHub(String user, String password, boolean clearFilter, WitnessResponseListener errorListener){
|
|
super(errorListener);
|
|
this.user = user;
|
|
this.password = password;
|
|
this.clearFilter = clearFilter;
|
|
this.mSubscriptionDeserializer = new SubscriptionResponse.SubscriptionResponseDeserializer();
|
|
GsonBuilder builder = new GsonBuilder();
|
|
builder.registerTypeAdapter(SubscriptionResponse.class, mSubscriptionDeserializer);
|
|
builder.registerTypeAdapter(Transaction.class, new Transaction.TransactionDeserializer());
|
|
builder.registerTypeAdapter(TransferOperation.class, new TransferOperation.TransferDeserializer());
|
|
builder.registerTypeAdapter(LimitOrderCreateOperation.class, new LimitOrderCreateOperation.LimitOrderCreateDeserializer());
|
|
builder.registerTypeAdapter(AssetAmount.class, new AssetAmount.AssetAmountDeserializer());
|
|
builder.registerTypeAdapter(UserAccount.class, new UserAccount.UserAccountSimpleDeserializer());
|
|
builder.registerTypeAdapter(DynamicGlobalProperties.class, new DynamicGlobalProperties.DynamicGlobalPropertiesDeserializer());
|
|
this.gson = builder.create();
|
|
}
|
|
|
|
/**
|
|
* Constructor used to create a subscription message hub that will call the set_subscribe_callback
|
|
<<<<<<< Updated upstream
|
|
* API with the clear_filter parameter set to true, meaning that it will receive automatic updates
|
|
* on all network events.
|
|
=======
|
|
* API with the clear_filter parameter set to false, meaning that it will only receive automatic updates
|
|
* from objects we register.
|
|
>>>>>>> Stashed changes
|
|
*
|
|
* @param user: User name, in case the node to which we're going to connect to requires authentication
|
|
* @param password: Password, same as above
|
|
* @param errorListener: Callback that will be fired in case there is an error.
|
|
*/
|
|
public SubscriptionMessagesHub(String user, String password, WitnessResponseListener errorListener){
|
|
this(user, password, false, errorListener);
|
|
}
|
|
|
|
@Override
|
|
public void addSubscriptionListener(SubscriptionListener listener){
|
|
this.mSubscriptionDeserializer.addSubscriptionListener(listener);
|
|
}
|
|
|
|
@Override
|
|
public void removeSubscriptionListener(SubscriptionListener listener){
|
|
this.mSubscriptionDeserializer.removeSubscriptionListener(listener);
|
|
}
|
|
|
|
@Override
|
|
public List<SubscriptionListener> getSubscriptionListeners() {
|
|
return this.mSubscriptionDeserializer.getSubscriptionListeners();
|
|
}
|
|
|
|
@Override
|
|
public void onConnected(WebSocket websocket, Map<String, List<String>> headers) throws Exception {
|
|
this.mWebsocket = websocket;
|
|
ArrayList<Serializable> loginParams = new ArrayList<>();
|
|
currentId = LOGIN_ID;
|
|
loginParams.add(user);
|
|
loginParams.add(password);
|
|
ApiCall loginCall = new ApiCall(1, RPC.CALL_LOGIN, loginParams, RPC.VERSION, currentId);
|
|
websocket.sendText(loginCall.toJsonString());
|
|
}
|
|
|
|
@Override
|
|
public void onTextFrame(WebSocket websocket, WebSocketFrame frame) throws Exception {
|
|
String message = frame.getPayloadText();
|
|
System.out.println("<< "+message);
|
|
if(currentId == LOGIN_ID){
|
|
ArrayList<Serializable> emptyParams = new ArrayList<>();
|
|
ApiCall getDatabaseId = new ApiCall(1, RPC.CALL_DATABASE, emptyParams, RPC.VERSION, currentId);
|
|
websocket.sendText(getDatabaseId.toJsonString());
|
|
currentId++;
|
|
}else if(currentId == GET_DATABASE_ID){
|
|
Type ApiIdResponse = new TypeToken<WitnessResponse<Integer>>() {}.getType();
|
|
WitnessResponse<Integer> witnessResponse = gson.fromJson(message, ApiIdResponse);
|
|
databaseApiId = witnessResponse.result;
|
|
|
|
ArrayList<Serializable> subscriptionParams = new ArrayList<>();
|
|
subscriptionParams.add(String.format("%d", SUBCRIPTION_NOTIFICATION));
|
|
subscriptionParams.add(clearFilter);
|
|
ApiCall getDatabaseId = new ApiCall(databaseApiId, RPC.CALL_SET_SUBSCRIBE_CALLBACK, subscriptionParams, RPC.VERSION, SUBCRIPTION_REQUEST);
|
|
websocket.sendText(getDatabaseId.toJsonString());
|
|
currentId++;
|
|
} else if(currentId == SUBCRIPTION_REQUEST){
|
|
List<SubscriptionListener> subscriptionListeners = mSubscriptionDeserializer.getSubscriptionListeners();
|
|
|
|
// If we haven't subscribed to all requested subscription channels yet,
|
|
// just send one more subscription
|
|
if(subscriptionListeners != null &&
|
|
subscriptionListeners.size() > 0 &&
|
|
subscriptionCounter < subscriptionListeners.size()){
|
|
|
|
ArrayList<Serializable> objects = new ArrayList<>();
|
|
ArrayList<Serializable> payload = new ArrayList<>();
|
|
for(SubscriptionListener listener : subscriptionListeners){
|
|
objects.add(listener.getInterestObjectType().getGenericObjectId());
|
|
}
|
|
|
|
payload.add(objects);
|
|
ApiCall subscribe = new ApiCall(databaseApiId, RPC.GET_OBJECTS, payload, RPC.VERSION, SUBSCRIPTION_ID);
|
|
websocket.sendText(subscribe.toJsonString());
|
|
subscriptionCounter++;
|
|
}else{
|
|
WitnessResponse witnessResponse = gson.fromJson(message, WitnessResponse.class);
|
|
if(witnessResponse.result != null){
|
|
// This is the response to a request that was submitted to the message hub
|
|
// and whose handler was stored in the "request id" -> "handler" map
|
|
BaseGrapheneHandler handler = mHandlerMap.get(witnessResponse.id);
|
|
handler.onTextFrame(websocket, frame);
|
|
mHandlerMap.remove(witnessResponse.id);
|
|
}else{
|
|
// If we've already subscribed to all requested subscription channels, we
|
|
// just proceed to deserialize content.
|
|
// The deserialization is handled by all those TypeAdapters registered in the class
|
|
// constructor while building the gson instance.
|
|
SubscriptionResponse response = gson.fromJson(message, SubscriptionResponse.class);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onFrameSent(WebSocket websocket, WebSocketFrame frame) throws Exception {
|
|
System.out.println(">> "+frame.getPayloadText());
|
|
}
|
|
|
|
public void reset(){
|
|
currentId = 0;
|
|
databaseApiId = -1;
|
|
subscriptionCounter = 0;
|
|
}
|
|
|
|
public void addRequestHandler(BaseGrapheneHandler handler) throws RepeatedRequestIdException {
|
|
if(mHandlerMap.get(handler.getRequestId()) != null){
|
|
throw new RepeatedRequestIdException("Already registered handler with id: "+handler.getRequestId());
|
|
}
|
|
|
|
System.out.println("Registering handler with id: "+handler.getRequestId());
|
|
mHandlerMap.put(handler.getRequestId(), handler);
|
|
|
|
try {
|
|
handler.onConnected(mWebsocket, null);
|
|
} catch (Exception e) {
|
|
System.out.println("Exception. Msg: "+e.getMessage());
|
|
System.out.println("Exception type: "+e);
|
|
}
|
|
}
|
|
}
|