如何在SSMS命令扩展中获取查询结果数据及识别IDE服务类型

针对你在SSMS扩展开发中遇到的两个核心问题，我结合社区实践和内部API探索经验给你梳理下可行方案：

一、在SSMS命令扩展中获取查询结果引用 #

SSMS基于Visual Studio Shell构建，但有自己专属的查询结果管理组件，核心是通过内部的SqlResultsControl类来访问结果数据。具体步骤如下：

1. 引用SSMS内部程序集
   你需要添加SSMS安装目录下的两个关键dll引用：

   • Microsoft.SqlServer.Management.UI.VSIntegration.dll
   • Microsoft.SqlServer.Management.UI.Grid.dll
     路径通常在C:\Program Files (x86)\Microsoft SQL Server Management Studio [版本号]\Common7\IDE（比如SSMS 18对应路径里的文件）

2. 获取当前查询结果控件实例
   通过SSMS的ServiceCache类可以快速定位到当前活动窗口的结果控件，示例代码如下：

   using Microsoft.SqlServer.Management.UI.VSIntegration;
   using Microsoft.SqlServer.Management.UI.Grid;
   
   // 获取当前活动窗口的查询结果控件
   var currentWindow = ServiceCache.ScriptFactory.CurrentWindow;
   if (currentWindow?.Frame != null)
   {
       var sqlResultsControl = currentWindow.Frame.GetSqlResultsControl();
       if (sqlResultsControl != null)
       {
           // 遍历所有结果集
           foreach (IResultInfo resultSet in sqlResultsControl.ResultSets)
           {
               // 获取列元数据
               var columnCount = resultSet.Columns.Count;
               var columnNames = resultSet.Columns.Cast<IColumnInfo>().Select(c => c.ColumnName).ToList();
               
               // 遍历行数据
               foreach (IRowInfo row in resultSet.Rows)
               {
                   for (int i = 0; i < columnCount; i++)
                   {
                       var cellValue = row[i];
                       // 这里可以处理单元格数据，用于可视化展示
                   }
               }
           }
       }
   }
   

   这里的IResultInfo和IRowInfo接口提供了对结果集结构和数据的访问能力，足够支撑你提取底层显示数据做可视化。

二、识别IDE元素对应的服务类型 #

由于SSMS官方扩展文档匮乏，识别IDE元素服务类型主要靠以下几种实用方法：

• 反编译核心程序集
  用ILSpy或dnSpy等工具反编译SSMS的核心dll（比如前面提到的VSIntegration.dll、Grid.dll），查看内部定义的接口和类。比如查询窗口对应的是SqlScriptEditorControl，结果面板对应SqlResultsControl，这些类通常会注册为特定服务类型。

• 利用Visual Studio服务定位机制
  SSMS继承了VS的IVsServiceProvider接口，你可以尝试通过GetService方法测试不同类型，比如：

  var serviceProvider = (IVsServiceProvider)ServiceCache.ServiceProvider;
  var scriptEditor = serviceProvider.GetService(typeof(SqlScriptEditorControl));
  

  这种试错法在找不到文档时很有效。

• 使用Spy++定位窗口类
  用Windows自带的Spy++工具找到IDE元素的窗口类名，再回到反编译的程序集中搜索对应的.NET类，比如查询结果窗口的类名是SqlResultsControl，就能快速定位到对应的服务类型。

• 参考社区开源扩展
  GitHub上有不少SSMS开源扩展（比如查询格式化、结果导出工具），查看它们的源码可以学习如何获取各种IDE元素的服务引用，这是最快捷的学习方式。

注意事项 #

这些方法依赖SSMS的内部未公开API，不同版本的SSMS可能会有接口变更，所以开发时需要针对目标版本做兼容性测试。

内容的提问来源于stack exchange，提问作者MartinHN