Refactoring code in com.baeldung.modelmapper package

This commit is contained in:
Sasa M 2020-04-25 21:57:43 +02:00
parent 32d1cc8a13
commit a5264182cd
5 changed files with 24 additions and 35 deletions

View File

@ -1,39 +1,27 @@
package com.baeldung.modelmapper; package com.baeldung.modelmapper;
import org.modelmapper.ModelMapper; import org.modelmapper.ModelMapper;
import org.modelmapper.convention.MatchingStrategies;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
/** /**
* This is a helper class that contains methods for generic mapping of the users list. * This is a helper class that contains methods for generic mapping of the users list.
* Initially, an instance of ModelMapper was created. In the static block we set the matching configuration to STRICT. * Initially, an instance of ModelMapper was created.
* *
* @author Sasa Milenkovic * @author Sasa Milenkovic
*/ */
public class MapperUtil { public class MapperUtil {
private static ModelMapper modelMapper = new ModelMapper(); private static ModelMapper modelMapper = new ModelMapper();
static {
modelMapper.getConfiguration().setMatchingStrategy(MatchingStrategies.STRICT);
}
private MapperUtil() { private MapperUtil() {
} }
public static <S, T> T mapTo(final S source, final Class<T> target) { public static <S, T> List<T> mapList(List<S> sourceList, Class<T> target) {
return modelMapper.map(source, target);
}
public static <S, T> List<T> mapList(final List<S> sourceList, final Class<T> target) {
List<T> targetList = new ArrayList<T>(); List<T> targetList = new ArrayList<T>();
for (S source : sourceList) { for (S source : sourceList) {

View File

@ -8,7 +8,7 @@ package com.baeldung.modelmapper;
public class User { public class User {
private String userId; private String userId;
private String userName; private String username;
private String email; private String email;
private String contactNumber; private String contactNumber;
private String userType; private String userType;
@ -18,9 +18,9 @@ public class User {
public User() { public User() {
} }
public User(String userId, String userName, String email, String contactNumber, String userType) { public User(String userId, String username, String email, String contactNumber, String userType) {
this.userId = userId; this.userId = userId;
this.userName = userName; this.username = username;
this.email = email; this.email = email;
this.contactNumber = contactNumber; this.contactNumber = contactNumber;
this.userType = userType; this.userType = userType;
@ -34,12 +34,12 @@ public class User {
this.userId = userId; this.userId = userId;
} }
public String getUserName() { public String getUsername() {
return userName; return username;
} }
public void setUserName(String userName) { public void setUsername(String userName) {
this.userName = userName; this.username = userName;
} }
public String getEmail() { public String getEmail() {

View File

@ -8,7 +8,7 @@ package com.baeldung.modelmapper;
public class UserDTO { public class UserDTO {
private String userId; private String userId;
private String userName; private String username;
private String email; private String email;
// getters and setters // getters and setters
@ -21,12 +21,12 @@ public class UserDTO {
this.userId = userId; this.userId = userId;
} }
public String getUserName() { public String getUsername() {
return userName; return username;
} }
public void setUserName(String userName) { public void setUsername(String username) {
this.userName = userName; this.username = username;
} }
public String getEmail() { public String getEmail() {

View File

@ -18,14 +18,15 @@ public class UserPropertyMap extends PropertyMap<UserList, UserListDTO> {
Converter<List<User>, List<String>> converter = new AbstractConverter<List<User>, List<String>>() { Converter<List<User>, List<String>> converter = new AbstractConverter<List<User>, List<String>>() {
List<String> usernames = new ArrayList<>(); protected List<String> usernames;
@Override
protected List<String> convert(List<User> users) { protected List<String> convert(List<User> users) {
users.forEach(user -> usernames.add(user.getUserName())); usernames = new ArrayList<String>();
users.forEach(user -> usernames.add(user.getUsername()));
return usernames; return usernames;
} }
}; };
@Override @Override

View File

@ -25,14 +25,14 @@ import static org.junit.Assert.assertThat;
*/ */
public class UsersListMappingUnitTest { public class UsersListMappingUnitTest {
private ModelMapper mapper; private ModelMapper modelMapper;
private List<User> users; private List<User> users;
@Before @Before
public void init() { public void init() {
mapper = new ModelMapper(); modelMapper = new ModelMapper();
mapper.addMappings(new UserPropertyMap()); modelMapper.addMappings(new UserPropertyMap());
users = new ArrayList(); users = new ArrayList();
users.add(new User("b100", "user1", "user1@baeldung.com", "111-222", "USER")); users.add(new User("b100", "user1", "user1@baeldung.com", "111-222", "USER"));
users.add(new User("b101", "user2", "user2@baeldung.com", "111-333", "USER")); users.add(new User("b101", "user2", "user2@baeldung.com", "111-333", "USER"));
@ -49,7 +49,7 @@ public class UsersListMappingUnitTest {
integers.add(2); integers.add(2);
integers.add(3); integers.add(3);
List<Character> characters = mapper.map(integers, new TypeToken<List<Character>>() { List<Character> characters = modelMapper.map(integers, new TypeToken<List<Character>>() {
}.getType()); }.getType());
assertThat(characters, hasItems('1', '2', '3')); assertThat(characters, hasItems('1', '2', '3'));
@ -66,14 +66,14 @@ public class UsersListMappingUnitTest {
assertThat(userDtoList, Matchers.<UserDTO>hasItem( assertThat(userDtoList, Matchers.<UserDTO>hasItem(
Matchers.both(hasProperty("userId", equalTo("b100"))) Matchers.both(hasProperty("userId", equalTo("b100")))
.and(hasProperty("email", equalTo("user1@baeldung.com"))) .and(hasProperty("email", equalTo("user1@baeldung.com")))
.and(hasProperty("userName", equalTo("user1"))))); .and(hasProperty("username", equalTo("user1")))));
// Mapping lists using PropertyMap and Converter // Mapping lists using PropertyMap and Converter
UserList userList = new UserList(); UserList userList = new UserList();
userList.setUsers(users); userList.setUsers(users);
UserListDTO dto = new UserListDTO(); UserListDTO dto = new UserListDTO();
mapper.map(userList, dto); modelMapper.map(userList, dto);
assertNotNull(dto); assertNotNull(dto);
assertThat(dto, Matchers.hasProperty("usernames")); assertThat(dto, Matchers.hasProperty("usernames"));