opensource/cloudbeaver
3
0
Fork 0
mirror of https://github.com/dbeaver/cloudbeaver.git synced 2026-09-24 16:04:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

CB-1112 Context list API

Browse Source
This commit is contained in:
Serge Rider
2021-07-12 16:46:58 +03:00
parent 41f30ee440
commit 9bccf6d9ce
3 changed files with 31 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/bundles/io.cloudbeaver.server/schema/service.sql.graphqls
+2 -1
View File
@@ -169,7 +169,8 @@ extend type Query {
sqlDialectInfo( connectionId: ID! ): SQLDialectInfo
sqlListContexts( connectionId: ID! ): [ SQLContextInfo ]
# Lists SQL contexts for a connection (optional) or returns the particular context info
sqlListContexts( connectionId: ID, contextId: ID ): [ SQLContextInfo ]
sqlCompletionProposals(
connectionId: ID!,
server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/DBWServiceSQL.java
+3
View File
@@ -34,6 +34,9 @@ import java.util.Map;
*/
public interface DBWServiceSQL extends DBWService {
@WebAction
WebSQLContextInfo[] listContexts(@NotNull WebSession session, @Nullable String connectionId, @Nullable String contextId) throws DBWebException;
@WebAction
WebSQLDialectInfo getDialectInfo(@NotNull WebSQLProcessor processor) throws DBWebException;
server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/impl/WebServiceSQL.java
+26
View File
@@ -20,8 +20,10 @@ package io.cloudbeaver.service.sql.impl;
import io.cloudbeaver.DBWebException;
import io.cloudbeaver.WebAction;
import io.cloudbeaver.model.WebAsyncTaskInfo;
import io.cloudbeaver.model.WebConnectionInfo;
import io.cloudbeaver.model.session.WebAsyncTaskProcessor;
import io.cloudbeaver.model.session.WebSession;
import io.cloudbeaver.service.WebServiceBindingBase;
import io.cloudbeaver.service.sql.*;
import org.eclipse.jface.text.Document;
import org.jkiss.code.NotNull;
@@ -45,6 +47,8 @@ import org.jkiss.dbeaver.model.struct.DBSDataContainer;
import org.jkiss.utils.CommonUtils;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
@@ -53,6 +57,28 @@ import java.util.Map;
*/
public class WebServiceSQL implements DBWServiceSQL {
@Override
public WebSQLContextInfo[] listContexts(@NotNull WebSession session, @Nullable String connectionId, @Nullable String contextId) throws DBWebException {
List<WebConnectionInfo> conToRead = new ArrayList<>();
if (connectionId != null) {
WebConnectionInfo webConnection = WebServiceBindingBase.getWebConnection(session, connectionId);
conToRead.add(webConnection);
} else {
conToRead.addAll(session.getConnections());
}
List<WebSQLContextInfo> contexts = new ArrayList<>();
for (WebConnectionInfo con : conToRead) {
WebSQLProcessor sqlProcessor = WebServiceBindingSQL.getSQLProcessor(con);
WebSQLContextInfo[] conContexts = sqlProcessor.getContexts();
contexts.addAll(Arrays.asList(conContexts));
}
if (contextId != null) {
contexts.removeIf(c -> c.getId().equals(contextId));
}
return contexts.toArray(new WebSQLContextInfo[0]);
}
@Override
@NotNull
public WebSQLDialectInfo getDialectInfo(@NotNull WebSQLProcessor processor) throws DBWebException {