Sql Server在存储过程里面使用游标遍历一个表

Posted 2021-02-22 幻梦飞羽

前：转自[https://www.cnblogs.com/SysoCjs/p/9842178.html]

文章很详细，就转过来收藏啦，以下是正文

---

这里关于SqlServer有两个知识点：一个是使用游标遍历表，另一个是使用if not exists的sql语句进行插入。

一、使用游标遍历表

　　这个表可以是数据库的表，也可以是外面DataTable类型的参数传进去，使用游标可以概括为以下步骤：声明游标、打开游标、读取数据、操作数据、读取数据、关闭游标、释放游标。

二、在insert语句使用if not exits

　　使用了if not exists的语句的insert操作，意思是，在找不到相关数据时才进行insert操作。不同数据库，有不同的使用语法。

下面是创建存储过程：

USE [CapacityManagement]
GO

/****** Object:  StoredProcedure [dbo].[USP_uploadResGpMaster]    Script Date: 2018/10/24 10:09:03 ******/
SET ANSI_NULLS ON
GO

SET QUOTED_IDENTIFIER ON
GO

--创建一张表，用来接收C#程序传过来的DataTable对象
--注意，这张表的字段必须跟传过来的DataTable对象的属性一致，名字可以不同

create type GS_Master as table
(
ResGp varchar(10),
Plant varchar(12),
Dept varchar(12),
Descrip nvarchar(50),
Step_mapping varchar(10),
CreateDate datetime,
CreateBy char(8),
ChangeDate datetime,
ChangeBy char(8)
)

go

CREATE PROCEDURE [dbo].[USP_uploadMaster] @Master GS_Master readonly,@OperationType nvarchar(3) 
AS
BEGIN    
    
    SET NOCOUNT ON;
    begin try
    begin transaction
    if(@OperationType =\'A\')
--声明一些local变量，用于接收查询表得到的数据，以便操作
    declare @ResGp varchar(10),
            @Plant varchar(12),
            @Dept varchar(12),
            @Descrip nvarchar(50),
            @Step_mapping varchar(10),
            @CreateDate datetime,
            @CreateBy char(8),
            @ChangeDate datetime,
            @ChangeBy char(8),
            @master_id int
      begin
       --一、声明游标
       declare master_cursor cursor for select * from @Master
       --二、打开游标
       open master_cursor
       --三、取第一条数据
       fetch next from master_cursor 
                  into @ResGp,
                        @Plant,
                        @Dept,
                        @Descrip,
                        @Step_mapping,
                        @CreateDate,
                        @CreateBy,
                        @ChangeDate,
                        @ChangeBy 
       while @@FETCH_STATUS = 0
       begin
        --四、操作数据,先操作[ResGp_Master]，后操作[Step_ResGp_Maping]
            --1、操作[ResGp_Master]
        SELECT @master_id=id FROM [CapacityManagement].[dbo].[ResGp_Master] 
                                where resgp = @ResGp
                                and    plant = @Plant
                                and    dept = @Dept

        update [CapacityManagement].[dbo].[ResGp_Master]
        set ChaDate = @ChangeDate,
            ChaBy = @ChangeBy,
            ResGpDesc = @Descrip
        where id = @master_id
        
        IF NOT EXISTS (SELECT id FROM [CapacityManagement].[dbo].[ResGp_Master] WHERE id = @master_id)  
        INSERT INTO [CapacityManagement].[dbo].[ResGp_Master] (ResGp, Dept, CreDate, CreBy, ChaDate, ChaBy, ResGpDesc, plant)
        values(@ResGp, @Dept, @CreateDate, @CreateBy, @ChangeDate, @ChangeBy, @Descrip, @Plant)

            --2、操作[Step_ResGp_Maping]
        update [CapacityManagement].[dbo].[Step_ResGp_Maping]
        set ChaDate = @ChangeDate,
            ChaBy = @ChangeBy
        where Master_id = @master_id
        and step = @Step_mapping

        if not exists(select master_id from [CapacityManagement].[dbo].[Step_ResGp_Maping] where Master_id = @master_id    and step = @Step_mapping)
        insert into [CapacityManagement].[dbo].[Step_ResGp_Maping](step, resgp, CreDate, CreBy, ChaDate, ChaBy, Master_id)
        values(@Step_mapping, @ResGp, @CreateDate, @CreateBy, @ChangeDate, @ChangeBy, @master_id)

        --五、取下一条数据
        fetch next from master_cursor 
                  into @ResGp,
                        @Plant,
                        @Dept,
                        @Descrip,
                        @Step_mapping,
                        @CreateDate,
                        @CreateBy,
                        @ChangeDate,
                        @ChangeBy 
       end      
       --六、关闭游标
       close master_cursor
       --七、释放游标
       deallocate master_cursor 
      end
      
     commit transaction
    end try 
    begin catch
    select ERROR_MESSAGE() as errorMessage
    rollback transaction
    end catch

END
GO

注释也是够详细了，操作数据那个过程，业务需求是对一堆数据进行插入，如果该数据已经存在，就进行更新，如果属于新数据，那么就进行插入（所以博主想到的是先遍历全部根据主键进行update，再根据主键查找是否存在该数据，若没有，则插入新数据），其实读者可以不细看，因为操作过程是根据业务需求，这里主要讲解游标的使用。

值得留意的一点：存储过程创建一个table类型的变量是用来存储程序传过来的DataTable对象，不仅字段个数要与DataTable的列数一致，类型也要特别注意，假设如果将DataTable的非数字列传给table类型里面的int类型字段，则该存储过程无法执行，会抛出异常。

注意：使用游标时，代码里使用一些局部变量存放查找的值（如@Master_id），如果第一次循环，@Master_id可以通过select查找到值，到了第二次select时却没有查找到值，那么@Master_id这个局部变量在第二次的循环里面的值不是null，也不是0或""，反而是第一次循环的值（即上一次循环的值）