数据中心分省厅

xiangpei

2024-09-06 a09c636267ea9e37296fa58d6619e43644e9eacd

 ycl-server/src/main/java/com/ycl/utils/MongoUtil.java

@@ -13,6 +13,8 @@
import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

/**
 * mongo工具类
@@ -26,36 +28,73 @@
     * 构造数据中心搜索的条件
     *
     * @param params 数据中心请求参数
     * @param gbFiled 国标码字段是哪一个
     * @param timeFiled 时间字段是哪一个
     * @param gbList 国标码
     * @param dataAuth 点位类型：null 所有  1 部级  2 省厅  3 重点  4 重点指挥  5 是否新设备
     * @param likeFileds 使用like查询的字段
     */
    public static Query getQuery(DataCenterQuery params, String gbFiled, String timeFiled, List<String> gbList) {
    public static Query getQuery(DataCenterQuery params, String timeFiled, List<String> likeFileds, Integer dataAuth) {
        // 关键字
        TextCriteria text = null;
        if (StringUtils.hasText(params.getKeyword())) {
            text = TextCriteria.forDefaultLanguage().matching(params.getKeyword());
        List<Criteria> likeList = null;
        if (StringUtils.hasText(params.getKeyword()) && !CollectionUtils.isEmpty(likeFileds)) {
            // 转义
            String keyword = Pattern.quote(params.getKeyword());
            likeList = likeFileds.stream().map(filed -> {
                return Criteria.where(filed).regex(".*?" + keyword + ".*");
            }).collect(Collectors.toList());
        }
        // 将like字段or起来
        Criteria or = null;
        if (! CollectionUtils.isEmpty(likeList)) {
            or = new Criteria();
            or.orOperator(likeList);
        }

        Criteria in = null;
        Criteria tag = null;
        // in
        if (! CollectionUtils.isEmpty(gbList)) {
            in = Criteria.where(gbFiled).in(gbList);
        if (Objects.nonNull(dataAuth)) {
            switch (dataAuth) {
                case 1:
                    tag = Criteria.where("deptTag").is(Boolean.TRUE);
                    break;
                case 2:
                    tag = Criteria.where("provinceTag").is(Boolean.TRUE);
                    break;
                case 3:
                    tag = Criteria.where("importantTag").is(Boolean.TRUE);
                    break;
                case 4:
                    tag = Criteria.where("importantCommandImageTag").is(Boolean.TRUE);
                    break;
                case 5:
                    tag = Criteria.where("newDevice").is(Boolean.TRUE);
                    break;
                default:
                    break;
            }
        }

        // 时间
        Criteria time = Criteria.where(timeFiled).gte(params.getStartTime()).lte(params.getEndTime());

        // 数据类型
        Criteria dType = null;
        if (params.getDataType().equals(1)) {
            dType = Criteria.where("provinceTag").is(Boolean.TRUE);
        }

        // 将查询条件and起来
        Criteria and = new Criteria();
        List<Criteria> qList = new ArrayList<>(3);
        if (Objects.nonNull(text)) {
            qList.add(Criteria.byExample(text));
        }
        if (Objects.nonNull(in)) {
            qList.add(in);
        }
        qList.add(time);
        if (Objects.nonNull(tag)) {
            qList.add(tag);
        }
        if (Objects.nonNull(or)) {
            qList.add(or);
        }
        if (Objects.nonNull(dType)) {
            qList.add(dType);
        }
        and.andOperator(qList);
        // 返回查询
        return Query.query(and);