ElasticSearch之script语法使用

需求回顾
1、业务需要根据ES内部多个字段进行比较运算
2、不能使用外部参数进行传入比较

技术回顾

1、script参数能够自由定义需要比较运算的字段
2、script参数能够适用各种业务场景，一般restApi实现不了的都能够解决

小试牛刀

本文直接适用java语言进行演示
本次测试查询 【年龄 -10 > 工龄 * 2 】的数据

1、定义项目pom以及application
1.1 pom.xml

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

1.2 application.yml

spring:
  elasticsearch:
    username: elastic
    password: VaHcSC3mOFfovLWTqW6E
    uris: 10.10.22.174:9200
    connection-timeout: 1s
    socket-timeout: 30s

2、创建es 索引及业务实体

2.1 es文档实体

/**
 *
 * 用户实体
 * @author senfel
 * @version 1.0
 * @date 2022/7/26 10:40
 */
@Data
@Document(indexName = "users")
public class Users {
​
    /**
     * id
     */
    private Long id;
    /**
     * 姓名
     */
    private String name;
    /**
     * 年龄
     */
    private Integer age;
​
    /**
     * 工龄
     */
    private Integer workAge;
}
​

2.2 创建es索引并增加测试数据

  @Test
    void contextLoads() {
        //创建索引
        boolean usersFlag = elasticsearchRestTemplate.indexOps(IndexCoordinates.of("users")).create();
        if(usersFlag) {
            //设置测试数据
            for (int i = 0; i < 100; i++) {
                Users users = new Users();
                users.setAge(25 + i / 10 + i % 10);
                users.setWorkAge(3 + i / 10 + i % 10);
                users.setName("senfel" + i);
                users.setId(Long.valueOf(i + ""));
                Users save = elasticsearchRestTemplate.save(users);
                System.err.println(save);
            }
        }
    }

2.3 测试数据

3、组装script语句,本次测试查询 <年龄-10 > 工龄 * 2> 的数据
3.1 java查询es语句

 /**
     * 测试es  script
     * @author senfel
     * @date 2022/7/26 10:48
     * @return void
     */
    @Test
    public void testEsByScript(){
        NativeSearchQueryBuilder nativeSearchQueryBuilder = new NativeSearchQueryBuilder();
        nativeSearchQueryBuilder.withSourceFilter(new FetchSourceFilter(new String[]{"id","name","age","workAge"},null)).withPageable(PageRequest.of(0,10));
        //查询条件
        BoolQueryBuilder queryBuilder = new BoolQueryBuilder();
        ScriptQueryBuilder builderByScript = new ScriptQueryBuilder(new Script("if (doc['age'].value -10 " +
                "> doc['workAge'].value * 2){ return true}else{return false}"));
        queryBuilder.must(builderByScript);
        nativeSearchQueryBuilder.withQuery(queryBuilder);
​
        String esInfo = nativeSearchQueryBuilder.build().getQuery().toString();
        System.err.println(esInfo);
        long count = elasticsearchRestTemplate.count(nativeSearchQueryBuilder.build(), Users.class);
        if(count > 0){
            SearchHits<Users> searchHits = elasticsearchRestTemplate.search(nativeSearchQueryBuilder.build(), Users.class);
            List<SearchHit<Users>> datas = searchHits.getSearchHits();
            datas.forEach(
                    System.err::println
            );
        }
    }

3.2 es query语句与查询结果

{
  "bool" : {
    "must" : [
      {
        "script" : {
          "script" : {
            "source" : "if (doc['age'].value -10 > doc['workAge'].value * 2){ return true}else{return false}",
            "lang" : "painless"
          },
          "boost" : 1.0
        }
      }
    ],
    "adjust_pure_negative" : true,
    "boost" : 1.0
  }
}

总条数45
SearchHit{id=‘0’, score=NaN, sortValues=[0], content=Users(id=0, name=senfel0, age=25, workAge=3), highlightFields={}}
SearchHit{id=‘1’, score=NaN, sortValues=[1], content=Users(id=1, name=senfel1, age=26, workAge=4), highlightFields={}}
SearchHit{id=‘2’, score=NaN, sortValues=[2], content=Users(id=2, name=senfel2, age=27, workAge=5), highlightFields={}}
SearchHit{id=‘3’, score=NaN, sortValues=[3], content=Users(id=3, name=senfel3, age=28, workAge=6), highlightFields={}}
SearchHit{id=‘4’, score=NaN, sortValues=[4], content=Users(id=4, name=senfel4, age=29, workAge=7), highlightFields={}}
SearchHit{id=‘5’, score=NaN, sortValues=[5], content=Users(id=5, name=senfel5, age=30, workAge=8), highlightFields={}}
SearchHit{id=‘6’, score=NaN, sortValues=[6], content=Users(id=6, name=senfel6, age=31, workAge=9), highlightFields={}}
SearchHit{id=‘7’, score=NaN, sortValues=[7], content=Users(id=7, name=senfel7, age=32, workAge=10), highlightFields={}}
SearchHit{id=‘8’, score=NaN, sortValues=[8], content=Users(id=8, name=senfel8, age=33, workAge=11), highlightFields={}}
SearchHit{id=‘10’, score=NaN, sortValues=[10], content=Users(id=10, name=senfel10, age=26, workAge=4), highlightFields={}}