Skip to content

Commit 679f99a

Browse files
xhaggixhaggi
committed
Rewrite persistent property converter for date and java.time
1 parent 66d1344 commit 679f99a

File tree

5 files changed

+244
-82
lines changed

5 files changed

+244
-82
lines changed

‎src/main/java/org/springframework/data/elasticsearch/core/convert/AbstractPersistentPropertyConverter.java

+40
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
/*
2+
* Copyright 2021 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package org.springframework.data.elasticsearch.core.convert;
17+
18+
import org.springframework.data.elasticsearch.core.mapping.ElasticsearchPersistentPropertyConverter;
19+
import org.springframework.data.mapping.PersistentProperty;
20+
import org.springframework.util.Assert;
21+
22+
/**
23+
* @author Sascha Woo
24+
* @since 4.3
25+
*/
26+
public abstract class AbstractPersistentPropertyConverter implements ElasticsearchPersistentPropertyConverter {
27+
28+
private final PersistentProperty<?> property;
29+
30+
public AbstractPersistentPropertyConverter(PersistentProperty<?> property) {
31+
32+
Assert.notNull(property, "property must not be null.");
33+
this.property = property;
34+
}
35+
36+
protected PersistentProperty<?> getProperty() {
37+
return property;
38+
}
39+
40+
}

‎src/main/java/org/springframework/data/elasticsearch/core/convert/DatePersistentPropertyConverter.java

+68
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,68 @@
1+
/*
2+
* Copyright 2021 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package org.springframework.data.elasticsearch.core.convert;
17+
18+
import java.util.Date;
19+
import java.util.List;
20+
21+
import org.slf4j.Logger;
22+
import org.slf4j.LoggerFactory;
23+
import org.springframework.data.mapping.PersistentProperty;
24+
25+
/**
26+
* @author Sascha Woo
27+
* @since 4.3
28+
*/
29+
public class DatePersistentPropertyConverter extends AbstractPersistentPropertyConverter {
30+
31+
private static final Logger LOGGER = LoggerFactory.getLogger(DatePersistentPropertyConverter.class);
32+
33+
private final List<ElasticsearchDateConverter> dateConverters;
34+
35+
public DatePersistentPropertyConverter(PersistentProperty<?> property,
36+
List<ElasticsearchDateConverter> dateConverters) {
37+
38+
super(property);
39+
this.dateConverters = dateConverters;
40+
}
41+
42+
@Override
43+
public Object read(String value) {
44+
45+
for (ElasticsearchDateConverter dateConverter : dateConverters) {
46+
try {
47+
return dateConverter.parse(value);
48+
} catch (Exception e) {
49+
LOGGER.trace(e.getMessage(), e);
50+
}
51+
}
52+
53+
throw new ConversionException(String.format("Unable to convert value '%s' to %s for property '%s'", value,
54+
getProperty().getActualType().getTypeName(), getProperty().getName()));
55+
}
56+
57+
@Override
58+
public String write(Object value) {
59+
60+
try {
61+
return dateConverters.get(0).format((Date) value);
62+
} catch (Exception e) {
63+
throw new ConversionException(
64+
String.format("Unable to convert value '%s' of property '%s'", value, getProperty().getName()), e);
65+
}
66+
}
67+
68+
}

‎src/main/java/org/springframework/data/elasticsearch/core/convert/TemporalPersistentPropertyConverter.java

+71
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,71 @@
1+
/*
2+
* Copyright 2021 the original author or authors.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package org.springframework.data.elasticsearch.core.convert;
17+
18+
import java.time.temporal.TemporalAccessor;
19+
import java.util.List;
20+
21+
import org.slf4j.Logger;
22+
import org.slf4j.LoggerFactory;
23+
import org.springframework.data.mapping.PersistentProperty;
24+
25+
/**
26+
* @author Sascha Woo
27+
* @since 4.3
28+
*/
29+
public class TemporalPersistentPropertyConverter extends AbstractPersistentPropertyConverter {
30+
31+
private static final Logger LOGGER = LoggerFactory.getLogger(TemporalPersistentPropertyConverter.class);
32+
33+
private final List<ElasticsearchDateConverter> dateConverters;
34+
35+
public TemporalPersistentPropertyConverter(PersistentProperty<?> property,
36+
List<ElasticsearchDateConverter> dateConverters) {
37+
38+
super(property);
39+
this.dateConverters = dateConverters;
40+
}
41+
42+
@SuppressWarnings("unchecked")
43+
@Override
44+
public Object read(String value) {
45+
46+
Class<?> actualType = getProperty().getActualType();
47+
48+
for (ElasticsearchDateConverter dateConverter : dateConverters) {
49+
try {
50+
return dateConverter.parse(value, (Class<? extends TemporalAccessor>) actualType);
51+
} catch (Exception e) {
52+
LOGGER.trace(e.getMessage(), e);
53+
}
54+
}
55+
56+
throw new ConversionException(String.format("Unable to convert value '%s' to %s for property '%s'", value,
57+
getProperty().getActualType().getTypeName(), getProperty().getName()));
58+
}
59+
60+
@Override
61+
public String write(Object value) {
62+
63+
try {
64+
return dateConverters.get(0).format((TemporalAccessor) value);
65+
} catch (Exception e) {
66+
throw new ConversionException(
67+
String.format("Unable to convert value '%s' of property '%s'", value, getProperty().getName()), e);
68+
}
69+
}
70+
71+
}

‎src/main/java/org/springframework/data/elasticsearch/core/mapping/SimpleElasticsearchPersistentProperty.java

+64-81
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,17 @@
3030
import org.springframework.data.elasticsearch.annotations.GeoShapeField;
3131
import org.springframework.data.elasticsearch.annotations.MultiField;
3232
import org.springframework.data.elasticsearch.core.completion.Completion;
33-
import org.springframework.data.elasticsearch.core.convert.ConversionException;
33+
import org.springframework.data.elasticsearch.core.convert.DatePersistentPropertyConverter;
3434
import org.springframework.data.elasticsearch.core.convert.ElasticsearchDateConverter;
35+
import org.springframework.data.elasticsearch.core.convert.TemporalPersistentPropertyConverter;
3536
import org.springframework.data.elasticsearch.core.geo.GeoJson;
3637
import org.springframework.data.elasticsearch.core.geo.GeoPoint;
3738
import org.springframework.data.elasticsearch.core.join.JoinField;
3839
import org.springframework.data.elasticsearch.core.query.SeqNoPrimaryTerm;
3940
import org.springframework.data.mapping.Association;
4041
import org.springframework.data.mapping.MappingException;
4142
import org.springframework.data.mapping.PersistentEntity;
43+
import org.springframework.data.mapping.PersistentProperty;
4244
import org.springframework.data.mapping.model.AnnotationBasedPersistentProperty;
4345
import org.springframework.data.mapping.model.FieldNamingStrategy;
4446
import org.springframework.data.mapping.model.Property;
@@ -92,7 +94,7 @@ public SimpleElasticsearchPersistentProperty(Property property,
9294
throw new MappingException("@Field annotation must not be used on a @MultiField property.");
9395
}
9496

95-
initDateConverter();
97+
initPropertyConverter();
9698

9799
storeNullValue = isField && getRequiredAnnotation(Field.class).storeNullValue();
98100
}
@@ -128,102 +130,82 @@ protected boolean hasExplicitFieldName() {
128130
}
129131

130132
/**
131-
* Initializes an {@link ElasticsearchPersistentPropertyConverter} if this property is annotated as a Field with type
132-
* {@link FieldType#Date}, has a {@link DateFormat} set and if the type of the property is one of the Java8 temporal
133-
* classes or java.util.Date.
133+
* Initializes the property converter for this {@link PersistentProperty}, if any.
134134
*/
135-
private void initDateConverter() {
136-
Field field = findAnnotation(Field.class);
135+
private void initPropertyConverter() {
137136

138137
Class<?> actualType = getActualTypeOrNull();
139-
140138
if (actualType == null) {
141139
return;
142140
}
143141

144-
boolean isTemporalAccessor = TemporalAccessor.class.isAssignableFrom(actualType);
145-
boolean isDate = Date.class.isAssignableFrom(actualType);
142+
Field field = findAnnotation(Field.class);
143+
if (field == null || field.type() == null) {
144+
return;
145+
}
146146

147-
if (field != null && (field.type() == FieldType.Date || field.type() == FieldType.Date_Nanos)
148-
&& (isTemporalAccessor || isDate)) {
147+
switch (field.type()) {
148+
case Date:
149+
case Date_Nanos:
150+
List<ElasticsearchDateConverter> dateConverters = getDateConverters(field, actualType);
149151

150-
DateFormat[] dateFormats = field.format();
151-
String[] dateFormatPatterns = field.pattern();
152+
if (dateConverters.isEmpty()) {
153+
LOGGER.warn("No date formatters configured for property '{}'.", getName());
154+
return;
155+
}
156+
157+
if (TemporalAccessor.class.isAssignableFrom(actualType)) {
158+
propertyConverter = new TemporalPersistentPropertyConverter(this, dateConverters);
159+
} else if (Date.class.isAssignableFrom(actualType)) {
160+
propertyConverter = new DatePersistentPropertyConverter(this, dateConverters);
161+
} else {
162+
LOGGER.warn("Unsupported type '{}' for date property '{}'.", actualType, getName());
163+
}
164+
break;
165+
default:
166+
break;
167+
}
168+
}
152169

153-
String property = getOwner().getType().getSimpleName() + "." + getName();
170+
private List<ElasticsearchDateConverter> getDateConverters(Field field, Class<?> actualType) {
154171

155-
if (dateFormats.length == 0 && dateFormatPatterns.length == 0) {
156-
LOGGER.warn(
157-
"Property '{}' has @Field type '{}' but has no built-in format or custom date pattern defined. Make sure you have a converter registered for type {}.",
158-
property, field.type().name(), actualType.getSimpleName());
159-
return;
160-
}
172+
DateFormat[] dateFormats = field.format();
173+
String[] dateFormatPatterns = field.pattern();
174+
List<ElasticsearchDateConverter> converters = new ArrayList<>();
161175

162-
List<ElasticsearchDateConverter> converters = new ArrayList<>();
163-
164-
// register converters for built-in formats
165-
for (DateFormat dateFormat : dateFormats) {
166-
switch (dateFormat) {
167-
case none:
168-
case custom:
169-
break;
170-
case weekyear:
171-
case weekyear_week:
172-
case weekyear_week_day:
173-
LOGGER.warn("No default converter available for '{}' and date format '{}'. Use a custom converter instead.",
174-
actualType.getName(), dateFormat.name());
175-
break;
176-
default:
177-
converters.add(ElasticsearchDateConverter.of(dateFormat));
178-
break;
179-
}
180-
}
176+
if (dateFormats.length == 0 && dateFormatPatterns.length == 0) {
177+
LOGGER.warn(
178+
"Property '{}' has @Field type '{}' but has no built-in format or custom date pattern defined. Make sure you have a converter registered for type {}.",
179+
getName(), field.type().name(), actualType.getSimpleName());
180+
return converters;
181+
}
181182

182-
// register converters for custom formats
183-
for (String dateFormatPattern : dateFormatPatterns) {
184-
if (!StringUtils.hasText(dateFormatPattern)) {
185-
throw new MappingException(String.format("Date pattern of property '%s' must not be empty", property));
186-
}
187-
converters.add(ElasticsearchDateConverter.of(dateFormatPattern));
183+
// register converters for built-in formats
184+
for (DateFormat dateFormat : dateFormats) {
185+
switch (dateFormat) {
186+
case none:
187+
case custom:
188+
break;
189+
case weekyear:
190+
case weekyear_week:
191+
case weekyear_week_day:
192+
LOGGER.warn("No default converter available for '{}' and date format '{}'. Use a custom converter instead.",
193+
actualType.getName(), dateFormat.name());
194+
break;
195+
default:
196+
converters.add(ElasticsearchDateConverter.of(dateFormat));
197+
break;
188198
}
199+
}
189200

190-
if (!converters.isEmpty()) {
191-
propertyConverter = new ElasticsearchPersistentPropertyConverter() {
192-
final List<ElasticsearchDateConverter> dateConverters = converters;
193-
194-
@SuppressWarnings("unchecked")
195-
@Override
196-
public Object read(String s) {
197-
for (ElasticsearchDateConverter dateConverter : dateConverters) {
198-
try {
199-
if (isTemporalAccessor) {
200-
return dateConverter.parse(s, (Class<? extends TemporalAccessor>) actualType);
201-
} else { // must be date
202-
return dateConverter.parse(s);
203-
}
204-
} catch (Exception e) {
205-
LOGGER.trace(e.getMessage(), e);
206-
}
207-
}
208-
209-
throw new ConversionException(String
210-
.format("Unable to parse date value '%s' of property '%s' with configured converters", s, property));
211-
}
212-
213-
@Override
214-
public String write(Object property) {
215-
ElasticsearchDateConverter dateConverter = dateConverters.get(0);
216-
if (isTemporalAccessor && TemporalAccessor.class.isAssignableFrom(property.getClass())) {
217-
return dateConverter.format((TemporalAccessor) property);
218-
} else if (isDate && Date.class.isAssignableFrom(property.getClass())) {
219-
return dateConverter.format((Date) property);
220-
} else {
221-
return property.toString();
222-
}
223-
}
224-
};
201+
for (String dateFormatPattern : dateFormatPatterns) {
202+
if (!StringUtils.hasText(dateFormatPattern)) {
203+
throw new MappingException(String.format("Date pattern of property '%s' must not be empty", getName()));
225204
}
205+
converters.add(ElasticsearchDateConverter.of(dateFormatPattern));
226206
}
207+
208+
return converters;
227209
}
228210

229211
@SuppressWarnings("ConstantConditions")
@@ -303,4 +285,5 @@ public boolean isJoinFieldProperty() {
303285
public boolean isCompletionProperty() {
304286
return getActualType() == Completion.class;
305287
}
288+
306289
}

‎src/test/java/org/springframework/data/elasticsearch/core/CriteriaQueryMappingUnitTests.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ void shouldMapNamesAndConvertValuesInCriteriaQueryForSubCriteriaWithDate() throw
196196
CriteriaQuery criteriaQuery = new CriteriaQuery( //
197197
Criteria.or().subCriteria(Criteria.where("birthDate") //
198198
.between(LocalDate.of(1989, 11, 9), LocalDate.of(1990, 11, 9))) //
199-
.subCriteria(Criteria.where("createdDate").is(383745721653L)) //
199+
.subCriteria(Criteria.where("createdDate").is(new Date(383745721653L))) //
200200
);
201201

202202
// mapped field name and converted parameter

0 commit comments

Comments
 (0)