极客笔记MySQL 存储过程中的 IF 语句
MySQL 中的 IF 语句在存储过程中使用非常频繁，它可以根据具体条件执行相应的 SQL 语句，实现了存储过程的灵活性和可控性。本文将详细介绍 MySQL 中 IF 语句在存储过程中的使用方法，并通过实例说明。
阅读更多：MySQL 教程
IF 语句的语法
MySQL 中，IF 语句有两种语法形式：
一、IF-THEN 语法
IF expression THEN
statement_list
END IF;
其中，expression 是一个布尔表达式，如果该表达式为真，则执行 statement_list 中的代码块。如果为假，则跳过 statement_list。
例如，下面的例子中，当参数 a 大于 10 时，执行 SELECT 'a 大于 10' 语句，否则什么也不做。
DELIMITER CREATE PROCEDURE `test_if` (IN `a` INT)
BEGIN
IF a>10 THEN
SELECT 'a 大于 10';
END IF;
END
DELIMITER ;
二、IF-THEN-ELSE 语法
IF expression THEN
statement_list
ELSE
statement_list
END IF;
其中，expression 是一个布尔表达式，如果该表达式为真，则执行第一条 statement_list 中的代码块；否则执行第二条 statement_list 中的代码块。
例如，下面的例子中，当参数 a 大于 10 时，执行 SELECT 'a 大于 10' 语句，否则执行 SELECT 'a 小于等于 10' 语句。
DELIMITER CREATE PROCEDURE `test_if_else` (IN `a` INT)
BEGIN
IF a>10 THEN
SELECT 'a 大于 10';
ELSE
SELECT 'a 小于等于 10';
END IF;
END
DELIMITER ;
IF 语句的使用方法
一、在存储过程中使用 IF 语句
在 MySQL 中，IF 语句常常被用于存储过程中，以实现具体条件下的 SQL 操作。下面以一个实例为例，说明如何在存储过程中使用 IF 语句。
例如，有一个 student 表，包含 id（学生 ID）、name（学生姓名）、score（学生分数）和 rank（学生排名）四个字段。现在需要编写一个存储过程，将学生的排名计算出来，然后更新 rank 字段。
首先，我们可以创建一个名为 update_rank 存储过程。
DELIMITER CREATE PROCEDURE `update_rank` ()
BEGIN
DECLARE i INT;
SELECT COUNT(*) INTO i FROM student;
WHILE i>0 DO
UPDATE student SET rank = 1 WHERE id = (SELECT id FROM student ORDER BY score DESC LIMIT i - 1, 1);
SET i = i - 1;
END WHILE;
END
DELIMITER ;
上面的存储过程实现了一个简单的计算排名的功能。它通过一个 WHILE 循环，依次更新每个学生的排名。其中：
- 定义了一个变量
i，用来记录学生的总数； - 在
WHILE循环中，依次取出排名第i名的学生 ID，并将其排名更新为 1； - 更新完成后，将
i减 1，即去掉排名最后一名学生。
但在实际应用中，有些学生分数可能相等，因此需要为这类学生设置相同的排名。我们可以在循环中，判断当前学生分数是否与上一名学生分数相等，如果相等，则将排名设置为上一名学生的排名；否则，将排名设置为当前循环次数，即 i。
下面是具体的实现代码：
DELIMITER CREATE PROCEDURE `update_rank` ()
BEGIN
DECLARE i INT;
DECLARE last_score INT DEFAULT 0;
DECLARE current_rank INT DEFAULT 0;
SELECT COUNT(*) INTO i FROM student;
WHILE i>0 DO
SELECT score INTO last_score FROM student ORDER BY score DESC LIMIT i - 1, 1;
IF last_score = (SELECT score FROM student ORDER BY score DESC LIMIT i - 2, 1) THEN
UPDATE student SET rank = current_rank WHERE score = last_score;
ELSE
SET current_rank = i;
UPDATE student SET rank = current_rank WHERE score = last_score;
END IF;
SET i = i - 1;
END WHILE;
END
DELIMITER ;
上面的代码中，我们在循环中先定义了两个变量 last_score 和 current_rank，分别用来记录上一名学生的分数以及当前排名。在每次循环中，用 SELECT 语句获取当前排名第 i 名学生的分数，并将其与上一名学生的分数进行比较。如果相等，则将排名设置为上一名学生的排名；否则，将排名设置为当前循环次数 i。
如果我们现在查询 student 表，会发现每个学生的排名已经被正确计算出来了。
SELECT * FROM student;
+----+--------+-------+------+
| id | name | score | rank |
+----+--------+-------+------+
| 1 | Tom | 80 | 2 |
| 2 | Jack | 90 | 1 |
| 3 | Peter | 70 | 3 |
| 4 | Lily | 90 | 1 |
| 5 | Alice | 85 | 4 |
| 6 | Bob | 60 | 6 |
| 7 | Wendy | 75 | 5 |
+----+--------+-------+------+
二、在存储过程中使用 IF-ELSE 语句
在存储过程中，有时我们需要根据不同的条件执行不同的 SQL 操作。这时就需要用到 IF-ELSE 语句。下面以一个实例为例，说明如何在存储过程中使用 IF-ELSE 语句。
假设我们有一个 employee 表，包含 id（员工 ID）、name（员工姓名）和 salary（员工工资）三个字段。现在需要编写一个存储过程，给每个员工涨工资。如果员工的工资低于 5000，那么涨 500 元的工资；否则，涨 300 元的工资。
首先，我们可以创建一个名为 increase_salary 存储过程。
DELIMITER CREATE PROCEDURE `increase_salary` ()
BEGIN
DECLARE e_id INT;
DECLARE e_salary INT;
DECLARE done INT DEFAULT 0;
DECLARE cur CURSOR FOR SELECT id, salary FROM employee;
DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = 1;
OPEN cur;
read_loop: LOOP
FETCH cur INTO e_id, e_salary;
IF done THEN
LEAVE read_loop;
END IF;
IF e_salary<5000 THEN
SET e_salary = e_salary + 500;
ELSE
SET e_salary = e_salary + 300;
END IF;
UPDATE employee SET salary = e_salary WHERE id = e_id;
END LOOP;
CLOSE cur;
END
DELIMITER ;
上面的代码中，我们使用了一个 CURSOR 来遍历 employee 表，并根据员工工资的不同情况，调整工资涨幅。具体实现方法是：在每次循环中，先判断是否读取完了所有的记录（通过 done 变量控制）；然后使用 IF-ELSE 语句判断员工工资是否低于 5000 元，根据不同情况涨工资；最后更新 employee 表中该员工的工资字段。
如果我们现在查询 employee 表，会发现每个员工的工资已经成功涨了。
SELECT * FROM employee;
+----+--------+--------+
| id | name | salary |
+----+--------+--------+
| 1 | Tom | 15000 |
| 2 | Jack | 10500 |
| 3 | Peter | 5500 |
| 4 | Lily | 6200 |
| 5 | Alice | 7800 |
| 6 | Bob | 4800 |
| 7 | Wendy | 5250 |
+----+--------+--------+
总结
IF 语句在 MySQL 存储过程中的使用非常灵活，可以帮助我们根据具体条件执行相应的 SQL 操作，实现了存储过程的灵活性和可控性。在实际应用中，我们可以结合具体需求，合理使用 IF 语句和 IF-ELSE 语句，提高存储过程的效率和准确性。