使用ES支持树状结构查询实战

背景

最近在做一个权限可见的功能。每一个订单都有创建人和创建人部门,而查询订单时可以查看自己的或者自己可见部门的。

比如你可以看电商组下的订单,那么自然可以看到电商A组下的订单。

使用ES写一个demo

搭建ES及数据初始化

java 复制代码
docker run -p 9200:9200 -p 9300:9300 --name es-container -e "discovery.type=single-node" docker.elastic.co/elasticsearch/elasticsearch:8.10.3

ES索引的映射

java 复制代码
{
    "order": {
        "mappings": {
            "properties": {
                "createrId": {
                    "type": "keyword"
                },
                "department": {
                    "type": "keyword"
                },
                "departmentTree": {  
                    "type": "keyword"
                },
                "id": {
                    "type": "keyword"
                },
                "name": {
                    "type": "text"
                }
            }
        }
    }
}

其中数据如下

方案1

场景:你可以看电商组(departMentId = 1)下的订单,那么自然可以看到电商A组下(departMentId = 2,3)的订单。

我通过查询你的权限可以获取大部门电商组(departMentId = 1),然后查询大部门的子部门,从而拿到了部门的全集。

代码略,不推荐,因为部门ID多。

方案2

场景:你可以看电商组(departMentId = 1)下的订单,那么自然可以看到电商A组下(departMentId = 2,3)的订单。

但是我在订单上存储的是这个人的部门链,如下图departmentTree

实操

java 复制代码
<dependency>
            <groupId>org.elasticsearch.client</groupId>
            <artifactId>elasticsearch-rest-high-level-client</artifactId>
            <version>7.17.3</version>
        </dependency>
java 复制代码
import org.apache.http.HttpHost;
import org.elasticsearch.action.search.SearchRequest;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.builder.SearchSourceBuilder;

import java.io.IOException;

/**
 * @author chaird
 * @create 2020-08-27 21:42
 */
public class Start {

    public static void main(String[] args) throws Exception {


        // 创建 REST 高级客户端
        RestHighLevelClient client = new RestHighLevelClient(
                RestClient.builder(new HttpHost("101.200.123.220", 9200, "http")));

        try {
            // 创建搜索请求
            SearchRequest searchRequest = new SearchRequest("order");
            SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
            BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();

            //id = 1
            //boolQueryBuilder.filter(QueryBuilders.termQuery("id", 1));
            //department = 1
            //boolQueryBuilder.filter(QueryBuilders.termQuery("department", 1));
            // id in (1,2)
            //boolQueryBuilder.filter(QueryBuilders.termsQuery("id", Arrays.asList(1,2)));
            // db departmentTree =[1,2,3]  命中一个就行,部门查询
            boolQueryBuilder.filter(QueryBuilders.termQuery("departmentTree", 1));
            // where id = 1 or department = 3
//            BoolQueryBuilder orFilter = QueryBuilders.boolQuery();
//            orFilter.should(QueryBuilders.termQuery("id", 1));
//            orFilter.should(QueryBuilders.termQuery("department", 3));
//            boolQueryBuilder.filter(orFilter);


            searchSourceBuilder.query(boolQueryBuilder);

            //System.out.println(searchSourceBuilder.toString());
            searchRequest.source(searchSourceBuilder);
            // 执行搜索请求
            SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);

            // 处理搜索结果
            for (SearchHit hit : searchResponse.getHits().getHits()) {
                System.out.println(hit.getSourceAsString());
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                // 关闭客户端
                client.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}
相关推荐
SelectDB19 分钟前
浩瀚深度:从 ClickHouse 到 Doris,支撑单表 13PB、534 万亿行的超大规模数据分析场景
大数据·数据库·apache
SelectDB24 分钟前
公开免费!Apache Doris & SelectDB 培训与认证课程正式上线
大数据·数据库·apache
mykyle34 分钟前
Elasticsearch-8.17.0 centos7安装
大数据·elasticsearch·jenkins
大视码垛机38 分钟前
协作机器人掀起工厂革命:码垛场景如何用数据重塑制造业命脉?
大数据·数据库·人工智能
StarRocks_labs1 小时前
Lakehouse x AI ,打造智能 BI 新体验
大数据·starrocks·人工智能·iceberg·lakehouse·智能bi·湖仓分析
IvanCodes2 小时前
三、Spark 运行环境部署:全面掌握四种核心模式
大数据·分布式·spark
躲在云朵里`3 小时前
Git的使用
大数据·git·elasticsearch
Aurora_NeAr3 小时前
大数据之路:阿里巴巴大数据实践——实时技术与数据服务
大数据·后端
牛客企业服务5 小时前
AI面试与传统面试的核心差异解析——AI面试如何提升秋招效率?
java·大数据·人工智能·python·面试·职场和发展·金融
武子康6 小时前
大数据-52 Kafka 架构全解析:高吞吐、高可用分布式消息系统的核心奥秘
大数据·后端·kafka