Package org.openmetadata.service.search.openSearch

Class OpenSearchClientImpl

java.lang.Object

org.openmetadata.service.search.openSearch.OpenSearchClientImpl

All Implemented Interfaces:

SearchClient

public class OpenSearchClientImpl
extends Object
implements SearchClient

Constructor Summary

Constructors

Constructor	Description
OpenSearchClientImpl(ElasticSearchConfiguration esConfig, CollectionDAO dao)

Method Summary

Modifier and Type	Method	Description
void	addTestCaseFromLogicalTestSuite(TestSuite testSuite, ChangeEvent event, SearchIndexDefinition.ElasticSearchIndexType indexType)
javax.ws.rs.core.Response	aggregate(String index, String fieldName, String value, String query)
org.opensearch.action.update.UpdateRequest	applyOSChangeEvent(ChangeEvent event)
org.opensearch.action.bulk.BulkResponse	bulk(org.opensearch.action.bulk.BulkRequest data, org.opensearch.client.RequestOptions options)
void	close()
boolean	createIndex(SearchIndexDefinition.ElasticSearchIndexType elasticSearchIndexType, String lang)
static org.opensearch.client.RestHighLevelClient	createOpenSearchClient(ElasticSearchConfiguration esConfig)
void	deleteIndex(SearchIndexDefinition.ElasticSearchIndexType elasticSearchIndexType)
CollectionDAO	getDao()
ElasticSearchConfiguration.SearchType	getSearchType()
TreeMap<Long,List<Object>>	getSortedDate(String team, Long scheduleTime, Long currentTime, DataInsightChartResult.DataInsightChartType chartType, String indexName)
int	getSuccessFromBulkResponse(org.opensearch.action.bulk.BulkResponse response)
javax.ws.rs.core.Response	listDataInsightChartResult(Long startTs, Long endTs, String tier, String team, DataInsightChartResult.DataInsightChartType dataInsightChartName, String dataReportIndex)
void	processTestCase(TestCase testCase, ChangeEvent event, SearchIndexDefinition.ElasticSearchIndexType indexType)
javax.ws.rs.core.Response	search(SearchRequest request)
javax.ws.rs.core.Response	searchBySourceUrl(String sourceUrl)
javax.ws.rs.core.Response	suggest(SearchRequest request)
void	updateClassification(ChangeEvent event)
void	updateDashboardService(ChangeEvent event)
void	updateDatabase(ChangeEvent event)
void	updateDatabaseSchema(ChangeEvent event)
void	updateDatabaseService(ChangeEvent event)
void	updateElasticSearch(org.opensearch.action.update.UpdateRequest updateRequest)
void	updateEntity(ChangeEvent event)
void	updateGlossary(ChangeEvent event)
void	updateGlossaryTerm(ChangeEvent event)
void	updateIndex(SearchIndexDefinition.ElasticSearchIndexType elasticSearchIndexType, String lang)
void	updateMessagingService(ChangeEvent event)
void	updateMlModelService(ChangeEvent event)
void	updatePipelineService(ChangeEvent event)
void	updateSearchForEntityCreated(SearchIndexDefinition.ElasticSearchIndexType indexType, String entityType, ChangeEvent event)
void	updateStorageService(ChangeEvent event)
void	updateTag(ChangeEvent event)
void	updateTeam(ChangeEvent event)
void	updateTestSuite(ChangeEvent event)
void	updateUser(ChangeEvent event)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.openmetadata.service.search.SearchClient

applyESChangeEvent, bulk, getIndexMapping, getScriptWithParams, getSuccessFromBulkResponse, updateElasticSearch, updateElasticSearchFailureStatus, updateSearchForEntityUpdated, updateTestCase

Constructor Detail

OpenSearchClientImpl

public OpenSearchClientImpl(ElasticSearchConfiguration esConfig,
                            CollectionDAO dao)

Method Detail

createIndex

public boolean createIndex(SearchIndexDefinition.ElasticSearchIndexType elasticSearchIndexType,
                           String lang)

Specified by:

createIndex in interface SearchClient

updateIndex

public void updateIndex(SearchIndexDefinition.ElasticSearchIndexType elasticSearchIndexType,
                        String lang)

Specified by:

updateIndex in interface SearchClient

deleteIndex

public void deleteIndex(SearchIndexDefinition.ElasticSearchIndexType elasticSearchIndexType)

Specified by:

deleteIndex in interface SearchClient

search

public javax.ws.rs.core.Response search(SearchRequest request)
                                 throws IOException

Specified by:

search in interface SearchClient

Throws:

IOException

searchBySourceUrl

public javax.ws.rs.core.Response searchBySourceUrl(String sourceUrl)
                                            throws IOException

Specified by:

searchBySourceUrl in interface SearchClient

Parameters:

sourceUrl -

Returns:

Throws:

IOException

aggregate

public javax.ws.rs.core.Response aggregate(String index,
                                           String fieldName,
                                           String value,
                                           String query)
                                    throws IOException

Specified by:

aggregate in interface SearchClient

Throws:

IOException

updateElasticSearch

public void updateElasticSearch(org.opensearch.action.update.UpdateRequest updateRequest)
                         throws IOException

Specified by:

updateElasticSearch in interface SearchClient

Throws:

IOException

suggest

public javax.ws.rs.core.Response suggest(SearchRequest request)
                                  throws IOException

Specified by:

suggest in interface SearchClient

Throws:

IOException

getSearchType

public ElasticSearchConfiguration.SearchType getSearchType()

Specified by:

getSearchType in interface SearchClient

updateEntity

public void updateEntity(ChangeEvent event)
                  throws IOException

Specified by:

updateEntity in interface SearchClient

Throws:

IOException

updateUser

public void updateUser(ChangeEvent event)
                throws IOException

Specified by:

updateUser in interface SearchClient

Throws:

IOException

updateSearchForEntityCreated

public void updateSearchForEntityCreated(SearchIndexDefinition.ElasticSearchIndexType indexType,
                                         String entityType,
                                         ChangeEvent event)
                                  throws IOException

Specified by:

updateSearchForEntityCreated in interface SearchClient

Throws:

IOException

updateTeam

public void updateTeam(ChangeEvent event)
                throws IOException

Specified by:

updateTeam in interface SearchClient

Throws:

IOException

updateGlossaryTerm

public void updateGlossaryTerm(ChangeEvent event)
                        throws IOException

Specified by:

updateGlossaryTerm in interface SearchClient

Throws:

IOException

updateGlossary

public void updateGlossary(ChangeEvent event)
                    throws IOException

Specified by:

updateGlossary in interface SearchClient

Throws:

IOException

updateTag

public void updateTag(ChangeEvent event)
               throws IOException

Specified by:

updateTag in interface SearchClient

Throws:

IOException

updateDatabase

public void updateDatabase(ChangeEvent event)
                    throws IOException

Specified by:

updateDatabase in interface SearchClient

Throws:

IOException

updateDatabaseSchema

public void updateDatabaseSchema(ChangeEvent event)
                          throws IOException

Specified by:

updateDatabaseSchema in interface SearchClient

Throws:

IOException

updateDatabaseService

public void updateDatabaseService(ChangeEvent event)
                           throws IOException

Specified by:

updateDatabaseService in interface SearchClient

Throws:

IOException

updatePipelineService

public void updatePipelineService(ChangeEvent event)
                           throws IOException

Specified by:

updatePipelineService in interface SearchClient

Throws:

IOException

updateMlModelService

public void updateMlModelService(ChangeEvent event)
                          throws IOException

Specified by:

updateMlModelService in interface SearchClient

Throws:

IOException

updateStorageService

public void updateStorageService(ChangeEvent event)
                          throws IOException

Specified by:

updateStorageService in interface SearchClient

Throws:

IOException

updateMessagingService

public void updateMessagingService(ChangeEvent event)
                            throws IOException

Specified by:

updateMessagingService in interface SearchClient

Throws:

IOException

updateDashboardService

public void updateDashboardService(ChangeEvent event)
                            throws IOException

Specified by:

updateDashboardService in interface SearchClient

Throws:

IOException

updateClassification

public void updateClassification(ChangeEvent event)
                          throws IOException

Specified by:

updateClassification in interface SearchClient

Throws:

IOException

updateTestSuite

public void updateTestSuite(ChangeEvent event)
                     throws IOException

Specified by:

updateTestSuite in interface SearchClient

Throws:

IOException

processTestCase

public void processTestCase(TestCase testCase,
                            ChangeEvent event,
                            SearchIndexDefinition.ElasticSearchIndexType indexType)
                     throws IOException

Specified by:

processTestCase in interface SearchClient

Throws:

IOException

addTestCaseFromLogicalTestSuite

public void addTestCaseFromLogicalTestSuite(TestSuite testSuite,
                                            ChangeEvent event,
                                            SearchIndexDefinition.ElasticSearchIndexType indexType)
                                     throws IOException

Specified by:

addTestCaseFromLogicalTestSuite in interface SearchClient

Throws:

IOException

close

public void close()

Specified by:

close in interface SearchClient

applyOSChangeEvent

public org.opensearch.action.update.UpdateRequest applyOSChangeEvent(ChangeEvent event)

Specified by:

applyOSChangeEvent in interface SearchClient

bulk

public org.opensearch.action.bulk.BulkResponse bulk(org.opensearch.action.bulk.BulkRequest data,
                                                    org.opensearch.client.RequestOptions options)
                                             throws IOException

Specified by:

bulk in interface SearchClient

Throws:

IOException

getSuccessFromBulkResponse

public int getSuccessFromBulkResponse(org.opensearch.action.bulk.BulkResponse response)

Specified by:

getSuccessFromBulkResponse in interface SearchClient

getSortedDate

public TreeMap<Long,List<Object>> getSortedDate(String team,
                                                      Long scheduleTime,
                                                      Long currentTime,
                                                      DataInsightChartResult.DataInsightChartType chartType,
                                                      String indexName)
                                               throws IOException,
                                                      ParseException

Specified by:

getSortedDate in interface SearchClient

Throws:

IOException

ParseException

listDataInsightChartResult

public javax.ws.rs.core.Response listDataInsightChartResult(Long startTs,
                                                            Long endTs,
                                                            String tier,
                                                            String team,
                                                            DataInsightChartResult.DataInsightChartType dataInsightChartName,
                                                            String dataReportIndex)
                                                     throws IOException,
                                                            ParseException

Specified by:

listDataInsightChartResult in interface SearchClient

Throws:

IOException

ParseException

getDao

public CollectionDAO getDao()

Specified by:

getDao in interface SearchClient

createOpenSearchClient

public static org.opensearch.client.RestHighLevelClient createOpenSearchClient(ElasticSearchConfiguration esConfig)