SearchSocket.java

package com.machine.learning.socket;

import java.io.IOException;
import java.util.List;

import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;

import org.apache.log4j.Logger;

import com.google.gson.Gson;
import com.learningmodule.association.conceptdrug.PredictionResult;
import com.machine.learning.LearningModulesPool;
import com.machine.learning.request.Request;
import com.machine.learning.request.SearchQueryRequest;

/*
 * Class implementing the WebSocket Interface
 */

@ServerEndpoint(value = "/ml/search")
public class SearchSocket {

	private static Gson gson = new Gson();
	private static Logger log = Logger.getLogger(SearchSocket.class);
	private Session session;

	// public static LearningModuleInterface algo = new
	// ConceptDrugAssociationModule(
	// new ConceptDrugLearningModule());

	@OnOpen
	public void start(Session session) {
		this.session = session;
	}

	@OnClose
	public void end() {

	}

	/*
	 * Method that is invoked when there is msg from client input from client
	 * for querying should be of form "query:<search string>" input from client
	 * to command for server to learn should be of form "learn:"
	 */
	@OnMessage
	public void onMsg(String query) {
		System.out.println(query);
		log.debug(query);
		Request req = gson.fromJson(query, Request.class);

		if (req.getSearchRequest() != null) {

			// send the results to client for search string
			sendResults(req.getSearchRequest());

		} else if (req.getLearningRequest() != null
				&& req.getLearningRequest().getLearningKey() == "raxalearn") {
			// if client commands to learn, start a new thread to run the
			// learning algo.

			new Thread(new Runnable() {

				@Override
				public void run() {
					LearningModulesPool.learn();

				}
			}).run();
		}
	}

	public void sendResults(SearchQueryRequest query) {

		try {
			// get the predicitions, convert the results into Json String and
			// send to the client
			session.getBasicRemote().sendText(
					getJson(LearningModulesPool.predict(query.getQuery(), query.getFeatures())));
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}

	}

	@OnError
	public void onError(Throwable t) throws Throwable {
		t.printStackTrace();
	}

	// method to convert the results into Json string;
	private String getJson(List<PredictionResult> list) {
		String result = "{\"results\":";
		result = result + gson.toJson(list);

		result = result + "}";
		System.out.println(result);
		log.debug(result);
		return result;
	}

	public static void main(String[] args) {
		String msg = "{searchRequest: { query: 'asthma', features:[{name: 'age', value: '24'}] }}";
		Request req = gson.fromJson(msg, Request.class);
		System.out.println(req);
	}
}