数据库基础(12) . 自定义函数

在 MySQL 中，你可以创建自定义函数来扩展其内置功能。自定义函数可以通过编写存储过程或函数来实现。这里我们将重点介绍如何创建自定义函数，并给出一些示例。

创建自定义函数的基本语法

创建自定义函数的基本语法如下：

CREATE FUNCTION function_name(args)
RETURNS return_type
BEGIN-- 函数体-- 执行计算并返回结果RETURN result;
END;

示例：创建一个简单的数学函数

假设我们要创建一个自定义函数来计算两个整数的和。

示例：计算两个整数的和

DELIMITER //
CREATE FUNCTION add_two_numbers(num1 INT, num2 INT)
RETURNS INT
BEGINDECLARE result INT;SET result = num1 + num2;RETURN result;
END//
DELIMITER ;-- 测试函数
SELECT add_two_numbers(5, 3); -- 输出 8

示例：创建一个字符串处理函数

接下来，我们创建一个函数来处理字符串，例如拼接两个字符串。

示例：拼接两个字符串

DELIMITER //
CREATE FUNCTION concatenate_strings(str1 VARCHAR(255), str2 VARCHAR(255))
RETURNS VARCHAR(510)
BEGINDECLARE result VARCHAR(510);SET result = CONCAT(str1, ' ', str2);RETURN result;
END//
DELIMITER ;-- 测试函数
SELECT concatenate_strings('Hello', 'World'); -- 输出 'Hello World'

示例：创建一个日期处理函数

我们还可以创建一个函数来处理日期，例如计算两个日期之间的天数差。

示例：计算两个日期之间的天数差

DELIMITER //
CREATE FUNCTION days_between_dates(date1 DATE, date2 DATE)
RETURNS INT
BEGINDECLARE days_diff INT;SET days_diff = DATEDIFF(date1, date2);RETURN days_diff;
END//
DELIMITER ;-- 测试函数
SELECT days_between_dates('2024-01-01', '2024-12-31'); -- 输出 364

示例：使用条件语句的函数

假设我们需要创建一个函数来判断一个数字是否为偶数，并返回相应的布尔值。

示例：判断一个数字是否为偶数

DELIMITER //
CREATE FUNCTION is_even(num INT)
RETURNS BOOLEAN
BEGINDECLARE result BOOLEAN;IF num % 2 = 0 THENSET result = TRUE;ELSESET result = FALSE;END IF;RETURN result;
END//
DELIMITER ;-- 测试函数
SELECT is_even(4); -- 输出 1 (表示 true)
SELECT is_even(5); -- 输出 0 (表示 false)

示例：使用循环的函数

我们还可以创建一个函数来计算斐波那契数列中的第 n 个数。

示例：计算斐波那契数列中的第 n 个数

DELIMITER //
CREATE FUNCTION fibonacci(n INT)
RETURNS INT
BEGINDECLARE a INT DEFAULT 0;DECLARE b INT DEFAULT 1;DECLARE i INT DEFAULT 2;IF n = 0 THEN RETURN a; END IF;IF n = 1 THEN RETURN b; END IF;WHILE i <= n DOSET a = b - a;SET b = b + a;SET i = i + 1;END WHILE;RETURN b;
END//
DELIMITER ;-- 测试函数
SELECT fibonacci(10); -- 输出 55