ModelMapperのcustom mappingについて - エキサイト TechBlog.

エキサイト株式会社　メディアサービスエンジニアの中尾です。前回に続いてModelMapperのちょっとした内容を説明します。現場ではほとんど使うことはないと思うので、「こんなこともできるんだ」ぐらいでお願いします。

コードは以下になります。

@ExtendWith(MockitoExtension.class)
public class ModelMapperTest {

    @Test
    public void testModelMapper() {
        final ModelMapper modelMapper = new ModelMapper();
        modelMapper.addMappings(new PropertyMap<TestModel, Model>() {
            @Override
            protected void configure() {
                using(toUserId).map(source, destination.getUserId());
                using(toFirstname).map(source, destination.getFirstname());
                using(toLastname).map(source, destination.getLastname());
            }
        });

        final TestModel testModel = new TestModel();
        testModel.setUserId("10000");
        testModel.setUserName("naka sho");

        final Model map = modelMapper.map(testModel, Model.class);
        Assertions.assertEquals(
                10000,
                map.getUserId()
        );
        Assertions.assertEquals(
                "naka",
                map.getFirstname()
        );
        Assertions.assertEquals(
                "sho",
                map.getLastname()
        );
    }

    private Converter<TestModel, Integer> toUserId = context ->
            Integer.valueOf(context.getSource().getUserId());

    private Converter<TestModel, String> toFirstname = context ->
    {
        final String[] s = context.getSource().getUserName().split(" ");
        return s[0];
    };

    private Converter<TestModel, String> toLastname = context ->
    {
        final String[] s = context.getSource().getUserName().split(" ");
        return s[1];
    };

    @Data
    public static class Model {
        private int userId;
        private String firstname;
        private String lastname;
    }

    @Data
    public static class TestModel {
        private String userId;
        private String userName;
    }
}