sql >> Base de Datos >  >> NoSQL >> MongoDB

Spring Data Mongo:¿cómo mapear entidades POJO heredadas?

Responderé a mi propia pregunta. Como se menciona en los comentarios, la solución es usar Converter .

Aquí hay un ejemplo de lo que pretendía lograr con mi modelo de clase:

Un Contact puede ser una Person o una Organisation .

Si está utilizando spring-data-mongodb MongoRepository para escribir datos en su base de datos de acuerdo con su modelo de entidad, una _class El campo se agregará a las raíces del documento y a los tipos de propiedades complejas (ver esta sección ). Estos campos almacenan el nombre completamente calificado de la clase Java y permite la eliminación de ambigüedades al mapear del documento MongoDb al modelo de datos Spring.

Si su aplicación acaba de leer el documento de la base de datos (sin _class campos), debe decirle a los datos de Spring qué clase instanciar al mapear un Contact . Spring-data le permite personalizar el comportamiento de mapeo de tipo predeterminado usando Converter . Usando un Converter explícito anular predeterminado mapeo para la clase . necesita mapear explícitamente toda su clase. Este es un ejemplo de mi ContactReadConverter :

@ReadingConverter
public class ContactReadConverter implements Converter<Document, Contact> {

    @Override
    public Contact convert(Document source) {
        if (source.get("firstName") == null) {
            Organisation organisation = new Organisation();
            I18n name = new I18n();
            name.setEn(source.get("name", Document.class).get("en", String.class));
            name.setFr(source.get("name", Document.class).get("fr", String.class));
            organisation.setName(name);
            organisation.setAcronym(source.get("acronym", String.class));
            organisation.setRole(source.get("role", String.class));
            return organisation;
        }
        Person person = new Person();
        person.setFirstName(source.get("firstName", String.class));
        person.setLastName(source.get("lastName", String.class));
        person.setRole(source.get("role", String.class));
        person.setEmail(source.get("email", String.class));
        person.setOrcId(source.get("orcId", String.class));
        if (source.get("organisation") != null) {
            Document sourceOrg = source.get("organisation", Document.class);
            Organisation organisation = new Organisation();
            organisation.setAcronym(sourceOrg.get("acronym", String.class));
            organisation.setRole(sourceOrg.get("role", String.class));
            if (sourceOrg.get("name") != null) {
                I18n name = new I18n();
                name.setFr(sourceOrg.get("name", Document.class).get("fr", String.class));
                name.setEn(sourceOrg.get("name", Document.class).get("en", String.class));
                organisation.setName(name);
            }
            person.setOrganisation(organisation);
        }
        return person;
    }
}

Luego, los convertidores recién definidos deben registrarse:

@Configuration
public class DataportalApplicationConfig extends AbstractMongoConfiguration {
    @Value("${spring.data.mongodb.uri}")
    private String uri;
    @Value("${spring.data.mongodb.database}")
    private String database;
    @Override
    public MongoClient mongoClient() {
        return new MongoClient(new MongoClientURI(uri));
    }
    @Override
    protected String getDatabaseName() {
        return database;
    }    
    @Bean
    @Override
    public MongoCustomConversions customConversions() {
        List<Converter<?, ?>> converterList = new ArrayList<>();
        converterList.add(new ContactReadConverter());
        return new MongoCustomConversions(converterList);
    }
}

Espero que ayude.