Nearly have StudentClient sending via JMS.

mark.george / democall3

Browse code Nearly have StudentClient sending via JMS. master
1 parent c4c6a9e commit 55923ae5101b100227fb3b308e649661e786c41d GEOMA48p authored on 20 Jul 2015

Patch

Showing 5 changed files

Ignore Space Show notes View src/clients/StudentClient.java
package clients; import constants.Constants; import discovery.computername.ComputerNameResolver; import discovery.computername.InvalidComputerNameException; import discovery.computername.OtagoComputerNameResolver; import discovery.jms.ServiceLocator; import discovery.server.multicast.ServiceNotFoundException; import gui.Lab; import java.awt.; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.util.Collection; import java.util.logging.Level; import java.util.logging.Logger; import javax.swing.JOptionPane; import network.MessageGenerator; import network.MessageType; import network.jms.RequestSender; / * The student's client for requesting help. * * It adds a system tray icon that has a pop-up menu that the student can use to * request help or cancel their request. Requests are automatically canceled * when the student clicks the "exit" menu item, or the client is terminated by * the OS (via a shutdown hook) meaning there should never be requests in the * queue for students who have gone home. * * This aims to be both simple and robust, therefor the StudentClient stores no * state relating to requests, uses no persistent connections to the server, and * receives no information from the server. * * The client uses a multi-cast broadcast to discover the IP of the server, * which it then uses to send requests to the server. * * The wire protocol is very simple. A string in the form "request 23" is a * request for from the machine with ID 23, and "cancel 23" will cancel that * request. Sending multiple "request" commands or multiple "cancel" commands to * the server should cause no duplication or problems on the server and are * effectively ignored. * * @author Mark / public class StudentClient { private static final Logger LOG = Logger.getLogger(StudentClient.class.getName()); private final PopupMenu trayPopopMenu = new PopupMenu(); private final Image trayIconImage = Toolkit.getDefaultToolkit().getImage(getClass().getResource("/icon.png")); private final TrayIcon systemTrayIcon = new TrayIcon(trayIconImage, "Demo Call " + Constants.VERSION, trayPopopMenu); private final MessageGenerator generator = new MessageGenerator(); private String machineId; private RequestSender requestSender; private boolean foundServer = false; private final Thread shutDownHook = new Thread() { @Override public void run() { cleanUp(); } }; public StudentClient(final ComputerNameResolver nameResolver) { try { createTrayMenu(); // not sure this is necessary but Windows is currently not exiting // when the shutdown hook kicks in... this.shutDownHook.setDaemon(true); Runtime.getRuntime().addShutdownHook(this.shutDownHook); try { nameResolver.resolve(); } catch (InvalidComputerNameException ex) { this.systemTrayIcon.displayMessage("Whoops", ex.getMessage(), TrayIcon.MessageType.ERROR); // give user a chance to read message Thread.sleep(5000); // shutdown exit(); } final String currentLab = nameResolver.getLabName(); this.machineId = nameResolver.getMachineId(); // find server IP (uses network multicast) final Collection<Lab> activeLabs = new ServiceLocator().getActiveLabs(); final Lab lab = (Lab) JOptionPane.showInputDialog(null, "Which lab?", "Which lab?", JOptionPane.QUESTION_MESSAGE, null, activeLabs.toArray(), currentLab); String labName=null; if (lab == null) { System.exit(0); } else { labName = lab.getLabName(); } this.foundServer = true; // this.systemTrayIcon.displayMessage("Yay", "Server found: " +serverIp, TrayIcon.MessageType.INFO); this.requestSender = new RequestSender(lab, machineId); } catch (ServiceNotFoundException ex) { this.systemTrayIcon.displayMessage("Whoops", "Could not connect to server. Please try again and let the supervisor know if it continues to happen.", TrayIcon.MessageType.ERROR); } catch (Exception ex) { LOG.log(Level.SEVERE, null, ex); } } private void createTrayMenu() { try { this.systemTrayIcon.setImageAutoSize(true); final MenuItem requestHelp = new MenuItem("Request Help"); requestHelp.addActionListener(new ActionListener() { @Override public void actionPerformed(final ActionEvent e) { helpRequest(); } }); final MenuItem requestMarking = new MenuItem("Request Marking"); requestMarking.addActionListener(new ActionListener() { @Override public void actionPerformed(final ActionEvent e) { markingRequest(); } }); final MenuItem cancelHelp = new MenuItem("Cancel Help"); cancelHelp.addActionListener(new ActionListener() { @Override public void actionPerformed(final ActionEvent e) { cancelHelp(); } }); final MenuItem cancelMarking = new MenuItem("Cancel Marking"); cancelMarking.addActionListener(new ActionListener() { @Override public void actionPerformed(final ActionEvent e) { cancelMarking(); } }); final MenuItem cancelAll = new MenuItem("Cancel All Requests"); cancelAll.addActionListener(new ActionListener() { @Override public void actionPerformed(final ActionEvent e) { cancelAll(); } }); final MenuItem exit = new MenuItem("Exit"); exit.addActionListener(new ActionListener() { @Override public void actionPerformed(final ActionEvent e) { exit(); } }); this.trayPopopMenu.add(requestHelp); this.trayPopopMenu.add(cancelHelp); this.trayPopopMenu.addSeparator(); this.trayPopopMenu.add(requestMarking); this.trayPopopMenu.add(cancelMarking); this.trayPopopMenu.addSeparator(); this.trayPopopMenu.add(cancelAll); this.trayPopopMenu.addSeparator(); this.trayPopopMenu.add(exit); SystemTray.getSystemTray().add(this.systemTrayIcon); } catch (Exception ex) { LOG.log(Level.SEVERE, null, ex); } } private void helpRequest() { try { requestSender.sendRequest(MessageType.HELP); systemTrayIcon.displayMessage("Help is on the way", "Your request is now in the queue.", TrayIcon.MessageType.INFO); } catch (Exception ex) { systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR); LOG.log(Level.SEVERE, "Error making request", ex); } } private void markingRequest() { try { requestSender.sendRequest(MessageType.MARKING); systemTrayIcon.displayMessage("Help is on the way", "Your request is now in the queue.", TrayIcon.MessageType.INFO); } catch (Exception ex) { systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR); LOG.log(Level.SEVERE, "Error making request", ex); } } private void cancelHelp() { // try { // requestSender.sendRequest(generator.cancelHelp(machineId)); // systemTrayIcon.displayMessage("Request removed", "Your request has been removed from the queue.", TrayIcon.MessageType.INFO); // } catch (Exception ex) { // systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR); // LOG.log(Level.SEVERE, "Error making request", ex); // } } private void cancelMarking() { // try { // requestSender.sendRequest(generator.cancelMarking(machineId)); // systemTrayIcon.displayMessage("Request removed", "Your request has been removed from the queue.", TrayIcon.MessageType.INFO); // } catch (Exception ex) { // systemTrayIcon.displayMessage("Whoops", "Failed to cancel request", TrayIcon.MessageType.ERROR); // LOG.log(Level.SEVERE, "Error cancelling request", ex); // } } private void cancelAll() { // try { // requestSender.sendRequest(generator.cancelHelp(machineId)); // requestSender.sendRequest(generator.cancelMarking(machineId)); // systemTrayIcon.displayMessage("All requests removeds", "All of your request have been removed from the queues.", TrayIcon.MessageType.INFO); // } catch (Exception ex) { // systemTrayIcon.displayMessage("Whoops", "Failed to cancel request", TrayIcon.MessageType.ERROR); // LOG.log(Level.SEVERE, "Error cancelling request", ex); // } } private void exit() { this.cleanUp(); // force an exit since they are several threads lurking around that could keep the process running System.exit(0); } private void cleanUp() { // cancel any requests the student has made if (this.foundServer) { cancelAll(); } // remove the tray icon from the system tray SystemTray.getSystemTray().remove(systemTrayIcon); } @SuppressWarnings("ResultOfObjectAllocationIgnored") public static void main(final String[] args) { // System.out.println("Student Client"); // final String name = "SB318-15"; // final String name = "SB317-10"; // final String name = "SBEASTCAL1-30"; // final String name = "SB316-15"; final String name = args.length > 0 ? args[0] : null; final ComputerNameResolver nameResolver = new OtagoComputerNameResolver(name, "CLIENTNAME"); nameResolver.resolve(); System.out.println(nameResolver); new StudentClient(nameResolver); } } package clients; import constants.Constants; import discovery.computername.ComputerNameResolver; import discovery.computername.InvalidComputerNameException; import discovery.computername.OtagoComputerNameResolver; import discovery.jms.ServiceLocator; import discovery.server.multicast.ServiceNotFoundException; import gui.Lab; import java.awt.; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.util.Collection; import java.util.logging.Level; import java.util.logging.Logger; import javax.swing.JOptionPane; import network.MessageGenerator; import network.jms.RequestSender; /* * The student's client for requesting help. * * It adds a system tray icon that has a pop-up menu that the student can use to * request help or cancel their request. Requests are automatically canceled * when the student clicks the "exit" menu item, or the client is terminated by * the OS (via a shutdown hook) meaning there should never be requests in the * queue for students who have gone home. * * This aims to be both simple and robust, therefor the StudentClient stores no * state relating to requests, uses no persistent connections to the server, and * receives no information from the server. * * The client uses a multi-cast broadcast to discover the IP of the server, * which it then uses to send requests to the server. * * The wire protocol is very simple. A string in the form "request 23" is a * request for from the machine with ID 23, and "cancel 23" will cancel that * request. Sending multiple "request" commands or multiple "cancel" commands to * the server should cause no duplication or problems on the server and are * effectively ignored. * * @author Mark */ public class StudentClient { private static final Logger LOG = Logger.getLogger(StudentClient.class.getName()); private final PopupMenu trayPopopMenu = new PopupMenu(); private final Image trayIconImage = Toolkit.getDefaultToolkit().getImage(getClass().getResource("/icon.png")); private final TrayIcon systemTrayIcon = new TrayIcon(trayIconImage, "Demo Call " + Constants.VERSION, trayPopopMenu); private final MessageGenerator generator = new MessageGenerator(); private String machineId; private RequestSender requestSender; private boolean foundServer = false; private final Thread shutDownHook = new Thread() { @Override public void run() { cleanUp(); } }; public StudentClient(final ComputerNameResolver nameResolver) { try { createTrayMenu(); // not sure this is necessary but Windows is currently not exiting // when the shutdown hook kicks in... this.shutDownHook.setDaemon(true); Runtime.getRuntime().addShutdownHook(this.shutDownHook); try { nameResolver.resolve(); } catch (InvalidComputerNameException ex) { this.systemTrayIcon.displayMessage("Whoops", ex.getMessage(), TrayIcon.MessageType.ERROR); // give user a chance to read message Thread.sleep(5000); // shutdown exit(); } final String currentLab = nameResolver.getLabName(); this.machineId = nameResolver.getMachineId(); // find server IP (uses network multicast) final Collection<String> activeServers = new ServiceLocator().getActiveServers(); final Lab lab = (Lab) JOptionPane.showInputDialog(null, "Which lab?", "Which lab?", JOptionPane.QUESTION_MESSAGE, null, activeServers.toArray(), currentLab); String labName=""; if (lab == null) { System.exit(0); } else { labName = lab.getLabName(); } this.foundServer = true; // this.systemTrayIcon.displayMessage("Yay", "Server found: " +serverIp, TrayIcon.MessageType.INFO); this.requestSender = new RequestSender(labName, machineId); } catch (ServiceNotFoundException ex) { this.systemTrayIcon.displayMessage("Whoops", "Could not connect to server. Please try again and let the supervisor know if it continues to happen.", TrayIcon.MessageType.ERROR); } catch (Exception ex) { LOG.log(Level.SEVERE, null, ex); } } private void createTrayMenu() { try { this.systemTrayIcon.setImageAutoSize(true); final MenuItem requestHelp = new MenuItem("Request Help"); requestHelp.addActionListener(new ActionListener() { public void actionPerformed(final ActionEvent e) { helpRequest(); } }); final MenuItem requestMarking = new MenuItem("Request Marking"); requestMarking.addActionListener(new ActionListener() { public void actionPerformed(final ActionEvent e) { markingRequest(); } }); final MenuItem cancelHelp = new MenuItem("Cancel Help"); cancelHelp.addActionListener(new ActionListener() { public void actionPerformed(final ActionEvent e) { cancelHelp(); } }); final MenuItem cancelMarking = new MenuItem("Cancel Marking"); cancelMarking.addActionListener(new ActionListener() { public void actionPerformed(final ActionEvent e) { cancelMarking(); } }); final MenuItem cancelAll = new MenuItem("Cancel All Requests"); cancelAll.addActionListener(new ActionListener() { public void actionPerformed(final ActionEvent e) { cancelAll(); } }); final MenuItem exit = new MenuItem("Exit"); exit.addActionListener(new ActionListener() { public void actionPerformed(final ActionEvent e) { exit(); } }); this.trayPopopMenu.add(requestHelp); this.trayPopopMenu.add(cancelHelp); this.trayPopopMenu.addSeparator(); this.trayPopopMenu.add(requestMarking); this.trayPopopMenu.add(cancelMarking); this.trayPopopMenu.addSeparator(); this.trayPopopMenu.add(cancelAll); this.trayPopopMenu.addSeparator(); this.trayPopopMenu.add(exit); SystemTray.getSystemTray().add(this.systemTrayIcon); } catch (Exception ex) { LOG.log(Level.SEVERE, null, ex); } } private void helpRequest() { try { requestSender.sendRequest(generator.requestHelp(machineId)); systemTrayIcon.displayMessage("Help is on the way", "Your request is now in the queue.", TrayIcon.MessageType.INFO); } catch (Exception ex) { systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR); LOG.log(Level.SEVERE, "Error making request", ex); } } private void markingRequest() { try { requestSender.sendRequest(generator.requestMarking(machineId)); systemTrayIcon.displayMessage("Help is on the way", "Your request is now in the queue.", TrayIcon.MessageType.INFO); } catch (Exception ex) { systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR); LOG.log(Level.SEVERE, "Error making request", ex); } } private void cancelHelp() { try { requestSender.sendRequest(generator.cancelHelp(machineId)); systemTrayIcon.displayMessage("Request removed", "Your request has been removed from the queue.", TrayIcon.MessageType.INFO); } catch (Exception ex) { systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR); LOG.log(Level.SEVERE, "Error making request", ex); } } private void cancelMarking() { try { requestSender.sendRequest(generator.cancelMarking(machineId)); systemTrayIcon.displayMessage("Request removed", "Your request has been removed from the queue.", TrayIcon.MessageType.INFO); } catch (Exception ex) { systemTrayIcon.displayMessage("Whoops", "Failed to cancel request", TrayIcon.MessageType.ERROR); LOG.log(Level.SEVERE, "Error cancelling request", ex); } } private void cancelAll() { try { requestSender.sendRequest(generator.cancelHelp(machineId)); requestSender.sendRequest(generator.cancelMarking(machineId)); systemTrayIcon.displayMessage("All requests removeds", "All of your request have been removed from the queues.", TrayIcon.MessageType.INFO); } catch (Exception ex) { systemTrayIcon.displayMessage("Whoops", "Failed to cancel request", TrayIcon.MessageType.ERROR); LOG.log(Level.SEVERE, "Error cancelling request", ex); } } private void exit() { this.cleanUp(); // force an exit since they are several threads lurking around that could keep the process running System.exit(0); } private void cleanUp() { // cancel any requests the student has made if (this.foundServer) { cancelAll(); } // remove the tray icon from the system tray SystemTray.getSystemTray().remove(systemTrayIcon); } @SuppressWarnings("ResultOfObjectAllocationIgnored") public static void main(final String[] args) { // System.out.println("Student Client"); // final String name = "SB318-15"; // final String name = "SB317-10"; // final String name = "SBEASTCAL1-30"; // final String name = "SB316-15"; final String name = args.length > 0 ? args[0] : null; final ComputerNameResolver nameResolver = new OtagoComputerNameResolver(name, "CLIENTNAME"); nameResolver.resolve(); System.out.println(nameResolver); new StudentClient(nameResolver); } }

Ignore Space Show notes View

src/clients/StudentClient.java

package clients;

import constants.Constants;
import discovery.computername.ComputerNameResolver;
import discovery.computername.InvalidComputerNameException;
import discovery.computername.OtagoComputerNameResolver;
import discovery.jms.ServiceLocator;
import discovery.server.multicast.ServiceNotFoundException;
import gui.Lab;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Collection;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.swing.JOptionPane;
import network.MessageGenerator;
import network.MessageType;
import network.jms.RequestSender;

/*
 * The student's client for requesting help.
 *
 * It adds a system tray icon that has a pop-up menu that the student can use to
 * request help or cancel their request. Requests are automatically canceled
 * when the student clicks the "exit" menu item, or the client is terminated by
 * the OS (via a shutdown hook) meaning there should never be requests in the
 * queue for students who have gone home.
 *
 * This aims to be both simple and robust, therefor the StudentClient stores no
 * state relating to requests, uses no persistent connections to the server, and
 * receives no information from the server.
 *
 * The client uses a multi-cast broadcast to discover the IP of the server,
 * which it then uses to send requests to the server.
 *
 * The wire protocol is very simple. A string in the form "request 23" is a
 * request for from the machine with ID 23, and "cancel 23" will cancel that
 * request. Sending multiple "request" commands or multiple "cancel" commands to
 * the server should cause no duplication or problems on the server and are
 * effectively ignored.
 *
 * @author Mark
 */
public class StudentClient {

private static final Logger LOG = Logger.getLogger(StudentClient.class.getName());
	private final PopupMenu trayPopopMenu = new PopupMenu();
	private final Image trayIconImage = Toolkit.getDefaultToolkit().getImage(getClass().getResource("/icon.png"));
	private final TrayIcon systemTrayIcon = new TrayIcon(trayIconImage, "Demo Call " + Constants.VERSION, trayPopopMenu);
	private final MessageGenerator generator = new MessageGenerator();
	private String machineId;
	private RequestSender requestSender;
	private boolean foundServer = false;
	private final Thread shutDownHook = new Thread() {

@Override
		public void run() {
			cleanUp();
		}
	};

public StudentClient(final ComputerNameResolver nameResolver) {
		try {
			createTrayMenu();

// not sure this is necessary but Windows is currently not exiting
			// when the shutdown hook kicks in...
			this.shutDownHook.setDaemon(true);

Runtime.getRuntime().addShutdownHook(this.shutDownHook);

try {
				nameResolver.resolve();
			} catch (InvalidComputerNameException ex) {
				this.systemTrayIcon.displayMessage("Whoops", ex.getMessage(), TrayIcon.MessageType.ERROR);

// give user a chance to read message
				Thread.sleep(5000);

// shutdown
				exit();
			}

final String currentLab = nameResolver.getLabName();
			this.machineId = nameResolver.getMachineId();

// find server IP (uses network multicast)
			final Collection<Lab> activeLabs = new ServiceLocator().getActiveLabs();

final Lab lab = (Lab) JOptionPane.showInputDialog(null, "Which lab?", "Which lab?", JOptionPane.QUESTION_MESSAGE, null, activeLabs.toArray(), currentLab);

String labName=null;
			
			if (lab == null) {
				System.exit(0);
			} else {
				labName = lab.getLabName();
			}			
			
			this.foundServer = true;

//         this.systemTrayIcon.displayMessage("Yay", "Server found: "  +serverIp, TrayIcon.MessageType.INFO);

this.requestSender = new RequestSender(lab, machineId);

} catch (ServiceNotFoundException ex) {
			this.systemTrayIcon.displayMessage("Whoops", "Could not connect to server.  Please try again and let the supervisor know if it continues to happen.", TrayIcon.MessageType.ERROR);
		} catch (Exception ex) {
			LOG.log(Level.SEVERE, null, ex);
		}

}

private void createTrayMenu() {
		try {

this.systemTrayIcon.setImageAutoSize(true);

final MenuItem requestHelp = new MenuItem("Request Help");
			requestHelp.addActionListener(new ActionListener() {

@Override
				public void actionPerformed(final ActionEvent e) {
					helpRequest();
				}
			});

final MenuItem requestMarking = new MenuItem("Request Marking");
			requestMarking.addActionListener(new ActionListener() {

@Override
				public void actionPerformed(final ActionEvent e) {
					markingRequest();
				}
			});

final MenuItem cancelHelp = new MenuItem("Cancel Help");
			cancelHelp.addActionListener(new ActionListener() {

@Override
				public void actionPerformed(final ActionEvent e) {
					cancelHelp();
				}
			});

final MenuItem cancelMarking = new MenuItem("Cancel Marking");
			cancelMarking.addActionListener(new ActionListener() {

@Override
				public void actionPerformed(final ActionEvent e) {
					cancelMarking();
				}
			});

final MenuItem cancelAll = new MenuItem("Cancel All Requests");
			cancelAll.addActionListener(new ActionListener() {

@Override
				public void actionPerformed(final ActionEvent e) {
					cancelAll();
				}
			});

final MenuItem exit = new MenuItem("Exit");
			exit.addActionListener(new ActionListener() {

@Override
				public void actionPerformed(final ActionEvent e) {
					exit();
				}
			});

this.trayPopopMenu.add(requestHelp);
			this.trayPopopMenu.add(cancelHelp);
			this.trayPopopMenu.addSeparator();
			this.trayPopopMenu.add(requestMarking);
			this.trayPopopMenu.add(cancelMarking);
			this.trayPopopMenu.addSeparator();
			this.trayPopopMenu.add(cancelAll);
			this.trayPopopMenu.addSeparator();
			this.trayPopopMenu.add(exit);

SystemTray.getSystemTray().add(this.systemTrayIcon);

} catch (Exception ex) {
			LOG.log(Level.SEVERE, null, ex);
		}
	}

private void helpRequest() {
		try {
			requestSender.sendRequest(MessageType.HELP);
			systemTrayIcon.displayMessage("Help is on the way", "Your request is now in the queue.", TrayIcon.MessageType.INFO);
		} catch (Exception ex) {
			systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR);
			LOG.log(Level.SEVERE, "Error making request", ex);
		}
	}

private void markingRequest() {
		try {
			requestSender.sendRequest(MessageType.MARKING);
			systemTrayIcon.displayMessage("Help is on the way", "Your request is now in the queue.", TrayIcon.MessageType.INFO);
		} catch (Exception ex) {
			systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR);
			LOG.log(Level.SEVERE, "Error making request", ex);
		}
	}

private void cancelHelp() {
//		try {
//			requestSender.sendRequest(generator.cancelHelp(machineId));
//			systemTrayIcon.displayMessage("Request removed", "Your request has been removed from the queue.", TrayIcon.MessageType.INFO);
//		} catch (Exception ex) {
//			systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR);
//			LOG.log(Level.SEVERE, "Error making request", ex);
//		}
	}

private void cancelMarking() {
//		try {
//			requestSender.sendRequest(generator.cancelMarking(machineId));
//			systemTrayIcon.displayMessage("Request removed", "Your request has been removed from the queue.", TrayIcon.MessageType.INFO);
//		} catch (Exception ex) {
//			systemTrayIcon.displayMessage("Whoops", "Failed to cancel request", TrayIcon.MessageType.ERROR);
//			LOG.log(Level.SEVERE, "Error cancelling request", ex);
//		}
	}

private void cancelAll() {
//		try {
//			requestSender.sendRequest(generator.cancelHelp(machineId));
//			requestSender.sendRequest(generator.cancelMarking(machineId));
//			systemTrayIcon.displayMessage("All requests removeds", "All of your request have been removed from the queues.", TrayIcon.MessageType.INFO);
//		} catch (Exception ex) {
//			systemTrayIcon.displayMessage("Whoops", "Failed to cancel request", TrayIcon.MessageType.ERROR);
//			LOG.log(Level.SEVERE, "Error cancelling request", ex);
//		}
	}

private void exit() {
		this.cleanUp();

// force an exit since they are several threads lurking around that could keep the process running
		System.exit(0);
	}

private void cleanUp() {
		// cancel any requests the student has made
		if (this.foundServer) {
			cancelAll();
		}

// remove the tray icon from the system tray
		SystemTray.getSystemTray().remove(systemTrayIcon);
	}

@SuppressWarnings("ResultOfObjectAllocationIgnored")
	public static void main(final String[] args) {

//		System.out.println("Student Client");
		
//      final String name = "SB318-15";
//      final String name = "SB317-10";
//      final String name = "SBEASTCAL1-30";
//		final String name = "SB316-15";

final String name = args.length > 0 ? args[0] : null;
		final ComputerNameResolver nameResolver = new OtagoComputerNameResolver(name, "CLIENTNAME");

nameResolver.resolve();
		System.out.println(nameResolver);
		
		new StudentClient(nameResolver);
	}
}

package clients;

@Override
		public void run() {
			cleanUp();
		}
	};

public StudentClient(final ComputerNameResolver nameResolver) {
		try {
			createTrayMenu();

// not sure this is necessary but Windows is currently not exiting
			// when the shutdown hook kicks in...
			this.shutDownHook.setDaemon(true);

Runtime.getRuntime().addShutdownHook(this.shutDownHook);

try {
				nameResolver.resolve();
			} catch (InvalidComputerNameException ex) {
				this.systemTrayIcon.displayMessage("Whoops", ex.getMessage(), TrayIcon.MessageType.ERROR);

// give user a chance to read message
				Thread.sleep(5000);

// shutdown
				exit();
			}

final String currentLab = nameResolver.getLabName();
			this.machineId = nameResolver.getMachineId();

// find server IP (uses network multicast)
			final Collection<String> activeServers = new ServiceLocator().getActiveServers();

final Lab lab = (Lab) JOptionPane.showInputDialog(null, "Which lab?", "Which lab?", JOptionPane.QUESTION_MESSAGE, null, activeServers.toArray(), currentLab);

String labName="";
			
			if (lab == null) {
				System.exit(0);
			} else {
				labName = lab.getLabName();
			}			
			
			this.foundServer = true;

//         this.systemTrayIcon.displayMessage("Yay", "Server found: "  +serverIp, TrayIcon.MessageType.INFO);

this.requestSender = new RequestSender(labName, machineId);

}

private void createTrayMenu() {
		try {

this.systemTrayIcon.setImageAutoSize(true);

final MenuItem requestHelp = new MenuItem("Request Help");
			requestHelp.addActionListener(new ActionListener() {

public void actionPerformed(final ActionEvent e) {
					helpRequest();
				}
			});

final MenuItem requestMarking = new MenuItem("Request Marking");
			requestMarking.addActionListener(new ActionListener() {

public void actionPerformed(final ActionEvent e) {
					markingRequest();
				}
			});

final MenuItem cancelHelp = new MenuItem("Cancel Help");
			cancelHelp.addActionListener(new ActionListener() {

public void actionPerformed(final ActionEvent e) {
					cancelHelp();
				}
			});

final MenuItem cancelMarking = new MenuItem("Cancel Marking");
			cancelMarking.addActionListener(new ActionListener() {

public void actionPerformed(final ActionEvent e) {
					cancelMarking();
				}
			});

final MenuItem cancelAll = new MenuItem("Cancel All Requests");
			cancelAll.addActionListener(new ActionListener() {

public void actionPerformed(final ActionEvent e) {
					cancelAll();
				}
			});

final MenuItem exit = new MenuItem("Exit");
			exit.addActionListener(new ActionListener() {

public void actionPerformed(final ActionEvent e) {
					exit();
				}
			});

SystemTray.getSystemTray().add(this.systemTrayIcon);

} catch (Exception ex) {
			LOG.log(Level.SEVERE, null, ex);
		}
	}

private void helpRequest() {
		try {
			requestSender.sendRequest(generator.requestHelp(machineId));
			systemTrayIcon.displayMessage("Help is on the way", "Your request is now in the queue.", TrayIcon.MessageType.INFO);
		} catch (Exception ex) {
			systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR);
			LOG.log(Level.SEVERE, "Error making request", ex);
		}
	}

private void markingRequest() {
		try {
			requestSender.sendRequest(generator.requestMarking(machineId));
			systemTrayIcon.displayMessage("Help is on the way", "Your request is now in the queue.", TrayIcon.MessageType.INFO);
		} catch (Exception ex) {
			systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR);
			LOG.log(Level.SEVERE, "Error making request", ex);
		}
	}

private void cancelHelp() {
		try {
			requestSender.sendRequest(generator.cancelHelp(machineId));
			systemTrayIcon.displayMessage("Request removed", "Your request has been removed from the queue.", TrayIcon.MessageType.INFO);
		} catch (Exception ex) {
			systemTrayIcon.displayMessage("Whoops", "Failed to make request", TrayIcon.MessageType.ERROR);
			LOG.log(Level.SEVERE, "Error making request", ex);
		}
	}

private void cancelMarking() {
		try {
			requestSender.sendRequest(generator.cancelMarking(machineId));
			systemTrayIcon.displayMessage("Request removed", "Your request has been removed from the queue.", TrayIcon.MessageType.INFO);
		} catch (Exception ex) {
			systemTrayIcon.displayMessage("Whoops", "Failed to cancel request", TrayIcon.MessageType.ERROR);
			LOG.log(Level.SEVERE, "Error cancelling request", ex);
		}
	}

private void cancelAll() {
		try {
			requestSender.sendRequest(generator.cancelHelp(machineId));
			requestSender.sendRequest(generator.cancelMarking(machineId));
			systemTrayIcon.displayMessage("All requests removeds", "All of your request have been removed from the queues.", TrayIcon.MessageType.INFO);
		} catch (Exception ex) {
			systemTrayIcon.displayMessage("Whoops", "Failed to cancel request", TrayIcon.MessageType.ERROR);
			LOG.log(Level.SEVERE, "Error cancelling request", ex);
		}
	}

private void exit() {
		this.cleanUp();

// force an exit since they are several threads lurking around that could keep the process running
		System.exit(0);
	}

private void cleanUp() {
		// cancel any requests the student has made
		if (this.foundServer) {
			cancelAll();
		}

// remove the tray icon from the system tray
		SystemTray.getSystemTray().remove(systemTrayIcon);
	}

@SuppressWarnings("ResultOfObjectAllocationIgnored")
	public static void main(final String[] args) {

final String name = args.length > 0 ? args[0] : null;
		final ComputerNameResolver nameResolver = new OtagoComputerNameResolver(name, "CLIENTNAME");

nameResolver.resolve();
		System.out.println(nameResolver);
		
		new StudentClient(nameResolver);
	}
}

Ignore Space Show notes View src/discovery/jms/ServiceLocator.java
package discovery.jms; import gui.Lab; import gui.LabRegistry; import java.util.ArrayList; import java.util.Collection; import server.jms.JmsConnection; /** * * @author Mark / public class ServiceLocator { public Collection<Lab> getActiveLabs() { Collection<String> labNames = new JmsConnection().peekAll("active_servers"); Collection<Lab> activeLabs = new ArrayList<Lab>(); LabRegistry registry = new LabRegistry(); for (String labName : labNames) { Lab lab = registry.getLab(labName); activeLabs.add(lab); } return activeLabs; } } package discovery.jms; import java.util.Collection; import server.jms.JmsConnection; /* * * @author Mark */ public class ServiceLocator { public Collection<String> getActiveServers() { return new JmsConnection().peekAll("active_servers"); } }

Ignore Space Show notes View

src/discovery/jms/ServiceLocator.java

Ignore Space Show notes View src/network/MessageType.java 0 → 100644
package network; /** * * @author Mark */ public enum MessageType { HELP, MARKING }

Ignore Space Show notes View src/network/jms/RequestSender.java
package network.jms; import gui.Lab; import java.net.Socket; import java.util.logging.Logger; import network.MessageGenerator; import network.MessageType; import server.jms.JmsConnection; /** * * @author Mark / public class RequestSender { private static final Logger LOG = Logger.getLogger(RequestSender.class.getName()); private final MessageGenerator messageGenerator = new MessageGenerator(); private final Lab lab; private final String machineId; public RequestSender(final Lab lab, String machineId) { this.lab = lab; this.machineId = machineId; } public void sendRequest(final MessageType type) { JmsConnection connection = new JmsConnection(); if(type == MessageType.HELP ) { connection.sendMessageToTopic(lab.getLabName()+"_help", machineId); } else if (type == MessageType.MARKING) { connection.sendMessageToTopic(lab.getLabName()+"_marking", machineId); } } // @SuppressWarnings({"rawtypes", "unchecked"}) // public Collection<Integer> requestQueue() { // try { // final Socket socket = new Socket(destinationIp, Constants.PORT); // final PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream())); // writer.println(messageGenerator.requestQueue()); // writer.flush(); // final ObjectInputStream ois = new ObjectInputStream(socket.getInputStream()); // final Collection<Integer> response = (Collection<Integer>) ois.readObject(); // writer.close(); // ois.close(); // socket.close(); // return response; // } catch (Exception ex) { // LOG.log(Level.SEVERE, null, ex); // } // return Collections.EMPTY_SET; // } @SuppressWarnings({"rawtypes", "unchecked"}) public Socket registerTutorClient() { // try { // final Socket socket = new Socket(destinationIp, Constants.PORT); // final PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream())); // // writer.println(messageGenerator.registerTutorClient()); // writer.flush(); // // return socket; // } catch (Exception ex) { // LOG.log(Level.SEVERE, "Error registering tutor client", ex); // } return null; } } package network.jms; import constants.Constants; import java.io.OutputStreamWriter; import java.io.PrintWriter; import java.net.Socket; import java.util.logging.Level; import java.util.logging.Logger; import network.MessageGenerator; /* * * @author Mark */ public class RequestSender { private static final Logger LOG = Logger.getLogger(RequestSender.class.getName()); private final MessageGenerator messageGenerator = new MessageGenerator(); private final String labName; private final String machineId; public RequestSender(final String labName, String machineId) { this.labName = labName; this.machineId = machineId; } public void sendRequest(final String request) { // try { // final Socket socket = new Socket(destinationIp, Constants.PORT); // final PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream())); // writer.println(request); // writer.close(); // socket.close(); // } catch (Exception ex) { // LOG.log(Level.SEVERE, null, ex); // } } // @SuppressWarnings({"rawtypes", "unchecked"}) // public Collection<Integer> requestQueue() { // try { // final Socket socket = new Socket(destinationIp, Constants.PORT); // final PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream())); // writer.println(messageGenerator.requestQueue()); // writer.flush(); // final ObjectInputStream ois = new ObjectInputStream(socket.getInputStream()); // final Collection<Integer> response = (Collection<Integer>) ois.readObject(); // writer.close(); // ois.close(); // socket.close(); // return response; // } catch (Exception ex) { // LOG.log(Level.SEVERE, null, ex); // } // return Collections.EMPTY_SET; // } @SuppressWarnings({"rawtypes", "unchecked"}) public Socket registerTutorClient() { // try { // final Socket socket = new Socket(destinationIp, Constants.PORT); // final PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream())); // // writer.println(messageGenerator.registerTutorClient()); // writer.flush(); // // return socket; // } catch (Exception ex) { // LOG.log(Level.SEVERE, "Error registering tutor client", ex); // } return null; } }

Ignore Space Show notes View

src/network/jms/RequestSender.java

package network.jms;

import gui.Lab;
import java.net.Socket;
import java.util.logging.Logger;
import network.MessageGenerator;
import network.MessageType;
import server.jms.JmsConnection;

/**
 *
 * @author Mark
 */
public class RequestSender {

private static final Logger LOG = Logger.getLogger(RequestSender.class.getName());

private final MessageGenerator messageGenerator = new MessageGenerator();
	private final Lab lab;
	private final String machineId;

public RequestSender(final Lab lab, String machineId) {
		this.lab = lab;
		this.machineId = machineId;
	}

public void sendRequest(final MessageType type) {
		JmsConnection connection = new JmsConnection();
		
		if(type == MessageType.HELP ) {
			connection.sendMessageToTopic(lab.getLabName()+"_help", machineId);
		} else if (type == MessageType.MARKING) {
			connection.sendMessageToTopic(lab.getLabName()+"_marking", machineId);
		}
	}

//   @SuppressWarnings({"rawtypes", "unchecked"})
//   public Collection<Integer> requestQueue() {
//      try {
//         final Socket socket = new Socket(destinationIp, Constants.PORT);
//         final PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream()));
//         writer.println(messageGenerator.requestQueue());
//         writer.flush();
//         final ObjectInputStream ois = new ObjectInputStream(socket.getInputStream());
//         final Collection<Integer> response = (Collection<Integer>) ois.readObject();
//         writer.close();
//         ois.close();
//         socket.close();
//         return response;
//      } catch (Exception ex) {
//         LOG.log(Level.SEVERE, null, ex);
//      }
//      return Collections.EMPTY_SET;
//   }
	@SuppressWarnings({"rawtypes", "unchecked"})
	public Socket registerTutorClient() {
//		try {
//			final Socket socket = new Socket(destinationIp, Constants.PORT);
//			final PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream()));
//			
//			writer.println(messageGenerator.registerTutorClient());
//			writer.flush();
//			
//			return socket;
//		} catch (Exception ex) {
//			LOG.log(Level.SEVERE, "Error registering tutor client", ex);
//		}
		return null;
	}

}

package network.jms;

import constants.Constants;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.net.Socket;
import java.util.logging.Level;
import java.util.logging.Logger;
import network.MessageGenerator;

/**
 *
 * @author Mark
 */
public class RequestSender {

private static final Logger LOG = Logger.getLogger(RequestSender.class.getName());

private final MessageGenerator messageGenerator = new MessageGenerator();
	private final String labName;
	private final String machineId;

public RequestSender(final String labName, String machineId) {
		this.labName = labName;
		this.machineId = machineId;
	}

public void sendRequest(final String request) {
//		try {
//			final Socket socket = new Socket(destinationIp, Constants.PORT);
//			final PrintWriter writer = new PrintWriter(new OutputStreamWriter(socket.getOutputStream()));
//			writer.println(request);
//			writer.close();
//			socket.close();
//		} catch (Exception ex) {
//			LOG.log(Level.SEVERE, null, ex);
//		}
	}

}

Ignore Space Show notes View src/server/jms/JmsConnection.java
package server.jms; import java.util.Collection; import java.util.Enumeration; import java.util.TreeSet; import java.util.logging.Level; import java.util.logging.Logger; import javax.jms.Connection; import javax.jms.JMSException; import javax.jms.MessageProducer; import javax.jms.Queue; import javax.jms.QueueBrowser; import javax.jms.Session; import javax.jms.TextMessage; import javax.jms.Topic; import org.apache.activemq.ActiveMQConnectionFactory; /** * * @author Mark / public class JmsConnection { private Connection connection; private Session session; public JmsConnection() { try { connection = new ActiveMQConnectionFactory().createConnection(); connection.start(); session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); } catch (JMSException ex) { Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex); } } public Collection<String> peekAll(String queueName) { Collection<String> messages = new TreeSet<String>(); try { QueueBrowser serverBrowser = session.createBrowser(session.createQueue(queueName)); Enumeration enumeration = serverBrowser.getEnumeration(); while(enumeration.hasMoreElements()) { String server = ((TextMessage)enumeration.nextElement()).getText(); messages.add(server); } } catch (JMSException ex) { Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex); } return messages; } public void sendMessageToTopic(String topicName, String message) { try { Topic topic = session.createTopic(topicName); TextMessage textMessage = session.createTextMessage(message); MessageProducer producer = session.createProducer(topic); producer.send(textMessage); } catch (JMSException ex) { Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex); } } public void sendMessageToQueue(String queueName, String message) { try { Queue queue = session.createQueue(message); TextMessage textMessage = session.createTextMessage(message); MessageProducer producer = session.createProducer(queue); producer.send(textMessage); } catch (JMSException ex) { Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex); } } } package server.jms; import java.util.Collection; import java.util.Enumeration; import java.util.TreeSet; import java.util.logging.Level; import java.util.logging.Logger; import javax.jms.Connection; import javax.jms.JMSException; import javax.jms.QueueBrowser; import javax.jms.Session; import javax.jms.TextMessage; import org.apache.activemq.ActiveMQConnectionFactory; /* * * @author Mark */ public class JmsConnection { private Connection connection; private Session session; public JmsConnection() { try { connection = new ActiveMQConnectionFactory().createConnection(); connection.start(); session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); } catch (JMSException ex) { Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex); } } public Collection<String> peekAll(String queueName) { Collection<String> messages = new TreeSet<String>(); try { QueueBrowser serverBrowser = session.createBrowser(session.createQueue(queueName)); Enumeration enumeration = serverBrowser.getEnumeration(); while(enumeration.hasMoreElements()) { String server = ((TextMessage)enumeration.nextElement()).getText(); messages.add(server); } } catch (JMSException ex) { Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex); } return messages; } }

Ignore Space Show notes View

src/server/jms/JmsConnection.java

package server.jms;

import java.util.Collection;
import java.util.Enumeration;
import java.util.TreeSet;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.jms.Connection;
import javax.jms.JMSException;
import javax.jms.MessageProducer;
import javax.jms.Queue;
import javax.jms.QueueBrowser;
import javax.jms.Session;
import javax.jms.TextMessage;
import javax.jms.Topic;
import org.apache.activemq.ActiveMQConnectionFactory;

/**
 *
 * @author Mark
 */
public class JmsConnection {

private Connection connection;
	private Session session;

public JmsConnection() {
		try {
			connection = new ActiveMQConnectionFactory().createConnection();
			connection.start();

session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
	
			
			
		} catch (JMSException ex) {
			Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex);
		}

}

public Collection<String> peekAll(String queueName) {
		
		Collection<String> messages = new TreeSet<String>();
		
		try {
			QueueBrowser serverBrowser = session.createBrowser(session.createQueue(queueName));
			Enumeration enumeration = serverBrowser.getEnumeration();

while(enumeration.hasMoreElements()) {
				String server = ((TextMessage)enumeration.nextElement()).getText();
				messages.add(server);
			}

} catch (JMSException ex) {
			Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex);
		}
		
		return messages;
	}
	
	public void sendMessageToTopic(String topicName, String message) {
		try {
			Topic topic = session.createTopic(topicName);
			TextMessage textMessage = session.createTextMessage(message);
			MessageProducer producer = session.createProducer(topic);
			producer.send(textMessage);
		} catch (JMSException ex) {
			Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex);
		}
	}
	
	public void sendMessageToQueue(String queueName, String message) {
		try {
			Queue queue = session.createQueue(message);
			TextMessage textMessage = session.createTextMessage(message);
			MessageProducer producer = session.createProducer(queue);
			producer.send(textMessage);
		} catch (JMSException ex) {
			Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex);
		}
	}	
	
}

package server.jms;

import java.util.Collection;
import java.util.Enumeration;
import java.util.TreeSet;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.jms.Connection;
import javax.jms.JMSException;
import javax.jms.QueueBrowser;
import javax.jms.Session;
import javax.jms.TextMessage;
import org.apache.activemq.ActiveMQConnectionFactory;

/**
 *
 * @author Mark
 */
public class JmsConnection {

private Connection connection;
	private Session session;

public JmsConnection() {
		try {
			connection = new ActiveMQConnectionFactory().createConnection();
			connection.start();

session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
	
			
			
		} catch (JMSException ex) {
			Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex);
		}

}

while(enumeration.hasMoreElements()) {
				String server = ((TextMessage)enumeration.nextElement()).getText();
				messages.add(server);
			}

} catch (JMSException ex) {
			Logger.getLogger(JmsConnection.class.getName()).log(Level.SEVERE, null, ex);
		}
		
		return messages;
	}
	
}

Show line notes below