1
0

refactor(web): 重构列表查询逻辑

- 修改 listPredicate 方法返回类型为 ImmutableList<Predicate>
-优化非管理员用户的数据查询条件- 统一使用 Eclipse Collections 的 Lists 工厂方法
-调整代码结构,提高可读性和可维护性
This commit is contained in:
2024-12-10 18:51:01 +08:00
parent 3d1970ec09
commit 6e4baed23f
7 changed files with 79 additions and 56 deletions

View File

@@ -22,7 +22,9 @@ import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root; import javax.persistence.criteria.Root;
import javax.transaction.Transactional; import javax.transaction.Transactional;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.eclipse.collections.api.factory.Lists;
import org.eclipse.collections.api.factory.Maps; import org.eclipse.collections.api.factory.Maps;
import org.eclipse.collections.api.list.ImmutableList;
import org.eclipse.collections.api.map.ImmutableMap; import org.eclipse.collections.api.map.ImmutableMap;
import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder; import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
@@ -48,12 +50,12 @@ public class AuthenticationService extends LogicDeleteService<Authentication> im
} }
@Override @Override
protected Predicate listPredicate(Root<Authentication> root, CriteriaQuery<?> query, CriteriaBuilder builder) { protected ImmutableList<Predicate> listPredicate(Root<Authentication> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
if (ignoreWhenAdministrator()) {
return null;
}
User loginUser = userService.currentLoginUser(); User loginUser = userService.currentLoginUser();
return builder.or( if (User.isAdministrator(loginUser)) {
return Lists.immutable.empty();
}
return Lists.immutable.of(builder.or(
builder.equal(root.get(Authentication_.createdUser), loginUser), builder.equal(root.get(Authentication_.createdUser), loginUser),
builder.and( builder.and(
builder.equal(root.get(Authentication_.order).get(CheckOrder_.target), CheckOrder.Target.ROLE), builder.equal(root.get(Authentication_.order).get(CheckOrder_.target), CheckOrder.Target.ROLE),
@@ -63,7 +65,7 @@ public class AuthenticationService extends LogicDeleteService<Authentication> im
builder.equal(root.get(Authentication_.order).get(CheckOrder_.target), CheckOrder.Target.USER), builder.equal(root.get(Authentication_.order).get(CheckOrder_.target), CheckOrder.Target.USER),
builder.equal(root.get(Authentication_.order).get(CheckOrder_.targetUser), loginUser) builder.equal(root.get(Authentication_.order).get(CheckOrder_.targetUser), loginUser)
) )
); ));
} }
@Override @Override

View File

@@ -13,6 +13,7 @@ import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root; import javax.persistence.criteria.Root;
import javax.transaction.Transactional; import javax.transaction.Transactional;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.eclipse.collections.api.list.ImmutableList;
/** /**
* @author lanyuanxiaoyao * @author lanyuanxiaoyao
@@ -28,8 +29,8 @@ public abstract class LogicDeleteService<ENTITY extends LogicDeleteEntity> exten
} }
@Override @Override
protected Predicate listPredicate(Root<ENTITY> root, CriteriaQuery<?> query, CriteriaBuilder builder) { protected ImmutableList<Predicate> listPredicate(Root<ENTITY> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
return builder.equal(root.get("deleted"), false); return super.listPredicate(root, query, builder).newWith(builder.equal(root.get("deleted"), false));
} }
@Transactional(rollbackOn = Throwable.class) @Transactional(rollbackOn = Throwable.class)

View File

@@ -6,7 +6,9 @@ import cn.hutool.core.util.EnumUtil;
import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import com.eshore.gringotts.web.domain.base.controller.query.Query; import com.eshore.gringotts.web.domain.base.controller.query.Query;
import com.eshore.gringotts.web.domain.base.entity.IdOnlyEntity_;
import com.eshore.gringotts.web.domain.base.entity.SimpleEntity; import com.eshore.gringotts.web.domain.base.entity.SimpleEntity;
import com.eshore.gringotts.web.domain.base.entity.SimpleEntity_;
import com.eshore.gringotts.web.domain.base.repository.SimpleRepository; import com.eshore.gringotts.web.domain.base.repository.SimpleRepository;
import com.eshore.gringotts.web.domain.user.entity.User; import com.eshore.gringotts.web.domain.user.entity.User;
import com.eshore.gringotts.web.domain.user.service.UserService; import com.eshore.gringotts.web.domain.user.service.UserService;
@@ -48,7 +50,13 @@ public abstract class SimpleServiceSupport<ENTITY extends SimpleEntity> implemen
entity, entity,
targetEntity, targetEntity,
CopyOptions.create() CopyOptions.create()
.setIgnoreProperties("id", "createdUser", "createdTime", "modifiedUser", "modifiedTime") .setIgnoreProperties(
IdOnlyEntity_.ID,
SimpleEntity_.CREATED_USER,
SimpleEntity_.CREATED_TIME,
SimpleEntity_.MODIFIED_USER,
SimpleEntity_.MODIFIED_TIME
)
); );
entity = targetEntity; entity = targetEntity;
} }
@@ -58,16 +66,24 @@ public abstract class SimpleServiceSupport<ENTITY extends SimpleEntity> implemen
@Override @Override
public ImmutableList<ENTITY> list() throws Exception { public ImmutableList<ENTITY> list() throws Exception {
return Lists.immutable.ofAll(repository.findAll(this::listPredicate)); return Lists.immutable.ofAll(repository.findAll(
(root, query, builder) ->
builder.and(
listPredicate(root, query, builder)
.reject(ObjectUtil::isNull)
.toArray(new Predicate[]{})
)
));
} }
@Override @Override
public ImmutableList<ENTITY> list(ImmutableSet<Long> ids) throws Exception { public ImmutableList<ENTITY> list(ImmutableSet<Long> ids) throws Exception {
return Lists.immutable.ofAll(repository.findAll( return Lists.immutable.ofAll(repository.findAll(
(root, query, builder) -> builder.and( (root, query, builder) -> {
listPredicate(root, query, builder), MutableList<Predicate> predicates = Lists.mutable.ofAll(listPredicate(root, query, builder));
builder.in(root.get("id")).value(ids) predicates.add(builder.in(root.get("id")).value(ids));
) return builder.and(predicates.reject(ObjectUtil::isNull).toArray(new Predicate[predicates.size()]));
}
)); ));
} }
@@ -152,27 +168,23 @@ public abstract class SimpleServiceSupport<ENTITY extends SimpleEntity> implemen
return predicates.toImmutable(); return predicates.toImmutable();
} }
protected Boolean ignoreWhenAdministrator() { protected ImmutableList<Predicate> listPredicate(Root<ENTITY> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
return User.isAdministrator(userService.currentLoginUser());
}
protected Predicate listPredicate(Root<ENTITY> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
User user = userService.currentLoginUser(); User user = userService.currentLoginUser();
if (User.isNotAdministrator(user)) { if (User.isNotAdministrator(user)) {
return builder.equal(root.get("createdUser"), user); return Lists.immutable.of(builder.equal(root.get(SimpleEntity_.createdUser), user));
} }
return null; return Lists.immutable.empty();
} }
@Override @Override
public ImmutableList<ENTITY> list(Query listQuery) throws Exception { public ImmutableList<ENTITY> list(Query listQuery) throws Exception {
return Lists.immutable.ofAll(repository.findAll( return Lists.immutable.ofAll(repository.findAll(
(root, query, builder) -> { (root, query, builder) -> {
MutableList<Predicate> predicates = Lists.mutable.of(listPredicate(root, query, builder)); MutableList<Predicate> predicates = Lists.mutable.ofAll(listPredicate(root, query, builder));
predicates.addAllIterable(queryPredicates(listQuery.getQuery(), root, query, builder)); predicates.addAllIterable(queryPredicates(listQuery.getQuery(), root, query, builder));
return builder.and(predicates.toArray(new Predicate[predicates.size()])); return builder.and(predicates.reject(ObjectUtil::isNull).toArray(new Predicate[predicates.size()]));
}, },
Sort.by("createdTime").descending() Sort.by(SimpleEntity_.CREATED_TIME).descending()
)); ));
} }
@@ -182,10 +194,11 @@ public abstract class SimpleServiceSupport<ENTITY extends SimpleEntity> implemen
return Optional.empty(); return Optional.empty();
} }
return repository.findOne( return repository.findOne(
(root, query, builder) -> builder.and( (root, query, builder) -> {
listPredicate(root, query, builder), MutableList<Predicate> predicates = Lists.mutable.ofAll(listPredicate(root, query, builder));
builder.equal(root.get("id"), id) predicates.add(builder.equal(root.get(IdOnlyEntity_.id), id));
) return builder.and(predicates.reject(ObjectUtil::isNull).toArray(new Predicate[predicates.size()]));
}
); );
} }

View File

@@ -5,6 +5,7 @@ import cn.hutool.core.util.StrUtil;
import com.eshore.gringotts.web.domain.base.service.CheckingService; import com.eshore.gringotts.web.domain.base.service.CheckingService;
import com.eshore.gringotts.web.domain.base.service.SimpleServiceSupport; import com.eshore.gringotts.web.domain.base.service.SimpleServiceSupport;
import com.eshore.gringotts.web.domain.check.entity.CheckOrder; import com.eshore.gringotts.web.domain.check.entity.CheckOrder;
import com.eshore.gringotts.web.domain.check.entity.CheckOrder_;
import com.eshore.gringotts.web.domain.check.repository.CheckOrderRepository; import com.eshore.gringotts.web.domain.check.repository.CheckOrderRepository;
import com.eshore.gringotts.web.domain.user.entity.User; import com.eshore.gringotts.web.domain.user.entity.User;
import com.eshore.gringotts.web.domain.user.service.UserService; import com.eshore.gringotts.web.domain.user.service.UserService;
@@ -16,6 +17,8 @@ import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate; import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root; import javax.persistence.criteria.Root;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.eclipse.collections.api.factory.Lists;
import org.eclipse.collections.api.list.ImmutableList;
import org.eclipse.collections.api.map.ImmutableMap; import org.eclipse.collections.api.map.ImmutableMap;
import org.springframework.context.ApplicationContext; import org.springframework.context.ApplicationContext;
import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder; import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder;
@@ -42,22 +45,22 @@ public class CheckOrderService extends SimpleServiceSupport<CheckOrder> {
} }
@Override @Override
protected Predicate listPredicate(Root<CheckOrder> root, CriteriaQuery<?> query, CriteriaBuilder builder) { protected ImmutableList<Predicate> listPredicate(Root<CheckOrder> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
User user = userService.currentLoginUser(); User user = userService.currentLoginUser();
if (ObjectUtil.equals(user.getRole(), User.Role.ADMINISTRATOR)) { if (User.isAdministrator(user)) {
return null; return Lists.immutable.empty();
} }
return builder.or( return Lists.immutable.of(builder.or(
builder.equal(root.get("createdUser"), user), builder.equal(root.get(CheckOrder_.createdUser), user),
builder.and( builder.and(
builder.equal(root.get("target"), CheckOrder.Target.USER), builder.equal(root.get(CheckOrder_.target), CheckOrder.Target.USER),
builder.equal(root.get("targetUser"), user) builder.equal(root.get(CheckOrder_.targetUser), user)
), ),
builder.and( builder.and(
builder.equal(root.get("target"), CheckOrder.Target.ROLE), builder.equal(root.get(CheckOrder_.target), CheckOrder.Target.ROLE),
builder.equal(root.get("targetRole"), user.getRole()) builder.equal(root.get(CheckOrder_.targetRole), user.getRole())
) )
); ));
} }
public void operation(Long id, CheckOrder.Operation operation) throws JsonProcessingException { public void operation(Long id, CheckOrder.Operation operation) throws JsonProcessingException {

View File

@@ -21,7 +21,9 @@ import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root; import javax.persistence.criteria.Root;
import javax.transaction.Transactional; import javax.transaction.Transactional;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.eclipse.collections.api.factory.Lists;
import org.eclipse.collections.api.factory.Maps; import org.eclipse.collections.api.factory.Maps;
import org.eclipse.collections.api.list.ImmutableList;
import org.eclipse.collections.api.map.ImmutableMap; import org.eclipse.collections.api.map.ImmutableMap;
import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder; import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
@@ -45,12 +47,12 @@ public class ConfirmationService extends SimpleServiceSupport<Confirmation> impl
} }
@Override @Override
protected Predicate listPredicate(Root<Confirmation> root, CriteriaQuery<?> query, CriteriaBuilder builder) { protected ImmutableList<Predicate> listPredicate(Root<Confirmation> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
if (ignoreWhenAdministrator()) {
return null;
}
User loginUser = userService.currentLoginUser(); User loginUser = userService.currentLoginUser();
return builder.or( if (User.isAdministrator(loginUser)) {
return Lists.immutable.empty();
}
return Lists.immutable.of(builder.or(
builder.equal(root.get(Confirmation_.createdUser), loginUser), builder.equal(root.get(Confirmation_.createdUser), loginUser),
builder.and( builder.and(
builder.equal(root.get(Confirmation_.order).get(CheckOrder_.target), CheckOrder.Target.ROLE), builder.equal(root.get(Confirmation_.order).get(CheckOrder_.target), CheckOrder.Target.ROLE),
@@ -60,7 +62,7 @@ public class ConfirmationService extends SimpleServiceSupport<Confirmation> impl
builder.equal(root.get(Confirmation_.order).get(CheckOrder_.target), CheckOrder.Target.USER), builder.equal(root.get(Confirmation_.order).get(CheckOrder_.target), CheckOrder.Target.USER),
builder.equal(root.get(Confirmation_.order).get(CheckOrder_.targetUser), loginUser) builder.equal(root.get(Confirmation_.order).get(CheckOrder_.targetUser), loginUser)
) )
); ));
} }
@Override @Override

View File

@@ -49,11 +49,11 @@ public class DataResourceService extends SimpleServiceSupport<DataResource> {
} }
@Override @Override
protected Predicate listPredicate(Root<DataResource> root, CriteriaQuery<?> query, CriteriaBuilder builder) { protected ImmutableList<Predicate> listPredicate(Root<DataResource> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
if (ignoreWhenAdministrator()) {
return null;
}
User loginUser = userService.currentLoginUser(); User loginUser = userService.currentLoginUser();
if (User.isAdministrator(loginUser)) {
return Lists.immutable.empty();
}
Subquery<Confirmation> confirmationSubquery = query.subquery(Confirmation.class); Subquery<Confirmation> confirmationSubquery = query.subquery(Confirmation.class);
Root<Confirmation> confirmationRoot = confirmationSubquery.from(Confirmation.class); Root<Confirmation> confirmationRoot = confirmationSubquery.from(Confirmation.class);
@@ -71,11 +71,11 @@ public class DataResourceService extends SimpleServiceSupport<DataResource> {
EntityHelper.checkNeededEntityPrediction(authenticationRoot, builder, loginUser) EntityHelper.checkNeededEntityPrediction(authenticationRoot, builder, loginUser)
); );
return builder.or( return Lists.immutable.of(builder.or(
builder.equal(root.get(DataResource_.createdUser), loginUser), builder.equal(root.get(DataResource_.createdUser), loginUser),
builder.exists(confirmationSubquery), builder.exists(confirmationSubquery),
builder.exists(authenticationSubquery) builder.exists(authenticationSubquery)
); ));
} }
public ImmutableList<DataResource> listNoConfirmation() { public ImmutableList<DataResource> listNoConfirmation() {

View File

@@ -17,6 +17,8 @@ import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root; import javax.persistence.criteria.Root;
import javax.persistence.criteria.Subquery; import javax.persistence.criteria.Subquery;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.eclipse.collections.api.factory.Lists;
import org.eclipse.collections.api.list.ImmutableList;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
/** /**
@@ -38,11 +40,11 @@ public class DataFileService extends SimpleServiceSupport<DataFile> {
} }
@Override @Override
protected Predicate listPredicate(Root<DataFile> root, CriteriaQuery<?> query, CriteriaBuilder builder) { protected ImmutableList<Predicate> listPredicate(Root<DataFile> root, CriteriaQuery<?> query, CriteriaBuilder builder) {
if (ignoreWhenAdministrator()) {
return null;
}
User loginUser = userService.currentLoginUser(); User loginUser = userService.currentLoginUser();
if (User.isAdministrator(loginUser)) {
return Lists.immutable.empty();
}
Subquery<Confirmation> confirmationSubquery = query.subquery(Confirmation.class); Subquery<Confirmation> confirmationSubquery = query.subquery(Confirmation.class);
Root<Confirmation> confirmationRoot = confirmationSubquery.from(Confirmation.class); Root<Confirmation> confirmationRoot = confirmationSubquery.from(Confirmation.class);
@@ -60,11 +62,11 @@ public class DataFileService extends SimpleServiceSupport<DataFile> {
EntityHelper.checkNeededEntityPrediction(authenticationRoot, builder, loginUser) EntityHelper.checkNeededEntityPrediction(authenticationRoot, builder, loginUser)
); );
return builder.or( return Lists.immutable.of(builder.or(
builder.equal(root.get(DataFile_.createdUser), loginUser), builder.equal(root.get(DataFile_.createdUser), loginUser),
builder.exists(confirmationSubquery), builder.exists(confirmationSubquery),
builder.exists(authenticationSubquery) builder.exists(authenticationSubquery)
); ));
} }
public Long initialDataFile(String filename) { public Long initialDataFile(String filename) {