1 files changed, 10 insertions, 0 deletions
diff --git a/src/main/java/de/bwlehrpool/bwlp_guac/WrappedConnection.java b/src/main/java/de/bwlehrpool/bwlp_guac/WrappedConnection.java
index a2ecb52..95a014b 100644
--- a/src/main/java/de/bwlehrpool/bwlp_guac/WrappedConnection.java
+++ b/src/main/java/de/bwlehrpool/bwlp_guac/WrappedConnection.java
@@ -11,6 +11,8 @@ public class WrappedConnection extends SimpleConnection {
 
 	private boolean valid = true;
 
+	private BwlpUserContext context;
+
 	public WrappedConnection(String name, AvailableClient ac) {
 		super(name, name, makeConfig(ac));
 		this.ac = ac.clone();
@@ -31,6 +33,14 @@ public class WrappedConnection extends SimpleConnection {
 
 	public void invalidate() {
 		this.valid = false;
+		if (this.context != null) {
+			this.context.clearConnections();
+			this.context = null;
+		}
+	}
+
+	public void setContext(BwlpUserContext context) {
+		this.context = context;
 	}
 
 }