脚本字段与运行时字段
Spring Data Elasticsearch 支持脚本字段和运行时字段。 请参阅 Elasticsearch 关于脚本(www.elastic.co/guide/en/elasticsearch/reference/current/modules-scripting.html)和运行时字段(www.elastic.co/guide/en/elasticsearch/reference/8.9/runtime.html)的文档,获取详细信息。 在 Spring Data Elasticsearch 的语境下,你可以使用
-
脚本字段用于返回在结果文档上计算并添加到返回文档中的字段。
-
运行时字段,这些字段在存储的文档中计算,可用于查询和/或在搜索结果中返回。
以下代码片段将展示你可以做什么(这些是命令式代码,但响应式实现的原理类似)。
人实体
这些例子中使用的enity是人实体。
该实体拥有生日以及年龄财产。
虽然出生日期是固定的,但年龄取决于查询发出的时间,需要动态计算。
import org.springframework.data.annotation.Id;
import org.springframework.data.elasticsearch.annotations.DateFormat;
import org.springframework.data.elasticsearch.annotations.Document;
import org.springframework.data.elasticsearch.annotations.Field;
import org.springframework.data.elasticsearch.annotations.ScriptedField;
import org.springframework.lang.Nullable;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import static org.springframework.data.elasticsearch.annotations.FieldType.*;
import java.lang.Integer;
@Document(indexName = "persons")
public record Person(
@Id
@Nullable
String id,
@Field(type = Text)
String lastName,
@Field(type = Text)
String firstName,
@Field(type = Keyword)
String gender,
@Field(type = Date, format = DateFormat.basic_date)
LocalDate birthDate,
@Nullable
@ScriptedField Integer age (1)
) {
public Person(String id,String lastName, String firstName, String gender, String birthDate) {
this(id, (2)
lastName,
firstName,
LocalDate.parse(birthDate, DateTimeFormatter.ISO_LOCAL_DATE),
gender,
null);
}
}
| 1 | 这年龄房产将被计算并填写搜索结果。 |
| 2 | 一个方便的构造器,用于设置测试数据。 |
注意年龄属性注释为@ScriptedField.
这会阻止在索引映射中写入对应的条目,并将该属性标记为搜索响应中计算字段的目标。
仓库接口
本例中使用的仓库:
public interface PersonRepository extends ElasticsearchRepository<Person, String> {
SearchHits<Person> findAllBy(ScriptedField scriptedField);
SearchHits<Person> findByGenderAndAgeLessThanEqual(String gender, Integer age, RuntimeField runtimeField);
}
服务级别
服务类注入了一个仓库,并且弹性搜索作实例展示了几种填充和使用年龄财产。
我们展示了代码拆分成不同部分以放入解释
import org.springframework.data.elasticsearch.core.ElasticsearchOperations;
import org.springframework.data.elasticsearch.core.SearchHits;
import org.springframework.data.elasticsearch.core.query.Criteria;
import org.springframework.data.elasticsearch.core.query.CriteriaQuery;
import org.springframework.data.elasticsearch.core.query.FetchSourceFilter;
import org.springframework.data.elasticsearch.core.query.RuntimeField;
import org.springframework.data.elasticsearch.core.query.ScriptData;
import org.springframework.data.elasticsearch.core.query.ScriptType;
import org.springframework.data.elasticsearch.core.query.ScriptedField;
import org.springframework.data.elasticsearch.core.query.StringQuery;
import org.springframework.stereotype.Service;
import java.util.List;
@Service
public class PersonService {
private final ElasticsearchOperations operations;
private final PersonRepository repository;
public PersonService(ElasticsearchOperations operations, SaRPersonRepository repository) {
this.operations = operations;
this.repository = repository;
}
public void save() { (1)
List<Person> persons = List.of(
new Person("1", "Smith", "Mary", "f", "1987-05-03"),
new Person("2", "Smith", "Joshua", "m", "1982-11-17"),
new Person("3", "Smith", "Joanna", "f", "2018-03-27"),
new Person("4", "Smith", "Alex", "m", "2020-08-01"),
new Person("5", "McNeill", "Fiona", "f", "1989-04-07"),
new Person("6", "McNeill", "Michael", "m", "1984-10-20"),
new Person("7", "McNeill", "Geraldine", "f", "2020-03-02"),
new Person("8", "McNeill", "Patrick", "m", "2022-07-04"));
repository.saveAll(persons);
}
| 1 | 一种用于在Elasticsearch中存储部分数据的工具方法。 |
脚本字段
接下来的部分展示了如何使用脚本字段来计算并返回人物的年龄。 脚本字段只能向返回的数据添加内容,查询时不能使用年龄(参见运行时字段)。
public SearchHits<Person> findAllWithAge() {
var scriptedField = ScriptedField.of("age", (1)
ScriptData.of(b -> b
.withType(ScriptType.INLINE)
.withScript("""
Instant currentDate = Instant.ofEpochMilli(new Date().getTime());
Instant startDate = doc['birth-date'].value.toInstant();
return (ChronoUnit.DAYS.between(startDate, currentDate) / 365);
""")));
// version 1: use a direct query
var query = new StringQuery("""
{ "match_all": {} }
""");
query.addScriptedField(scriptedField); (2)
query.addSourceFilter(FetchSourceFilter.of(b -> b.withIncludes("*"))); (3)
var result1 = operations.search(query, Person.class); (4)
// version 2: use the repository
var result2 = repository.findAllBy(scriptedField); (5)
return result1;
}
| 1 | 定义脚本场它计算一个人的年龄。 |
| 2 | 当使用查询,将脚本字段添加到查询中。 |
| 3 | 在向查询,还需要一个额外的源过滤器来从文档源检索法线字段。 |
| 4 | 获取人实体现在在其年龄财产。 |
| 5 | 使用仓库时,只需将脚本字段添加为方法参数即可。 |
运行时字段
使用运行时字段时,计算出来的值可以在查询本身中使用。 在以下代码中,这用于对给定性别和最大年龄的人员进行查询:
public SearchHits<Person> findWithGenderAndMaxAge(String gender, Integer maxAge) {
var runtimeField = new RuntimeField("age", "long", """ (1)
Instant currentDate = Instant.ofEpochMilli(new Date().getTime());
Instant startDate = doc['birthDate'].value.toInstant();
emit (ChronoUnit.DAYS.between(startDate, currentDate) / 365);
""");
// variant 1 : use a direct query
var query = CriteriaQuery.builder(Criteria
.where("gender").is(gender)
.and("age").lessThanEqual(maxAge))
.withRuntimeFields(List.of(runtimeField)) (2)
.withFields("age") (3)
.withSourceFilter(FetchSourceFilter.of(b -> b.withIncludes("*"))) (4)
.build();
var result1 = operations.search(query, Person.class); (5)
// variant 2: use the repository (6)
var result2 = repository.findByGenderAndAgeLessThanEqual(gender, maxAge, runtimeField);
return result1;
}
}
| 1 | 定义计算一个人年龄的运行时字段。关于内置属性,请参见 asciidoctor.org/docs/user-manual/#builtin-attributes。 |
| 2 | 使用查询,添加运行时字段。 |
| 3 | 在向查询需要额外的字段参数以返回计算值。 |
| 4 | 在向查询,还需要一个额外的源过滤器来从文档源检索法线字段。 |
| 5 | 用查询过滤数据,并且返回的ent有设置age属性。 |
| 6 | 使用仓库时,只需添加运行时字段作为方法参数即可。 |
除了在查询上定义运行时字段外,还可以在索引中定义它们,通过设置runtimeFieldsPath的属性@Mapping注释指向包含运行时字段定义的JSON文件。