Examples with SearchModule org.opensearch.search.SearchModule used on opensource projects

Example 21 with SearchModule

use of org.opensearch.search.SearchModule in project OpenSearch by opensearch-project.

the class SortBuilderTests method init.

@BeforeClass
public static void init() {
    SearchModule searchModule = new SearchModule(Settings.EMPTY, emptyList());
    xContentRegistry = new NamedXContentRegistry(searchModule.getNamedXContents());
}

Example 22 with SearchModule

use of org.opensearch.search.SearchModule in project OpenSearch by opensearch-project.

the class ExplainRequestTests method setUp.

public void setUp() throws Exception {
    super.setUp();
    IndicesModule indicesModule = new IndicesModule(Collections.emptyList());
    SearchModule searchModule = new SearchModule(Settings.EMPTY, Collections.emptyList());
    List<NamedWriteableRegistry.Entry> entries = new ArrayList<>();
    entries.addAll(indicesModule.getNamedWriteables());
    entries.addAll(searchModule.getNamedWriteables());
    namedWriteableRegistry = new NamedWriteableRegistry(entries);
}

Example 23 with SearchModule

use of org.opensearch.search.SearchModule in project OpenSearch by opensearch-project.

the class BasePipelineAggregationTestCase method setUp.

/**
 * Setup for the whole base test class.
 */
@Override
public void setUp() throws Exception {
    super.setUp();
    Settings settings = Settings.builder().put("node.name", AbstractQueryTestCase.class.toString()).put(Environment.PATH_HOME_SETTING.getKey(), createTempDir()).build();
    IndicesModule indicesModule = new IndicesModule(Collections.emptyList());
    SearchModule searchModule = new SearchModule(settings, plugins());
    List<NamedWriteableRegistry.Entry> entries = new ArrayList<>();
    entries.addAll(indicesModule.getNamedWriteables());
    entries.addAll(searchModule.getNamedWriteables());
    entries.addAll(additionalNamedWriteables());
    namedWriteableRegistry = new NamedWriteableRegistry(entries);
    List<NamedXContentRegistry.Entry> xContentEntries = searchModule.getNamedXContents();
    xContentEntries.addAll(additionalNamedContents());
    xContentRegistry = new NamedXContentRegistry(xContentEntries);
    // create some random type with some default field, those types will stick around for all of the subclasses
    currentTypes = new String[randomIntBetween(0, 5)];
    for (int i = 0; i < currentTypes.length; i++) {
        String type = randomAlphaOfLengthBetween(1, 10);
        currentTypes[i] = type;
    }
}

Example 24 with SearchModule

use of org.opensearch.search.SearchModule in project OpenSearch by opensearch-project.

the class ClusterSearchShardsResponseTests method testSerialization.

public void testSerialization() throws Exception {
    Map<String, AliasFilter> indicesAndFilters = new HashMap<>();
    Set<DiscoveryNode> nodes = new HashSet<>();
    int numShards = randomIntBetween(1, 10);
    ClusterSearchShardsGroup[] clusterSearchShardsGroups = new ClusterSearchShardsGroup[numShards];
    for (int i = 0; i < numShards; i++) {
        String index = randomAlphaOfLengthBetween(3, 10);
        ShardId shardId = new ShardId(index, randomAlphaOfLength(12), i);
        String nodeId = randomAlphaOfLength(10);
        ShardRouting shardRouting = TestShardRouting.newShardRouting(shardId, nodeId, randomBoolean(), ShardRoutingState.STARTED);
        clusterSearchShardsGroups[i] = new ClusterSearchShardsGroup(shardId, new ShardRouting[] { shardRouting });
        DiscoveryNode node = new DiscoveryNode(shardRouting.currentNodeId(), new TransportAddress(TransportAddress.META_ADDRESS, randomInt(0xFFFF)), VersionUtils.randomVersion(random()));
        nodes.add(node);
        AliasFilter aliasFilter;
        if (randomBoolean()) {
            aliasFilter = new AliasFilter(RandomQueryBuilder.createQuery(random()), "alias-" + index);
        } else {
            aliasFilter = new AliasFilter(null, Strings.EMPTY_ARRAY);
        }
        indicesAndFilters.put(index, aliasFilter);
    }
    ClusterSearchShardsResponse clusterSearchShardsResponse = new ClusterSearchShardsResponse(clusterSearchShardsGroups, nodes.toArray(new DiscoveryNode[nodes.size()]), indicesAndFilters);
    SearchModule searchModule = new SearchModule(Settings.EMPTY, Collections.emptyList());
    List<NamedWriteableRegistry.Entry> entries = new ArrayList<>();
    entries.addAll(searchModule.getNamedWriteables());
    NamedWriteableRegistry namedWriteableRegistry = new NamedWriteableRegistry(entries);
    Version version = VersionUtils.randomIndexCompatibleVersion(random());
    try (BytesStreamOutput out = new BytesStreamOutput()) {
        out.setVersion(version);
        clusterSearchShardsResponse.writeTo(out);
        try (StreamInput in = new NamedWriteableAwareStreamInput(out.bytes().streamInput(), namedWriteableRegistry)) {
            in.setVersion(version);
            ClusterSearchShardsResponse deserialized = new ClusterSearchShardsResponse(in);
            assertArrayEquals(clusterSearchShardsResponse.getNodes(), deserialized.getNodes());
            assertEquals(clusterSearchShardsResponse.getGroups().length, deserialized.getGroups().length);
            for (int i = 0; i < clusterSearchShardsResponse.getGroups().length; i++) {
                ClusterSearchShardsGroup clusterSearchShardsGroup = clusterSearchShardsResponse.getGroups()[i];
                ClusterSearchShardsGroup deserializedGroup = deserialized.getGroups()[i];
                assertEquals(clusterSearchShardsGroup.getShardId(), deserializedGroup.getShardId());
                assertArrayEquals(clusterSearchShardsGroup.getShards(), deserializedGroup.getShards());
            }
            assertEquals(clusterSearchShardsResponse.getIndicesAndFilters(), deserialized.getIndicesAndFilters());
        }
    }
}

Example 25 with SearchModule

use of org.opensearch.search.SearchModule in project OpenSearch by opensearch-project.

the class QueryBuilderBWCIT method testQueryBuilderBWC.

public void testQueryBuilderBWC() throws Exception {
    final String type = getOldClusterVersion().before(LegacyESVersion.V_7_0_0) ? "doc" : "_doc";
    String index = "queries";
    if (isRunningAgainstOldCluster()) {
        XContentBuilder mappingsAndSettings = jsonBuilder();
        mappingsAndSettings.startObject();
        {
            mappingsAndSettings.startObject("settings");
            mappingsAndSettings.field("number_of_shards", 1);
            mappingsAndSettings.field("number_of_replicas", 0);
            mappingsAndSettings.endObject();
        }
        {
            mappingsAndSettings.startObject("mappings");
            if (isRunningAgainstAncientCluster()) {
                mappingsAndSettings.startObject(type);
            }
            mappingsAndSettings.startObject("properties");
            {
                mappingsAndSettings.startObject("query");
                mappingsAndSettings.field("type", "percolator");
                mappingsAndSettings.endObject();
            }
            {
                mappingsAndSettings.startObject("keyword_field");
                mappingsAndSettings.field("type", "keyword");
                mappingsAndSettings.endObject();
            }
            {
                mappingsAndSettings.startObject("long_field");
                mappingsAndSettings.field("type", "long");
                mappingsAndSettings.endObject();
            }
            mappingsAndSettings.endObject();
            mappingsAndSettings.endObject();
            if (isRunningAgainstAncientCluster()) {
                mappingsAndSettings.endObject();
            }
        }
        mappingsAndSettings.endObject();
        Request request = new Request("PUT", "/" + index);
        request.setOptions(allowTypesRemovalWarnings());
        request.setJsonEntity(Strings.toString(mappingsAndSettings));
        Response rsp = client().performRequest(request);
        assertEquals(200, rsp.getStatusLine().getStatusCode());
        for (int i = 0; i < CANDIDATES.size(); i++) {
            request = new Request("PUT", "/" + index + "/" + type + "/" + Integer.toString(i));
            request.setJsonEntity((String) CANDIDATES.get(i)[0]);
            rsp = client().performRequest(request);
            assertEquals(201, rsp.getStatusLine().getStatusCode());
        }
    } else {
        NamedWriteableRegistry registry = new NamedWriteableRegistry(new SearchModule(Settings.EMPTY, Collections.emptyList()).getNamedWriteables());
        for (int i = 0; i < CANDIDATES.size(); i++) {
            QueryBuilder expectedQueryBuilder = (QueryBuilder) CANDIDATES.get(i)[1];
            Request request = new Request("GET", "/" + index + "/_search");
            request.setJsonEntity("{\"query\": {\"ids\": {\"values\": [\"" + Integer.toString(i) + "\"]}}, " + "\"docvalue_fields\": [{\"field\":\"query.query_builder_field\"}]}");
            Response rsp = client().performRequest(request);
            assertEquals(200, rsp.getStatusLine().getStatusCode());
            Map<?, ?> hitRsp = (Map<?, ?>) ((List<?>) ((Map<?, ?>) toMap(rsp).get("hits")).get("hits")).get(0);
            String queryBuilderStr = (String) ((List<?>) ((Map<?, ?>) hitRsp.get("fields")).get("query.query_builder_field")).get(0);
            byte[] qbSource = Base64.getDecoder().decode(queryBuilderStr);
            try (InputStream in = new ByteArrayInputStream(qbSource, 0, qbSource.length)) {
                try (StreamInput input = new NamedWriteableAwareStreamInput(new InputStreamStreamInput(in), registry)) {
                    input.setVersion(getOldClusterVersion());
                    QueryBuilder queryBuilder = input.readNamedWriteable(QueryBuilder.class);
                    assert in.read() == -1;
                    assertEquals(expectedQueryBuilder, queryBuilder);
                }
            }
        }
    }
}