Get point out of query and into value

Signed-off-by: Harsha Vamsi Kalluri <[email protected]>
opensearch-project · Jan 24, 2024 · 3934e9c · 3934e9c
1 parent 802c462
commit 3934e9c
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/server/src/main/java/org/opensearch/index/mapper/IpFieldMapper.java b/server/src/main/java/org/opensearch/index/mapper/IpFieldMapper.java
@@ -243,21 +243,22 @@ public Query termQuery(Object value, @Nullable QueryShardContext context) {
             }
             if (isSearchable() && hasDocValues()) {
                 String term = value.toString();
-                if (value instanceof String) value = new BytesRef(((String) value));
                 if (term.contains("/")) {
                     final Tuple<InetAddress, Integer> cidr = InetAddresses.parseCidr(term);
                     return InetAddressPoint.newPrefixQuery(name(), cidr.v1(), cidr.v2());
                 }
-                return new IndexOrDocValuesQuery(query, SortedSetDocValuesField.newSlowExactQuery(name(), (BytesRef) (value)));
+                return new IndexOrDocValuesQuery(
+                    query,
+                    SortedSetDocValuesField.newSlowExactQuery(name(), new BytesRef(((PointRangeQuery) query).getLowerPoint()))
+                );
             }
             if (hasDocValues()) {
                 String term = value.toString();
-                if (value instanceof String) value = new BytesRef(((String) value));
                 if (term.contains("/")) {
                     final Tuple<InetAddress, Integer> cidr = InetAddresses.parseCidr(term);
                     return InetAddressPoint.newPrefixQuery(name(), cidr.v1(), cidr.v2());
                 }
-                return SortedSetDocValuesField.newSlowExactQuery(name(), ((BytesRef) value));
+                return SortedSetDocValuesField.newSlowExactQuery(name(), new BytesRef(((PointRangeQuery) query).getLowerPoint()));
             }
             return query;
         }