services/org.eclipse.sensinact.studio.http.services/src/org/eclipse/sensinact/studio/http/services/client/websockets/SensinactSocket.java

/**
 * Copyright (c) 2018 CEA.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 *  Contributors:
 *     CEA - initial API and implementation and/or initial documentation
 */
package org.eclipse.sensinact.studio.http.services.client.websockets;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import org.apache.log4j.Logger;
import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketClose;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage;
import org.eclipse.jetty.websocket.api.annotations.WebSocket;
import org.eclipse.sensinact.studio.http.services.client.subscribe.agent.Agent;
import org.eclipse.sensinact.studio.preferences.ConfigurationManager;
import org.eclipse.sensinact.studio.preferences.GatewayHttpConfig;
import org.json.JSONException;
import org.json.JSONObject;

/**
 * @author Jander Nascimento
 */
@WebSocket(maxTextMessageSize = 64 * 1024)
public class SensinactSocket {

	private static final Logger logger = Logger.getLogger(SensinactSocket.class);
	
    private final CountDownLatch closeLatch;
    private Session session;
    private String gatewayID;
	
    public SensinactSocket(String gatewayID) {
        this.closeLatch = new CountDownLatch(1);
        this.gatewayID=gatewayID;
    }

    public boolean awaitClose(int duration, TimeUnit unit) throws InterruptedException {
        return this.closeLatch.await(duration, unit);
    }

    @OnWebSocketClose
    public void onClose(int statusCode, String reason) {
        session=null;
        System.out.println("WebSocket connection closed from the gateway " + gatewayID);
        this.closeLatch.countDown();
        try {
        	GatewayHttpConfig gwConfig = ConfigurationManager.getGateway(gatewayID);
            Agent.getInstance().unsubscribe(gwConfig);
    		SensinactWebSocketConnectionManager.getInstance().disconnect(gwConfig.getName());
    		//ModelEditor.getInstance().clearGatewayContent(gwConfig.getName());	
        }catch(Exception e){
        	logger.debug("Websocket clients was disconnected from the server " + gatewayID);
        }
    }

    @OnWebSocketConnect
    public void onConnect(Session session) {
    	logger.info("WebSocket connected to the gateway " + gatewayID);
        this.session = session;
    }

    @OnWebSocketMessage
    public void onMessage(String msg) {
    	logger.info("Websocket message received from " + gatewayID);
    	String uri;
		try {
			JSONObject jsonMessage=new JSONObject(msg);
			uri = jsonMessage.getString("uri");		
			//SubscriptionManager sm=SubscriptionManager.getInstance();
			
			// TODO update this with new subscription model
			
			/*
			// List<String> subscriptionids=sm.getSubscriptionIDFromResourceURI(uri);
			//Find subscription with a given destination
			if(subscriptionids.size()!=0)
			for(String subscriptionID:subscriptionids){
				logger.debug("Sending notification to subscriptionID:"+subscriptionID);
				sm.callbackRecieved(msg, subscriptionID);
			}
			else {
				//Case so far the notification was not dispatched, dispatch system message if that correspond to one
				String subscriptionID=sm.getSubscriptionId(new ResourceDescriptor(gatewayID, null, null, null));
				sm.callbackRecieved(msg, subscriptionID);
			}
			*/
		} catch (JSONException e) {
			logger.debug("Error receiving message:" + e.getMessage());
		}    	
    }
    
    public Session getSession() {
    	return session;
    }
}