update

Signed-off-by: Peng Huo <penghuo@gmail.com>
opensearch-project · Oct 28, 2022 · 26fc997 · 26fc997
1 parent 2422a5b
commit 26fc997
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 8 deletions.
diff --git a/integ-test/src/test/java/org/opensearch/sql/ppl/StandaloneIT.java b/integ-test/src/test/java/org/opensearch/sql/ppl/StandaloneIT.java
@@ -182,10 +182,16 @@ QueryManager queryManager() {
     @Bean
     @Scope(value = ConfigurableBeanFactory.SCOPE_PROTOTYPE)
     QueryPlanFactory queryExecutionFactory(BuiltinFunctionRepository functionRepository) {
-      catalogService.getCatalogs()
-          .forEach(catalog -> catalogService.getStorageEngine(catalog)
-              .getFunctions()
-              .forEach(functionResolver -> functionRepository.register(catalog, functionResolver)));
+      catalogService
+          .getCatalogs()
+          .forEach(
+              catalog ->
+                  catalog
+                      .getStorageEngine()
+                      .getFunctions()
+                      .forEach(
+                          functionResolver ->
+                              functionRepository.register(catalog.getName(), functionResolver)));
       Analyzer analyzer = new Analyzer(new ExpressionAnalyzer(functionRepository),
           catalogService, functionRepository);
       Planner planner =

diff --git a/plugin/src/main/java/org/opensearch/sql/plugin/config/OpenSearchPluginConfig.java b/plugin/src/main/java/org/opensearch/sql/plugin/config/OpenSearchPluginConfig.java
@@ -102,10 +102,16 @@ public QueryManager queryManager() {
   @Bean
   @Scope(value = ConfigurableBeanFactory.SCOPE_PROTOTYPE)
   public QueryPlanFactory queryExecutionFactory(BuiltinFunctionRepository functionRepository) {
-    catalogService.getCatalogs()
-        .forEach(catalog -> catalogService.getStorageEngine(catalog)
-            .getFunctions()
-            .forEach(functionResolver -> functionRepository.register(catalog, functionResolver)));
+    catalogService
+        .getCatalogs()
+        .forEach(
+            catalog ->
+                catalog
+                    .getStorageEngine()
+                    .getFunctions()
+                    .forEach(
+                        functionResolver ->
+                            functionRepository.register(catalog.getName(), functionResolver)));
     Analyzer analyzer = new Analyzer(new ExpressionAnalyzer(functionRepository),
         catalogService, functionRepository);
     Planner planner =