MySQL全面瓦解20：可编程性之流程控制语句

背景

说到流程控制语句，我们在程序语法中用的比较多，比如C#的if..else...，while...，?: 等。同样的，在MySQL中，也有一些流程控制的语法，方便我们在写函数、存储过程的时候对逻辑进行控制和处理。

常见的过程式SQL语句可以用在存储过程或者函数体中。其中包括：IF函数、IF条件语句、CASE语句、LOOP语句、WHILE语句、REPEAT语句、LEAVE语句和ITERATE语句，它们极大的方便了我们进行流程控制。

下面我们一个一个来看。

流程语句分解

数据基础

 1 mysql> select *  from students;
 2 +-----------+-------------+-------+---------+-----+
 3 | studentid | studentname | score | classid | sex |
 4 +-----------+-------------+-------+---------+-----+
 5 |         1 | brand       | 105.5 |       1 |   1 |
 6 |         2 | helen       | 98.5  |       1 |   0 |
 7 |         3 | lyn         | 97    |       1 |   0 |
 8 |         4 | sol         | 97    |       1 |   1 |
 9 |         5 | b1          | 89    |       2 |   1 |
10 |         6 | b2          | 90    |       2 |   1 |
11 |         7 | c1          | 76    |       3 |   0 |
12 |         8 | c2          | 73.5  |       3 |   0 |
13 |         9 | lala        | 73    |       0 |   0 |
14 |        10 | A           | 100   |       3 |   1 |
15 |        16 | test1       | 100   |       0 |   1 |
16 |        17 | trigger2    | 107   |       0 |   1 |
17 |        22 | trigger1    | 100   |       0 |   0 |
18 +-----------+-------------+-------+---------+-----+
19 13 rows in set
20
21 mysql> select *  from scores;
22 +-----------+---------+-------+
23 | scoregrad | downset | upset |
24 +-----------+---------+-------+
25 | A         |      81 |    90 |
26 | B         |      71 |    80 |
27 | C         |      61 |    70 |
28 | D         |      51 |    60 |
29 | S         |      91 |   100 |
30 | S+        |     101 |   120 |
31 +-----------+---------+-------+
32 6 rows in set 

IF函数

有点类似C#语法中的三元表达式，有3个参数，第一个参数是表达式，后面两个是值，当表达式成立的时候取第一个值，表达式不成立的时候取第二个值。

1 if(expr,val1,val2);   --语法 

输出学生信息中的名称和性别（1为男，0为女，这边用if函数进行转换）

 1 mysql> select studentname,if(sex=0,'女','男') from students where classid<>0;
 2 +-------------+---------------------+
 3 | studentname | if(sex=0,'女','男') |
 4 +-------------+---------------------+
 5 | brand       | 男                  |
 6 | helen       | 女                  |
 7 | lyn         | 女                  |
 8 | sol         | 男                  |
 9 | b1          | 男                  |
10 | b2          | 男                  |
11 | c1          | 女                  |
12 | c2          | 女                  |
13 | A           | 男                  |
14 +-------------+---------------------+
15 9 rows in set 

IF条件语句

IF语句用来进行条件判断，根据不同的条件执行不同的操作。该语句在执行时首先判断IF后的条件是否为真，则执行THEN后的语句，如果为假则继续判断IF语句直到为真为止，当以上都不满足时则执行ELSE语句后的内容。

1 IF condition THEN
2     ...
3 ELSEIF condition THEN
4     ...
5 ELSE
6     ...
7 END IF 

代码示例，根据考试成绩来分布不同的成绩等级

 1 mysql>
 2 /*如果存在函数func_test2，则删除*/
 3 DROP FUNCTION IF EXISTS fun_if;
 4 /*声明结束符为$*/
 5 DELIMITER $
 6 /*创建函数*/
 7 CREATE FUNCTION fun_if(score DECIMAL(10,2))
 8   RETURNS CHAR
 9 BEGIN
10 DECLARE score_grad VARCHAR(5) DEFAULT '';
11 IF score>100 THEN SET score_grad='S';
12 ELSEIF (score BETWEEN 91 AND 100) THEN SET score_grad='A';
13 ELSEIF (score BETWEEN 81 AND 90) THEN SET score_grad='B';
14 ELSEIF (score BETWEEN 71 AND 80) THEN SET score_grad='C' ;
15 ELSE set score_grad='D';
16 END IF;
17 return score_grad;
18 END $
19 /*重置结束符为;*/
20 DELIMITER ;
21 Query OK, 0 rows affected 

执行结果

1 mysql> select fun_if(101),fun_if(100),fun_if(90),fun_if(80),fun_if(70);
2 +-------------+-------------+------------+------------+------------+
3 | fun_if(101) | fun_if(100) | fun_if(90) | fun_if(80) | fun_if(70) |
4 +-------------+-------------+------------+------------+------------+
5 | S           | A           | B          | C          | D          |
6 +-------------+-------------+------------+------------+------------+
7 1 row in set 

CASE语句

CASE语句为多分支语句结构，该语句首先从WHEN后的VALUE中查找与CASE后的VALUE相等的值，如果查找到则执行该分支的内容，否则执行ELSE后的内容。CASE语句表示形式如下，类似C#中switch：

1 CASE expr
2 WHEN val1 THEN result1 or state1[;]（可选项，如果是语句需要加分号,结果值可以加）
3 WHEN val2 THEN result2 or state2
4 ...
5 ELSE resultn or staten
6 END [CASE] （可选项，在begin end之间需加case，select后就不需要） 

在select中使用示例

 1 mysql> select studentname,case sex WHEN 0 THEN '女' WHEN 1 THEN '男' end as sex
 2 from students where classid<>0;
 3 +-------------+-----+
 4 | studentname | sex |
 5 +-------------+-----+
 6 | brand       | 男  |
 7 | helen       | 女  |
 8 | lyn         | 女  |
 9 | sol         | 男  |
10 | b1          | 男  |
11 | b2          | 男  |
12 | c1          | 女  |
13 | c2          | 女  |
14 | A           | 男  |
15 +-------------+-----+
16 9 rows in set 

在函数或存储过程中使用示例

 1 mysql>
 2 /*如果存在函数func_test2，则删除*/
 3 DROP FUNCTION IF EXISTS fun_case;
 4 /*声明结束符为$*/
 5 DELIMITER $
 6 /*创建函数*/
 7 CREATE FUNCTION fun_case(sex INT)
 8 RETURNS VARCHAR(20)
 9 BEGIN
10 DECLARE sexStr VARCHAR(20) DEFAULT '';
11 CASE sex
12 WHEN 0 then set sexStr='女';
13 WHEN 1 then set sexStr='男';
14 ELSE set sexStr='不确定';
15 END CASE;
16 return sexStr;
17 END $
18 /*重置结束符为;*/
19 DELIMITER ;
20
21 Query OK, 0 rows affected 

函数执行结果

 1 mysql> select studentname,fun_case(sex) from students where classid<>0;
 2 +-------------+---------------+
 3 | studentname | fun_case(sex) |
 4 +-------------+---------------+
 5 | brand       | 男            |
 6 | helen       | 女            |
 7 | lyn         | 女            |
 8 | sol         | 男            |
 9 | b1          | 男            |
10 | b2          | 男            |
11 | c1          | 女            |
12 | c2          | 女            |
13 | A           | 男            |
14 +-------------+---------------+
15 9 rows in set 

循环语句while

循环语句while 类似于C#中的while循环，我们知道在C#的while 或者 for 语句中，经常有用到两个关键语法：跳过当前循环（continue） 和 结束循环（break）。

同样的，在MySQL中也有两个语法对应跳过和结束循环。

1  ITERATE  loop_label;   --跳过当前循环

1 LEAVE  loop_label;   --结束循环

while 语法

1 [loop_label:]while condition do
2  --Todo:loop body
3 end while [loop_label]; 

loop_label：循环标签，和iterate、leave结合用于在循环内部对循环进行控制：如：跳过本次循环、结束循环。

condition：循环条件，当满足条件的时候，就会执行循环体，条件不成立的时候结束循环。

while示例

下面脚本代码演示了将students表中studentid在给定数值范围内的数据存储到另外一张表中。

 1 /*先清除studentCount表记录*/
 2 truncate table studentcount;
 3 /*存储过程如果存在则删除*/
 4 DROP PROCEDURE IF EXISTS sp_while1;
 5 /*声明结束符为$*/
 6 DELIMITER $
 7 /*创建存储过程*/
 8 CREATE PROCEDURE sp_while1(varial_count int)
 9   BEGIN
10     DECLARE idx int DEFAULT 1;
11     DECLARE uname VARCHAR(30) DEFAULT '';
12     loop_label:WHILE idx<=varial_count DO
13       select studentname into uname from students where studentid = idx;
14       INSERT into studentCount values (idx,uname);
15       SET idx=idx+1;
16     END WHILE;
17   END $
18 /*结束符置为;*/
19 DELIMITER ; 

调用存储过程，给定数值范围是10，所以这边取出1~10的数据存储到studentCount表中

 1 mysql> CALL sp_while1(10);
 2 Query OK, 1 row affected
 3
 4 mysql> select * from studentCount;
 5 +-----------+-------------+
 6 | studentid | studentname |
 7 +-----------+-------------+
 8 |         1 | brand       |
 9 |         2 | helen       |
10 |         3 | lyn         |
11 |         4 | sol         |
12 |         5 | b1          |
13 |         6 | b2          |
14 |         7 | c1          |
15 |         8 | c2          |
16 |         9 | lala        |
17 |        10 | A           |
18 +-----------+-------------+
19 10 rows in set 

while示例：包含iterate/leave

前面我们说明过了，iterate 和 leave 分别代表跳过本次循环，类似于C#中的continue和break。我们在例子中测试下吧：

遇到studentname=lala时，结束循环，遇到偶数时候跳过单次循环。

 1 /*先清除studentCount表记录*/
 2 truncate table studentcount;
 3 /*存储过程如果存在则删除*/
 4 DROP PROCEDURE IF EXISTS sp_while2;
 5 /*声明结束符为$*/
 6 DELIMITER $
 7 /*创建存储过程*/
 8 CREATE PROCEDURE sp_while2(varial_count int)
 9   BEGIN
10     DECLARE idx int DEFAULT 0;
11     DECLARE uname VARCHAR(30) DEFAULT '';
12     loop_label:WHILE idx<=varial_count DO
13       SET idx=idx+1;
14       select studentname into uname from students where studentid = idx;
15       /*如果遇到studentname为lala的同学，结束循环*/
16       IF uname='lala' THEN
17         LEAVE loop_label;
18       /*如果idx为偶数，则跳过本次循环*/
19       ELSEIF idx%2=0 THEN
20         ITERATE loop_label;
21       END IF;
22       INSERT into studentCount values (idx,uname);
23     END WHILE;
24   END $
25 /*结束符置为;*/
26 DELIMITER ; 

调用存储过程，输出符合要求的数据：

 1 mysql> CALL sp_while2(10);
 2 Query OK, 1 row affected
 3
 4 mysql> select * from studentCount;
 5 +-----------+-------------+
 6 | studentid | studentname |
 7 +-----------+-------------+
 8 |         1 | brand       |
 9 |         3 | lyn         |
10 |         5 | b1          |
11 |         7 | c1          |
12 +-----------+-------------+
13 4 rows in set

循环语句repeat

repeat语法

1 [loop_label:]repeat
2 -- Todo loop body
3 until condition
4 end repeat [loop_label]; 

可以对比下上面while的语法，while是先判断条件是否成立再执行循环体，repeat循环更像是的do...while循环，就是循环始终都会先执行一次，然后再判断结束循环的条件，不满足结束条件，循环体继续执行。

 1 /*先清除studentCount表记录*/
 2 truncate table studentcount;
 3 /*存储过程如果存在则删除*/
 4 DROP PROCEDURE IF EXISTS sp_repeat;
 5 /*声明结束符为$*/
 6 DELIMITER $
 7 /*创建存储过程*/
 8 CREATE PROCEDURE sp_repeat(varial_count int)
 9   BEGIN
10     DECLARE idx int DEFAULT 0;
11     DECLARE uname VARCHAR(30) DEFAULT '';
12     loop_label:REPEAT
13       SET idx=idx+1;
14       select studentname into uname from students where studentid = idx;
15       /*如果遇到studentname为lala的同学，结束循环*/
16       IF uname='lala' THEN
17         LEAVE loop_label;
18       /*如果idx为偶数，则跳过本次循环*/
19       ELSEIF idx%2=0 THEN
20         ITERATE loop_label;
21       END IF;
22       INSERT into studentCount values (idx,uname);
23     UNTIL idx>varial_count
24     END REPEAT;
25   END $
26 /*结束符置为;*/
27 DELIMITER ; 

注意条件的变化，下面是调用存储过程，输出需要的数据：

 1 mysql> CALL sp_repeat(10);
 2 Query OK, 1 row affected
 3
 4 mysql> select * from studentCount;
 5 +-----------+-------------+
 6 | studentid | studentname |
 7 +-----------+-------------+
 8 |         1 | brand       |
 9 |         3 | lyn         |
10 |         5 | b1          |
11 |         7 | c1          |
12 +-----------+-------------+
13 4 rows in set 

循环语句loop

loop语法

1 [loop_label:]loop
2 --Todo loop body
3 end loop [loop label]; 

loop不像while和repeat那样有控制条件，条件不符合的时候会跳出。所以它实际上是会一直执行的，如果不主动中断或者跳出的话，类似于一个死循环，需要在循环体中使用iterate或者leave来控制循环的执行。

 1 /*先清除studentCount表记录*/
 2 truncate table studentcount;
 3 /*存储过程如果存在则删除*/
 4 DROP PROCEDURE IF EXISTS sp_loop;
 5 /*声明结束符为$*/
 6 DELIMITER $
 7 /*创建存储过程*/
 8 CREATE PROCEDURE sp_loop(varial_count int)
 9   BEGIN
10     DECLARE idx int DEFAULT 0;
11     DECLARE uname VARCHAR(30) DEFAULT '';
12     loop_label:LOOP
13       SET idx=idx+1;
14       select studentname into uname from students where studentid = idx;
15       /*如果遇到studentname为lala的同学，结束循环*/
16       IF uname='lala' THEN
17         LEAVE loop_label;
18       /*如果idx为偶数，则跳过本次循环*/
19       ELSEIF idx%2<>0 THEN
20         ITERATE loop_label;
21       /*这边加一个终结计数跳出的条件*/
22       ELSEIF idx>varial_count THEN
23         LEAVE loop_label;
24       END IF;
25       INSERT into studentCount values (idx,uname);
26
27     END LOOP;
28   END $
29 /*结束符置为;*/
30 DELIMITER ; 

调用存储过程，并输出你需要的数据：

 1 mysql> CALL sp_loop(6);
 2 Query OK, 1 row affected
 3
 4 mysql> select * from studentCount;
 5 +-----------+-------------+
 6 | studentid | studentname |
 7 +-----------+-------------+
 8 |         2 | helen       |
 9 |         4 | sol         |
10 |         6 | b2          |
11 +-----------+-------------+
12 3 rows in set

总结

1、了解了IF函数，它常用在SELECT语句中，类似于C#中的三元表达式。
2、IF条件表达式，类似于C#中的IF... ELSE...，多用于函数或存储过程中的判断选择逻辑。
3、了解CASE语句的两种用法，一种用在SELECT中使用，一种用在函数和存储过程中。
4、了解了三种循环体的使用，while、repeat分别对应C#中的while 和 do while循环，loop类似于一个while(true)的死循环。
5、循环体都包含在begin end中，循环体的控制依靠leave和iterate，leave相当于break，即退出整个循环体，iterate类似于continue，即跳过本次循环。