Fix some minor handling issues with transfers wrt global image exchange

1 files changed, 8 insertions, 0 deletions

diff --git a/src/main/java/org/openslx/filetransfer/util/AbstractTransfer.java b/src/main/java/org/openslx/filetransfer/util/AbstractTransfer.java
index 75c68e8..636f880 100644
--- a/src/main/java/org/openslx/filetransfer/util/AbstractTransfer.java
+++ b/src/main/java/org/openslx/filetransfer/util/AbstractTransfer.java
@@ -1,6 +1,7 @@
 package org.openslx.filetransfer.util;
 
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicLong;
 
 import org.openslx.bwlp.thrift.iface.TransferInformation;
@@ -43,6 +44,8 @@ public abstract class AbstractTransfer
 	protected final AtomicLong lastActivityTime = new AtomicLong( System.currentTimeMillis() );
 
 	private final String transferId;
+	
+	protected AtomicInteger connectFails = new AtomicInteger();
 
 	public AbstractTransfer( String transferId )
 	{
@@ -77,6 +80,11 @@ public abstract class AbstractTransfer
 	{
 		return getActiveConnectionCount() > 0 || lastActivityTime.get() + HOT_IDLE_TIMEOUT > now;
 	}
+	
+	public final int connectFailCount()
+	{
+		return connectFails.get();
+	}
 
 	public final String getId()
 	{