|
19 | 19 | import org.hibernate.search.engine.search.predicate.spi.SpatialWithinBoundingBoxPredicateBuilder;
|
20 | 20 | import org.hibernate.search.engine.search.predicate.spi.SpatialWithinCirclePredicateBuilder;
|
21 | 21 | import org.hibernate.search.engine.search.predicate.spi.SpatialWithinPolygonPredicateBuilder;
|
| 22 | +import org.hibernate.search.engine.search.reference.predicate.SpatialPredicateFieldReference; |
22 | 23 | import org.hibernate.search.engine.spatial.DistanceUnit;
|
23 | 24 | import org.hibernate.search.engine.spatial.GeoBoundingBox;
|
24 | 25 | import org.hibernate.search.engine.spatial.GeoPoint;
|
25 | 26 | import org.hibernate.search.engine.spatial.GeoPolygon;
|
26 | 27 | import org.hibernate.search.util.common.impl.CollectionHelper;
|
27 | 28 | import org.hibernate.search.util.common.impl.Contracts;
|
28 | 29 |
|
29 |
| -class SpatialWithinPredicateFieldMoreStepImpl |
| 30 | +class SpatialWithinPredicateFieldMoreStepImpl<SR> |
30 | 31 | implements
|
31 |
| - SpatialWithinPredicateFieldMoreStep<SpatialWithinPredicateFieldMoreStepImpl, SpatialWithinPredicateOptionsStep<?>>, |
| 32 | + SpatialWithinPredicateFieldMoreStep<SR, |
| 33 | + SpatialWithinPredicateFieldMoreStepImpl<SR>, |
| 34 | + SpatialWithinPredicateOptionsStep<?>>, |
32 | 35 | AbstractBooleanMultiFieldPredicateCommonState.FieldSetState {
|
33 | 36 |
|
34 |
| - private final CommonState commonState; |
| 37 | + private final CommonState<SR> commonState; |
35 | 38 |
|
36 | 39 | private final List<String> fieldPaths;
|
37 | 40 |
|
38 | 41 | private final List<SearchPredicateBuilder> predicateBuilders;
|
39 | 42 |
|
40 | 43 | private Float fieldSetBoost;
|
41 | 44 |
|
42 |
| - SpatialWithinPredicateFieldMoreStepImpl(CommonState commonState, List<String> fieldPaths) { |
| 45 | + SpatialWithinPredicateFieldMoreStepImpl(CommonState<SR> commonState, List<String> fieldPaths) { |
43 | 46 | this.commonState = commonState;
|
44 | 47 | this.commonState.add( this );
|
45 | 48 | this.fieldPaths = CollectionHelper.toImmutableList( fieldPaths );
|
46 | 49 | this.predicateBuilders = new ArrayList<>( fieldPaths.size() );
|
47 | 50 | }
|
48 | 51 |
|
49 | 52 | @Override
|
50 |
| - public SpatialWithinPredicateFieldMoreStepImpl fields(String... fieldPaths) { |
51 |
| - return new SpatialWithinPredicateFieldMoreStepImpl( commonState, Arrays.asList( fieldPaths ) ); |
| 53 | + public SpatialWithinPredicateFieldMoreStepImpl<SR> fields(String... fieldPaths) { |
| 54 | + return new SpatialWithinPredicateFieldMoreStepImpl<>( commonState, Arrays.asList( fieldPaths ) ); |
52 | 55 | }
|
53 | 56 |
|
| 57 | + @SuppressWarnings("unchecked") |
54 | 58 | @Override
|
55 |
| - public SpatialWithinPredicateFieldMoreStepImpl boost(float boost) { |
| 59 | + public SpatialWithinPredicateFieldMoreStepImpl<SR> fields(SpatialPredicateFieldReference<? super SR>... fieldReferences) { |
| 60 | + List<String> fieldPaths = new ArrayList<>( fieldReferences.length ); |
| 61 | + for ( SpatialPredicateFieldReference<? super SR> fieldReference : fieldReferences ) { |
| 62 | + fieldPaths.add( fieldReference.absolutePath() ); |
| 63 | + } |
| 64 | + return new SpatialWithinPredicateFieldMoreStepImpl<>( commonState, fieldPaths ); |
| 65 | + } |
| 66 | + |
| 67 | + @Override |
| 68 | + public SpatialWithinPredicateFieldMoreStepImpl<SR> boost(float boost) { |
56 | 69 | this.fieldSetBoost = boost;
|
57 | 70 | return this;
|
58 | 71 | }
|
@@ -120,40 +133,40 @@ private void generateWithinBoundingBoxQueryBuilders(GeoBoundingBox boundingBox)
|
120 | 133 | }
|
121 | 134 | }
|
122 | 135 |
|
123 |
| - static class CommonState |
124 |
| - extends AbstractBooleanMultiFieldPredicateCommonState<CommonState, SpatialWithinPredicateFieldMoreStepImpl> |
125 |
| - implements SpatialWithinPredicateOptionsStep<CommonState> { |
| 136 | + static class CommonState<SR> |
| 137 | + extends AbstractBooleanMultiFieldPredicateCommonState<CommonState<SR>, SpatialWithinPredicateFieldMoreStepImpl<SR>> |
| 138 | + implements SpatialWithinPredicateOptionsStep<CommonState<SR>> { |
126 | 139 |
|
127 | 140 | CommonState(SearchPredicateDslContext<?> dslContext) {
|
128 | 141 | super( dslContext );
|
129 | 142 | }
|
130 | 143 |
|
131 | 144 | SpatialWithinPredicateOptionsStep<?> circle(GeoPoint center, double radius, DistanceUnit unit) {
|
132 |
| - for ( SpatialWithinPredicateFieldMoreStepImpl fieldSetState : getFieldSetStates() ) { |
| 145 | + for ( SpatialWithinPredicateFieldMoreStepImpl<SR> fieldSetState : getFieldSetStates() ) { |
133 | 146 | fieldSetState.generateWithinCircleQueryBuilders( center, radius, unit );
|
134 | 147 | }
|
135 | 148 |
|
136 | 149 | return this;
|
137 | 150 | }
|
138 | 151 |
|
139 | 152 | SpatialWithinPredicateOptionsStep<?> polygon(GeoPolygon polygon) {
|
140 |
| - for ( SpatialWithinPredicateFieldMoreStepImpl fieldSetState : getFieldSetStates() ) { |
| 153 | + for ( SpatialWithinPredicateFieldMoreStepImpl<SR> fieldSetState : getFieldSetStates() ) { |
141 | 154 | fieldSetState.generateWithinPolygonQueryBuilders( polygon );
|
142 | 155 | }
|
143 | 156 |
|
144 | 157 | return this;
|
145 | 158 | }
|
146 | 159 |
|
147 | 160 | SpatialWithinPredicateOptionsStep<?> boundingBox(GeoBoundingBox boundingBox) {
|
148 |
| - for ( SpatialWithinPredicateFieldMoreStepImpl fieldSetState : getFieldSetStates() ) { |
| 161 | + for ( SpatialWithinPredicateFieldMoreStepImpl<SR> fieldSetState : getFieldSetStates() ) { |
149 | 162 | fieldSetState.generateWithinBoundingBoxQueryBuilders( boundingBox );
|
150 | 163 | }
|
151 | 164 |
|
152 | 165 | return this;
|
153 | 166 | }
|
154 | 167 |
|
155 | 168 | @Override
|
156 |
| - protected CommonState thisAsS() { |
| 169 | + protected CommonState<SR> thisAsS() { |
157 | 170 | return this;
|
158 | 171 | }
|
159 | 172 | }
|
|
0 commit comments