ElasticSearch使用 co.elastic.clients 把普通的SQL筛选条件转为ES的查询条件的工具类

原创

已于 2023-10-08 10:06:47 修改 · 1.3k 阅读

标签

#elasticsearch #sql #java

于 2023-10-08 10:04:12 首次发布

文章介绍了如何在将公司公共平台数据接入Elasticsearch(ES)后，根据原有的SQL数据权限控制逻辑，使用Java工具类将SQL条件转换为ES的查询条件，以便在ES中执行数据过滤。

开发板推荐：天空星STM32F407VET6开发板

超高性价比 STM32主控 | 超高主频 | 一板兼容百芯 | 比赛神器 | 沉金彩色丝印

点击查看

引言

公司公共平台数据全部接入了ES中，但是公司曾经有数据权限，是通过sql语句做的过滤，数据进入了ES中也需要按照同样的方式做数据过滤，这里我贴一下工具类

数据SQL转换工具类

package org.jeecg.common.es;

import co.elastic.clients.elasticsearch._types.FieldValue;
import co.elastic.clients.elasticsearch._types.query_dsl.*;
import org.jeecg.common.system.vo.SysPermissionDataRuleModel;
import co.elastic.clients.json.JsonData;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * ES权限控制工具类
 * Author: WangLiang
 * Date: 2023/10/7 11:12
 */
public class ESDataPermissionUtil {
   
   


    /**
     * 根据提供的条件map和构造的query拼接过滤条件
     *
     * @param dataRoleMap
     * @param boolQuery
     */
    public static void addFilterRule(Map<String, List<SysPermissionDataRuleModel>> dataRoleMap, BoolQuery.Builder boolQuery) {
   
   
        // 过滤数据权限
        // 对拿到的数据进行分组，每一组之间用 and 分割，组内用 or 分割；
        // 例： where (condition1 = x or condition2 < x or condition3 < x) and (condition4 = x or condition5 = x)

        for (Map.Entry<String, List<SysPermissionDataRuleModel>> entry : dataRoleMap.entrySet()) {
   
   
            BoolQuery.Builder boolGroupQuery = new BoolQuery.Builder();
            List<SysPermissionDataRuleModel> dataList = entry.getValue();
            List<Query> queryList = new ArrayList<>();
            for (SysPermissionDataRuleModel model : dataList) {
   
   
                queryList.add(ruleConvertFilter(model));
            }
            boolGroupQuery.should(queryList);
            boolQuery.filter(f->f.bool(boolGroupQuery.build()));
        }
    }

    /**
     * 根据系统存入的数据过滤规则转换为ES的过滤规则
     * @param model 系统model
     * @return ES的query对象
     */
    private static Query ruleConvertFilter(SysPermissionDataRuleModel model) {
   
   
        String condition = model.getRuleConditions();

        Query.Builder query = new Query.Builder();
        if (condition.equals("=")) {
   
   
            query.term(t -> t.