|
6 | 6 | import com.fasterxml.jackson.databind.type.TypeBindings; |
7 | 7 | import com.fasterxml.jackson.databind.type.TypeFactory; |
8 | 8 | import com.fasterxml.jackson.databind.type.TypeModifier; |
9 | | - |
10 | 9 | import com.google.common.base.Optional; |
11 | 10 | import com.google.common.collect.FluentIterable; |
12 | 11 | import com.google.common.collect.Multimap; |
13 | 12 | import com.google.common.collect.Range; |
| 13 | +import com.google.common.collect.Table; |
14 | 14 |
|
15 | 15 | /** |
16 | 16 | * We need somewhat hacky support for following Guava types: |
@@ -42,18 +42,24 @@ public class GuavaTypeModifier extends TypeModifier |
42 | 42 | public JavaType modifyType(JavaType type, Type jdkType, TypeBindings context, TypeFactory typeFactory) |
43 | 43 | { |
44 | 44 | final Class<?> raw = type.getRawClass(); |
| 45 | + // First: make Multimaps look more Map-like |
45 | 46 | if (Multimap.class.isAssignableFrom(raw)) { |
46 | | - JavaType keyType = type.containedType(0); |
47 | | - JavaType contentType = type.containedType(1); |
48 | | - |
49 | | - if (keyType == null) { |
50 | | - keyType = TypeFactory.unknownType(); |
51 | | - } |
52 | | - if (contentType == null) { |
53 | | - contentType = TypeFactory.unknownType(); |
| 47 | + JavaType[] types = typeFactory.findTypeParameters(type, Multimap.class); |
| 48 | + return typeFactory.constructMapLikeType(raw, _type(types, 0), _type(types, 1)); |
| 49 | + } |
| 50 | + // 14-Sep-2015, tatu: Pre-resolve type parameters for Tables as well |
| 51 | + if (Table.class.isAssignableFrom(raw)) { |
| 52 | + JavaType[] types = typeFactory.findTypeParameters(type, Table.class); |
| 53 | + if (types == null || types.length != 3) { |
| 54 | + types = new JavaType[] { |
| 55 | + _type(types, 0), |
| 56 | + _type(types, 1), |
| 57 | + _type(types, 2) |
| 58 | + }; |
54 | 59 | } |
55 | | - return typeFactory.constructMapLikeType(type.getRawClass(), keyType, contentType); |
| 60 | + return typeFactory.constructParametrizedType(raw, Table.class, types); |
56 | 61 | } |
| 62 | + |
57 | 63 | /* Guava 12 changed the implementation of their {@link FluentIterable} to include a method named "isEmpty." This method |
58 | 64 | * causes Jackson to treat FluentIterables as a Bean instead of an {@link Iterable}. Serialization of FluentIterables by |
59 | 65 | * default result in a string like "{\"empty\":true}." This module modifies the JavaType of FluentIterable to be |
@@ -108,4 +114,11 @@ public JavaType modifyType(JavaType type, Type jdkType, TypeBindings context, Ty |
108 | 114 | } |
109 | 115 | return type; |
110 | 116 | } |
| 117 | + |
| 118 | + private static JavaType _type(JavaType[] types, int index) { |
| 119 | + if (types == null || types.length <= index) { |
| 120 | + return TypeFactory.unknownType(); |
| 121 | + } |
| 122 | + return types[index]; |
| 123 | + } |
111 | 124 | } |
0 commit comments