MySQL存储过程中处理数组 UDF代码分享

最新推荐文章于 2024-07-03 08:24:40 发布

cq377078944

最新推荐文章于 2024-07-03 08:24:40 发布

阅读量166

点赞数 1

在存储过程中，传入的参数经常可能会有数组，而mysql没有自带处理数组的方法或函数。

需求是这样的：
传入数组形如：【aaa|bbb|ccc】
返回值形如：【aaa】、【bbb】、【ccc】

自己写了一个自定义函数来完成上述需求：

使用例子：

mysql> select FUN_GET_ELEMENT('第一个|第二个|第三个', '|', -1);
+------------------------------------------------------------------------------+
| FUN_GET_ELEMENT('第一个|第二个|第三个', '|', -1) |
+------------------------------------------------------------------------------+
| 上越界 |
+------------------------------------------------------------------------------+
1 row in set (0.00 sec)
mysql> select FUN_GET_ELEMENT('第一个|第二个|第三个', '|', 0);
+------------------------------------------------------------------------------+
| FUN_GET_ELEMENT('第一个|第二个|第三个', '|', 0) |
+------------------------------------------------------------------------------+
| 第一个 |
+------------------------------------------------------------------------------+
1 row in set (0.00 sec)
mysql> select FUN_GET_ELEMENT('第一个|第二个|第三个', '|', 1);
+-----------------------------------------------------------------------------+
| FUN_GET_ELEMENT('第一个|第二个|第三个', '|', 1) |
+-----------------------------------------------------------------------------+
| 第二个 |
+-----------------------------------------------------------------------------+
1 row in set (0.01 sec)
mysql> select FUN_GET_ELEMENT('第一个|第二个|第三个', '|', 2);
+-----------------------------------------------------------------------------+
| FUN_GET_ELEMENT('第一个|第二个|第三个', '|', 2) |
+-----------------------------------------------------------------------------+
| 第三个 |
+-----------------------------------------------------------------------------+
1 row in set (0.00 sec)
mysql> select FUN_GET_ELEMENT('第一个/第二个/第三个', '/', 5);
+-----------------------------------------------------------------------------+
| FUN_GET_ELEMENT('第一个/第二个/第三个', '/', 5) |
+-----------------------------------------------------------------------------+
| 下越界 |
+-----------------------------------------------------------------------------+
1 row in set (0.00 sec)

其中传入的值有三个：
第一个是要处理的数组，
第二个是数组内元素与元素的分隔符，比如 | /
第三个是要取出的元素索引，比如，0为第一个元素，1为第二个元素，以此类推，若输入的索引 没有值，则返回上越界或下越界

FUN_GET_ELEMENT函数定义：

DROP FUNCTION IF EXISTS FUN_GET_ELEMENT;
delimiter ;;
CREATE FUNCTION `FUN_GET_ELEMENT`(
`mainstring` varchar(1024) CHARSET utf8,
`mark` varchar(4) CHARSET utf8,
`count` int
)
RETURNS varchar(65) CHARSET utf8
BEGIN
DECLARE v_string varchar(1024) CHARSET utf8 DEFAULT mainstring;
DECLARE v_element varchar(64) CHARSET utf8;
DECLARE i int DEFAULT 0;
IF(count > 0 and count <= FUN_GET_COUNT(mainstring, mark)) THEN
WHILE(i < count) DO
SET v_string = SUBSTRING(v_string, INSTR(v_string, mark) + 1, CHAR_LENGTH(v_string));
SET i = i + 1;
END WHILE;
END IF;
IF(count < 0) THEN
SET v_element = '上越界';
ELSEIF (count < FUN_GET_COUNT(mainstring, mark)) THEN
SET v_element = SUBSTRING(v_string, 1, INSTR(v_string, mark) - 1);
ELSEIF (count = FUN_GET_COUNT(mainstring, mark)) THEN
SET v_element = v_string;
ELSE
SET v_element = '下越界';
END IF;
RETURN(v_element);
END;;
delimiter ;

上述UDF调用了另外一个函数FUN_GET_COUNT()，这一个函数同样也是自定义的。
用于获取传入的字符串数组的最大索引（元素数目-1）。

该函数使用例子：

mysql> select FUN_GET_COUNT('a|b|c','|');
+----------------------------+
| FUN_GET_COUNT('a|b|c','|') |
+----------------------------+
| 2 |
+----------------------------+
1 row in set (0.00 sec)

FUN_GET_COUNT 函数定义：

DROP FUNCTION IF EXISTS FUN_GET_COUNT;
delimiter ;;
CREATE FUNCTION `FUN_GET_COUNT`(
`mainstring` varchar(1024) CHARSET utf8,
`mark` varchar(200) CHARSET utf8
) RETURNS int
BEGIN
DECLARE i int;
SET i = (CHAR_LENGTH(mainstring) - CHAR_LENGTH(REPLACE(mainstring, mark, ''))) / CHAR_LENGTH(mark);
RETURN i;
END;;
delimiter ;