PostgreSQL中有哪些钩子函数
PostgreSQL 中有哪些钩子函数,针对这个问题,这篇文章详细介绍了相对应的分析和解答,希望可以帮助更多想解决这个问题的小伙伴找到更简单易行的方法。
专注于为中小企业提供网站设计制作、成都网站制作服务,电脑端+手机端+微信端的三站合一,更高效的管理,为中小企业绵竹免费做网站提供优质的服务。我们立足成都,凝聚了一批互联网行业人才,有力地推动了上1000+企业的稳健成长,帮助中小企业通过网站建设实现规模扩充和转变。
一、需求
删除数据库pg12db时,只能使用pg12用户删除,其他用户(包括超级用户)均不能删除此数据库。
二、实现步骤
删除数据库的命令是drop database,属于Utility命令,而PG提供了ProcessUtility_hook钩子可供使用.
实现一个钩子函数,判断SQL语句是否为T_DropdbStmt,如是,则判断数据库名称和用户名称是否匹配,如不匹配,则报错,源码如下:
[pg12@localhost hookdemo_dbrestrict]$ cat hookdemo_dbrestrict.c /* * This is a hook demo. * */ #include "postgres.h" #include "miscadmin.h" #include "tcop/utility.h" PG_MODULE_MAGIC; void _PG_init(void); void _PG_fini(void); static char *undroppabledb = "pg12db"; static char *hooksuperuser = "pg12"; static ProcessUtility_hook_type prev_utility_hook = NULL; static void hookdemodbrestrict_ProcessUtility(PlannedStmt *pstmt, const char *queryString, ProcessUtilityContext context, ParamListInfo params, QueryEnvironment *queryEnv, DestReceiver *dest, char *completionTag) { /* Do our custom process on drop database */ switch(nodeTag(pstmt->utilityStmt)) { case T_DropdbStmt: { DropdbStmt *stmt = (DropdbStmt *)pstmt->utilityStmt; char *username = GetUserNameFromId(GetUserId(),false); /* * only user pg12 can drop pg12db. */ if (strcmp(stmt->dbname, undroppabledb) == 0 && strcmp(username, hooksuperuser) != 0) ereport(ERROR,(errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),errmsg("Only superuser \"%s\" can drop database \"%s\"",hooksuperuser,undroppabledb))); break; } default: break; } /*Standard process*/ standard_ProcessUtility(pstmt, queryString, context, params, queryEnv, dest, completionTag); } /* _PG_init */ void _PG_init(void) { prev_utility_hook = ProcessUtility_hook; ProcessUtility_hook = hookdemodbrestrict_ProcessUtility; } /* Uninstall */ void _PG_fini(void) { ProcessUtility_hook = prev_utility_hook; } [pg12@localhost hookdemo_dbrestrict]$
三、实际效果
创建超级用户superx,使用该用户登录,创建pg12db数据库,删除数据库,报错.
[local:/data/run/pg12]:5120 pg12@testdb=# create user superx with superuser password 'root'; CREATE ROLE [local:/data/run/pg12]:5120 pg12@testdb=# \q [pg12@localhost pg122db]$ psql -U superx Expanded display is used automatically. psql (12.2) Type "help" for help. [local:/data/run/pg12]:5120 superx@testdb=# create database pg12db; CREATE DATABASE [local:/data/run/pg12]:5120 superx@testdb=# drop database pg12db; ERROR: Only superuser "pg12" can drop database "pg12db" [local:/data/run/pg12]:5120 superx@testdb=#
关于PostgreSQL 中有哪些钩子函数问题的解答就分享到这里了,希望以上内容可以对大家有一定的帮助,如果你还有很多疑惑没有解开,可以关注创新互联行业资讯频道了解更多相关知识。
网站标题:PostgreSQL中有哪些钩子函数
文章链接:http://ybzwz.com/article/pseipp.html