sql-server - select语句中的变量表名

Question

我有一些表用于存储不同的文件信息，例如拇指、图像、数据表......

我正在编写一个存储过程来检索特定 ID 的文件名。像这样:

    CREATE PROCEDURE get_file_name(
    @id int,
    @table nvarchar(50)
    )as
    if @table='images'
        select [filename] from images
        where id = @id
    if @table='icons'
        select [filename] from icons
        where id = @id
....

如何使用 case when 语句重写此过程，还是应该使用表名作为变量？

Answer 1

您不能使用 case .. when 在 FROM 子句中的表之间进行切换(就像在 conditional ORDER BY 中一样)。即如下:

select * from 
    case when 1=1
      then t1
      else t2
    end;

不起作用。

因此您需要使用动态 SQL。最好尽可能参数化查询，例如可以参数化@id值:

-- Validate @table is E ['images', 'icons', ... other valid names here]
DECLARE @sql NVARCHAR(MAX)
SET @sql = 'select [filename] from **TABLE** where id = @id';
SET @sql = REPLACE(@sql, '**TABLE**', @table);

sp_executesql @sql, N'@id INT', @id = @id;

与所有动态 Sql 一样，请注意，替换到查询中的未参数化值(例如@table)会使查询容易受到 Sql 注入(inject)攻击。因此，我建议您确保 @table 来自可信来源，或者更好的是，将 @table 的值与允许的白名单进行比较执行查询之前的表。