org.modelmapper.flattening.example2.FlatteningExample2 Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of modelmapper-examples Show documentation

There is a newer version: 3.2.1

package org.modelmapper.flattening.example2;

import static org.testng.Assert.assertEquals;

import org.modelmapper.ModelMapper;
import org.modelmapper.PropertyMap;
import org.modelmapper.convention.MatchingStrategies;

public class FlatteningExample2 {
  public static void main(String... args) throws Exception {
    Person person = new Person();
    Address address = new Address();
    address.setStreet("1234 Main street");
    address.setCity("San Francisco");
    person.setAddress(address);

    // Option 1
    ModelMapper modelMapper = new ModelMapper();
    PropertyMap personMap = new PropertyMap() {
      protected void configure() {
        map().setStreet(source.getAddress().getStreet());
        map(source.getAddress().city, destination.city);
      }
    };

    modelMapper.addMappings(personMap);
    PersonDTO dto = modelMapper.map(person, PersonDTO.class);

    assertEquals(dto.getStreet(), person.getAddress().getStreet());
    assertEquals(dto.getCity(), person.getAddress().getCity());

    // Option 2
    modelMapper.getConfiguration().setMatchingStrategy(MatchingStrategies.LOOSE);
    dto = modelMapper.map(person, PersonDTO.class);

    assertEquals(dto.getStreet(), person.getAddress().getStreet());
    assertEquals(dto.getCity(), person.getAddress().getCity());
  }
}