diff --git a/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/SQLServerUtils.java b/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/SQLServerUtils.java index 3f0e7c88cf15bb26c37d1ecce3e5f49bf927915a..a753caba48152678e4ea15f82538d830f48201d8 100644 --- a/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/SQLServerUtils.java +++ b/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/SQLServerUtils.java @@ -144,13 +144,11 @@ public class SQLServerUtils { } } - public static String extractSource(@NotNull DBRProgressMonitor monitor, @NotNull SQLServerDatabase database, @Nullable SQLServerSchema schema, @NotNull String objectName) throws DBException { - SQLServerDataSource dataSource = database.getDataSource(); - String systemSchema = getSystemSchemaFQN(dataSource, database.getName(), SQLServerConstants.SQL_SERVER_SYSTEM_SCHEMA); + public static String extractSource(@NotNull DBRProgressMonitor monitor, @NotNull SQLServerSchema schema, @NotNull String objectName) throws DBException { + SQLServerDataSource dataSource = schema.getDataSource(); + String systemSchema = getSystemSchemaFQN(dataSource, schema.getDatabase().getName(), SQLServerConstants.SQL_SERVER_SYSTEM_SCHEMA); try (JDBCSession session = DBUtils.openMetaSession(monitor, dataSource, "Read source code")) { - String objectFQN = schema == null ? - DBUtils.getQuotedIdentifier(dataSource, objectName) : - DBUtils.getQuotedIdentifier(dataSource, schema.getName()) + "." + DBUtils.getQuotedIdentifier(dataSource, objectName); + String objectFQN = DBUtils.getQuotedIdentifier(dataSource, schema.getName()) + "." + DBUtils.getQuotedIdentifier(dataSource, objectName); try (JDBCPreparedStatement dbStat = session.prepareStatement(systemSchema + ".sp_helptext '" + objectFQN + "'")) { try (JDBCResultSet dbResult = dbStat.executeQuery()) { StringBuilder sql = new StringBuilder(); diff --git a/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerDataType.java b/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerDataType.java index 30ff795e942ae85bf7d326a9453f5bfd85eadb9a..030a8cb8ec81eaaa513efcda48ba2560448e840f 100644 --- a/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerDataType.java +++ b/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerDataType.java @@ -96,6 +96,7 @@ public class SQLServerDataType implements DBSDataType, SQLServerObject, DBPQuali this.valueType = valueType; } + @NotNull @Override public SQLServerDataSource getDataSource() { return (SQLServerDataSource) owner.getDataSource(); diff --git a/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerObject.java b/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerObject.java index d2cc328dca2df01dde01b719beff8dfde87e9bed..a562159c92764f9619052add7b1bee023461ede0 100644 --- a/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerObject.java +++ b/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerObject.java @@ -16,6 +16,7 @@ */ package org.jkiss.dbeaver.ext.mssql.model; +import org.jkiss.code.NotNull; import org.jkiss.dbeaver.model.DBPNamedObject; import org.jkiss.dbeaver.model.DBPObjectWithLongId; import org.jkiss.dbeaver.model.struct.DBSObject; @@ -26,5 +27,6 @@ import org.jkiss.dbeaver.model.struct.DBSObject; */ public interface SQLServerObject extends DBPNamedObject, DBSObject, DBPObjectWithLongId { - + @NotNull + SQLServerDataSource getDataSource(); } diff --git a/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerProcedure.java b/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerProcedure.java index c08a3b5650bac268eac93cbc7a3646523a7312c5..71285c918edb7c6b003d27c9c344e8def6af2b92 100644 --- a/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerProcedure.java +++ b/plugins/org.jkiss.dbeaver.ext.mssql/src/org/jkiss/dbeaver/ext/mssql/model/SQLServerProcedure.java @@ -137,7 +137,7 @@ public class SQLServerProcedure extends AbstractProcedure