如何修改存储过程，使多表查询结果返回对应数据表名称

核心修改方案 #

在每个查询语句中新增一个自定义列，硬编码指定当前查询的数据源表名称；如果希望返回单一结果集（方便Python直接存入DataFrame），可以用UNION ALL合并三个查询结果。

修改后的存储过程代码 #

procedure [dbo].[getUser](@username nvarchar(225))
AS
-- 合并三个查询为单一结果集，每行携带来源表名
SELECT username, user_id, user_type, 'db1.dublin_table' AS source_table
FROM db1.dublin_table
WHERE username LIKE '%' + TRIM(@username) + '%'
UNION ALL
SELECT username, user_id, user_type, 'db2.nyc_table' AS source_table
FROM db2.nyc_table
WHERE username LIKE '%' + TRIM(@username) + '%'
UNION ALL
SELECT username, user_id, user_type, 'db3.tokyo_table' AS source_table
FROM db3.tokyo_table
WHERE username LIKE '%' + TRIM(@username) + '%'
ORDER BY source_table, username; -- 可按需调整排序规则

关键说明 #

1. 新增来源列：通过'表名' AS source_table给每一行数据添加明确的来源标识，用户在TreeView中就能直接看到该行数据来自哪个表。
2. 合并结果集：使用UNION ALL替代原存储过程的三次独立查询，将结果合并为一个数据集。这样Python端无需处理多个结果集的拆分与合并，直接读取即可存入DataFrame。
3. 兼容原逻辑：保留了原有的TRIM(@username)和模糊匹配逻辑，确保查询效果不变。
4. 可选方案：如果需要保留原存储过程返回多个独立结果集的形式，也可以单独给每个SELECT添加source_table列，无需UNION ALL，但Python读取时需要遍历所有结果集并手动合并到同一个DataFrame中。

注意事项 #

• 确保三个表的username, user_id, user_type字段数据类型完全一致，否则UNION ALL会抛出类型不匹配错误。
• 排序规则可根据业务需求调整，比如仅按username排序，或者按source_table分组后再排序。

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