Make MessagePackGenerator be able to serialize extended type

komamitsu · komamitsu · commit 584e9d27256e · 2015-06-14T18:58:18.000+09:00
diff --git a/msgpack-jackson/src/main/java/org/msgpack/jackson/dataformat/MessagePackExtendedType.java b/msgpack-jackson/src/main/java/org/msgpack/jackson/dataformat/MessagePackExtendedType.java
@@ -1,10 +1,15 @@
 package org.msgpack.jackson.dataformat;
 
+import com.fasterxml.jackson.core.JsonGenerator;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.JsonSerializer;
+import com.fasterxml.jackson.databind.SerializerProvider;
+import com.fasterxml.jackson.databind.annotation.JsonSerialize;
+
+import java.io.IOException;
 import java.nio.ByteBuffer;
 
-/**
- * Created by komamitsu on 3/7/15.
- */
+@JsonSerialize(using = MessagePackExtendedType.Serializer.class)
 public class MessagePackExtendedType {
     private final int extType;
     private final ByteBuffer byteBuffer;
@@ -22,4 +27,18 @@ public int extType() {
     public ByteBuffer byteBuffer() {
         return byteBuffer;
     }
+
+    public static class Serializer extends JsonSerializer<MessagePackExtendedType> {
+        @Override
+        public void serialize(MessagePackExtendedType value, JsonGenerator gen, SerializerProvider serializers)
+                throws IOException, JsonProcessingException {
+            if (gen instanceof MessagePackGenerator) {
+                MessagePackGenerator msgpackGenerator = (MessagePackGenerator)gen;
+                msgpackGenerator.writeExtendedType(value);
+            }
+            else {
+                throw new IllegalStateException("gen is expected to be MessagePackGenerator but it's " + gen.getClass());
+            }
+        }
+    }
 }
diff --git a/msgpack-jackson/src/main/java/org/msgpack/jackson/dataformat/MessagePackGenerator.java b/msgpack-jackson/src/main/java/org/msgpack/jackson/dataformat/MessagePackGenerator.java
@@ -175,6 +175,12 @@ else if (v instanceof BigDecimal) {
         else if (v instanceof Boolean) {
             messagePacker.packBoolean((Boolean) v);
         }
+        else if (v instanceof MessagePackExtendedType) {
+            MessagePackExtendedType extendedType = (MessagePackExtendedType) v;
+            ByteBuffer buf = extendedType.byteBuffer();
+            messagePacker.packExtendedTypeHeader(extendedType.extType(), buf.remaining());
+            messagePacker.writePayload(buf);
+        }
         else {
             throw new IllegalArgumentException(v.toString());
         }
@@ -324,6 +330,10 @@ public void writeNull() throws IOException, JsonGenerationException {
         addValueToStackTop(null);
     }
 
+    public void writeExtendedType(MessagePackExtendedType extendedType) throws IOException {
+        addValueToStackTop(extendedType);
+    }
+
     @Override
     public void close() throws IOException {
         try {
diff --git a/msgpack-jackson/src/test/java/org/msgpack/jackson/dataformat/MessagePackGeneratorTest.java b/msgpack-jackson/src/test/java/org/msgpack/jackson/dataformat/MessagePackGeneratorTest.java
@@ -19,6 +19,7 @@
 import com.fasterxml.jackson.core.JsonGenerator;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import org.junit.Test;
+import org.msgpack.core.ExtendedTypeHeader;
 import org.msgpack.core.MessagePack;
 import org.msgpack.core.MessageUnpacker;
 import org.msgpack.core.buffer.ArrayBufferInput;
@@ -29,8 +30,10 @@
 import java.io.IOException;
 import java.io.OutputStream;
 import java.math.BigDecimal;
+import java.nio.ByteBuffer;
 import java.util.*;
 
+import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
@@ -63,6 +66,10 @@ public void testGeneratorShouldWriteObject() throws IOException {
         childArray.add("child#1");
         childArray.add(1.23f);
         hashMap.put("childArray", childArray);
+        // #10
+        byte[] hello = "hello".getBytes("UTF-8");
+        ByteBuffer buffer = ByteBuffer.wrap(hello);
+        hashMap.put("ext", new MessagePackExtendedType(17, buffer));
 
         long bitmap = 0;
         byte[] bytes = objectMapper.writeValueAsBytes(hashMap);
@@ -134,11 +141,24 @@ else if (key.equals("childArray")) {
                 assertEquals(1.23f, messageUnpacker.unpackFloat(), 0.01f);
                 bitmap |= 0x1 << 9;
             }
+            else if (key.equals("ext")) {
+                // #9
+                ExtendedTypeHeader header = messageUnpacker.unpackExtendedTypeHeader();
+                assertEquals(17, header.getType());
+                assertEquals(5, header.getLength());
+                ByteBuffer payload = ByteBuffer.allocate(header.getLength());
+                payload.flip();
+                payload.limit(payload.capacity());
+                messageUnpacker.readPayload(payload);
+                payload.flip();
+                assertArrayEquals("hello".getBytes(), payload.array());
+                bitmap |= 0x1 << 10;
+            }
             else {
                 assertTrue(false);
             }
         }
-        assertEquals(0x03FF, bitmap);
+        assertEquals(0x07FF, bitmap);
     }
 
     @Test
@@ -270,7 +290,6 @@ public void testBigDecimal() throws IOException {
         }
 
         {
-
             BigDecimal decimal = new BigDecimal("1234.567890123456789012345678901234567890");
             List<BigDecimal> bigDecimals = Arrays.asList(
                     decimal
