added BufferUnpacker#getBufferSize()

frsyuki · frsyuki · commit 548ddafdad36 · 2011-12-15T09:52:48.000-08:00
diff --git a/src/main/java/org/msgpack/io/LinkedBufferInput.java b/src/main/java/org/msgpack/io/LinkedBufferInput.java
@@ -369,6 +369,14 @@ public void copyReferencedBuffer() {
         }
     }
 
+    public int getSize() {
+        int size = 0;
+        for(ByteBuffer bb : link) {
+            size += bb.remaining();
+        }
+        return size;
+    }
+
     public void close() {
     }
 }
diff --git a/src/main/java/org/msgpack/unpacker/BufferUnpacker.java b/src/main/java/org/msgpack/unpacker/BufferUnpacker.java
@@ -44,6 +44,8 @@ public interface BufferUnpacker extends Unpacker {
 
     public BufferUnpacker feed(ByteBuffer buf, boolean reference);
 
+    public int getBufferSize();
+
     public void copyReferencedBuffer();
 
     public void clear();
diff --git a/src/main/java/org/msgpack/unpacker/MessagePackBufferUnpacker.java b/src/main/java/org/msgpack/unpacker/MessagePackBufferUnpacker.java
@@ -89,6 +89,11 @@ public MessagePackBufferUnpacker feed(ByteBuffer buf, boolean reference) {
         return this;
     }
 
+    @Override
+    public int getBufferSize() {
+        return ((LinkedBufferInput) in).getSize();
+    }
+
     @Override
     public void copyReferencedBuffer() {
         ((LinkedBufferInput) in).copyReferencedBuffer();
diff --git a/src/main/java/org/msgpack/util/json/JSONBufferUnpacker.java b/src/main/java/org/msgpack/util/json/JSONBufferUnpacker.java
@@ -97,6 +97,12 @@ public JSONBufferUnpacker feed(ByteBuffer buf, boolean reference) {
                 "JSONBufferUnpacker doesn't support feed()");
     }
 
+    @Override
+    public int getBufferSize() {
+        throw new UnsupportedOperationException(
+                "JSONBufferUnpacker doesn't support getBufferSize()");
+    }
+
     @Override
     public void copyReferencedBuffer() {
         throw new UnsupportedOperationException(

Original file line number	Diff line number	Diff line change
`@@ -369,6 +369,14 @@ public void copyReferencedBuffer() {`
`369`	`369`	`}`
`370`	`370`	`}`
`371`	`371`
	`372`	`+ public int getSize() {`
	`373`	`+ int size = 0;`
	`374`	`+ for(ByteBuffer bb : link) {`
	`375`	`+ size += bb.remaining();`
	`376`	`+ }`
	`377`	`+ return size;`
	`378`	`+ }`
	`379`	`+`
`372`	`380`	`public void close() {`
`373`	`381`	`}`
`374`	`382`	`}`