diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/DatasourceContextServiceCEImpl.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/DatasourceContextServiceCEImpl.java index adc1d2ea82bb..69632105e0d3 100644 --- a/app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/DatasourceContextServiceCEImpl.java +++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/DatasourceContextServiceCEImpl.java @@ -384,6 +384,11 @@ public boolean isValidDatasourceContextAvailable( && !isInErrorState; } + /** + * This method is deprecated and should not be used. Use the method {@link #getDatasourceContext(DatasourceStorage, Plugin)} instead. + * such entities as possible candidates for domain mapping. + */ + @Deprecated @Override public Mono> getDatasourceContext(DatasourceStorage datasourceStorage) { final String datasourceId = datasourceStorage.getDatasourceId(); @@ -452,9 +457,12 @@ public Mono> getDatasourceContext(DatasourceStorage datasou @Override public Mono retryOnce(DatasourceStorage datasourceStorage, Function, Mono> task) { - final Mono taskRunnerMono = Mono.justOrEmpty(datasourceStorage) - .flatMap(this::getDatasourceContext) + .flatMap(ds -> { + return pluginService + .findById(datasourceStorage.getPluginId()) + .flatMap(plugin -> getDatasourceContext(datasourceStorage, plugin)); + }) // Now that we have the context (connection details), call the task. .flatMap(task);