MongoDB
 sql >> Base de Dados >  >> NoSQL >> MongoDB

Spring Data Mongo - Como mapear entidades POJO herdadas?


Vou responder minha própria pergunta. Conforme mencionado nos comentários, a solução é usar o Converter .

Aqui está um exemplo do que eu pretendia alcançar com meu modelo de classe:



Um Contact pode ser uma Person ou uma Organisation .

Se você estiver usando spring-data-mongodb MongoRepository para gravar dados em seu banco de dados de acordo com seu modelo de entidade, um _class será adicionado às raízes do documento e aos tipos de propriedade complexos (veja esta seção ). Esses campos armazenam o nome totalmente qualificado da classe Java e permitem a desambiguação ao mapear do Documento MongoDb para o modelo de dados Spring.

Se seu aplicativo acabou de ler o documento do banco de dados (sem _class campos), você precisa informar aos dados do Spring qual classe instanciar ao mapear um Contact . Spring-data permite que você personalize o comportamento de mapeamento de tipo padrão usando Converter . Usando o Converter explícito substituir padrão mapeamento para a classe . você precisa mapear explicitamente toda a sua classe. Aqui está um exemplo do meu 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;
    }
}

Em seguida, os conversores recém-definidos precisam ser registrados:
@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 ajude.