11package restx .jackson ;
22
3- import com .fasterxml .jackson .core .JsonGenerator ;
43import com .fasterxml .jackson .core .JsonParser ;
54import com .fasterxml .jackson .core .JsonProcessingException ;
65import com .fasterxml .jackson .databind .DeserializationContext ;
76import com .fasterxml .jackson .databind .JsonDeserializer ;
8- import com .fasterxml .jackson .databind .JsonSerializer ;
97import com .fasterxml .jackson .databind .SerializerProvider ;
108import com .fasterxml .jackson .databind .module .SimpleModule ;
119import de .undercouch .bson4jackson .BsonGenerator ;
12- import org . joda . time . DateMidnight ;
10+ import de . undercouch . bson4jackson . serializers . BsonSerializer ;
1311import org .joda .time .DateTime ;
1412import org .joda .time .DateTimeZone ;
1513
@@ -25,31 +23,25 @@ public class BsonJodaTimeModule extends SimpleModule {
2523 public BsonJodaTimeModule () {
2624 super ("BsonJodaTimeModule" );
2725
28- addSerializer (org .joda .time .DateMidnight .class , new JsonSerializer < DateMidnight >() {
26+ addSerializer (org .joda .time .DateMidnight .class , new BsonSerializer < org . joda . time . DateMidnight >() {
2927 @ Override
30- public void serialize (org .joda .time .DateMidnight date , JsonGenerator gen , SerializerProvider provider )
28+ public void serialize (org .joda .time .DateMidnight date , BsonGenerator bsonGenerator , SerializerProvider serializerProvider )
3129 throws IOException {
3230 if (date == null ) {
33- provider .defaultSerializeNull (gen );
34- } else if (gen instanceof BsonGenerator ) {
35- BsonGenerator bgen = (BsonGenerator )gen ;
36- bgen .writeDateTime (date .toDate ());
31+ serializerProvider .defaultSerializeNull (bsonGenerator );
3732 } else {
38- gen . writeNumber (date .getMillis ());
33+ bsonGenerator . writeDateTime (date .toDate ());
3934 }
4035 }
4136 });
42- addSerializer (DateTime .class , new JsonSerializer <DateTime >() {
37+ addSerializer (DateTime .class , new BsonSerializer <DateTime >() {
4338 @ Override
44- public void serialize (DateTime date , JsonGenerator gen , SerializerProvider provider )
39+ public void serialize (DateTime date , BsonGenerator bsonGenerator , SerializerProvider serializerProvider )
4540 throws IOException {
4641 if (date == null ) {
47- provider .defaultSerializeNull (gen );
48- } else if (gen instanceof BsonGenerator ) {
49- BsonGenerator bgen = (BsonGenerator )gen ;
50- bgen .writeDateTime (date .toDate ());
42+ serializerProvider .defaultSerializeNull (bsonGenerator );
5143 } else {
52- gen . writeNumber (date .getMillis ());
44+ bsonGenerator . writeDateTime (date .toDate ());
5345 }
5446 }
5547 });
0 commit comments