com/wentch/redkale/convert/MapEncoder.java

@@ -0,0 +1,62 @@
+/*
+ * To change this license header, choose License Headers in Project Properties.
+ * To change this template file, choose Tools | Templates
+ * and open the template in the editor.
+ */
+package com.wentch.redkale.convert;
+
+import java.lang.reflect.ParameterizedType;
+import java.lang.reflect.Type;
+import java.util.Map;
+
+/**
+ *
+ * @author zhangjx
+ * @param <K>
+ * @param <V>
+ */
+@SuppressWarnings("unchecked")
+public final class MapEncoder<K, V> implements Encodeable<Writer, Map<K, V>> {
+
+    private final Type type;
+
+    private final Encodeable<Writer, K> keyencoder;
+
+    private final Encodeable<Writer, V> valencoder;
+
+    public MapEncoder(final Factory factory, final Type type) {
+        this.type = type;
+        if (type instanceof ParameterizedType) {
+            final Type[] pt = ((ParameterizedType) type).getActualTypeArguments();
+            this.keyencoder = factory.loadEncoder(pt[0]);
+            this.valencoder = factory.loadEncoder(pt[1]);
+        } else {
+            this.keyencoder = factory.getAnyEncoder();
+            this.valencoder = factory.getAnyEncoder();
+        }
+    }
+
+    @Override
+    public void convertTo(Writer out, Map<K, V> value) {
+        final Map<K, V> values = value;
+        if (values == null) {
+            out.writeNull();
+            return;
+        }
+        out.writeMapB(values.size());
+        boolean first = true;
+        for (Map.Entry<K, V> en : values.entrySet()) {
+            if (!first) out.writeArrayMark();
+            this.keyencoder.convertTo(out, en.getKey());
+            out.writeMapMark();
+            this.valencoder.convertTo(out, en.getValue());
+            if (first) first = false;
+        }
+        out.writeMapE();
+    }
+
+    @Override
+    public Type getType() {
+        return type;
+    }
+}