VSIX以SQL Server Management Studio扩展名获取查询结果

问题描述

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

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

我已使用以下指南为SSMS创建了扩展程序：https://stackoverflow.com/a/55661807/2972

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

但是没有发现有用的东西。

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

通常，我如何发现IDE的“元素”后面是哪种“服务类型”？

解决方法

万一其他人想这样做，可以通过读取私有字段等方式以一种很怪异的方式完成。

我的代码在这里：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);
}

,

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

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

也，请参阅this document about how to create a menu。

此外，SSMS sdk api在这里：https://docs.microsoft.com/en-us/dotnet/api/?view=sql-smo-160](https://docs.microsoft.com/en-us/dotnet/api/?view=sql-smo-160)，您也可以参考this link。