lihongjie/coal
1
0
Fork 0
mirror of https://codeup.aliyun.com/64f7d6b8ce01efaafef1e678/coal/coal.git synced 2026-01-25 07:46:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 用户相关接口确保要有一个管理员账户

Browse Source
This commit is contained in:
lihongjie0209
2024-11-13 20:44:56 +08:00
parent 6cd141c343
commit 8fcdccc8c0
1 changed files with 29 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/main/java/cn/lihongjie/coal/user/service/UserService.java
View File

@@ -81,8 +81,7 @@ public class UserService extends BaseService<UserEntity, UserRepository> {
@Autowired PasswordDictService passwordDictService;
private Pbkdf2PasswordEncoder passwordEncoder;
@Autowired
UserService that;
@Autowired UserService that;
@PostConstruct
public void init() {
@@ -193,14 +192,41 @@ public class UserService extends BaseService<UserEntity, UserRepository> {
UserEntity user = this.repository.get(request.getId());
this.mapper.updateEntity(user, request);
this.repository.save(user);
String organizationId = user.getOrganizationId();
assertAdmin(organizationId);
clearUserCache(request.getId());
return getById(user.getId());
}
private void assertAdmin(String organizationId) {
long adminCnt =
this.repository.count(
new Specification<UserEntity>() {
@Override
public Predicate toPredicate(
Root<UserEntity> root,
CriteriaQuery<?> query,
CriteriaBuilder criteriaBuilder) {
return criteriaBuilder.and(
criteriaBuilder.equal(
root.get("organizationId"), organizationId),
criteriaBuilder.equal(root.get("orgAdmin"), true));
}
});
if (adminCnt == 0) {
throw new BizException("组织必须有一个管理员");
}
}
public void delete(IdRequest request) {
this.repository.deleteAllById(request.getIds());
request.getIds().forEach(this::clearUserCache);
this.assertAdmin(Ctx.currentUser().getOrganizationId());
}
public UserDto getById(String id) {
@@ -541,7 +567,7 @@ public class UserService extends BaseService<UserEntity, UserRepository> {
stopWatch.start("contains");
boolean contains = ids.contains(resourceId);
stopWatch.stop();
// log.info(stopWatch.prettyPrint());
// log.info(stopWatch.prettyPrint());
return contains;
}
@@ -549,7 +575,5 @@ public class UserService extends BaseService<UserEntity, UserRepository> {
Try.run(() -> cacheManager.getCache(Constants.CACHE_USER_RESOURCES).clear());
Try.run(() -> cacheManager.getCache(Constants.CACHE_USER_PERMISSIONS).clear());
Try.run(() -> cacheManager.getCache(Constants.CACHE_USER_RESOURCE_IDS).clear());
}
}