package com.nova.sankuai.service.impl;
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.nova.sankuai.domain.entity.Role;
|
import com.nova.sankuai.domain.entity.RoleUser;
|
import com.nova.sankuai.infra.mapper.RoleUserMapper;
|
import com.nova.sankuai.service.IRoleService;
|
import com.nova.sankuai.service.IRoleUserService;
|
import lombok.AllArgsConstructor;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* description
|
* </p>
|
*
|
* @author denglb 2021/11/19
|
*/
|
@Service
|
@AllArgsConstructor
|
public class RoleUserServiceImpl extends ServiceImpl<RoleUserMapper, RoleUser> implements IRoleUserService {
|
|
private RoleUserMapper roleUserMapper;
|
|
private IRoleService roleService;
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void assignRoles(List<Long> roleIds, Long userId) {
|
roleUserMapper.deleteByUserId(userId);
|
if (roleIds.size() > 0) {
|
roleIds = roleIds.stream().distinct().collect(Collectors.toList());
|
roleUserMapper.assignRoles(roleIds, userId);
|
}
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void setDefaultRole(Long userId, Integer type) {
|
Role role = roleService.lambdaQuery().eq(Role::getRoleCode, type).one();
|
List<Long> roleIds = new ArrayList<Long>() {{
|
add(role.getId());
|
}};
|
roleUserMapper.assignRoles(roleIds, userId);
|
}
|
}
|
