dbeaver/dbeaver#35475 Fix sql script execution

dbeaver · Oct 7, 2024 · 1f2abd3 · 1f2abd3
1 parent 0e32052
commit 1f2abd3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/WebSQLProcessor.java b/server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/WebSQLProcessor.java
@@ -89,7 +89,7 @@ public class WebSQLProcessor implements WebSessionProvider {
             connection.getDataSource().getSQLDialect(), connection.getDataSourceContainer().getPreferenceStore());
 
         ruleManager = new SQLRuleManager(syntaxManager);
-        ruleManager.loadRules(connection.getDataSource(), false);
+        ruleManager.loadRules(connection.getDataSourceContainer(), false);
     }
 
     void dispose() {
@@ -192,7 +192,7 @@ public WebSQLExecuteInfo processQuery(
             document.set(sql);
 
             SQLParserContext parserContext = new SQLParserContext(
-                context.getDataSource(),
+                context.getDataSource().getContainer(),
                 syntaxManager,
                 ruleManager,
                 document);

diff --git a/server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/impl/WebServiceSQL.java b/server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/impl/WebServiceSQL.java
@@ -149,8 +149,9 @@ public WebSQLCompletionProposal[] getCompletionProposals(
             SQLScriptElement activeQuery;
 
             if (position != null) {
+                DBPDataSource source = sqlContext.getProcessor().getConnection().getDataSource();
                 SQLParserContext parserContext = new SQLParserContext(
-                    sqlContext.getProcessor().getConnection().getDataSource(),
+                    source != null ? source.getContainer() : null,
                     completionContext.getSyntaxManager(),
                     completionContext.getRuleManager(),
                     document);