xyz.thepathfinder.android.ModelMessageHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pathfinder-android Show documentation
Android/Java client library for the Pathfinder service
There is a newer version: 1.0.1
package xyz.thepathfinder.android;

import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Routes all web socket messages received to the receiving models.
 *
 * @author David Robinson
 */
class ModelMessageHandler implements MessageHandler {

    /**
     * Logs actions performed by the class.
     */
    private static final Logger logger = LoggerFactory.getLogger(MessageHandler.class);

    /**
     * Holds access to all the models.
     */
    private PathfinderServices services;

    /**
     * Number of messages received, helps with testing.
     */
    private int receivedMessageCount;

    /**
     * Constructs the message handler.
     *
     * @param services to find models.
     */
    protected ModelMessageHandler(PathfinderServices services) {
        this.services = services;
        this.receivedMessageCount = 0;
    }

    /**
     * Invoked when the web socket receives a message.
     *
     * @param message the message received as a String.
     */
    @Override
    public void onMessage(String message) {
        try {
            logger.info("Received json: " + message);
            this.receivedMessageCount++;
            JsonObject json = new JsonParser().parse(message).getAsJsonObject();

            if (!json.has("message") || !json.has("model")) {
                logger.warn("Ignoring invalid message: " + json.toString());
            } else {
                String type = json.get("message").getAsString();

                JsonObject value = json.getAsJsonObject("value");

                ModelType modelType = ModelType.getModelType(json.get("model").getAsString());
                logger.info("Model Type : " + modelType);

                Path path = new Path(this.getPath(json), modelType);

                Model model = this.services.getRegistry().getModel(path);
                if (model == null && type.equals("Created") && ModelType.CLUSTER != modelType) {
                    model = this.services.getRegistry().findInCreateBacklog(value, modelType);
                    if (model != null) {
                        this.services.getRegistry().removeCreateBacklog(model);
                        model.setPathName(path.getPathName());
                    }
                }

                if (model != null) {
                    logger.info("Notifying " + model.getPathName() + " Type: " + model.getModelType() + " of message");

                    model.notifyUpdate(type, json);
                } else {
                    Path parentPath = path.getParentPath();

                    if (parentPath != null && this.services.getRegistry().isModelRegistered(parentPath)) {
                        if (modelType == ModelType.CLUSTER) {
                            Cluster.getInstance(value, this.services);
                            return;
                        } else if (modelType == ModelType.COMMODITY) {
                            Commodity.getInstance(value, this.services);
                            return;
                        } else if (modelType == ModelType.TRANSPORT) {
                            Transport.getInstance(value, this.services);
                            return;
                        }
                    }

                    logger.warn("Received message that couldn't be routed to a model: " + message);
                }
            }
        } catch (Exception e) { // catch any exception that occured while serving a message
            logger.error(e.getMessage());
            e.printStackTrace();
        }
    }

    /**
     * Returns the path of a received message.
     *
     * @param json message received.
     * @return the string path name.
     */
    private String getPath(JsonObject json) {
        JsonObject value = json.getAsJsonObject("value");
        String path = "";
        if (json.has("id")) {
            path += json.get("id").getAsString();
        } else if (json.has("clusterId")) {
            path += json.get("clusterId").getAsString();
        } else if (!json.get("model").getAsString().equals(ModelType.CLUSTER.toString())) {
            path += value.get("clusterId").getAsString();
            path += "/" + value.get("id").getAsString();
        } else {
            path += value.get("id").getAsString();
        }
        return path;
    }

    /**
     * Returns the number of message received by the web socket.
     *
     * @return number of messages received by the web socket.
     */
    @Override
    public int getReceivedMessageCount() {
        return this.receivedMessageCount;
    }
}