communicator.java

无刷新功能的聊天室工具 java实现,本人课程设计的作业,附源代码

/*
 * Created on 2003. 2. 20.
 */
package chipchat;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

/**
 * Communicate with client.
 * @author Mr. Lee
 *
 */
public final class Communicator {
	/*
	 * Env values.
	 */

	/** Server name */
	private static final String SERVERNAME =
		Env.getInstance().getProperty("Communicator.serverName");

	/** Admin Password */
	private static final String ADMINPW =
		Env.getInstance().getProperty("Communicator.adminpasswd");

	/**
	 * Constructor.
	 */
	public Communicator() {
	}

	/**
	 * Service chipchat function.
	 * @param req ServletRequest
	 * @param resp ServletResponse
	 */
	public void service(
		final HttpServletRequest req,
		final HttpServletResponse resp) {
		try {
			// Input Stream and output Stream...
			InputStream inputStream = req.getInputStream();
			OutputStream outputStream = resp.getOutputStream();

			// Get User Information from session.
			HttpSession session = req.getSession(false);

			if (session == null) {
				outputStream.write("ERROR:SessionNotExist".getBytes());
				outputStream.flush();
				return;
			}

			session.setMaxInactiveInterval(60 * 60 * 2);

			String name = (String) session.getAttribute("chipchat_username");
			String channel = (String) session.getAttribute("chipchat_channel");
			String roompw = (String) session.getAttribute("chipchat_roompw");
			int userid;
			Long roomid;
			try {
				userid =
					Integer.parseInt(
						(String) session.getAttribute("chipchat_userid"));
				roomid = (Long) session.getAttribute("chipchat_roomid");
				session.setAttribute("chipchat_roomid", null);
			} catch (NumberFormatException e) {
				outputStream.write("ERROR:WrongSessionValue".getBytes());
				outputStream.flush();
				return;
			}

			User user = new User(name, userid, outputStream);

			service(user, channel, roomid, roompw, inputStream);

		} catch (Exception e) {
			System.err.println("Exception : 93");
			e.printStackTrace();
		}
	}

	/**
	 * 
	 * @param conName
	 * @param input
	 * @param output
	 */
	public void service(
		String conName,
		final InputStream input,
		final OutputStream output) {
		try {
			ConnectionInfo info = ConnectionWaiter.getInstance().get(conName);
			if (info == null) {
				output.write("ERROR:SessionNotExist".getBytes());
				output.flush();
				return;
			}

			//			session.setMaxInactiveInterval(60 * 60 * 2);

			User user = new User(info.getName(), info.getUserid(), output);
			service(
				user,
				info.getChannel(),
				info.getRoomid(),
				info.getRoompw(),
				input);
		} catch (Exception e) {
			System.err.println("Exception : 93");
			e.printStackTrace();
		}
	}

	/**
	 * Service chipchat function.
	 * @param user User
	 * @param channel Channel name
	 * @param roomid Room id
	 * @param passwd Password
	 * @param inputStream Input Stream
	 */
	public void service(
		final User user,
		final String channel,
		final Long roomid,
		final String passwd,
		final InputStream inputStream) {
		try {
			// Input Stream and output Stream...
			BufferedReader reader =
				new BufferedReader(new InputStreamReader(inputStream));
			OutputStream outputStream = user.getOutputStream();

			outputStream.write(
				("ConnectID:" + user.getUserid() + "\r\n").getBytes());
			outputStream.write(
				("ConnectName:" + user.getUsername() + "\r\n").getBytes());

			final Channel channel2 = ChipChat.getInstance().getChannel(channel);

			final Room room = channel2.getRoom(roomid);

			if (room == null) {
				outputStream.write("ERROR:RoomIsNotExist\r\n".getBytes());
				outputStream.flush();
				return;
			}

			int enter;
			if ("00admin".equals(user.getUsername())
				&& ("\t" + ADMINPW).equals(passwd)) {
				enter = room.enterAdmin(user, passwd);
			} else {
				enter = room.enterUser(user, passwd);
			}

			if (enter != 0) {
				if (enter == -1) {
					outputStream.write("ERROR:RoomIsFull\r\n".getBytes());
				} else if (enter == -2) {
					outputStream.write("ERROR:PasswordNotMatch\r\n".getBytes());
				} else {
					outputStream.write("ERROR:Unknown\r\n".getBytes());
				}
				outputStream.flush();
				return;
			}

			outputStream.write(("Connected:" + SERVERNAME + "\r\n").getBytes());
			outputStream.write(
				("ADMIN:" + room.getMaster() + "\r\n").getBytes());
			outputStream.flush();

			boolean connected = true;

			room.inputRoomInfo();

			try {
				while (connected) {
					String r = reader.readLine();
					if (r == null) {
						// Connection Closed...
						break;
					}
					try {
						int index = r.indexOf(":");
						if (index < 0) {
							System.out.println(
								"NO Command Found .... Skipping....:" + r);
							continue;
						}
						String cmd = r.substring(0, index);
						if (cmd.equalsIgnoreCase("MSG")) { // message
							room.inputMsg(
								-1,
								r.substring(index + 1),
								user.getUsername());
						} else if (cmd.equalsIgnoreCase("WSP")) { // whisper
							try {
								int index2 = r.indexOf(":", index + 1);
								int to =
									Integer.parseInt(
										r.substring(index + 1, index2));
								room.inputWhisper(
									user.getUserid().intValue(),
									to,
									r.substring(index2 + 1),
									user.getUsername());
							} catch (NumberFormatException e) {
								System.out.println("Wrong User ID:" + r);
							}
						} else if (cmd.equalsIgnoreCase("ACK")) {
							// Acknowlege
							outputStream.write("ACK:\r\n".getBytes());
							outputStream.flush();
							continue;
						} else if (cmd.equalsIgnoreCase("KEEPQUIET")) {
							try {
								int to =
									Integer.parseInt(r.substring(index + 1));
								room.inputKeepQuiet(
									user.getUserid().intValue(),
									to);
							} catch (NumberFormatException e) {
								System.out.println("Wrong User ID:" + r);
							}
							continue;
						} else if (cmd.equalsIgnoreCase("KICKOUT")) {
							try {
								int to =
									Integer.parseInt(r.substring(index + 1));
								room.inputKickOut(
									user.getUserid().intValue(),
									to);
							} catch (NumberFormatException e) {
								System.out.println("Wrong User ID:" + r);
							}
							continue;
						} else if (cmd.equalsIgnoreCase("ENTRUST")) {
							try {
								int to =
									Integer.parseInt(r.substring(index + 1));
								room.inputEntrust(
									user.getUserid().intValue(),
									to);
							} catch (NumberFormatException e) {
								System.out.println("Wrong User ID:" + r);
							}
							continue;
						} else if (cmd.equalsIgnoreCase("GETOUT")) {
							// Acknowlege
							break;
						} else if (cmd.equalsIgnoreCase("CHGPASSWORD")) {
							String str = r.substring(index + 1);
							room.changePasswd(user.getUserid().intValue(), str);
						} else if (cmd.equalsIgnoreCase("CHGROOMNAME")) {
							String str = r.substring(index + 1);
							room.changeRoomName(
								user.getUserid().intValue(),
								str);
						} else if (cmd.equalsIgnoreCase("CHGMAX")) {
							try {
								int num =
									Integer.parseInt(r.substring(index + 1));
								room.changeMaxMan(
									user.getUserid().intValue(),
									num);
							} catch (NumberFormatException e) {
								System.out.println("Wrong Number Format:" + r);
							}
						} else if (cmd.equalsIgnoreCase("CUSTOM")) {
							try {
								int index2 = r.indexOf(":", index + 1);
								int to =
									Integer.parseInt(
										r.substring(index + 1, index2));
								room.inputCustomMsg(
									to,
									r.substring(index2 + 1),
									user.getUsername());
							} catch (NumberFormatException e) {
								System.out.println("Wrong Number Format:" + r);
							}
						} else {
							System.out.println(
								"Wrong Command.... Skipping....:" + r);
							continue;
						}
					} catch (Exception e) {
						e.printStackTrace();
					}
				}
			} catch (IOException e) {
				e.printStackTrace();
			} catch (Exception e) {
				e.printStackTrace();
			} finally {
				room.exitUser(user);
				connected = false;
				try {
					if (outputStream != null) {
						outputStream.write("CLOSED:\r\n".getBytes());
						outputStream.flush();
					}
				} catch (IOException e) {
					System.err.println("IOException : 203");
					e.printStackTrace();
				}
			}
			System.out.println(
				"Listen thread ended....[" + user.getUserid() + "]");
		} catch (IOException e) {
			System.err.println("IOException : 155");
			e.printStackTrace();
		}
	}
}