Alawn/Chronicle-Map
1
0
Fork 0
mirror of https://github.com/OpenHFT/Chronicle-Map.git synced 2021-09-13 01:26:12 +00:00
Code Releases Activity

HCOLL-167 added support for keyset

Browse Source
This commit is contained in:
Rob Austin
2014-10-27 20:09:53 +00:00
parent 8136b5e298
commit accebbfeea
2 changed files with 81 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/net/openhft/chronicle/map/TcpReplicator.java
+45 -35
View File
@@ -548,7 +548,7 @@ class TcpReplicator extends AbstractChannelReplicator implements Closeable {
if (completed)
attached.entryWriter.workCompleted();
attached.hasRemoteHeartbeatInterval = false;
attached.hasRemoteHeartbeatInterval = false;
} else if (attached.remoteModificationIterator != null)
attached.entryWriter.entriesToBuffer(attached.remoteModificationIterator, key);
@@ -1573,40 +1573,6 @@ class StatelessServerConnector<K, V> {
return null;
}
private Work keySet(Bytes reader, final Bytes writer) {
final long sizeLocation = reflectTransactionId(reader, writer);
final Set<K> ks;
try {
ks = map.keySet();
} catch (RuntimeException e) {
return sendException(writer, sizeLocation, e);
}
writer.writeStopBit(ks.size());
final Iterator<K> iterator = ks.iterator();
// this allows us to write more data than the buffer will allow
return new Work() {
@Override
public boolean doWork(@NotNull final Bytes writer) {
while (iterator.hasNext()) {
// we've filled up the buffer, so lets give another channel a chance to send
// some data, we don't know the max key size, we will use the entrySize instead
if (writer.remaining() <= maxEntrySizeBytes)
return false;
writeKey(iterator.next(), writer);
}
writeSizeAndFlags(sizeLocation, false, writer);
return true;
}
};
}
private Work values(Bytes reader, Bytes writer) {
final long sizeLocation = reflectTransactionId(reader, writer);
@@ -1644,6 +1610,50 @@ class StatelessServerConnector<K, V> {
};
}
private Work keySet(Bytes reader, final Bytes writer) {
final long transactionId = reader.readLong();
Set<K> ks;
try {
ks = map.keySet();
} catch (RuntimeException e) {
return sendException(reader, writer, e);
}
final Iterator<K> iterator = ks.iterator();
// this allows us to write more data than the buffer will allow
return new Work() {
@Override
public boolean doWork(Bytes writer) {
final long sizeLocation = header(writer, transactionId);
int count = 0;
while (iterator.hasNext()) {
// we've filled up the buffer, so lets give another channel a chance to send
// some data, we don't know the max key size, we will use the entrySize instead
if (writer.remaining() <= maxEntrySizeBytes) {
writeHeader(writer, sizeLocation, count, true);
return false;
}
count++;
K key = iterator.next();
writeKey(key, writer);
}
writeHeader(writer, sizeLocation, count, false);
return true;
}
};
}
private Work entrySet(Bytes reader, Bytes writer) {
final long transactionId = reader.readLong();