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

Question

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

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

我使用本指南为 SSMS 创建了一个扩展:https://stackoverflow.com/a/55661807/2972

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

但没有找到有用的东西。

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

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

Answer 1

如果其他人想要这样做，可以通过读取私有(private)字段等以相当 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);
}