bitmessage/Jabit
bitmessage/Jabit
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed synchronisation

Browse Source
This commit is contained in:
Christian Basler 2016-07-09 16:37:12 +02:00
parent d130080df2
commit 50f2c7e080
3 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
networking/src/main/java/ch/dissem/bitmessage/networking/AbstractConnection.java
View File

@ -61,7 +61,7 @@ public abstract class AbstractConnection {
protected long lastObjectTime;
private final long syncTimeout;
private int readTimeoutCounter;
private long syncReadTimeout = Long.MAX_VALUE;
protected long peerNonce;
protected int version;
@ -305,16 +305,13 @@ public abstract class AbstractConnection {
return true;
}
if (!sendingQueue.isEmpty()) {
syncReadTimeout = System.currentTimeMillis() + 1000;
return false;
}
if (msg == null) {
if (requestedObjects.isEmpty())
return true;
readTimeoutCounter++;
return readTimeoutCounter > 1;
return syncReadTimeout < System.currentTimeMillis();
} else {
readTimeoutCounter = 0;
syncReadTimeout = System.currentTimeMillis() + 1000;
return false;
}
}

11
networking/src/main/java/ch/dissem/bitmessage/networking/nio/NioNetworkHandler.java
View File

@ -77,7 +77,6 @@ public class NioNetworkHandler implements NetworkHandler, InternalContext.Contex
public Void call() throws Exception {
Set<InventoryVector> requestedObjects = new HashSet<>();
try (SocketChannel channel = SocketChannel.open(new InetSocketAddress(server, port))) {
channel.finishConnect();
channel.configureBlocking(false);
ConnectionInfo connection = new ConnectionInfo(ctx, SYNC,
new NetworkAddress.Builder().ip(server).port(port).stream(1).build(),
@ -88,6 +87,7 @@ public class NioNetworkHandler implements NetworkHandler, InternalContext.Contex
read(channel, connection);
Thread.sleep(10);
}
connections.remove(connection);
LOG.info("Synchronization finished");
}
return null;
@ -225,7 +225,6 @@ public class NioNetworkHandler implements NetworkHandler, InternalContext.Contex
while (selector.isOpen()) {
selector.select(1000);
Iterator<SelectionKey> keyIterator = selector.selectedKeys().iterator();
while (keyIterator.hasNext()) {
SelectionKey key = keyIterator.next();
if (key.attachment() instanceof ConnectionInfo) {
@ -253,6 +252,14 @@ public class NioNetworkHandler implements NetworkHandler, InternalContext.Contex
}
keyIterator.remove();
}
for (SelectionKey key : selector.keys()) {
if ((key.interestOps() & OP_WRITE) == 0) {
if (key.attachment() instanceof ConnectionInfo &&
!((ConnectionInfo) key.attachment()).getSendingQueue().isEmpty()) {
key.interestOps(OP_READ | OP_WRITE);
}
}
}
}
selector.close();
} catch (ClosedSelectorException ignore) {

2
networking/src/test/java/ch/dissem/bitmessage/networking/NetworkHandlerTest.java
View File

@ -254,7 +254,7 @@ public class NetworkHandlerTest {
Future<?> future = nodeNetworkHandler.synchronize(peerAddress.toInetAddress(), peerAddress.getPort(),
mock(NetworkHandler.MessageListener.class),
100);
10);
future.get();
assertInventorySize(1, nodeInventory);
assertInventorySize(1, peerInventory);