MySQL自定义函数（二）- 进阶篇

MySQL自定义函数：

使用自定义函数，可以简化很多的操作，在上一回，我们练习了无参函数，有参函数，其实在函数中，还可以完成很多复杂的功能，这里我们继续了解下。

在函数中执行多条SQL

上一篇，我们只是简单的使用return 返回了字符串，其实，这里是可以执行多条SQL的。

create function do_some_delete(p_emp_no int) returns varchar(20)
begin
    delete from titles where emp_no=p_emp_no;
    delete from employees where emp_no=p_emp_no;
    return 'ok';
end

当我们有多条SQL要执行的时候，需要使用begin ... end，将多条SQL写在里面，这个函数通过传入的emp_no，删除了2张表中的数据，就是说函数中封装了我们想要执行的一系列SQL。

变量的使用

关于变量，以前写了篇文章，可以看考下：MySQL-变量的使用

MySQL中的变量和其他编程语言中的变量差不多，可以理解成临时存储数据的

begin
    declare total_record int;

    select count(1) from titles into total_record;

    return concat('total record of table titles is ',total_record);
end

使用变量的时候，需要提前声明，使用declare 变量名类型
......
我想错了，并不需要提前定义，直接去给变量赋值也是可以的，比如这样：

drop function if exists query_total_record;
create function query_total_record() returns varchar(100) charset utf8
begin

    select count(1) from titles into @total_record_titles;
    select count(1) from employees into @total_record_employees;

    return concat('titles表总记录数为： ',@total_record,' , employees表总记录数为：',@total_record_employees);
end

注意一下，我这里使用了@变量名，这个是比较标准的用户自定义变量定义，而且使用@变量名的时候，可以不用提前声明，但是如果不适用@的话，直接使用变量名，是没有办法不声明就使用的，这一点大家注意下。

就是说，下面这种写法是会报错的：

create function query_total_record() returns varchar(100) charset utf8
begin

    select count(1) from titles into total_record_titles;
    select count(1) from employees into total_record_employees;

    return concat('titles表总记录数为： ',total_record,' , employees表总记录数为：',total_record_employees);
end

所以后面，标准起见，我们就使用@变量名的方式了。

上面，我们使用select xxx into 变量的方式，给变量初始化，我们还可以使用 set 的方式给变量赋值。

create function do_some_add(p_1 int) returns varchar(100)
begin 
    set @a=10;
    set @b=10,@c=20;
    
    set @d=@a+@b-@c+p_1;
    
    return concat(@a,' + ',@b,' - ',@c,' + ',p_1,' = ',@d);
end

IF语句

在begin end里面，我们还可以使用流程控制语句，我们先从IF 说起

create function parse_age(age int ) returns varchar(100) charset utf8
begin
    if age<=20 then set @msg='嘿，你还不到20岁';
    else set @msg='嘿，你超过20岁了！';
    end if;
    
    return @msg;
end

我们还可以使用 elseif做多重判断

create function parse_age(age int ) returns varchar(100) charset utf8
begin
    if age<=20 then set @msg='嘿，你还不到20岁';
    ELSEIF age<=30 then set @msg=concat(@msg,'嘿，你20多岁了');
    else set @msg='嘿，你超过20岁了！';
    end if;
    
    return @msg;
end

记得在结尾使用end if

case 语句

这个和我们平时使用的case when类似，结尾需要使用 end case;

create function parse_age(age int ) returns varchar(100) charset utf8
begin
    case 
        when age<=20 then set @msg='嘿，你还不到20岁';
        when age<=30 then set @msg='嘿，你20多岁了';
        else set @msg='嘿，你超过20岁了！';
    end case;
    
    return @msg;
end

while 循环

函数里面也是可以写循环的，我们先来看while循环方式。

create function while_demo(num int) returns int
begin 
    set @count=0;
    
    while num>0 do 
        set @count=@count+num;
        set num=num-1;
    end while;
    
    return @count;
end

本来还在想这个传入的参数能不能修改，试了下，也是一样的，就是个传入变量，这里就是做了一个累加

repeat语句

这个和while类似，只是个repeat是先执行循环体，满足条件后就退出

create function repeat_demo(num int) returns int 
begin 
        set @count=0;
    
        repeat 
            set @count=@count+num;
            set num=num-1;
        until num=0
        end repeat;

        return @count;
end

Loop语句

这是一个基本的循环语句，用来执行循环内容，但是Loop本身没有退出循环的方法，需要借助其他的方式，比如 LEAVE，或者ITERATE

create function loop_demo(num int) returns int
begin 
    set @count=0;
    
    loop1:Loop 
    
        set @count=@count+num;
        set num=num-1;
        
        if num=0 then LEAVE loop1; end if;
    end Loop loop1 ;

    return @count;
end

像上面的代码一样，我们要通过自己写逻辑判断来跳出Loop循环

好了，进阶篇，也到这里位置，这里介绍了变量和一些控制语句的使用，在实际使用中灵活运用即可。