Merge pull request #38 from PM2-IT21bWIN-ruiz-mach-krea/ConnectionHandlerClass
Connection handler class
This commit is contained in:
@@ -1,7 +1,8 @@
|
||||
package ch.zhaw.pm2.multichat.client;
|
||||
|
||||
import ch.zhaw.pm2.multichat.client.ClientConnectionHandler.State;
|
||||
import ch.zhaw.pm2.multichat.protocol.ConnectionHandler.State;
|
||||
import ch.zhaw.pm2.multichat.protocol.ChatProtocolException;
|
||||
import ch.zhaw.pm2.multichat.protocol.ConnectionHandler;
|
||||
import ch.zhaw.pm2.multichat.protocol.NetworkHandler;
|
||||
import javafx.application.Platform;
|
||||
import javafx.beans.value.ChangeListener;
|
||||
@@ -15,10 +16,8 @@ import javafx.scene.layout.Pane;
|
||||
import javafx.stage.WindowEvent;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
import static ch.zhaw.pm2.multichat.client.ClientConnectionHandler.State.*;
|
||||
import static ch.zhaw.pm2.multichat.protocol.ConnectionHandler.State.*;
|
||||
|
||||
public class ChatWindowController {
|
||||
private ClientConnectionHandler connectionHandler;
|
||||
@@ -133,7 +132,7 @@ public class ChatWindowController {
|
||||
}
|
||||
}
|
||||
|
||||
public void stateChanged(State newState) {
|
||||
public void stateChanged(ConnectionHandler.State newState) {
|
||||
// update UI (need to be run in UI thread: see Platform.runLater())
|
||||
Platform.runLater(new Runnable() {
|
||||
@Override
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
package ch.zhaw.pm2.multichat.client;
|
||||
|
||||
import ch.zhaw.pm2.multichat.protocol.ChatProtocolException;
|
||||
import ch.zhaw.pm2.multichat.protocol.ConnectionHandler;
|
||||
import ch.zhaw.pm2.multichat.protocol.NetworkHandler;
|
||||
import javafx.beans.property.SimpleIntegerProperty;
|
||||
import javafx.beans.property.SimpleObjectProperty;
|
||||
@@ -12,21 +13,9 @@ import java.net.SocketException;
|
||||
import java.util.Scanner;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
import static ch.zhaw.pm2.multichat.protocol.ConnectionHandler.State.*;
|
||||
|
||||
import static ch.zhaw.pm2.multichat.client.ClientConnectionHandler.State.*;
|
||||
|
||||
public class ClientConnectionHandler implements Runnable {
|
||||
private final NetworkHandler.NetworkConnection<String> connection;
|
||||
|
||||
// Data types used for the Chat Protocol
|
||||
private static final String DATA_TYPE_CONNECT = "CONNECT";
|
||||
private static final String DATA_TYPE_CONFIRM = "CONFIRM";
|
||||
private static final String DATA_TYPE_DISCONNECT = "DISCONNECT";
|
||||
private static final String DATA_TYPE_MESSAGE = "MESSAGE";
|
||||
private static final String DATA_TYPE_ERROR = "ERROR";
|
||||
|
||||
public static final String USER_NONE = "";
|
||||
public static final String USER_ALL = "*";
|
||||
public class ClientConnectionHandler extends ConnectionHandler implements Runnable {
|
||||
|
||||
private final Pattern messagePattern = Pattern.compile( "^(?:@(\\S*))?\\s*(.*)$" );
|
||||
|
||||
@@ -36,17 +25,13 @@ public class ClientConnectionHandler implements Runnable {
|
||||
private SimpleStringProperty serverAddress;
|
||||
private SimpleIntegerProperty serverPort;
|
||||
|
||||
enum State {
|
||||
NEW, CONFIRM_CONNECT, CONNECTED, CONFIRM_DISCONNECT, DISCONNECTED;
|
||||
}
|
||||
|
||||
public ClientConnectionHandler(NetworkHandler.NetworkConnection<String> connection,
|
||||
String userName,
|
||||
ClientMessageList messages) {
|
||||
this.connection = connection;
|
||||
super(connection);
|
||||
this.userName = new SimpleStringProperty((userName == null || userName.isBlank())? USER_NONE : userName);
|
||||
this.messages = messages;
|
||||
state = new SimpleObjectProperty<>(NEW);
|
||||
state = new SimpleObjectProperty<>(State.NEW);
|
||||
serverAddress = new SimpleStringProperty();
|
||||
serverPort = new SimpleIntegerProperty();
|
||||
}
|
||||
@@ -69,12 +54,12 @@ public class ClientConnectionHandler implements Runnable {
|
||||
startReceiving();
|
||||
}
|
||||
|
||||
public void startReceiving() {
|
||||
private void startReceiving() {
|
||||
System.out.println("Starting Connection Handler");
|
||||
try {
|
||||
System.out.println("Start receiving data...");
|
||||
while (connection.isAvailable()) {
|
||||
String data = connection.receive();
|
||||
while (getConnection().isAvailable()) {
|
||||
String data = getConnection().receive();
|
||||
processData(data);
|
||||
}
|
||||
System.out.println("Stopped recieving data");
|
||||
@@ -98,7 +83,7 @@ public class ClientConnectionHandler implements Runnable {
|
||||
System.out.println("Closing Connection Handler to Server");
|
||||
try {
|
||||
System.out.println("Stop receiving data...");
|
||||
connection.close();
|
||||
getConnection().close();
|
||||
System.out.println("Stopped receiving data.");
|
||||
} catch (IOException e) {
|
||||
System.err.println("Failed to close connection." + e.getMessage());
|
||||
@@ -133,13 +118,13 @@ public class ClientConnectionHandler implements Runnable {
|
||||
payload = scanner.nextLine();
|
||||
}
|
||||
// dispatch operation based on type parameter
|
||||
if (type.equals(DATA_TYPE_CONNECT)) {
|
||||
if (type.equals(getDataTypeConnect())) {
|
||||
System.err.println("Illegal connect request from server");
|
||||
} else if (type.equals(DATA_TYPE_CONFIRM)) {
|
||||
} else if (type.equals(getDataTypeConfirm())) {
|
||||
if (state.get() == CONFIRM_CONNECT) {
|
||||
this.userName.set(reciever);
|
||||
this.serverPort.set(connection.getRemotePort());
|
||||
this.serverAddress.set(connection.getRemoteHost());
|
||||
this.serverPort.set(getConnection().getRemotePort());
|
||||
this.serverAddress.set(getConnection().getRemoteHost());
|
||||
messages.addMessage(new Message(Message.MessageType.INFO,sender,reciever,payload));
|
||||
System.out.println("CONFIRM: " + payload);
|
||||
this.setState(CONNECTED);
|
||||
@@ -150,7 +135,7 @@ public class ClientConnectionHandler implements Runnable {
|
||||
} else {
|
||||
System.err.println("Got unexpected confirm message: " + payload);
|
||||
}
|
||||
} else if (type.equals(DATA_TYPE_DISCONNECT)) {
|
||||
} else if (type.equals(getDataTypeDisconnect())) {
|
||||
if (state.get() == DISCONNECTED) {
|
||||
System.out.println("DISCONNECT: Already in disconnected: " + payload);
|
||||
return;
|
||||
@@ -158,14 +143,14 @@ public class ClientConnectionHandler implements Runnable {
|
||||
messages.addMessage(new Message(Message.MessageType.INFO,sender,reciever,payload));
|
||||
System.out.println("DISCONNECT: " + payload);
|
||||
this.setState(DISCONNECTED);
|
||||
} else if (type.equals(DATA_TYPE_MESSAGE)) {
|
||||
} else if (type.equals(getDataTypeMessage())) {
|
||||
if (state.get() != CONNECTED) {
|
||||
System.out.println("MESSAGE: Illegal state " + state + " for message: " + payload);
|
||||
return;
|
||||
}
|
||||
messages.addMessage(new Message(Message.MessageType.MESSAGE,sender,reciever,payload));
|
||||
System.out.println("MESSAGE: From " + sender + " to " + reciever + ": "+ payload);
|
||||
} else if (type.equals(DATA_TYPE_ERROR)) {
|
||||
} else if (type.equals(getDataTypeError())) {
|
||||
messages.addMessage(new Message(Message.MessageType.ERROR,sender,reciever,payload));
|
||||
System.out.println("ERROR: " + payload);
|
||||
} else {
|
||||
@@ -173,12 +158,12 @@ public class ClientConnectionHandler implements Runnable {
|
||||
}
|
||||
} catch (ChatProtocolException e) {
|
||||
System.err.println("Error while processing data: " + e.getMessage());
|
||||
sendData(USER_NONE, userName.get(), DATA_TYPE_ERROR, e.getMessage());
|
||||
sendData(USER_NONE, userName.get(), getDataTypeError(), e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
public void sendData(String sender, String receiver, String type, String payload) {
|
||||
if (connection.isAvailable()) {
|
||||
private void sendData(String sender, String receiver, String type, String payload) {
|
||||
if (getConnection().isAvailable()) {
|
||||
new StringBuilder();
|
||||
String data = new StringBuilder()
|
||||
.append(sender+"\n")
|
||||
@@ -187,7 +172,7 @@ public class ClientConnectionHandler implements Runnable {
|
||||
.append(payload+"\n")
|
||||
.toString();
|
||||
try {
|
||||
connection.send(data);
|
||||
getConnection().send(data);
|
||||
} catch (SocketException e) {
|
||||
System.err.println("Connection closed: " + e.getMessage());
|
||||
} catch (EOFException e) {
|
||||
@@ -200,13 +185,13 @@ public class ClientConnectionHandler implements Runnable {
|
||||
|
||||
public void connect() throws ChatProtocolException {
|
||||
if (state.get() != NEW) throw new ChatProtocolException("Illegal state for connect: " + state);
|
||||
this.sendData(userName.get(), USER_NONE, DATA_TYPE_CONNECT,null);
|
||||
this.sendData(userName.get(), USER_NONE, getDataTypeConnect(),null);
|
||||
this.setState(CONFIRM_CONNECT);
|
||||
}
|
||||
|
||||
public void disconnect() throws ChatProtocolException {
|
||||
if (state.get() != NEW && state.get() != CONNECTED) throw new ChatProtocolException("Illegal state for disconnect: " + state);
|
||||
this.sendData(userName.get(), USER_NONE, DATA_TYPE_DISCONNECT,null);
|
||||
this.sendData(userName.get(), USER_NONE, getDataTypeDisconnect(),null);
|
||||
this.setState(CONFIRM_DISCONNECT);
|
||||
}
|
||||
|
||||
@@ -221,7 +206,7 @@ public class ClientConnectionHandler implements Runnable {
|
||||
return false;
|
||||
}
|
||||
if (receiver == null || receiver.isBlank()) receiver = ClientConnectionHandler.USER_ALL;
|
||||
this.sendData(userName.get(), receiver, DATA_TYPE_MESSAGE,message);
|
||||
this.sendData(userName.get(), receiver, getDataTypeMessage(),message);
|
||||
return true;
|
||||
} else {
|
||||
return false;
|
||||
|
||||
Reference in New Issue
Block a user