From db4cbebd2ec87d4cbbd2853918870c71c6b62168 Mon Sep 17 00:00:00 2001
From: thinkphp <thinkphp@qq.com>
Date: Wed, 25 May 2016 11:32:40 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9BPgsql=E9=A9=B1=E5=8A=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 library/think/db/connector/Pgsql.php |   2 +-
 library/think/db/connector/pgsql.sql | 117 +++++++++++++++++++++++++++
 2 files changed, 118 insertions(+), 1 deletion(-)
 create mode 100644 library/think/db/connector/pgsql.sql

diff --git a/library/think/db/connector/Pgsql.php b/library/think/db/connector/Pgsql.php
index a2c40b17..350803a8 100644
--- a/library/think/db/connector/Pgsql.php
+++ b/library/think/db/connector/Pgsql.php
@@ -45,7 +45,7 @@ class Pgsql extends Connection
     {
         $this->initConnect(true);
         list($tableName) = explode(' ', $tableName);
-        $sql             = 'select fields_name as "field",fields_type as "type",fields_not_null as "null",fields_key_name as "key",fields_default as "default",fields_default as "extra" from table_msg(' . $tableName . ');';
+        $sql             = 'select fields_name as "field",fields_type as "type",fields_not_null as "null",fields_key_name as "key",fields_default as "default",fields_default as "extra" from table_msg(\'' . $tableName . '\');';
         $pdo             = $this->linkID->query($sql);
         $result          = $pdo->fetchAll(PDO::FETCH_ASSOC);
         $info            = [];
diff --git a/library/think/db/connector/pgsql.sql b/library/think/db/connector/pgsql.sql
new file mode 100644
index 00000000..e1a09a30
--- /dev/null
+++ b/library/think/db/connector/pgsql.sql
@@ -0,0 +1,117 @@
+CREATE OR REPLACE FUNCTION pgsql_type(a_type varchar) RETURNS varchar AS
+$BODY$
+DECLARE
+     v_type varchar;
+BEGIN
+     IF a_type='int8' THEN
+          v_type:='bigint';
+     ELSIF a_type='int4' THEN
+          v_type:='integer';
+     ELSIF a_type='int2' THEN
+          v_type:='smallint';
+     ELSIF a_type='bpchar' THEN
+          v_type:='char';
+     ELSE
+          v_type:=a_type;
+     END IF;
+     RETURN v_type;
+END;
+$BODY$
+LANGUAGE PLPGSQL;
+
+CREATE TYPE "public"."tablestruct" AS (
+  "fields_key_name" varchar(100),
+  "fields_name" VARCHAR(200),
+  "fields_type" VARCHAR(20),
+  "fields_length" BIGINT,
+  "fields_not_null" VARCHAR(10),
+  "fields_default" VARCHAR(500),
+  "fields_comment" VARCHAR(1000)
+);
+
+CREATE OR REPLACE FUNCTION "public"."table_msg" (a_schema_name varchar, a_table_name varchar) RETURNS SETOF "public"."tablestruct" AS
+$body$
+DECLARE
+     v_ret tablestruct;
+     v_oid oid;
+     v_sql varchar;
+     v_rec RECORD;
+     v_key varchar;
+BEGIN
+     SELECT
+           pg_class.oid  INTO v_oid
+     FROM
+           pg_class
+           INNER JOIN pg_namespace ON (pg_class.relnamespace = pg_namespace.oid AND lower(pg_namespace.nspname) = a_schema_name)
+     WHERE
+           pg_class.relname=a_table_name;
+     IF NOT FOUND THEN
+         RETURN;
+     END IF;
+
+     v_sql='
+     SELECT
+           pg_attribute.attname AS fields_name,
+           pg_attribute.attnum AS fields_index,
+           pgsql_type(pg_type.typname::varchar) AS fields_type,
+           pg_attribute.atttypmod-4 as fields_length,
+           CASE WHEN pg_attribute.attnotnull  THEN ''not null''
+           ELSE ''''
+           END AS fields_not_null,
+           pg_attrdef.adsrc AS fields_default,
+           pg_description.description AS fields_comment
+     FROM
+           pg_attribute
+           INNER JOIN pg_class  ON pg_attribute.attrelid = pg_class.oid
+           INNER JOIN pg_type   ON pg_attribute.atttypid = pg_type.oid
+           LEFT OUTER JOIN pg_attrdef ON pg_attrdef.adrelid = pg_class.oid AND pg_attrdef.adnum = pg_attribute.attnum
+           LEFT OUTER JOIN pg_description ON pg_description.objoid = pg_class.oid AND pg_description.objsubid = pg_attribute.attnum
+     WHERE
+           pg_attribute.attnum > 0
+           AND attisdropped <> ''t''
+           AND pg_class.oid = ' || v_oid || '
+     ORDER BY pg_attribute.attnum' ;
+
+     FOR v_rec IN EXECUTE v_sql LOOP
+         v_ret.fields_name=v_rec.fields_name;
+         v_ret.fields_type=v_rec.fields_type;
+         IF v_rec.fields_length > 0 THEN
+            v_ret.fields_length:=v_rec.fields_length;
+         ELSE
+            v_ret.fields_length:=NULL;
+         END IF;
+         v_ret.fields_not_null=v_rec.fields_not_null;
+         v_ret.fields_default=v_rec.fields_default;
+         v_ret.fields_comment=v_rec.fields_comment;
+         SELECT constraint_name INTO v_key FROM information_schema.key_column_usage WHERE table_schema=a_schema_name AND table_name=a_table_name AND column_name=v_rec.fields_name;
+         IF FOUND THEN
+            v_ret.fields_key_name=v_key;
+         ELSE
+            v_ret.fields_key_name='';
+         END IF;
+         RETURN NEXT v_ret;
+     END LOOP;
+     RETURN ;
+END;
+$body$
+LANGUAGE 'plpgsql' VOLATILE CALLED ON NULL INPUT SECURITY INVOKER;
+
+COMMENT ON FUNCTION "public"."table_msg"(a_schema_name varchar, a_table_name varchar)
+IS '获得表信息';
+
+---重载一个函数
+CREATE OR REPLACE FUNCTION "public"."table_msg" (a_table_name varchar) RETURNS SETOF "public"."tablestruct" AS
+$body$
+DECLARE
+    v_ret tablestruct;
+BEGIN
+    FOR v_ret IN SELECT * FROM table_msg('public',a_table_name) LOOP
+        RETURN NEXT v_ret;
+    END LOOP;
+    RETURN;
+END;
+$body$
+LANGUAGE 'plpgsql' VOLATILE CALLED ON NULL INPUT SECURITY INVOKER;
+
+COMMENT ON FUNCTION "public"."table_msg"(a_table_name varchar)
+IS '获得表信息';
\ No newline at end of file