VSIX 在 SQL Server Management Studio 的扩展中获取查询结果

Question

我想在 SQL Server Management Studio 中可视化查询结果。

为此，我需要访问查询结果的数据（正在显示的基础数据）。

我使用本指南为 SSMS 创建了一个扩展：

但是，扩展开发的文档很差。我试图在一些 Interop 命名空间中发现类，例如：https://docs.microsoft.com/en-us/dotnet/api/microsoft.visualstudio.textmanager.interop?view=visualstudiosdk-2019

但是没有找到有用的东西。

如何从命令扩展中获取对 SSMS 中查询结果本身的引用？

一般来说，我如何发现 IDE 的“元素”背后的“服务类型”？

Answer 1

其实，SSMS与ide有一些相似之处，例如，它们使用了相同的菜单控件的一部分。

如果你想在ssms上创建一个菜单命令，你可以参考this document about GUIDs and IDs。

还有，参考this document about how to create a menu。

Besides，SSMS sdk api在这里：

.NET API 列表，筛选到 sql-smo 库：https://docs.microsoft.com/en-us/dotnet/api/?view=sql-smo-160
.NET API 列表，筛选到 SQL Server 2019 库：https://docs.microsoft.com/en-us/dotnet/api/?view=sqlserver-2019

Answer 2

万一其他人想要这样做，可以通过读取私有字段等以相当 hacky 的方式来完成。

我的代码在这里：https://github.com/martinnormark/DataDive/blob/f1faa0621a7438cbe4f8d8b3455f7b3a16635be6/src/DataDive/FacetsDiveCommand.cs#L136-L192

复制此处供参考：

var objType = ServiceCache.ScriptFactory.GetType();
var method1 = objType.GetMethod("GetCurrentlyActiveFrameDocView", BindingFlags.NonPublic | BindingFlags.Instance);
var Result = method1.Invoke(ServiceCache.ScriptFactory, new object[] { ServiceCache.VSMonitorSelection, false, null });


var objType2 = Result.GetType();
var field = objType2.GetField("m_sqlResultsControl", BindingFlags.NonPublic | BindingFlags.Instance);
var SQLResultsControl = field.GetValue(Result);


var m_gridResultsPage = GetNonPublicField(SQLResultsControl, "m_gridResultsPage");
CollectionBase gridContainers = GetNonPublicField(m_gridResultsPage, "m_gridContainers") as CollectionBase;


foreach (var gridContainer in gridContainers)
{
    var grid = GetNonPublicField(gridContainer, "m_grid") as GridControl;
    var gridStorage = grid.GridStorage;
    var schemaTable = GetNonPublicField(gridStorage, "m_schemaTable") as DataTable;


    var data = new DataTable();


    for (long i = 0; i < gridStorage.NumRows(); i++)
    {
        var rowItems = new List<object>();


        for (int c = 0; c < schemaTable.Rows.Count; c++)
        {
            var columnName = schemaTable.Rows[c][0].ToString();
            var columnType = schemaTable.Rows[c][12] as Type;


            if (!data.Columns.Contains(columnName))
            {
                data.Columns.Add(columnName, columnType);
            }


            var cellData = gridStorage.GetCellDataAsString(i, c + 1);


            if (cellData == "NULL")
            {
                rowItems.Add(null);


                continue;
            }


            if (columnType == typeof(bool))
            {
                cellData = cellData == "0" ? "False" : "True";
            }


            Console.WriteLine($"Parsing {columnName} with '{cellData}'");
            var typedValue = Convert.ChangeType(cellData, columnType, CultureInfo.InvariantCulture);


            rowItems.Add(typedValue);
        }


        data.Rows.Add(rowItems.ToArray());
    }


    data.AcceptChanges();
}

public object GetNonPublicField(object obj, string field)
{
    FieldInfo f = obj.GetType().GetField(field, BindingFlags.NonPublic | BindingFlags.Instance);

    return f.GetValue(obj);
}

VSIX 在 SQL Server Management Studio 的扩展中获取查询结果

VSIX Get query results in extension for SQL Server Management Studio

ssms

visual-studio

vsix

visual-studio-extensions