MapStruct处理Java中实体与模型间不匹配属性转换-蒲公英云

MapStruct处理Java中实体与模型间不匹配属性转换

摘要： 前面介绍了MapStrut简单用法，MapStrut的最重要的特点就是处理Java中实体与模型间不匹配属性的转换。

实体模型

有一个User对象：

public class User { 
    private Integer id;
    private String name;
    private double account;
    private boolean married;
// setters, getters, toString()
}

有一个Employee 对象：

public class Employee { 
    private int id;
    private String ename;
    private String position;
    private String married;
// setters, getters, toString()
}

业务场景

需要User 与Employee 对象之间转换。
User 的name属性对应Employee 的ename属性，其取值相同，类型相同，名称不同
User 的married属性(取值true和false)对应Employee 的married属性(取值Y和N)，其取值不同，类型不同，名称相同。

分析与实现

最愚蠢的方式是自己写一堆的setter方法与getter方法，大量get/set代码堆积，增加了代码长度和阅读代码的难度。利用工具BeanUtils是可以处理第一个需求的，但第三种需求就无能为力了。这时MapStrut就派上用场了，最简单的配置可以像下面这样：

@Mapper
public interface UserMapper { 
    UserMapper INSTANCE = Mappers.getMapper(UserMapper.class);
    Employee userToEmployee(User user);
    User employeeToUser(Employee employee);
}

对于第二个需求，可以通过下面方式实现，注解@Mapping可以指定需要把哪个字段source转换为哪个字段target。

@Mapper
public interface UserMapper { 
    UserMapper INSTANCE = Mappers.getMapper(UserMapper.class);
    @Mappings({
        @Mapping(source="name", target="ename")
    })
    Employee userToEmployee(User user);
    @Mappings({
        @Mapping(source="ename", target="name")
    })
    User employeeToUser(Employee employee);
}

第三个需求有点变态，但是真实发生在我们的项目中，实现起来确实繁琐一些：
首先，自定义转化逻辑，布尔值到字符串，布尔的true对应字符串的Y，布尔的false对应字符串的N：

public class UserTransform { 
    public String booleanToString(boolean value){
        if(value){
            return "Y";
        }
        return "N";
    }
    public boolean strToBoolean(String str){
        if ("Y".equals(str)) {
            return true;
        }
        return false;
    }
}

使用很简单，在接口的注解Mapper添加uses参数，值就是需要刚才的转换逻辑类。

@Mapper(uses = UserTransform.class)
public interface UserMapper { ...}

结果与分析

用Junit Test写两个测试方法，分别测试User 对象转换Employee ，Employee 对象转换User。

public class MidTest { 
    @Test
    public void midTest(){
        User user = new User();
        user.setId(125);
        user.setName("Lee");
        user.setMarried(true);
        Employee e = UserMapper.INSTANCE.userToEmployee(user);
        System.out.println(e);
    }
    @Test
    public void midTest2(){
        Employee e = new Employee();
        e.setId(222);
        e.setEname("Chao");
        e.setMarried("N");
        User u = UserMapper.INSTANCE.employeeToUser(e);
        System.out.println(u);
    }
}

结果如下：
User [id=222, name=Chao, account=0.0, married=false]
Employee [id=125, ename=Lee, position=null, married=Y]

转换结果符合预期，转化期间不存在的属性，有了默认值(account和position)，包装类也能识别(int和Integer)，从自动生成的UserMapperImpl.java中，可以看到，
employee.setMarried( userTransform.booleanToString( user.isMarried() ) );，用到了刚才自定义的转换逻辑。第三种需求是很少的，但是遇到了也是很难解决的，MapStruct的自定义函数确实方便不少，不过与其他的转换工具相比，上手难度确实大，配置也稍显繁琐。

项目代码托管在码云。