2

在下面提供的函数find_value_in_table()中,我试图找到任何列的名称,这些列的记录中的列值与字符串“255.255.255.255”匹配。

简而言之,是这样的:

SELECT column_name
  FROM dynamic_table
  WHERE column_value = '255.255.255.255';

笔记:

  • 表的名称作为参数传递给函数,因此“dynamic_table”
  • 我正在尝试确定列的名称,而不是数据值。

这是第一步。稍后我还将参数化 column_value。我知道有一个表在"255.255.255.255"某处存储该值,并且想通过查找存储该值的表和列名来证明该函数的功能。

这一切的目的是:我有大数据库,我是逆向工程。我知道它在某处包含应用程序的一些配置参数。在当前配置中,我知道其中一些参数的精确值(在应用程序配置 GUI 中可以看到:例如计算机名称、IP 地址)。我想浏览整个数据库以确定哪些表存储此配置信息。

我一直在构建find_value()返回这些线索的函数。

如何才能做到这一点?

create or replace function find_all_columns(tablename in text) 
    return setof record as
$func$
    declare r record;
    begin 
        return select   a.attname as "Column",
            pg_catalog.format_type(a.atttypid, a.atttypmod) as "Datatype"
            from
            pg_catalog.pg_attribute a
            where
            a.attnum > 0
            and not a.attisdropped
            and a.attrelid = (  select c.oid from pg_catalog.pg_class c left join pg_catalog.pg_namespace n on n.oid = c.relnamespace where c.relname ~ '^('  ||   quote_ident(tablename)    ||   ')$' and pg_catalog.pg_table_is_visible(c.oid);
        end loop;
    end;
$func$ language 'plpgsql';  

create or replace function find_value_in_table(tablename text) 
    returns setof record as 
$func$
    declare r record;
        return select 
    begin 

        for r in (select find_all_columns(tablename)) loop
            return select * from tablename t where t... = "255.255.255.255" /*  here column would be the value in the record: r.Column*/ 
        end loop;
    end;
$func$ language 'plpgsql';

create or replace function find_tables_name(_username text)
  returns setof record as
$func$
declare
   tbl text;
begin
   for tbl in 
      select t.tablename from pg_tables t
      where  t.tableowner = _username and t.schemaname = 'public'
   loop
      return quote_ident(tbl);
   end loop;
end;
$func$ language 'plpgsql';

create or replace function find_value(_username text, valuetofind text) 
    returns setof record as 
$func$
    declare r record;
    begin 
        for r in (select find_tables_name(_username)) loop
            return find_value_in_table( r.tablename );
        end loop;
    end;
$func$ language 'plpgsql';
4

1 回答 1

3

实现此目的的一种原始方法是进行纯文本转储并使用您选择的编辑器(在我的情况下为 vim)来搜索字符串。

但是这个功能做得更好。:)

CREATE OR REPLACE FUNCTION find_columns(_owner text
                                       ,_valuetofind text
                                       ,_part bool = FALSE)
  RETURNS TABLE (tbl text, col text, typ text) LANGUAGE plpgsql STRICT AS
$func$
DECLARE
   _go         bool;
   _search_row text := '%' || _search || '%';  -- Search row for part of string
BEGIN
   IF _part THEN   -- search col for part of string?
      valuetofind := '%' || valuetofind || '%';
   END IF;

FOR tbl IN 
   SELECT quote_ident(t.schemaname) || '.' || quote_ident(t.tablename)
   FROM pg_tables t
   WHERE  t.tableowner = _owner
-- AND t.schemaname = 'public' -- uncomment to only search one schema
LOOP
   EXECUTE '
   SELECT EXISTS (
      SELECT 1 FROM ' || tbl || ' t WHERE t::text ~~ $1)' -- check whole row
   INTO  _go
   USING _search_row;

   IF _go THEN
      FOR col, typ IN 
         SELECT quote_ident(a.attname) -- AS col
               ,pg_catalog.format_type(a.atttypid, a.atttypmod) -- AS typ
         FROM   pg_catalog.pg_attribute a
         WHERE  a.attnum > 0
         AND    NOT a.attisdropped
         AND    a.attrelid = tbl::regclass
      LOOP
         EXECUTE '
         SELECT EXISTS (
            SELECT 1
            FROM ' || tbl || ' WHERE ' || col || '::text ~~ $1)' -- check col
         INTO   _go
         USING  valuetofind;

         IF _go THEN
            RETURN NEXT;
         END IF;
      END LOOP;
   END IF;

END LOOP;
END;
$func$;
COMMENT ON FUNCTION x.find_columns(text, text, boolean) IS 'Search all tables
 owned by "_owner" user for a value "_search" (text representation).
 Match full or partial (_part)';

称呼:

SELECT * FROM find_columns('postgres', '255.255.255.255');
SELECT * FROM find_columns('fadmin', '255.255.255.255', TRUE);

回报:

       tbl       |     col     | typ
-----------------+-------------+------
 event.eventkat  | eventkat    | text
 public.foo      | description | text
 public.bar      | filter      | text

用 PostgreSQL 9.1 测试

要点

  • 该功能是一站式商店。

  • 我构建了一个选项来搜索部分值 ( _part)。默认是搜索整列。

  • 我对整行进行了快速测试,以消除根本没有表格的表格valuetofind。为此,我使用 PostgreSQL 将整行快速转换为文本的能力。这应该使函数更快- 除非所有或几乎所有表都符合条件或表只有一列。

  • 我将返回类型定义为并立即分配隐式定义的RETURNS TABLE (tbl text, col text, typ text)变量。因此,我不需要额外的变量,并且可以在列符合条件时立即进行。tblcoltypRETURN NEXT

  • 好好利用EXISTS这里!这是最快的选择,因为您只对列是否具有值感兴趣

  • 使用LIKE(或~~简称)代替正则表达式。更简单,更快。

  • quote_ident()立即所有标识符。

  • EXECUTE *command* INTO USING是工具性的。

于 2012-08-25T04:41:41.563 回答