在 case 语句中更新值并设置局部变量

Question

我正在尝试更新数据库中的值，但也想设置一个局部变量VAR_IS_RATE_LIMITED。这样做的原因是因为我不想使用 select 语句并希望在一个语句中执行它。如何设置VAR_IS_RATE_LIMITED？我查看了其他问题，但他们的 CASE 语句没有嵌入到更新语句中。

DELIMITER //
CREATE FUNCTION F_RATE_LIMITED(P_IP varchar(45),
                               P_MAX_RATE int unsigned
                            )
RETURNS INT UNSIGNED
BEGIN

    DECLARE VAR_IS_RATE_LIMITED INT UNSIGNED DEFAULT 0;

    INSERT INTO rate_limit (ip, rate)
        VALUES (P_IP, 1)
    ON DUPLICATE KEY UPDATE 
        rate = 
            CASE
                WHEN (rate + 1) > P_MAX_RATE THEN
                    SET VAR_IS_RATE_LIMITED = 1;
                    rate
                ELSE
                    rate + 1
            END;

    RETURN VAR_IS_RATE_LIMITED;
END; //
DELIMITER ;

Answer 1

要在数据修改语句期间设置某些内容，它必须是表达式的一部分，而不是语句。据我所知，您不能在表达式中设置存储的函数变量。但是你可以设置一个用户变量：

DELIMITER //
CREATE FUNCTION F_RATE_LIMITED(P_IP varchar(45),
                               P_MAX_RATE int unsigned
                            )
RETURNS INT UNSIGNED
BEGIN

    SET @VAR_IS_RATE_LIMITED = 0;
    INSERT INTO rate_limit (ip, rate)
        VALUES (P_IP, 1)
    ON DUPLICATE KEY UPDATE
        rate =
            CASE
                WHEN (rate + 1) > P_MAX_RATE THEN
                    CASE WHEN @VAR_IS_RATE_LIMITED := 1 THEN rate END
                ELSE
                    rate + 1
            END;

    RETURN @VAR_IS_RATE_LIMITED;
END; //
DELIMITER ;

这里的变量是在 CASE 测试中设置的（您也可以使用 IF 表达式），该测试始终为真，因此始终从 CASE 返回 rate。