有没有办法自动创建Mongo编解码器?

时间:2015-09-05 20:12:05

标签: java mongodb asynchronous codec

我愿意将我的代码从mongojack迁移到支持新异步mongo驱动程序的代码。然而,我发现新的编码/解码方式是通过Codec,我不会看到自己为模型中的每个类编写Codec。这就是为什么我宁愿写一个给定类的库创建一个Codec。但是,我不知道如何,也不知道是否已经有一些努力试图实现同样的目标。 有些库可以实现我想要的吗?如果没有,那么实现它的最佳方法是什么。

(我知道我应该在那里的某处使用CodecProvider,但我仍然不知道从哪里开始)

2 个答案:

答案 0 :(得分:3)

是的,如果您使用杰克逊,您可以使用https://github.com/ylemoigne/mongo-jackson-codec中的mongo-jackson-codec,它将自动为您处理。

答案 1 :(得分:0)

以下是我们如何解决这个问题(最终结果是Lombok,Jackson和MongoDB之间的超级流畅):

提供者:

public class JacksonCodecProvider implements CodecProvider {
    private final ObjectMapper objectMapper;

    public JacksonCodecProvider(final ObjectMapper bsonObjectMapper) {
        this.objectMapper = bsonObjectMapper;
    }

    @Override
    public <T> Codec<T> get(final Class<T> type, final CodecRegistry registry) {

            return new JacksonCodec<>(objectMapper, registry, type);

    }
}

编解码器本身:

class JacksonCodec<T> implements Codec<T> {
    private final ObjectMapper objectMapper;
    private final Codec<RawBsonDocument> rawBsonDocumentCodec;
    private final Class<T> type;

    public JacksonCodec(ObjectMapper objectMapper,
                        CodecRegistry codecRegistry,
                        Class<T> type) {
        this.objectMapper = objectMapper;
        this.rawBsonDocumentCodec = codecRegistry.get(RawBsonDocument.class);
        this.type = type;
    }

    @Override
    public T decode(BsonReader reader, DecoderContext decoderContext) {
        try {

            RawBsonDocument document = rawBsonDocumentCodec.decode(reader, decoderContext);
            String json = document.toJson();
            return objectMapper.readValue(json, type);
        } catch (IOException e) {
            throw new UncheckedIOException(e);
        }
    }

    @Override
    public void encode(BsonWriter writer, Object value, EncoderContext encoderContext) {
        try {

            String json = objectMapper.writeValueAsString(value);

            rawBsonDocumentCodec.encode(writer, RawBsonDocument.parse(json), encoderContext);

        } catch (IOException e) {
            throw new UncheckedIOException(e);
        }
    }

    @Override
    public Class<T> getEncoderClass() {
        return this.type;
    }
}

当与Lombok和最新的Jackson注释结合使用时,它允许我们做这样的事情(几乎看起来不像Java代码,嗯?):

@JsonIgnoreProperties(ignoreUnknown=true)
@JsonDeserialize(builder = Account.AccountBuilder.class)
@Builder(toBuilder=true)
@Value
public class Account {

    @JsonProperty private String _id;
    @JsonProperty private long _version;
    @JsonProperty private String organizationName;

    @JsonPOJOBuilder(withPrefix = "")
    public static final class AccountBuilder {
    }

}

然后:

Account account = collection.find(eq("_id", id)).first();
System.out.println(account.getOrganizationName());