Weiba's Blog
0%

ES 随机查询

发表于 更新于 分类于

根据官方文档使用随机评分查询, 随机因子设置为当前时间, 即可实现指定条件的随机数据查询.

参考文档:

ELK第七篇:spring-boot-starter-data-elasticsearch使用
es官方文档-随机评分

es 随机查询

查询语句:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
{
  "query": {
    "function_score": {
      "query": {
        "bool": {
          "must": [
            {
              "term": {
                "typeCode": "01"
              }
            },
            {
              "term": {
                "gradeCode": "3101"
              }
            }
          ],
          "must_not": [],
          "should": []
        }
      },
      "functions": [
        {
          "random_score": {
            "seed": "15:38:48"
          }
        }
      ],
      "score_mode": "sum"
    }
  },
  "size": 1,
  "from": 0
}

Java 实现

maven 依赖:

1
2
3
4
5
6
7
8
9
10
11
12
<dependency>
	  <groupId>io.searchbox</groupId>
	  <artifactId>jest</artifactId>
	  <version>5.3.3</version>
</dependency>

<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>
		spring-boot-starter-data-elasticsearch
	</artifactId>
</dependency>

spring boot 配置:

1
2
3
4
5
6
7
8
9
10
11
spring:
  elasticsearch:
    jest:
      uris: http://xxxx:9200
      username: xxx
      password: xxx
      read-timeout: 20000
      connection-timeout: 20000
    content:
      index: xxx
    enable: true

test code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
@Autowired
public JestClient jestClient;
	
public List<EsInfoPO> recommendationQuestion(Integer count) {
	List<EsInfoPO> retList = new ArrayList<EsInfoPO>();

	FunctionScoreQueryBuilder functionScoreQueryBuilder =
			QueryBuilders.functionScoreQuery(
					boolQuery().must(QueryBuilders.matchQuery("typeCode", "01")), ScoreFunctionBuilders.randomFunction(System.currentTimeMillis()));

	SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
	searchSourceBuilder.query(functionScoreQueryBuilder).size(count);
	Search search = new Search.Builder(searchSourceBuilder.toString()).addIndex(getIndexName()).addType(getTypeName()).build();

	SearchResult execute = null;

	try {
		execute = jestClient.execute(search);
		if (execute.isSucceeded()) {
			List<SearchResult.Hit<EsInfoPO, Void>> hits = execute.getHits(getClazz());
			for (Hit<EsInfoPO, Void> hit : hits) {
				retList.add(hit.source);
			}
		}
	} catch (Exception e) {
		e.printStackTrace();
	}

	return retList;
}