Examples with AnalyzerProvider org.opensearch.index.analysis.AnalyzerProvider used on opensource projects

Example 1 with AnalyzerProvider

use of org.opensearch.index.analysis.AnalyzerProvider in project OpenSearch by opensearch-project.

the class IndexModuleTests method testIndexAnalyzersCleanedUpIfIndexServiceCreationFails.

public void testIndexAnalyzersCleanedUpIfIndexServiceCreationFails() {
    Settings settings = Settings.builder().put(Environment.PATH_HOME_SETTING.getKey(), createTempDir().toString()).put(IndexMetadata.SETTING_VERSION_CREATED, Version.CURRENT).build();
    final IndexSettings indexSettings = IndexSettingsModule.newIndexSettings("foo", settings);
    final HashSet<Analyzer> openAnalyzers = new HashSet<>();
    final AnalysisModule.AnalysisProvider<AnalyzerProvider<?>> analysisProvider = (i, e, n, s) -> new AnalyzerProvider<Analyzer>() {

        @Override
        public String name() {
            return "test";
        }

        @Override
        public AnalyzerScope scope() {
            return AnalyzerScope.INDEX;
        }

        @Override
        public Analyzer get() {
            final Analyzer analyzer = new Analyzer() {

                @Override
                protected TokenStreamComponents createComponents(String fieldName) {
                    return new TokenStreamComponents(new StandardTokenizer());
                }

                @Override
                public void close() {
                    super.close();
                    openAnalyzers.remove(this);
                }
            };
            openAnalyzers.add(analyzer);
            return analyzer;
        }
    };
    final AnalysisRegistry analysisRegistry = new AnalysisRegistry(environment, emptyMap(), emptyMap(), emptyMap(), singletonMap("test", analysisProvider), emptyMap(), emptyMap(), emptyMap(), emptyMap(), emptyMap());
    IndexModule module = createIndexModule(indexSettings, analysisRegistry);
    // causes index service creation to fail
    threadPool.shutdown();
    expectThrows(OpenSearchRejectedExecutionException.class, () -> newIndexService(module));
    assertThat(openAnalyzers, empty());
}