计算所有表中所有模式(已授予权限)中的行

时间:2019-03-07 08:33:51

标签: postgresql count schema database-table

我正在尝试计算数据库中所有表中的行。由于我想得到的答案必须区分不同的模式,因此我还要考虑到特定表所在的模式。

answer最有用,但是事实证明我没有数据库中所有模式的访问权限。

我知道我可以通过执行以下查询来检查特定表或模式的特权:

select count(*) from (
SELECT grantee, privilege_type 
FROM information_schema.role_table_grants 
WHERE table_name='data' and privilege_type = 'SELECT') as foo

并检查输出是否等于或大于一。

此刻我的代码如下:

CREATE or replace function rowcount_all(schema_name text default 'public')
  RETURNS table(table_name_var text, cnt bigint) as
$$
declare
 table_name_var text;
begin
  for table_name_var in SELECT c.relname FROM pg_class c
    JOIN pg_namespace s ON (c.relnamespace=s.oid)
    WHERE c.relkind = 'r' AND s.nspname=schema_name
  loop
    if (
    select count(*) from (
    SELECT grantee, privilege_type 
    FROM information_schema.role_table_grants 
    WHERE table_name=table_name_var and privilege_type = 'SELECT') as foo
  ) >= 1 then 
    RETURN QUERY EXECUTE format('select cast(%L as text),count(*) from %I.%I',
       table_name_var, schema_name, table_name_var);
    end if;
  END loop;
end
$$ language plpgsql;

在执行以下查询时

WITH rc(schema_name,tbl) AS (
  select s.n,rowcount_all(s.n) from (values ('schema1'),
    ('schema2'), ('schema3'), ('schema4')) as s(n)
)
SELECT schema_name,(tbl).* FROM rc;

我收到错误消息ERROR: permission denied for relation table1,其中table1在我无权访问的架构中。我假设我在IF语句中的逻辑某种程度上不会过滤掉我无权访问的表。

1 个答案:

答案 0 :(得分:1)

information_schema是数据库元数据的SQL标准表示形式,如果您尝试构建可移植的内容,则很有用,但如果您只是尝试管理Postgres服务器,通常会有些笨拙。< / p>

检查特权的最简单方法是使用privilege check functions。我相信这只会返回您能够查询的表:

select oid::regclass::text
from pg_class
where relkind = 'r'
  and relnamespace = schema_name::regnamespace
  and has_schema_privilege(relnamespace, 'USAGE')
  and has_any_column_privilege(oid, 'SELECT')

请注意,has_any_column_privilege()是有用的罕见情况之一(与更明显的has_table_privilege()相比),因为您不需要select count(*)的全表特权,只需访问其中一列即可(但这无关紧要)。

还请注意,oid::regclass::text将返回一个已被引用且具有模式限定的表名(如有必要),因此您的format()调用可以使用简单的%s而不是{ {1}}。

如果您可以处理大约和稍微过时的记录数,则可以通过查询最后一个{{1}的统计信息来完全绕过特权检查(并为自己节省很多表扫描) }运行:

%I.%I