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循环

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容