Examples with QueryShardException org.opensearch.index.query.QueryShardException used on opensource projects

Example 16 with QueryShardException

use of org.opensearch.index.query.QueryShardException in project OpenSearch by opensearch-project.

the class FieldSortBuilder method build.

@Override
public SortFieldAndFormat build(QueryShardContext context) throws IOException {
    if (DOC_FIELD_NAME.equals(fieldName)) {
        return order == SortOrder.DESC ? SORT_DOC_REVERSE : SORT_DOC;
    }
    MappedFieldType fieldType = context.fieldMapper(fieldName);
    Nested nested = nested(context, fieldType);
    if (fieldType == null) {
        fieldType = resolveUnmappedType(context);
    }
    boolean reverse = order == SortOrder.DESC;
    IndexFieldData<?> fieldData = context.getForField(fieldType);
    if (fieldData instanceof IndexNumericFieldData == false && (sortMode == SortMode.SUM || sortMode == SortMode.AVG || sortMode == SortMode.MEDIAN)) {
        throw new QueryShardException(context, "we only support AVG, MEDIAN and SUM on number based fields");
    }
    final SortField field;
    boolean isNanosecond = false;
    if (numericType != null) {
        if (fieldData instanceof IndexNumericFieldData == false) {
            throw new QueryShardException(context, "[numeric_type] option cannot be set on a non-numeric field, got " + fieldType.typeName());
        }
        IndexNumericFieldData numericFieldData = (IndexNumericFieldData) fieldData;
        NumericType resolvedType = resolveNumericType(numericType);
        field = numericFieldData.sortField(resolvedType, missing, localSortMode(), nested, reverse);
        isNanosecond = resolvedType == NumericType.DATE_NANOSECONDS;
    } else {
        field = fieldData.sortField(missing, localSortMode(), nested, reverse);
        if (fieldData instanceof IndexNumericFieldData) {
            isNanosecond = ((IndexNumericFieldData) fieldData).getNumericType() == NumericType.DATE_NANOSECONDS;
        }
    }
    DocValueFormat format = fieldType.docValueFormat(null, null);
    if (isNanosecond) {
        format = DocValueFormat.withNanosecondResolution(format);
    }
    return new SortFieldAndFormat(field, format);
}

Example 17 with QueryShardException

use of org.opensearch.index.query.QueryShardException in project OpenSearch by opensearch-project.

the class ExceptionSerializationTests method testWriteThrowable.

public void testWriteThrowable() throws IOException {
    final QueryShardException queryShardException = new QueryShardException(new Index("foo", "_na_"), "foobar", null);
    final UnknownException unknownException = new UnknownException("this exception is unknown", queryShardException);
    final Exception[] causes = new Exception[] { new IllegalStateException("foobar"), new IllegalArgumentException("alalaal"), new NullPointerException("boom"), new EOFException("dadada"), new OpenSearchSecurityException("nono!"), new NumberFormatException("not a number"), new CorruptIndexException("baaaam booom", "this is my resource"), new IndexFormatTooNewException("tooo new", 1, 2, 3), new IndexFormatTooOldException("tooo new", 1, 2, 3), new IndexFormatTooOldException("tooo new", "very old version"), new ArrayIndexOutOfBoundsException("booom"), new StringIndexOutOfBoundsException("booom"), new FileNotFoundException("booom"), new NoSuchFileException("booom"), new AlreadyClosedException("closed!!", new NullPointerException()), new LockObtainFailedException("can't lock directory", new NullPointerException()), unknownException };
    for (final Exception cause : causes) {
        OpenSearchException ex = new OpenSearchException("topLevel", cause);
        OpenSearchException deserialized = serialize(ex);
        assertEquals(deserialized.getMessage(), ex.getMessage());
        assertTrue("Expected: " + deserialized.getCause().getMessage() + " to contain: " + ex.getCause().getClass().getName() + " but it didn't", deserialized.getCause().getMessage().contains(ex.getCause().getMessage()));
        if (ex.getCause().getClass() != UnknownException.class) {
            // unknown exception is not directly mapped
            assertEquals(deserialized.getCause().getClass(), ex.getCause().getClass());
        } else {
            assertEquals(deserialized.getCause().getClass(), NotSerializableExceptionWrapper.class);
        }
        assertArrayEquals(deserialized.getStackTrace(), ex.getStackTrace());
        assertTrue(deserialized.getStackTrace().length > 1);
    }
}

Example 18 with QueryShardException

use of org.opensearch.index.query.QueryShardException in project OpenSearch by opensearch-project.

the class ExceptionsHelperTests method createShardFailureQueryShardException.

private static ShardSearchFailure createShardFailureQueryShardException(String error, String indexName, String clusterAlias) {
    Index index = new Index(RemoteClusterAware.buildRemoteIndexName(clusterAlias, indexName), "uuid");
    QueryShardException queryShardException = new QueryShardException(index, error, new IllegalArgumentException("parse error"));
    return new ShardSearchFailure(queryShardException, null);
}

Example 19 with QueryShardException

use of org.opensearch.index.query.QueryShardException in project OpenSearch by opensearch-project.

the class QueryParserHelper method resolveMappingField.

/**
 * Resolves the provided pattern or field name from the {@link QueryShardContext} and return a map of
 * the expanded fields with their original boost.
 * @param context The context of the query
 * @param fieldOrPattern The field name or the pattern to resolve
 * @param weight The weight for the field
 * @param acceptAllTypes Whether all field type should be added when a pattern is expanded.
 *                       If false, only searchable field types are added.
 * @param acceptMetadataField Whether metadata fields should be added when a pattern is expanded.
 * @param fieldSuffix The suffix name to add to the expanded field names if a mapping exists for that name.
 *                    The original name of the field is kept if adding the suffix to the field name does not point to a valid field
 *                    in the mapping.
 */
static Map<String, Float> resolveMappingField(QueryShardContext context, String fieldOrPattern, float weight, boolean acceptAllTypes, boolean acceptMetadataField, String fieldSuffix) {
    Set<String> allFields = context.simpleMatchToIndexNames(fieldOrPattern);
    Map<String, Float> fields = new HashMap<>();
    for (String fieldName : allFields) {
        if (fieldSuffix != null && context.fieldMapper(fieldName + fieldSuffix) != null) {
            fieldName = fieldName + fieldSuffix;
        }
        MappedFieldType fieldType = context.getMapperService().fieldType(fieldName);
        if (fieldType == null) {
            continue;
        }
        if (acceptMetadataField == false && fieldType.name().startsWith("_")) {
            // Ignore metadata fields
            continue;
        }
        if (acceptAllTypes == false) {
            try {
                fieldType.termQuery("", context);
            } catch (QueryShardException | UnsupportedOperationException e) {
                // field type is never searchable with term queries (eg. geo point): ignore
                continue;
            } catch (IllegalArgumentException | OpenSearchParseException e) {
            // other exceptions are parsing errors or not indexed fields: keep
            }
        }
        // Deduplicate aliases and their concrete fields.
        String resolvedFieldName = fieldType.name();
        if (allFields.contains(resolvedFieldName)) {
            fieldName = resolvedFieldName;
        }
        float w = fields.getOrDefault(fieldName, 1.0F);
        fields.put(fieldName, w * weight);
    }
    return fields;
}

Example 20 with QueryShardException

use of org.opensearch.index.query.QueryShardException in project OpenSearch by opensearch-project.

the class FieldSortBuilderTests method testModeNonNumericField.

/**
 * Test that MIN, MAX mode work on non-numeric fields, but other modes throw exception
 */
public void testModeNonNumericField() throws IOException {
    QueryShardContext shardContextMock = createMockShardContext();
    FieldSortBuilder sortBuilder = new FieldSortBuilder(MAPPED_STRING_FIELDNAME).sortMode(SortMode.MIN);
    SortField sortField = sortBuilder.build(shardContextMock).field;
    assertThat(sortField, instanceOf(SortedSetSortField.class));
    assertEquals(SortedSetSelector.Type.MIN, ((SortedSetSortField) sortField).getSelector());
    sortBuilder = new FieldSortBuilder(MAPPED_STRING_FIELDNAME).sortMode(SortMode.MAX);
    sortField = sortBuilder.build(shardContextMock).field;
    assertThat(sortField, instanceOf(SortedSetSortField.class));
    assertEquals(SortedSetSelector.Type.MAX, ((SortedSetSortField) sortField).getSelector());
    String expectedError = "we only support AVG, MEDIAN and SUM on number based fields";
    QueryShardException e = expectThrows(QueryShardException.class, () -> new FieldSortBuilder(MAPPED_STRING_FIELDNAME).sortMode(SortMode.AVG).build(shardContextMock));
    assertEquals(expectedError, e.getMessage());
    e = expectThrows(QueryShardException.class, () -> new FieldSortBuilder(MAPPED_STRING_FIELDNAME).sortMode(SortMode.SUM).build(shardContextMock));
    assertEquals(expectedError, e.getMessage());
    e = expectThrows(QueryShardException.class, () -> new FieldSortBuilder(MAPPED_STRING_FIELDNAME).sortMode(SortMode.MEDIAN).build(shardContextMock));
    assertEquals(expectedError, e.getMessage());
}