Spring Boot用户管理应用更新时角色嵌套问题及代码修改咨询

问题描述 #

开发Spring Boot用户管理应用时，每次更新用户都会出现角色嵌套的情况，示例如下：

roles=[Role{name='Role{name='ROLE_USER'}'}]
roles=[Role{name='Role{name='Role{name='ROLE_USER'}'}'}]

需要修改代码让角色显示为正常的Role对象。

问题根源 #

1. 多对多级联策略不当：User类中@ManyToMany使用了CascadeType.ALL，导致更新用户时会级联保存角色。如果传入的Role对象name字段被错误设置为旧Role的toString结果，会不断在数据库中创建嵌套名称的角色。
2. 更新时未复用数据库现有角色：更新用户时直接使用传入的新Role实例（而非从数据库查询已存在的角色），导致新角色的name被错误赋值为旧角色的字符串表示。

具体修改步骤 #

1. 调整User类的多对多级联策略 #

修改User类中roles字段的注解，移除CascadeType.ALL（若需支持创建新用户时级联保存角色，可改为CascadeType.PERSIST）：

@Entity
@Table(name="user")
public class User {
    // ... 其他属性不变

    @ManyToMany(fetch = FetchType.LAZY) // 移除CascadeType.ALL
    @JoinTable(name = "users_roles",
            joinColumns = @JoinColumn(name = "user_id"),
            inverseJoinColumns = @JoinColumn(name = "role_id"))
    private Collection<Role> roles;

    // ... 其他方法不变
}

2. 新增角色查询DAO，支持从数据库获取角色 #

RoleDAO接口

public interface RoleDAO {
    Role findByName(String name);
    void save(Role role);
}

RoleDAO实现类

@Repository
public class RoleDAOImplementation implements RoleDAO {

    @Autowired
    private EntityManager entityManager;

    @Override
    public Role findByName(String name) {
        Session currentSession = entityManager.unwrap(Session.class);
        Query<Role> query = currentSession.createQuery("from Role where name=:roleName", Role.class);
        query.setParameter("roleName", name);
        try {
            return query.getSingleResult();
        } catch (NoResultException e) {
            return null;
        }
    }

    @Override
    public void save(Role role) {
        Session currentSession = entityManager.unwrap(Session.class);
        currentSession.saveOrUpdate(role);
    }
}

3. 修改用户更新逻辑，关联数据库中正确的角色实例 #

在Service层的用户更新方法中，替换传入的角色实例为数据库中已存在的角色：

@Service
public class UserService {

    @Autowired
    private UserDAO userDAO;
    @Autowired
    private RoleDAO roleDAO;

    public void updateUser(User updatedUser) {
        User existingUser = userDAO.findByUserName(updatedUser.getUsername());
        if (existingUser == null) {
            throw new RuntimeException("用户不存在");
        }

        // 复制用户基本信息
        existingUser.setFirstName(updatedUser.getFirstName());
        existingUser.setLastName(updatedUser.getLastName());
        existingUser.setEmail(updatedUser.getEmail());
        existingUser.setEnabled(updatedUser.isEnabled());
        existingUser.setPassword(updatedUser.getPassword());

        // 处理角色：查询数据库中已存在的角色
        Collection<Role> validRoles = new HashSet<>();
        for (Role role : updatedUser.getRoles()) {
            // 确保传入的role.getName()是纯角色名，如"ROLE_USER"
            Role dbRole = roleDAO.findByName(role.getName());
            if (dbRole != null) {
                validRoles.add(dbRole);
            }
            // 如需支持创建新角色，可添加以下逻辑
            // else {
            //     dbRole = new Role(role.getName());
            //     roleDAO.save(dbRole);
            //     validRoles.add(dbRole);
            // }
        }
        existingUser.setRoles(validRoles);

        userDAO.save(existingUser);
    }
}

4. 修复前端传参问题（可选） #

如果前端传递的角色名称是嵌套字符串（如"Role{name='ROLE_USER'}"），需要在后端提取真实角色名：

// 在处理角色时添加解析逻辑
String rawName = role.getName();
// 提取引号中的内容
String realRoleName = rawName.substring(rawName.indexOf("'") + 1, rawName.lastIndexOf("'"));
// 处理多层嵌套情况
while (realRoleName.startsWith("Role{")) {
    realRoleName = realRoleName.substring(realRoleName.indexOf("'") + 1, realRoleName.lastIndexOf("'"));
}
Role dbRole = roleDAO.findByName(realRoleName);

说明 #

• 移除CascadeType.ALL：避免错误角色被级联保存到数据库，系统角色通常为预定义，无需随用户更新修改。
• 复用数据库角色：确保用户关联的是数据库中正确的角色记录，从根源杜绝嵌套问题。
• 前端参数校验：确保传递的角色名称是纯标识，避免传入toString格式的字符串。

内容的提问来源于stack exchange，提问作者Palfi Robert