Fix flushedBytes computation in writePayload

xerial · xerial · commit f762b10ac152 · 2015-02-11T00:50:13.000+09:00
diff --git a/msgpack-core/src/main/java/org/msgpack/core/MessagePacker.java b/msgpack-core/src/main/java/org/msgpack/core/MessagePacker.java
@@ -549,8 +549,8 @@ else if(len < (1 << 16)) {
 
 
     public MessagePacker writePayload(ByteBuffer src) throws IOException {
-        flushedBytes += src.remaining();
-        if(src.remaining() >= config.getPackerRawDataCopyingThreshold()) {
+        int len = src.remaining();
+        if(len >= config.getPackerRawDataCopyingThreshold()) {
             // Use the source ByteBuffer directly to avoid memory copy
 
             // First, flush the current buffer contents
@@ -561,6 +561,7 @@ public MessagePacker writePayload(ByteBuffer src) throws IOException {
             // Then, dump the source data to the output
             out.flush(wrapped);
             src.position(src.limit());
+            flushedBytes += len;
         }
         else {
             // If the input source is small, simply copy the contents to the buffer
@@ -593,6 +594,7 @@ public MessagePacker writePayload(byte[] src, int off, int len) throws IOExcepti
             MessageBuffer wrapped = MessageBuffer.wrap(src).slice(off, len);
             // Dump the source data to the output
             out.flush(wrapped);
+            flushedBytes += len;
         }
         else {
             int cursor = 0;
