存储过程、函数之流程控制

流程控制

解决复杂问题不可能通过一个 SQL 语句完成，我们需要执行多个 SQL 操作。流程控制语句的作用就是控制存储过程中 SQL 语句的执行顺序，是我们完成复杂操作必不可少的一部分。只要是执行的程序，流程就分为三大类：

顺序结构 ：程序从上往下依次执行
分支结构 ：程序按条件进行选择执行，从两条或多条路径中选择一条执行
循环结构 ：程序满足一定条件下，重复执行一组语句

针对于MySQL 的流程控制语句主要有 3 类。注意：只能用于存储程序。

条件判断语句 ：IF 语句和 CASE 语句
循环语句 ：LOOP、WHILE 和 REPEAT 语句
跳转语句 ：ITERATE 和 LEAVE 语句

1. 分支结构之 IF

IF 语句的语法结构是：
```
IF 表达式1 THEN 操作1 
[ELSEIF 表达式2 THEN 操作2]…… 
[ELSE 操作N] 
END IF
```
根据表达式的结果为TRUE或FALSE执行相应的语句。这里“[]”中的内容是可选的。
- 特点：① 不同的表达式对应不同的操作 ② 使用在begin end中

举例1：

CREATE PROCEDURE test_if()
BEGIN 
    DECLARE val varchar(15);
    IF val IS NULL 
        THEN SELECT 'val is null';
    ELSE SELECT 'val is not null'; 
    
    END IF;
END;

# 调用
CALL test_if();

举例2：声明存储过程“update_salary_by_eid1”，定义IN参数emp_id，输入员工编号。判断该员工薪资如果低于8000元并且入职时间超过5年，就涨薪500元；否则就不变。

DELIMITER // 

CREATE PROCEDURE update_salary_by_eid1(IN emp_id INT) 
    BEGIN
    DECLARE emp_salary DOUBLE; 
    DECLARE hire_year DOUBLE; 

    SELECT salary INTO emp_salary 
    FROM employees WHERE employee_id = emp_id; 

    SELECT DATEDIFF(CURDATE(),hire_date)/365 INTO hire_year 
    FROM employees WHERE employee_id = emp_id; 

    IF emp_salary < 8000 AND hire_year > 5 
    THEN UPDATE employees SET salary = salary + 500 
    WHERE employee_id = emp_id; 
        END IF; 
    END // 
DELIMITER ;

# 调用
call update_salary_by_eid1(104)

举例3：声明存储过程“update_salary_by_eid2”，定义IN参数emp_id，输入员工编号。判断该员工薪资如果低于9000元并且入职时间超过5年，就涨薪500元；否则就涨薪100元。

DELIMITER // 
CREATE PROCEDURE update_salary_by_eid2(IN emp_id INT) 
BEGIN
    DECLARE emp_salary DOUBLE; 
    DECLARE hire_year DOUBLE; 
    
    SELECT salary INTO emp_salary FROM employees WHERE employee_id = emp_id; 
    
    SELECT DATEDIFF(CURDATE(),hire_date)/365 INTO hire_year 
    FROM employees WHERE employee_id = emp_id; 
    
    IF emp_salary < 8000 AND hire_year > 5 
        THEN UPDATE employees SET salary = salary + 500 WHERE employee_id = emp_id; 
    ELSE
        UPDATE employees SET salary = salary + 100 WHERE employee_id = emp_id; 
    END IF; 
END // 

DELIMITER ;

# 调用
call update_salary_by_eid2(103);
call update_salary_by_eid2(104);

举例4：声明存储过程“update_salary_by_eid3”，定义IN参数emp_id，输入员工编号。判断该员工薪资如果低于9000元，就更新薪资为9000元；薪资如果大于等于9000元且低于10000的，但是奖金比例为NULL的，就更新奖金比例为0.01；其他的涨薪100元。

DELIMITER // 
CREATE PROCEDURE update_salary_by_eid3(IN emp_id INT) 
BEGIN
    DECLARE emp_salary DOUBLE; 
    DECLARE bonus DECIMAL(3,2); 
    
    SELECT salary INTO emp_salary FROM employees WHERE employee_id = emp_id;
  SELECT commission_pct INTO bonus FROM employees WHERE employee_id = emp_id;
  IF 
    emp_salary < 9000 
  THEN 
    UPDATE employees SET salary = 9000 WHERE employee_id = emp_id; 
  ELSEIF emp_salary < 10000 AND bonus IS NULL 
  THEN 
    UPDATE employees SET commission_pct = 0.01 WHERE employee_id = emp_id; ELSEUPDATE employees SET salary = salary + 100 WHERE employee_id = emp_id; 
  END IF; 
END // 
DELIMITER ;

# 调用
call update_salary_by_eid3(102);
call update_salary_by_eid3(103);
call update_salary_by_eid3(104);

2. 分支结构之 CASE

CASE 语句的语法结构1：

#情况一：类似于switch 
CASE 表达式 
WHEN 值1 THEN 结果1或语句1(如果是语句，需要加分号) 
WHEN 值2 THEN 结果2或语句2(如果是语句，需要加分号) 
... 
ELSE 结果n或语句n(如果是语句，需要加分号) 
END [case]（如果是放在begin end中需要加上case，如果放在select后面不需要）

CASE 语句的语法结构2：

#情况二：类似于多重if
CASE 
WHEN 条件1 THEN 结果1或语句1(如果是语句，需要加分号) 
WHEN 条件2 THEN 结果2或语句2(如果是语句，需要加分号) 
... 
ELSE 结果n或语句n(如果是语句，需要加分号) 
END [case]（如果是放在begin end中需要加上case，如果放在select后面不需要）

举例1：使用CASE流程控制语句的第1种格式，判断val值等于1、等于2，或者两者都不等。

CASE val 
    WHEN 1 THEN SELECT 'val is 1'; 
    WHEN 2 THEN SELECT 'val is 2'; 
    ELSE SELECT 'val is not 1 or 2'; 
END CASE;

举例2：使用CASE流程控制语句的第2种格式，判断val是否为空、小于0、大于0或者等于0。

CASE
    WHEN val IS NULL THEN SELECT 'val is null'; 
    WHEN val < 0 THEN SELECT 'val is less than 0'; 
    WHEN val > 0 THEN SELECT 'val is greater than 0'; 
    ELSE SELECT 'val is 0'; 
END CASE;

举例3：声明存储过程“update_salary_by_eid4”，定义IN参数emp_id，输入员工编号。判断该员工薪资如果低于9000元，就更新薪资为9000元；薪资大于等于9000元且低于10000的，但是奖金比例为NULL的，就更新奖金比例为0.01；其他的涨薪100元。

DELIMITER //
BEGIN
    DECLARE emp_sal DOUBLE; 
    DECLARE bonus DECIMAL(3,2); 
    
    SELECT salary INTO emp_sal FROM employees WHERE employee_id = emp_id;
  SELECT commission_pct INTO bonus FROM employees WHERE employee_id=emp_id; 
  
  CASE
    WHEN emp_sal<9000 
        THEN UPDATE employees SET salary=9000 WHERE employee_id = emp_id; 
    WHEN emp_sal<10000 AND bonus IS NULL 
        THEN UPDATE employees SET commission_pct=0.01 WHERE employee_id = emp_id; 
    ELSE
    UPDATE employees SET salary=salary+100 WHERE employee_id = emp_id; 
    END CASE; 
  END // 
 DELIMITER ;

举例4：声明存储过程update_salary_by_eid5，定义IN参数emp_id，输入员工编号。判断该员工的入职年限，如果是0年，薪资涨50；如果是1年，薪资涨100；如果是2年，薪资涨200；如果是3年，薪资涨300；如果是4年，薪资涨400；其他的涨薪500。

DELIMITER // 

CREATE PROCEDURE update_salary_by_eid5(IN emp_id INT) 
BEGIN
    DECLARE emp_sal DOUBLE; 
    DECLARE hire_year DOUBLE; 
    
    SELECT salary INTO emp_sal FROM employees WHERE employee_id = emp_id;
  
  SELECT ROUND(DATEDIFF(CURDATE(),hire_date)/365) INTO hire_year FROM employees WHERE employee_id = emp_id;
  
  CASE hire_year 
  WHEN 0 THEN UPDATE employees SET salary=salary+50 WHERE employee_id = emp_id; 
  WHEN 1 THEN UPDATE employees SET salary=salary+100 WHERE employee_id = emp_id; 
  WHEN 2 THEN UPDATE employees SET salary=salary+200 WHERE employee_id = emp_id; 
  WHEN 3 THEN UPDATE employees SET salary=salary+300 WHERE employee_id = emp_id; 
  WHEN 4 THEN UPDATE employees SET salary=salary+400 WHERE employee_id = emp_id; 
  ELSE UPDATE employees SET salary=salary+500 WHERE employee_id = emp_id; 
  END CASE; 
  
END //

DELIMITER ;

3. 循环结构之 LOOP

LOOP循环语句用来重复执行某些语句。LOOP内的语句一直重复执行直到循环被退出（使用LEAVE子句），跳出循环过程。

LOOP语句的基本格式如下：

[loop_label:] LOOP 循环执行的语句 
END LOOP [loop_label]

其中，loop_label表示LOOP语句的标注名称，该参数可以省略。

举例1：使用LOOP语句进行循环操作，id值小于10时将重复执行循环过程。

DECLARE id INT DEFAULT 0; 
add_loop:LOOP 
    SET id = id +1; 
    IF id >= 10 THEN LEAVE add_loop; 
    END IF; 

END LOOP add_loop;

举例2：当市场环境变好时，公司为了奖励大家，决定给大家涨工资。声明存储过程“update_salary_loop()”，声明OUT参数num，输出循环次数。存储过程中实现循环给大家涨薪，薪资涨为原来的1.1倍。直到全公司的平均薪资达到12000结束。并统计循环次数。

DELIMITER // 

CREATE PROCEDURE update_salary_loop(OUT num INT) 
BEGIN
    DECLARE avg_salary DOUBLE; 
    DECLARE loop_count INT DEFAULT 0; 
    
    SELECT AVG(salary) INTO avg_salary FROM employees;
  
  label_loop:LOOP 
    IF avg_salary >= 12000 THEN LEAVE label_loop; 
    END IF; 
    
    UPDATE employees SET salary = salary * 1.1; 
    SET loop_count = loop_count + 1; 
    SELECT AVG(salary) INTO avg_salary FROM employees; 
  END LOOP label_loop;
  SET num = loop_count; 
  
END // 
DELIMITER ;

4. 循环结构之 WHILE

WHILE语句创建一个带条件判断的循环过程。WHILE在执行语句执行时，先对指定的表达式进行判断，如果为真，就执行循环内的语句，否则退出循环。WHILE语句的基本格式如下：

[while_label:] WHILE 循环条件 DO 
    循环体 
END WHILE [while_label];

while_label为WHILE语句的标注名称；如果循环条件结果为真，WHILE语句内的语句或语句群被执行，直至循环条件为假，退出循环。

举例1： WHILE语句示例，i值小于10时，将重复执行循环过程，代码如下：

DELIMITER // 

CREATE PROCEDURE test_while() 
BEGIN
    DECLARE i INT DEFAULT 0; 
    
    WHILE i < 10 DO 
        SET i = i + 1; 
    END WHILE; 
    
    SELECT i; 
END // 

DELIMITER ; 
#调用 
CALL test_while();

举例2：市场环境不好时，公司为了渡过难关，决定暂时降低大家的薪资。声明存储过程“update_salary_while()”，声明OUT参数num，输出循环次数。存储过程中实现循环给大家降薪，薪资降为原来的90%。直到全公司的平均薪资达到5000结束。并统计循环次数。

DELIMITER // 

CREATE PROCEDURE update_salary_while(OUT num INT) 
BEGIN
    DECLARE avg_sal DOUBLE ;
  DECLARE while_count INT DEFAULT 0; 
  
  SELECT AVG(salary) INTO avg_sal FROM employees; 
  
  WHILE avg_sal > 5000 DO 
    UPDATE employees SET salary = salary * 0.9; 
    
    SET while_count = while_count + 1; 
    
    SELECT AVG(salary) INTO avg_sal FROM employees; 
  END WHILE; 
  
  SET num = while_count; 
  
END //

DELIMITER ;

5. 循环结构之 REPEAT

REPEAT语句创建一个带条件判断的循环过程。与WHILE循环不同的是，REPEAT 循环首先会执行一次循环，然后在 UNTIL 中进行表达式的判断，如果满足条件就退出，即 END REPEAT；如果条件不满足，则会就继续执行循环，直到满足退出条件为止。REPEAT语句的基本格式如下：

[repeat_label:] REPEAT 循环体的语句 
UNTIL 结束循环的条件表达式 
END REPEAT [repeat_label]

repeat_label为REPEAT语句的标注名称，该参数可以省略；REPEAT语句内的语句或语句群被重复，直至expr_condition为真。

举例1：

DELIMITER // 

CREATE PROCEDURE test_repeat() 
BEGIN
    DECLARE i INT DEFAULT 0; 
    
    REPEAT
        SET i = i + 1; 
    UNTIL i >= 10 
    END REPEAT;
  
  SELECT i; 
END // 

DELIMITER ;

举例2：当市场环境变好时，公司为了奖励大家，决定给大家涨工资。声明存储过程“update_salary_repeat()”，声明OUT参数num，输出循环次数。存储过程中实现循环给大家涨薪，薪资涨为原来的1.15倍。直到全公司的平均薪资达到13000结束。并统计循环次数。

DELIMITER // 

CREATE PROCEDURE update_salary_repeat(OUT num INT) 
BEGIN
    DECLARE avg_sal DOUBLE ; 
    DECLARE repeat_count INT DEFAULT 0; 
    
    SELECT AVG(salary) INTO avg_sal FROM employees;
  
  REPEAT
    UPDATE employees SET salary = salary * 1.15; 
    SET repeat_count = repeat_count + 1;
    SELECT AVG(salary) INTO avg_sal FROM employees; 
  UNTIL avg_sal >= 13000 
  END REPEAT; 
  
    SET num = repeat_count;
END // 
DELIMITER ;

对比三种循环结构：

1、这三种循环都可以省略名称，但如果循环中添加了循环控制语句（LEAVE或ITERATE）则必须添加名称。

2、 LOOP：一般用于实现简单的"死"循环

    WHILE：先判断后执行 

    REPEAT：先执行后判断，无条件至少执行一次

6. 跳转语句之 LEAVE 语句

LEAVE语句：可以用在循环语句内，或者以 BEGIN 和 END 包裹起来的程序体内，表示跳出循环或者跳出程序体的操作。如果你有面向过程的编程语言的使用经验，你可以把 LEAVE 理解为 break。基本格式如下：

LEAVE 标记名

其中，label参数表示循环的标志。LEAVE和BEGIN ... END或循环一起被使用。

举例1：创建存储过程 “leave_begin()”，声明INT类型的IN参数num。给BEGIN...END加标记名，并在BEGIN...END中使用IF语句判断num参数的值。

如果num<=0，则使用LEAVE语句退出BEGIN...END；
如果num=1，则查询“employees”表的平均薪资；
如果num=2，则查询“employees”表的最低薪资；
如果num>2，则查询“employees”表的最高薪资。

IF语句结束后查询“employees”表的总人数。

DELIMITER // 

CREATE PROCEDURE leave_begin(IN num INT) 
    begin_label: BEGIN 
        IF num<=0 THEN LEAVE begin_label; 
        ELSEIF num=1 THEN SELECT AVG(salary) FROM employees; 
        ELSEIF num=2 THEN SELECT MIN(salary) FROM employees; 
        ELSE SELECT MAX(salary) FROM employees; 
        END IF; 
        
        SELECT COUNT(*) FROM employees; 
    END // 

DELIMITER ;

举例2：

当市场环境不好时，公司为了渡过难关，决定暂时降低大家的薪资。声明存储过程“leave_while()”，声明OUT参数num，输出循环次数，存储过程中使用WHILE循环给大家降低薪资为原来薪资的90%，直到全公司的平均薪资小于等于10000，并统计循环次数。

DELIMITER // 

CREATE PROCEDURE leave_while(OUT num INT) 

BEGIN
    #
    DECLARE avg_sal DOUBLE;#记录平均工资
    DECLARE while_count INT DEFAULT 0; #记录循环次数 
    SELECT AVG(salary) INTO avg_sal FROM employees; #① 初始化条件
    while_label:WHILE TRUE DO #② 循环条件 
    
        #③ 循环体 
        IF avg_sal <= 10000 THEN 
            LEAVE while_label; 
        END IF;
    
    UPDATE employees SET salary = salary * 0.9; 
    SET while_count = while_count + 1; 
    
    #④ 迭代条件
    SELECT AVG(salary) INTO avg_sal FROM employees;
    
   END WHILE; 
   
   #赋值 
   SET num = while_count; 
  
 END // 
 
 DELIMITER ;

7. 跳转语句之 ITERATE 语句

ITERATE语句：只能用在循环语句（LOOP、REPEAT和WHILE语句）内，表示重新开始循环，将执行顺序转到语句段开头处。如果你有面向过程的编程语言的使用经验，你可以把 ITERATE 理解为 continue，意思为“再次循环”。语句基本格式如下：

ITERATE label

label参数表示循环的标志。ITERATE语句必须跟在循环标志前面。

举例： 定义局部变量num，初始值为0。循环结构中执行num + 1操作。

如果num < 10，则继续执行循环；
如果num > 15，则退出循环结构；

DELIMITER // 

CREATE PROCEDURE test_iterate() 
BEGIN
    DECLARE num INT DEFAULT 0;
  
  my_loop:LOOP 
    SET num = num + 1; 
    IF num < 10 THEN ITERATE my_loop;
    ELSEIF num > 15
    THEN LEAVE my_loop; 
    END IF; 
    SELECT '尚硅谷：让天下没有难学的技术'; 
    END LOOP my_loop; 
 
END // 
DELIMITER ;

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 213,558评论 6赞 492
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 91,002评论 3赞 387
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 159,036评论 0赞 349
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 57,024评论 1赞 285
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 66,144评论 6赞 385
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 50,255评论 1赞 292
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 39,295评论 3赞 412
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 38,068评论 0赞 268
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 44,478评论 1赞 305
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 36,789评论 2赞 327
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 38,965评论 1赞 341
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 34,649评论 4赞 336
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 40,267评论 3赞 318
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 30,982评论 0赞 21
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 32,223评论 1赞 267
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 46,800评论 2赞 365
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 43,847评论 2赞 351