Гибернация, двунаправленное обновление

Я пытаюсь обновить записи пользователей с несколькими ролями в 2 таблицах (пользователь и пользовательская роль) с помощью спящего режима. Когда я делаю сохранение, оно работает должным образом, однако, когда я пытаюсь обновить, если объект содержит роль, которая сохранена в базе данных, это терпит неудачу. Я не нашел хорошего ответа, но, полагаю, это невозможно сделать напрямую с помощью метода "update", двунаправленность должна быть реализована вручную при обновлении, не так ли? Я следовал за структурой Spring Security для пользователей.

Это мой класс пользователя:

@Entity
@Table(name = "users", schema = "cmsdb")
public class User {

private String username;
private String password;
private boolean enabled;

private Set<UserRole> userRole = new HashSet<UserRole>(0);

public User() {
}

public User(String username, String password, boolean enabled) {
    this.username = username;
    this.password = password;
    this.enabled = enabled;
}

public User(String username, String password, 
    boolean enabled, Set<UserRole> userRole) {
    this.username = username;
    this.password = password;
    this.enabled = enabled;
    this.userRole = userRole;
}

@Id
@Column(name = "username", unique = true, nullable = false, length = 45)
public String getUsername() {
    return username;
}

public void setUsername(String username) {
    this.username = username;
}

@Column(name = "password", nullable = false, length = 60)
public String getPassword() {
    return password;
}

public void setPassword(String password) {
    this.password = password;
}

@Column(name = "enabled", nullable = false)
public boolean isEnabled() {
    return enabled;
}

public void setEnabled(boolean enabled) {
    this.enabled = enabled;
}

@OneToMany(fetch = FetchType.EAGER, mappedBy = "user")  //,cascade = {CascadeType.ALL}
@Cascade({CascadeType.SAVE_UPDATE,CascadeType.DELETE})
@OnDelete(action = OnDeleteAction.CASCADE)
public Set<UserRole> getUserRole() {
    return userRole;
}

public void setUserRole(Set<UserRole> userRole) {
    this.userRole = userRole;
}
}

Это мой класс UserRole:

@Entity
@Table(name = "user_roles", schema = "cmsdb", 
uniqueConstraints = @UniqueConstraint(
    columnNames = { "role", "username" }))
public class UserRole {

private Integer userRoleId;
private User user;
@ValidRole
private String role;

public UserRole () {

}

public UserRole(User user, String role) {
    this.user = user;
    this.role = role;
}

@Id
@GeneratedValue(strategy = IDENTITY)
@Column(name = "user_role_id", 
    unique = true, nullable = false)
public Integer getUserRoleId() {
    return userRoleId;
}

public void setUserRoleId(Integer userRoleId) {
    this.userRoleId = userRoleId;
}

@ManyToOne(fetch = FetchType.EAGER)
@JoinColumn(name = "username", nullable = false)
public User getUser() {
    return user;
}

public void setUser(User user) {
    this.user = user;
}

@Column(name = "role", nullable = false, length = 45)
public String getRole() {
    return role;
}

public void setRole(String role) {
    this.role = role;
}

public String toString () {
    return role;
}

@Override
public int hashCode() {
    return userRoleId;
}

@Override
public boolean equals(Object obj) {
    UserRole role = (UserRole) obj;
    return (userRoleId == role.getUserRoleId());
}
}

Это мои методы DAO:

// IT WORKS PROPERLY
@Override
@Transactional
public boolean createUser(User user) {
    String hashedPassword = passwordEncoder.encode(user.getPassword());
    user.setPassword(hashedPassword);

    Set<UserRole> roles = new HashSet<UserRole> ();
    for(UserRole role: user.getUserRole()) {
        roles.add(new UserRole(user,role.getRole()));   
    }

    user.setUserRole(roles);
    sessionFactory.getCurrentSession().save(user);
    return true;
}

// It doesn't work if it tries to update an User with an UserRole stored in the UserRole table in the database
@Override
@Transactional
public boolean updateUser(User user) {

// Get user by id
    User userpersis = (User) getSessionFactory().getCurrentSession().load(User.class, user.getUsername());

  // Primary key shouldn't be modified
  userpersis.setEnabled(user.isEnabled());
  userpersis.getUserRole().retainAll( user.getUserRole() );
  userpersis.getUserRole().addAll( user.getUserRole() );
  sessionFactory.getCurrentSession().update(userpersis);
  return true;
}

Ошибка:

WARN : org.hibernate.engine.jdbc.spi.SqlExceptionHelper - SQL Error: 1062, SQLState: 23000
ERROR: org.hibernate.engine.jdbc.spi.SqlExceptionHelper - Duplicate entry  'ROLE_MAINTENANCE-asdf' for key 'uni_username_role'
WARN : org.hibernate.engine.jdbc.spi.SqlExceptionHelper - SQL Warning Code: 1062, SQLState: 23000 ...

Заранее спасибо.

1 ответ

Решение

В User, добавлять orphanRemoval на каскад:

@Cascade(cascade={CascadeType.ALL}, orphanRemoval=true)
@OnDelete(action = OnDeleteAction.CASCADE)
public Set<UserRole> getUserRole() { ...

(см., например, JPA 2.0 orphanRemoval=true VS при удалении каскада)

Тогда вместо

Set<UserRole> roles = new HashSet<UserRole> ();
// Guess for roles, should be needed to load them also from the db
for(UserRole role : user.getUserRole()) {
    roles.add(new UserRole(userpersis,role.getRole())); 
}

userpersis.setUserRole(roles);

пытаться

userpersis.getUserRole().clear(); // Let Hibernate know that we're really removing all the references

for(UserRole role : user.getUserRole()) {
    userpersis.getUserRole().add(new UserRole(userpersis,role.getRole())); 
}

Хотя на самом деле вы, вероятно, даже не хотите удалять все пользовательские роли, а затем вставлять все новые данные. Вы также можете сделать что-то вроде

userpersis.getUserRole().retainAll( user.getUserRole() );
userpersis.getUserRole().addAll( user.getUserRole() );

Это, однако, требует, чтобы вы реализовали equals() а также hashCode() в UserRole,

Изменить: Конечно, не работает, когда, как в вашем случае, userролевые объекты не могут быть такими же, как userpersisиз-за обратной ссылки на владеющий объект.

Другие вопросы по тегам