mysql 存储过程中的if 多条件判断

原创

mob649e8168f1bb 2024-08-08 11:09:42 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8168f1bb的原创作品，请联系作者获取转载授权，否则将追究法律责任

MySQL 存储过程中的 IF 多条件判断

在 MySQL 中，存储过程是一种将一组 SQL 语句封装在一起，以便于重复使用的程序。存储过程内的控制流语句（如 IF、CASE、LOOP）使我们能够编写更为复杂的数据库逻辑。本文将重点介绍如何在 MySQL 存储过程中使用 IF 语句进行多条件判断，并通过代码示例加以说明。

IF 语句的基本用法

IF 语句的语法比较简单，基本结构如下：

IF condition THEN
    -- 逻辑处理
ELSEIF condition2 THEN
    -- 逻辑处理
ELSE
    -- 逻辑处理
END IF;

在上述语法中，可以通过 ELSEIF和 ELSE 来处理多种条件。接下来，我们将讨论一个具体的例子，帮助理解其应用场景。

示例：处理订单状态

假设我们有一个简单的订单表 orders，包含 order_id、order_status 和 customer_id 字段。我们需要根据订单状态来更新每个订单的处理结果。订单状态可能有如下几种：'Pending'、'Shipped' 和 'Cancelled'。

首先，我们需要创建一个订单表并插入一些数据：

CREATE TABLE orders (
    order_id INT PRIMARY KEY,
    order_status VARCHAR(20),
    customer_id INT
);

INSERT INTO orders (order_id, order_status, customer_id)
VALUES 
(1, 'Pending', 101),
(2, 'Shipped', 102),
(3, 'Cancelled', 101),
(4, 'Pending', 103);

接着，可以定义一个存储过程，用以处理订单状态：

DELIMITER //

CREATE PROCEDURE UpdateOrderStatus(IN orderId INT)
BEGIN
    DECLARE status VARCHAR(20);
    
    -- 查找当前订单状态
    SELECT order_status INTO status FROM orders WHERE order_id = orderId;
    
    -- 根据状态进行处理
    IF status = 'Pending' THEN
        UPDATE orders SET order_status = 'Processed' WHERE order_id = orderId;
    ELSEIF status = 'Shipped' THEN
        UPDATE orders SET order_status = 'Delivered' WHERE order_id = orderId;
    ELSEIF status = 'Cancelled' THEN
        SELECT 'Order has been cancelled, no action needed.';
    ELSE
        SELECT 'Unknown order status!';
    END IF;
END //

DELIMITER ;

在这个存储过程中，我们首先根据 order_id 获取当前订单的状态，并根据不同状态进行相应的更新或反馈。

逻辑关系图

为了更好地理解流程，我们可以用 ER 图表示出两个表之间的关系。

erDiagram
    ORDERS {
        INT order_id PK
        VARCHAR order_status
        INT customer_id
    }
    CUSTOMERS {
        INT customer_id PK
        VARCHAR customer_name
    }
    ORDERS ||--o| CUSTOMERS : "belongs to"