support create encoder

taowen · taowen · commit 4defb13fa862 · 2017-01-03T13:56:33.000+08:00
diff --git a/src/main/java/com/jsoniter/output/Codegen.java b/src/main/java/com/jsoniter/output/Codegen.java
@@ -2,6 +2,7 @@
 
 import com.jsoniter.JsonException;
 import com.jsoniter.spi.Encoder;
+import com.jsoniter.spi.Extension;
 import com.jsoniter.spi.JsoniterSpi;
 import com.jsoniter.spi.TypeLiteral;
 
@@ -77,6 +78,14 @@ private static synchronized Encoder gen(String cacheKey, Type type) {
         if (encoder != null) {
             return encoder;
         }
+        List<Extension> extensions = JsoniterSpi.getExtensions();
+        for (Extension extension : extensions) {
+            encoder = extension.createEncoder(cacheKey, type);
+            if (encoder != null) {
+                JsoniterSpi.addNewEncoder(cacheKey, encoder);
+                return encoder;
+            }
+        }
         encoder = CodegenImplNative.NATIVE_ENCODERS.get(type);
         if (encoder != null) {
             JsoniterSpi.addNewEncoder(cacheKey, encoder);
diff --git a/src/main/java/com/jsoniter/spi/EmptyExtension.java b/src/main/java/com/jsoniter/spi/EmptyExtension.java
@@ -24,6 +24,11 @@ public Decoder createDecoder(String cacheKey, Type type) {
         return null;
     }
 
+    @Override
+    public Encoder createEncoder(String cacheKey, Type type) {
+        return null;
+    }
+
     @Override
     public void updateClassDescriptor(ClassDescriptor desc) {
     }
diff --git a/src/main/java/com/jsoniter/spi/Extension.java b/src/main/java/com/jsoniter/spi/Extension.java
@@ -30,16 +30,23 @@ public interface Extension {
     /**
      * Customize type decoding
      *
-     * @param cacheKey cacheKey parameter
+     * @param cacheKey name of the decoder
      * @param type     change how to decode the type
      * @return null, if no special customization needed
      */
     Decoder createDecoder(String cacheKey, Type type);
 
-    // TODO: add createEncoder
+    /**
+     * Customize type encoding
+     *
+     * @param cacheKey name of the encoder
+     * @param type     change how to encode the type
+     * @return null, if not special customization needed
+     */
+    Encoder createEncoder(String cacheKey, Type type);
 
     /**
-     * Update binding is done for the class
+     * Update how binding is done for the class
      *
      * @param desc binding information
      */

Original file line number	Diff line number	Diff line change
`@@ -24,6 +24,11 @@ public Decoder createDecoder(String cacheKey, Type type) {`
`24`	`24`	`return null;`
`25`	`25`	`}`
`26`	`26`
	`27`	`+ @Override`
	`28`	`+ public Encoder createEncoder(String cacheKey, Type type) {`
	`29`	`+ return null;`
	`30`	`+ }`
	`31`	`+`
`27`	`32`	`@Override`
`28`	`33`	`public void updateClassDescriptor(ClassDescriptor desc) {`
`29`	`34`	`}`