StressStream.java

//	---------------------------------------------------------------------------
//	jWebSocket - stress Stream
//	Copyright (c) 2010 Alexander Schulze, Innotrade GmbH
//	---------------------------------------------------------------------------
//	This program is free software; you can redistribute it and/or modify it
//	under the terms of the GNU Lesser General Public License as published by the
//	Free Software Foundation; either version 3 of the License, or (at your
//	option) any later version.
//	This program is distributed in the hope that it will be useful, but WITHOUT
//	ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
//	FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for
//	more details.
//	You should have received a copy of the GNU Lesser General Public License along
//	with this program; if not, see <http://www.gnu.org/licenses/lgpl.html>.
//	---------------------------------------------------------------------------
package org.jwebsocket.plugins.streaming;

import java.util.Calendar;
import java.util.Date;

import org.apache.log4j.Logger;
import org.jwebsocket.logging.Logging;
import org.jwebsocket.server.TokenServer;
import org.jwebsocket.token.Token;

/**
* implements the stressStream, primarily for demonstration purposes but it can
* also be used for client/server stress synchronization. It implements an
* internal thread which broadcasts the current system stress of the server to
* the registered clients once per second.
* @author aschulze
*/
public class StressStream extends TokenStream {

	private static Logger log = Logging.getLogger(StressStream.class);
	private Boolean isRunning = false;
	private StressProcess stressProcess = null;
	private Thread stressThread = null;

	/**
	 *
	 *
	 * @param aStreamID
	 * @param aServer
	 */
	public StressStream(String aStreamID, TokenServer aServer) {
		super(aStreamID, aServer);
		startStream(-1);
	}

	/**
	 *
	 */
	@Override
	public void startStream(long aTimeout) {
		super.startStream(aTimeout);

		if (log.isDebugEnabled()) {
			log.debug("Starting stress stream...");
		}
		stressProcess = new StressProcess();
		stressThread = new Thread(stressProcess);
		stressThread.start();
	}

	/**
	 *
	 */
	@Override
	public void stopStream(long aTimeout) {
		if (log.isDebugEnabled()) {
			log.debug("Stopping stress stream...");
		}
		long lStarted = new Date().getTime();
		isRunning = false;
		try {
			stressThread.join(aTimeout);
		} catch (Exception ex) {
			log.error(ex.getClass().getSimpleName() + ": " + ex.getMessage());
		}
		if (log.isDebugEnabled()) {
			long lDuration = new Date().getTime() - lStarted;
			if (stressThread.isAlive()) {
				log.warn("stress stream did not stopped after " + lDuration + "ms.");
			} else {
				log.debug("stress stream stopped after " + lDuration + "ms.");
			}
		}

		super.stopStream(aTimeout);
	}

	private class StressProcess implements Runnable {

		@Override
		public void run() {
			if (log.isDebugEnabled()) {
				log.debug("Running stress stream...");
			}
			isRunning = true;
			while (isRunning) {
				try {
					Thread.sleep(50);

					Token lToken = new Token("event");
					lToken.put("name", "stream");
					lToken.put("msg", new Date().getTime());
					lToken.put("seconds", Calendar.getInstance().get(Calendar.SECOND));					

					put(lToken);
				} catch (InterruptedException ex) {
					log.error("(run) " + ex.getClass().getSimpleName() + ": " + ex.getMessage());
				}
			}
			if (log.isDebugEnabled()) {
				log.debug("stress stream stopped.");
			}
		}
	}
}

///**
// * implements the stressStream, primarily for demonstration purposes but it can
// * also be used for client/server stress synchronization. It implements an
// * internal thread which broadcasts the current system stress of the server to
// * the registered clients once per second.
// * @author aschulze
// */
//public class StressStream extends TokenStream {
//
//	private static Logger log = Logging.getLogger(StressStream.class);
//	private Boolean isRunning = false;
//	private StressProcess stressProcess = null;
//	private Thread stressThread = null;
//
//	/**
//	 *
//	 *
//	 * @param aStreamID
//	 * @param aServer
//	 */
//	public StressStream(String aStreamID, TokenServer aServer) {
//		super(aStreamID, aServer);
//		startStream(-1);
//	}
//
//	/**
//	 *
//	 */
//	@Override
//	public void startStream(long aTimeout) {
//		super.startStream(aTimeout);
//
//		if (log.isDebugEnabled()) {
//			log.debug("Starting stress stream...");
//		}
//		stressProcess = new StressProcess();
//		stressThread = new Thread(stressProcess);
//		stressThread.start();
//	}
//
//	/**
//	 *
//	 */
//	@Override
//	public void stopStream(long aTimeout) {
//		if (log.isDebugEnabled()) {
//			log.debug("Stopping stress stream...");
//		}
//		long lStarted = new Date().getTime();
//		isRunning = false;
//		try {
//			stressThread.join(aTimeout);
//		} catch (Exception ex) {
//			log.error(ex.getClass().getSimpleName() + ": " + ex.getMessage());
//		}
//		if (log.isDebugEnabled()) {
//			long lDuration = new Date().getTime() - lStarted;
//			if (stressThread.isAlive()) {
//				log.warn("stress stream did not stopped after " + lDuration + "ms.");
//			} else {
//				log.debug("stress stream stopped after " + lDuration + "ms.");
//			}
//		}
//
//		super.stopStream(aTimeout);
//	}
//
//	private class StressProcess implements Runnable {
//
//		@Override
//		public void run() {
//			if (log.isDebugEnabled()) {
//				log.debug("Running stress stream...");
//			}
//			isRunning = true;
//			while (isRunning) {
//				try {
//					Thread.sleep(50);
//
//					Token lToken = new Token("event");
//					lToken.put("name", "stream");
//					lToken.put("msg", new Date().getTime());
//					lToken.put("seconds", Calendar.getInstance().get(Calendar.SECOND));					
//
//					put(lToken);
//				} catch (InterruptedException ex) {
//					log.error("(run) " + ex.getClass().getSimpleName() + ": " + ex.getMessage());
//				}
//			}
//			if (log.isDebugEnabled()) {
//				log.debug("stress stream stopped.");
//			}
//		}
//	}
//}