拆分字符串并循环遍历MySql过程中的值

时间:2016-05-13 15:22:42

标签: mysql sql stored-procedures

我遇到这种情况,我必须将逗号分隔的字符串传递给MySQL过程并拆分该字符串并将这些值作为行插入到表中。

如下所示

例如,如果我将'jhon,swetha,sitha'字符串传递给mysql过程,则必须用逗号分割该字符串,并将这些值作为3条记录插入表中。

  CREATE PROCEDURE new_routine (IN str varchar(30))   
   BEGIN
       DECLARE tmp varchar(10);
       DECLARE inc INT DEFAULT 0; 
       WHILE INSTR(str, ',') DO
         SET tmp = SUBSTRING(SUBSTRING_INDEX(str,',',inc),LENGTH(SUBSTRING_INDEX(str,',',inc-1))+1),',','');
         SET str = REPLACE(str, tmp, '');
         //insert tmp into a table.
       END WHILE;
    END

但这不能解决任何问题。

1 个答案:

答案 0 :(得分:34)

你需要对字符串操作更加小心。您不能使用REPLACE(),因为这将替换多次出现,如果逗号分隔列表中的一个元素是另一个元素的子字符串,则会破坏您的数据。 INSERT() string function对此更好,不要与用于插入表的INSERT语句混淆。

DELIMITER $$

DROP PROCEDURE IF EXISTS `insert_csv` $$
CREATE PROCEDURE `insert_csv`(_list MEDIUMTEXT)
BEGIN

DECLARE _next TEXT DEFAULT NULL;
DECLARE _nextlen INT DEFAULT NULL;
DECLARE _value TEXT DEFAULT NULL;

iterator:
LOOP
  -- exit the loop if the list seems empty or was null;
  -- this extra caution is necessary to avoid an endless loop in the proc.
  IF LENGTH(TRIM(_list)) = 0 OR _list IS NULL THEN
    LEAVE iterator;
  END IF;

  -- capture the next value from the list
  SET _next = SUBSTRING_INDEX(_list,',',1);

  -- save the length of the captured value; we will need to remove this
  -- many characters + 1 from the beginning of the string 
  -- before the next iteration
  SET _nextlen = LENGTH(_next);

  -- trim the value of leading and trailing spaces, in case of sloppy CSV strings
  SET _value = TRIM(_next);

  -- insert the extracted value into the target table
  INSERT INTO t1 (c1) VALUES (_value);

  -- rewrite the original string using the `INSERT()` string function,
  -- args are original string, start position, how many characters to remove, 
  -- and what to "insert" in their place (in this case, we "insert"
  -- an empty string, which removes _nextlen + 1 characters)
  SET _list = INSERT(_list,1,_nextlen + 1,'');
END LOOP;

END $$

DELIMITER ;

接下来是一个测试表:

CREATE TABLE `t1` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `c1` varchar(64) NOT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

新表格为空。

mysql> SELECT * FROM t1;
Empty set (0.00 sec)

调用程序。

mysql> CALL insert_csv('foo,bar,buzz,fizz');
Query OK, 1 row affected (0.00 sec)

注意" 1行受影响"并不意味着你所期望的。它指的是我们做的最后一个插入。由于我们一次插入一行,如果过程插入至少一行,那么您的行数总是为1;如果程序什么都不插入,那么你会受到0行的影响。

有效吗?

mysql> SELECT * FROM t1;
+----+------+
| id | c1   |
+----+------+
|  1 | foo  |
|  2 | bar  |
|  3 | buzz |
|  4 | fizz |
+----+------+
4 rows in set (0.00 sec)