在SQL Server中按游标（获取表名称）删除所有表

Question

我有一个stg_table列的元数据表。 这是dbo架构中STAGE数据库表的名称。 如果insert_type列为'select'，我希望在游标存在的情况下删除表。

元数据示例： https ： //imgur.com/a/TfyIWpv

我的游标代码：

declare kursor cursor for 
     select * 
     from METADATA.dbo.META_SOURCESTAGE;

declare @insert_type varchar(15), 
        @stg_table varchar(30), 
        @src varchar(80), 
        @SQL varchar(254);

open kursor

fetch next from kursor into @insert_type, @stg_table, @src;

while @@FETCH_STATUS = 0
begin
    if @insert_type = 'select'
        begin
        select @SQL = 'IF OBJECT_ID('+@stg_table+') IS NOT NULL DROP TABLE '+@stg_table
    exec(@SQL)

    fetch next from kursor into @insert_type, @stg_table, @src;
    end
    if @insert_type = 'bulk'
        execute ('truncate table ' + @stg_table);

    print @stg_table;

    fetch next from kursor into @insert_type, @stg_table, @src;
end

close kursor;
deallocate kursor;

我想通过在METADATA表创建的游标从STAGE.dbo。*中删除所有现有表。

我收到一个错误：

Msg 4104，Level 16，State 1，Line 2
无法绑定多部分标识符“STAGE.dbo.STG_KLIENT”。 STAGE.dbo.STG_MELDUNEK

Msg 4104，Level 16，State 1，Line 2
无法绑定多部分标识符“STAGE.dbo.STG_MIEJSCOWOSC”。
STAGE.dbo.STG_PRACOWNIK

@EDIT更改了代码 - 添加了开始/结束但仍然出现相同的错误。

Answer 1

它抱怨无法绑定OBJECT_ID调用，您的代码会产生如下结果：

IF OBJECT_ID(STAGE.dbo.STG_KLIENT) IS NOT NULL DROP TABLE STAGE.dbo.STG_KLIENT

DROP TABLE部分很好，但是你需要在调用OBJECT_ID引用表名，因为它需要一个(N)VARCHAR输入，所以请使用：

select @sql = 'IF OBJECT_ID('''+@stg_table+''') IS NOT NULL DROP TABLE '+@stg_table
--                           ^^            ^^

此外，您的代码有点遍布。 它会跳过一些表，因为你正在做的是：

1. 从光标获取（第一个结果，让我们假设选择类型）
2. 检查@insert_type ='select'是否为true，让我们进入块
   • 删除表（如果存在）
   • 从光标获取（第二个结果，让我们再次假设选择类型
   • 离开，如果阻止
3. 检查@insert_type ='insert'，false，不要截断
4. 打印表名
5. 从光标 - >第3个结果获取下一个结果，你刚刚对第二个结果做了什么，你应该放弃它。

这是一个稍微改进的版本：

while @@FETCH_STATUS = 0
begin
    if @insert_type = 'select'
    begin
       select @sql = 'IF OBJECT_ID('''+@stg_table+''') IS NOT NULL DROP TABLE '+@stg_table
    end
    else if @insert_type = 'insert'
    begin
       select @sql = 'TRUNCATE TABLE ' + @stg_table
    end

    print @sql -- "debugging"

    -- only do those once if you can, so you don't get lost in how the code branches
    exec sp_executesql @sql
    fetch next from kursor into @insert_type, @stg_table, @src;
end

此外，你真的不需要光标。 您只需生成一个大查询并执行一次即可。

DECLARE @sql NVARCHAR(MAX) = ''
SELECT @sql += CONCAT(CASE
                        WHEN meta.insert_table = 'select'
                         AND OBJECT_ID(meta.stg_table) IS NOT NULL
                        THEN CONCAT('DROP TABLE ', meta.stg_table)

                        WHEN meta.insert_table = 'insert'
                        THEN CONCAT('TRUNCATE TABLE', meta.stg_table)
                      END,
                      ';', CHAR(13), CHAR(10)) -- add newline for easier reading
  FROM METADATA.dbo.META_SOURCESTAGE meta

PRINT @sql
EXEC sp_executesql @sql

这应该工作，虽然我没有测试过，但我希望你能得到这个想法。