WebConsoleImpl.java example

Explorer
kraken-master
/*
 * Copyright 2011 Future Systems
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.krakenapps.webconsole.impl;

import java.net.InetSocketAddress;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import org.apache.felix.ipojo.annotations.Component;
import org.apache.felix.ipojo.annotations.Invalidate;
import org.apache.felix.ipojo.annotations.Provides;
import org.apache.felix.ipojo.annotations.Requires;
import org.apache.felix.ipojo.annotations.Validate;
import org.krakenapps.httpd.HttpContext;
import org.krakenapps.httpd.HttpContextRegistry;
import org.krakenapps.httpd.HttpService;
import org.krakenapps.httpd.WebSocket;
import org.krakenapps.httpd.WebSocketFrame;
import org.krakenapps.httpd.WebSocketListener;
import org.krakenapps.msgbus.Message;
import org.krakenapps.msgbus.MessageBus;
import org.krakenapps.msgbus.Session;
import org.krakenapps.msgbus.handler.CallbackType;
import org.krakenapps.msgbus.handler.MsgbusMethod;
import org.krakenapps.msgbus.handler.MsgbusPlugin;
import org.krakenapps.webconsole.WebConsole;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@MsgbusPlugin
@Component(name = "webconsole")
@Provides(specifications = { WebConsole.class })
public class WebConsoleImpl implements WebConsole, WebSocketListener {
	private final Logger logger = LoggerFactory.getLogger(WebConsoleImpl.class.getName());

	@Requires
	private HttpService httpd;

	@Requires
	private MessageBus msgbus;

	private ConcurrentMap<InetSocketAddress, WebSocketSession> sessions;

	@Validate
	public void start() {
		sessions = new ConcurrentHashMap<InetSocketAddress, WebSocketSession>();
		HttpContextRegistry contextRegistry = httpd.getContextRegistry();
		HttpContext ctx = contextRegistry.ensureContext("webconsole");
		ctx.getWebSocketManager().addListener(this);
	}

	@Invalidate
	public void stop() {
		if (httpd != null) {
			HttpContextRegistry contextRegistry = httpd.getContextRegistry();
			HttpContext ctx = contextRegistry.ensureContext("webconsole");
			ctx.getWebSocketManager().removeListener(this);
		}
	}

	@Override
	public void onConnected(WebSocket socket) {
		logger.trace("kraken webconsole: websocket connected [{}]", socket);
		WebSocketSession webSocketSession = new WebSocketSession(socket);
		sessions.put(socket.getRemoteAddress(), webSocketSession);
		msgbus.openSession(webSocketSession);
	}

	@Override
	public void onDisconnected(WebSocket socket) {
		WebSocketSession session = sessions.get(socket.getRemoteAddress());
		if (session == null)
			return;

		logger.trace("kraken webconsole: websocket disconnected [{}]", session);
		msgbus.closeSession(session);
	}

	@Override
	public void onMessage(WebSocket socket, WebSocketFrame frame) {
		WebSocketSession session = sessions.get(socket.getRemoteAddress());
		if (session == null) {
			logger.error("kraken webconsole: session not found for [{}]", socket);
			return;
		}

		logger.trace("kraken webconsole: websocket frame [{}]", frame);
		Message msg = KrakenMessageDecoder.decode(session, frame.getTextData());
		if (msg != null)
			msgbus.dispatch(session, msg);
	}

	@MsgbusMethod(type = CallbackType.SessionClosed)
	public void onSessionClose(Session session) {
		logger.debug("kraken webconsole: session [{}] closed", session);

		for (InetSocketAddress key : sessions.keySet()) {
			WebSocketSession wss = sessions.get(key);
			if (wss.getGuid().equals(session.getGuid())) {
				logger.info("kraken webconsole: kill websocket session [{}]", wss);
				wss.close();
				sessions.remove(key);
				break;
			}
		}
	}
}