海山数据库(He3DB)源码解读：T_DropRoleSetStmt原理浅析

一、概述

   DropRoleStmt 表示删除角色的 DDL 语句。。

二、DropRole 命令的执行流程

1. PostgresMain
2. exec_simple_query →执行简单的 SQL 查询;
3. StartTransactionCommand → 开始事务;
4. pg_parse_query →解析为内部的抽象语法树（AST）；
5. PortalRun
6. standard_ProcessUtility →权限检查和准备；
7. DropRole→删除角色；
8. CommandCounterIncrement→增量更新当前的命令计数器；
9. CommitTransactionCommand→ 提交当前事务；
10. finish_xact_command→ 在事务结束时，执行必要的清理和关闭操作；

图1 DropRole命令的执行流程图

三、核心结构体介绍

 (一)  DropRoleStmt 是一个结构体，定义在 He3DB中，用于表示删除角色的 SQL 命令。下面是对这个结构体中各个字段的详细解释：

typedef struct DropRoleStmt
{
   
	NodeTag		type;
	List	   *roles;			/* List of roles to remove */
	bool		missing_ok;		/* skip error if a role is missing? */
} DropRoleStmt;

NodeTag type;
这是一个枚举类型，表示节点类型。NodeTag 通常用于区分不同的 SQL 语句类型或查询树中的不同节点类型。对于 DropRoleStmt，这个字段通常会被设置为 T_DropRoleStmt，表示这是一个删除角色的语句。

List *roles;
这是一个链表（List），包含了要删除的角色的名称列表。每个角色名称通常是一个字符串（char *），表示要删除的角色。List 是一个在He3DB内部使用的链表结构，用于存储一组相关的数据。

bool missing_ok;
这是一个布尔类型（bool），表示在尝试删除不存在的角色时是否忽略错误。如果设置为 true，当某个角色不存在时，系统不会抛出错误，而是继续执行操作；如果设置为 false，当某个角色不存在时，系统会抛出一个错误并停止执行。

四、核心代码解析

   在 He3DB 的源代码中， DropRole（@src\backend\commands\user.c） 用于删除数据库角色的函数 。它处理了从数据库中删除角色的各种步骤，包括权限检查、元组删除、依赖关系检查等。

void DropRole(DropRoleStmt *stmt)

定义了一个名为 DropRole 的函数，该函数接受一个 DropRoleStmt 类型的指针作为参数。

  // 定义两个关系变量，用于打开两个系统目录表：pg_authid 和 pg_auth_members
    Relation    pg_authid_rel,
                pg_auth_members_rel;

声明了两个 Relation 类型的变量，分别用于操作 pg_authid 和 pg_auth_members 系统目录表。

   // 定义一个列表单元变量，用于遍历 `stmt->roles` 列表
   ListCell   *item = NULL;

声明了一个 ListCell 类型的指针变量 item，用于遍历 stmt->roles 列表。

   // 检查当前用户是否有创建角色的权限，如果没有则抛出错误
   if (!have_createrole_privilege())
       ereport(ERROR,
               (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
                errmsg("permission denied to drop role")));

检查当前用户是否具有创建角色的权限，如果没有，则抛出错误并中止操作。

   // 打开 pg_authid 和 pg_auth_members 表，获取行独占锁
   pg_authid_rel = table_open(AuthIdRelationId, RowExclusiveLock);
   pg_auth_members_rel = table_open(AuthMemRelationId, RowExclusiveLock);

打开 pg_authid 和 pg_auth_members 表，并获取行独占锁，以防止其他操作修改这些表。

   // 遍历 `stmt->roles` 列表中的每一个角色
   foreach(item, stmt->roles)
   {
   
        RoleSpec   *rolspec = lfirst(item);
       char       *role = NULL;
       HeapTuple  tuple,
                  tmp_tuple;
       Form_pg_authid roleform;
       ScanKeyData scankey;
       char       *detail = NULL;
       char       *detail_log = NULL;
       SysScanDesc sscan;
       Oid         roleid = 0;