SQL Server 存储过程流程控制：IF、CASE与WHILE实战

一、条件判断：IF ... ELSE

IF...ELSE 用于根据条件的真假执行不同的 SQL 语句块。虽然在不嵌套且语句简单时 BEGIN...END 可省略，但为了保证代码层次清晰和后续扩展性，强烈建议始终使用 BEGIN...END 包裹语句块。

第1步：声明变量并初始化

使用 DECLARE 声明变量，并使用 SET 赋予初始值。这是执行逻辑前的必要准备。

-- 声明整型变量 @id
DECLARE @id INT;
-- 设置初始值为 5
SET @id = 5;

第2步：构建 IF-ELSE 逻辑树

通过 IF (条件) 判断，随后紧跟 BEGIN ... END 块。若为多分支判断，可使用 ELSE IF；最后以 ELSE 覆盖默认情况。

-- 开始条件判断
IF (@id = 1)
BEGIN
    PRINT 'right';
END
ELSE IF (@id = 0)
BEGIN
    PRINT 'error';
END
ELSE
BEGIN
    -- 当 @id 不为 1 或 0 时执行
    PRINT 'default';
END;

结果说明：由于 @id 被设为 5，上述代码将输出字符串 'default'。

二、循环控制：WHILE

WHILE 循环用于重复执行 SQL 语句，直到指定条件变为假。常用于批量处理、计数器累加等场景。

第1步：初始化计数器与累加器

在进入循环前，必须确保计数变量和结果变量的初始值正确，否则可能导致死循环或结果错误。

DECLARE @i INT;
DECLARE @sum INT;

-- 初始化 i 为 0，sum 为 0
SET @i = 0;
SET @sum = 0;

第2步：编写循环体

在 WHILE (条件) 后使用 BEGIN ... END 包裹循环内的操作。注意必须在循环体内修改循环变量（如自增），否则将导致无限循环。

-- 当 i 小于等于 100 时持续循环
WHILE @i <= 100
BEGIN
    -- i 自增 1
    SET @i = @i + 1;
    -- 累加到 sum
    SET @sum = @sum + @i;
END;

-- 输出最终结果
PRINT @sum;

结果说明：该逻辑计算了从 1 加到 100 的和，最终输出结果为 5050。

三、多重选择：CASE ... WHEN

CASE 表达式允许根据多个条件返回不同的值，非常适合用于数据转换、状态映射或简单分类。

第1步：定义 CASE 选择逻辑

使用 SELECT @variable = CASE WHEN 条件 THEN 结果 ... END 的语法将匹配结果赋值给变量。最后务必包含 ELSE 子句以防未匹配项报错或为空。

DECLARE @iret INT;
DECLARE @var VARCHAR(10);

-- 设置测试值
SET @var = 'A';

-- 根据 @var 的值选择对应的整数
SELECT @iret =
    CASE
        WHEN @var = 'A' THEN 0
        WHEN @var = 'B' THEN 1
        WHEN @var = 'C' THEN 2
        WHEN @var = 'D' THEN 3
        ELSE -1
    END;

-- 打印匹配到的整数
PRINT @iret;

结果说明：因 @var 为 'A'，对应第一个 WHEN 条件成立，@iret 被赋值为 0，输出结果为 0。

四、最佳实践与注意事项

• 代码规范性：即便单条语句，也建议使用 BEGIN...END 明确代码块范围，提升可读性并避免维护时的逻辑漏洞。
• 循环安全：使用 WHILE 时，务必检查循环内部是否有更新循环条件的操作，防止出现“死循环”耗尽服务器资源。
• 变量作用域：在存储过程中，局部变量以 @ 开头，需注意命名不与其他全局对象冲突。

以上就是 SQL Server 存储过程流程控制（IF、CASE、WHILE）的详细内容，更多关于 SQL 高级查询与优化的资料请关注本站其它相关文章！