PostgreSql | 数据库 |自定义函数的设计和实现

本文涉及的产品
云原生数据库 PolarDB MySQL 版,通用型 2核4GB 50GB
云原生数据库 PolarDB PostgreSQL 版,标准版 2核4GB 50GB
简介: PostgreSql | 数据库 |自定义函数的设计和实现

前言:

数据库中的函数概念

和开发语言,Java,PHP,Python等等类似,关系型数据库也是有函数的,函数指的是动态的封装一部分特定功能的集合。

例如,查询任意表的记录数,复制指定表,这些函数可以简化一些数据库的操作

那么,最为简单直观的方法还是用一些演示示例来说明函数是什么以及函数的特性。

一,

查询任意表的记录数

create or replace function getsum(in talename VARCHAR)
RETURNS int as $$
DECLARE
   stmt VARCHAR;
   count int;
begin
  stmt:=format('select count(1) from %s', talename);
  raise notice '%',stmt;
  EXECUTE stmt into count;
  return count;
  EXCEPTION --捕获异常
  WHEN OTHERS THEN
    RETURN 1111;
end; $$ LANGUAGE plpgsql;

可以看到,该函数是一个带参函数,参数就是要查询的表的名称

调用函数:

select getsum('emp');

结果如下:

test=# select getsum('emp');
NOTICE:  select count(1) from emp
 getsum 
--------
     14
(1 row)

如果查询的是不存在的表,那么,将会返回函数定义的1111:

test=# select getsum('erwe');
NOTICE:  select count(1) from erwe
 getsum 
--------
   1111
(1 row)

查询指定表的记录数:

CREATE OR REPLACE FUNCTION "public"."totalrecords"()
  RETURNS "pg_catalog"."int4" AS $BODY$  
declare  
    total integer;  
BEGIN  
   SELECT count(*) into total FROM emp;  
   RETURN total;  
   EXCEPTION --捕获异常
    WHEN OTHERS THEN
    RETURN 1111;
END;  
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100

调用该函数:

test=# select totalrecords();
 totalrecords 
--------------
           14
(1 row)

二,

利用函数优雅的创建用户

CREATE OR REPLACE FUNCTION "public"."__tmp_create_user"()
  RETURNS "pg_catalog"."void" AS $BODY$
BEGIN
  IF NOT EXISTS (
          SELECT                       -- SELECT list can stay empty for this
          FROM   pg_catalog.pg_user
          WHERE  usename = 'postgres_exporter') THEN
    CREATE USER postgres_exporter;
  END IF;
END;
$BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100

该函数有if逻辑,其它没有什么太特殊的地方。

三,

复制指定表emp的数据不包括索引,约束,外键,指定份数:

create or replace function copytable(tablename varchar,times int)
RETURNS INT as $$
DECLARE
   stmt VARCHAR='create table %s_%s as select * from emp;';--
begin
   for i in 1..times loop
       raise notice 'd当前次数%',i;
       raise notice '%',format(stmt,tablename,i,tablename);
       EXECUTE format('drop table if EXISTS %s_%s;',tablename,i) ;
       EXECUTE format(stmt,tablename,i,tablename) ;
   end loop;
   return 100 
END; $$  LANGUAGE plpgsql;

该函数是for循环参数,循环创建指定名称的复制表,源表的名称是emp,emp表是真实存在的

四,

函数做加法

CREATE OR REPLACE FUNCTION "public"."add"("a" int4, "b" numeric)
  RETURNS "pg_catalog"."numeric" AS $BODY$
  SELECT a+b;
$BODY$
  LANGUAGE sql VOLATILE
  COST 100
select add(3,4);

调用函数:

test=# select add(55,6);
 add 
-----
  61
(1 row)

五,

函数复制表--同时复制索引,主键,约束,tablename1是新表名称加复制次数,tablename2是要复制的源表名称

CREATE OR REPLACE FUNCTION "public"."copytable"("tablename1" varchar, "tablename2" varchar, "times" int4)
  RETURNS "pg_catalog"."int4" AS $BODY$
DECLARE
   stmt1 varchar(1000);
   stmt2 varchar(1000);
begin
   for i in 1..times loop
       raise notice '当前次数%',i;
       stmt1 := format('create table "%1$s_%3$s" (like "%2$s" INCLUDING INDEXES INCLUDING DEFAULTS);',tablename1,tablename2,i);
       stmt2 := format('insert into "%1$s_%3$s" select * from "%2$s" ;',tablename1,tablename2,i);
       raise notice '%',stmt1;
       raise notice '%',stmt2;
       EXECUTE format('drop table if EXISTS "%1$s_%3$s";',tablename1,tablename2,i) ;
       EXECUTE stmt1 ;
       EXECUTE stmt2 ;
   end loop;
   return 110;      
END; $BODY$
  LANGUAGE plpgsql VOLATILE
  COST 100

该函数的调用示例(emp是源表,test表是新表,此函数将会建立两个表,表名分别是test_1和test_2)

另外,需要注意的是,该函数是冥等函数,也就是说,可以反复执行任意次,结果都是不变的。

test=# select copytable('test','emp',2);
NOTICE:  当前次数1
NOTICE:  create table "test_1" (like "emp" INCLUDING INDEXES INCLUDING DEFAULTS);
NOTICE:  insert into "test_1" select * from "emp" ;
NOTICE:  当前次数2
NOTICE:  create table "test_2" (like "emp" INCLUDING INDEXES INCLUDING DEFAULTS);
NOTICE:  insert into "test_2" select * from "emp" ;
 copytable 
-----------
       110
(1 row)
相关实践学习
使用PolarDB和ECS搭建门户网站
本场景主要介绍基于PolarDB和ECS实现搭建门户网站。
阿里云数据库产品家族及特性
阿里云智能数据库产品团队一直致力于不断健全产品体系,提升产品性能,打磨产品功能,从而帮助客户实现更加极致的弹性能力、具备更强的扩展能力、并利用云设施进一步降低企业成本。以云原生+分布式为核心技术抓手,打造以自研的在线事务型(OLTP)数据库Polar DB和在线分析型(OLAP)数据库Analytic DB为代表的新一代企业级云原生数据库产品体系, 结合NoSQL数据库、数据库生态工具、云原生智能化数据库管控平台,为阿里巴巴经济体以及各个行业的企业客户和开发者提供从公共云到混合云再到私有云的完整解决方案,提供基于云基础设施进行数据从处理、到存储、再到计算与分析的一体化解决方案。本节课带你了解阿里云数据库产品家族及特性。
目录
相关文章
|
8天前
|
关系型数据库 分布式数据库 数据库
开源云原生数据库PolarDB PostgreSQL 15兼容版本正式发布
PolarDB进行了深度的内核优化,从而实现以更低的成本提供商业数据库的性能。
|
30天前
|
存储 Ubuntu 关系型数据库
如何创建数据库 PostgreSQL
【8月更文挑战第12天】
38 4
如何创建数据库 PostgreSQL
|
20天前
|
关系型数据库 MySQL Linux
在Linux中,如何配置数据库服务器(如MySQL或PostgreSQL)?
在Linux中,如何配置数据库服务器(如MySQL或PostgreSQL)?
|
4天前
|
关系型数据库 数据库 网络虚拟化
Docker环境下重启PostgreSQL数据库服务的全面指南与代码示例
由于时间和空间限制,我将在后续的回答中分别涉及到“Python中采用lasso、SCAD、LARS技术分析棒球运动员薪资的案例集锦”以及“Docker环境下重启PostgreSQL数据库服务的全面指南与代码示例”。如果你有任何一个问题的优先顺序或需要立即回答的,请告知。
12 0
|
28天前
|
存储 Ubuntu 关系型数据库
如何在 Ubuntu VPS 上备份 PostgreSQL 数据库
如何在 Ubuntu VPS 上备份 PostgreSQL 数据库
15 1
|
30天前
|
关系型数据库 数据库 PostgreSQL
Linux 环境手动备份postgresql数据库
【8月更文挑战第12天】在Docker环境中使用命令行工具对PostgreSQL数据库进行备份和恢复。首先,通过dockerexec进入容器,使用pg_dump进行数据库模式的备份,然后使用dockercp将备份文件导出。接着,若需导入数据到另一数据库,先将备份文件复制到目标容器,再利用psql命令进行数据恢复。整个过程需确保目标数据库无同名模式,以防止导入失败
24 3
|
11天前
|
SQL 关系型数据库 MySQL
SQL Server、MySQL、PostgreSQL:主流数据库SQL语法异同比较——深入探讨数据类型、分页查询、表创建与数据插入、函数和索引等关键语法差异,为跨数据库开发提供实用指导
【8月更文挑战第31天】SQL Server、MySQL和PostgreSQL是当今最流行的关系型数据库管理系统,均使用SQL作为查询语言,但在语法和功能实现上存在差异。本文将比较它们在数据类型、分页查询、创建和插入数据以及函数和索引等方面的异同,帮助开发者更好地理解和使用这些数据库。尽管它们共用SQL语言,但每个系统都有独特的语法规则,了解这些差异有助于提升开发效率和项目成功率。
70 0
|
19天前
|
安全 关系型数据库 数据库
跟我来学如何保护PostgreSQL数据库
跟我来学如何保护PostgreSQL数据库
20 0
|
19天前
|
SQL 关系型数据库 数据库
手把手教你管理PostgreSQL数据库及其对象
手把手教你管理PostgreSQL数据库及其对象
20 0
|
26天前
|
关系型数据库 MySQL 数据库
探究数据库开源协议:PostgreSQL vs MySQL
探究数据库开源协议:PostgreSQL vs MySQL