Adapt the tests to read/write Bytes

OpenHFT · Feb 5, 2015 · 562e6df · 562e6df
1 parent 0fe82ce
commit 562e6df
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/chronicle-queue/src/main/java/net/openhft/chronicle/queue/impl/DirectChronicle.java b/chronicle-queue/src/main/java/net/openhft/chronicle/queue/impl/DirectChronicle.java
@@ -12,7 +12,7 @@
 public interface DirectChronicle extends Chronicle {
     void appendDocument(Bytes buffer);
 
-    void readDocument(AtomicLong offset, Bytes buffer);
+    boolean readDocument(AtomicLong offset, Bytes buffer);
 
     Bytes bytes();
 

diff --git a/chronicle-queue/src/main/java/net/openhft/chronicle/queue/impl/SingleChronicle.java b/chronicle-queue/src/main/java/net/openhft/chronicle/queue/impl/SingleChronicle.java
@@ -23,6 +23,8 @@
 import java.nio.file.Paths;
 import java.util.concurrent.atomic.AtomicLong;
 
+import static net.openhft.chronicle.wire.BinaryWire.isDocument;
+
 /**
  * SingleChronicle implements Chronicle over a single streaming file
  *
@@ -77,7 +79,7 @@ public void appendDocument(Bytes buffer) {
     }
 
     @Override
-    public void readDocument(AtomicLong offset, Bytes buffer) {
+    public boolean readDocument(AtomicLong offset, Bytes buffer) {
         buffer.clear();
         long lastByte = offset.get();
         for (; ; ) {
@@ -88,7 +90,7 @@ public void readDocument(AtomicLong offset, Bytes buffer) {
                 buffer.write(bytes, lastByte, length2);
                 lastByte += length2;
                 offset.set(lastByte);
-                return;
+                return isDocument(length);
             }
             Jvm.checkInterrupted();
         }