From bbcbd32a050078a4a551ce84ca1bde587c5b49d5 Mon Sep 17 00:00:00 2001
From: xuyuxiang <YJLP2014>
Date: Mon, 19 Apr 2021 19:31:36 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E6=9B=B4=E6=96=B0=E3=80=91=E4=BF=AE?=
 =?UTF-8?q?=E5=A4=8D=E4=BB=A3=E7=A0=81=E7=94=9F=E6=88=90oracle=E7=9A=84bug?=
 =?UTF-8?q?=E3=80=82=E6=96=B0=E5=A2=9E=E6=94=AF=E6=8C=81=E5=90=84=E7=B1=BB?=
 =?UTF-8?q?=E5=9E=8B=E6=95=B0=E6=8D=AE=E5=BA=93=E4=BB=A3=E7=A0=81=E7=94=9F?=
 =?UTF-8?q?=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../mapper/mapping/CodeGenerateMapper.xml     | 236 +++++++++++++++---
 1 file changed, 205 insertions(+), 31 deletions(-)

diff --git a/snowy-base/snowy-gen/src/main/java/vip/xiaonuo/generate/modular/mapper/mapping/CodeGenerateMapper.xml b/snowy-base/snowy-gen/src/main/java/vip/xiaonuo/generate/modular/mapper/mapping/CodeGenerateMapper.xml
index c9882158..449c7c9a 100644
--- a/snowy-base/snowy-gen/src/main/java/vip/xiaonuo/generate/modular/mapper/mapping/CodeGenerateMapper.xml
+++ b/snowy-base/snowy-gen/src/main/java/vip/xiaonuo/generate/modular/mapper/mapping/CodeGenerateMapper.xml
@@ -16,52 +16,226 @@
         <result column="column_key" property="columnKey" />
     </resultMap>
 
-    <!-- 查询指定库中所有表 -->
+    <!-- 查询指定库中所有表 mysql -->
     <select id="selectInformationTable" parameterType="String" resultMap="informationResult" databaseId = "mysql">
         select table_name,create_time,update_time,table_comment
-                from information_schema.tables
-            where
-                table_schema = '${dbName}'
+        from information_schema.tables
+        where
+            table_schema = '${dbName}'
     </select>
+
+    <!-- 查询指定库中所有表 oracle -->
     <select id="selectInformationTable" parameterType="String" resultMap="informationResult" databaseId = "oracle">
         select table_name, comments as table_comment
-            from user_tab_comments
+        from user_tab_comments
     </select>
 
-    <!-- 查询指定表中所有字段 -->
+    <!-- 查询指定库中所有表 mssql -->
+    <select id="selectInformationTable" parameterType="String" resultMap="informationResult" databaseId = "mssql">
+        SELECT DISTINCT
+            d.name as table_name,
+            f.value as table_comment
+        FROM
+            syscolumns a
+                LEFT JOIN systypes b ON a.xusertype= b.xusertype
+                INNER JOIN sysobjects d ON a.id= d.id
+                AND d.xtype= 'U'
+                AND d.name != 'dtproperties'
+                LEFT JOIN syscomments e ON a.cdefault= e.id
+                LEFT JOIN sys.extended_properties g ON a.id= G.major_id
+                AND a.colid= g.minor_id
+                LEFT JOIN sys.extended_properties f ON d.id= f.major_id
+                AND f.minor_id= 0
+    </select>
+
+    <!-- 查询指定库中所有表 pgsql -->
+    <select id="selectInformationTable" parameterType="String" resultMap="informationResult" databaseId = "pgsql">
+        SELECT
+            relname AS TABLE_NAME,
+            col_description ( C.oid, 0 ) AS TABLE_COMMENT
+        FROM
+            pg_class C
+        WHERE
+            relkind = 'r'
+          AND relname NOT LIKE'pg_%'
+          AND relname NOT LIKE'sql_%'
+        ORDER BY
+            relname
+    </select>
+
+    <!-- 查询指定库中所有表 达梦数据库 -->
+    <select id="selectInformationTable" parameterType="String" resultMap="informationResult" databaseId = "dm">
+        select table_name, comments as table_comment
+        from user_tab_comments
+    </select>
+
+    <!-- 查询指定库中所有表 人大金仓数据库 -->
+    <select id="selectInformationTable" parameterType="String" resultMap="informationResult" databaseId = "kingbasees">
+        select table_name, comments as table_comment
+        from user_tab_comments
+    </select>
+
+    <!-- 查询指定表中所有字段 mysql -->
     <select id="selectInformationColumns" parameterType="String" resultMap="inforMationColumnsResult" databaseId = "mysql">
-	    select
-	        column_name,data_type,column_comment,column_key
-	        from information_schema.columns
+        select
+            column_name,data_type,column_comment,column_key
+        from information_schema.columns
         where
             table_schema = '${dbName}' and table_name = '${tableName}';
-	</select>
+    </select>
+
+    <!-- 查询指定表中所有字段 oracle -->
     <select id="selectInformationColumns" parameterType="String" resultMap="inforMationColumnsResult" databaseId = "oracle">
-	    select
-	        a.column_name as column_name,
+        select
+            a.column_name as column_name,
             a.data_type as data_type,
             b.comments as column_comment,
-        case
-            when c.position>0 then 'PRI'
-        else ''
-        end column_key
+            case
+                when c.position>0 then 'PRI'
+                else ''
+                end column_key
         from
-            (select * from user_tab_columns where table_name='${tableName}') a
-        left join
-            (select * from user_col_comments where table_name='${tableName}') b
-        on a.column_name=b.column_name
-        left join
-             (
-              select table_name,column_name,position from user_cons_columns
-              where
+                (select * from user_tab_columns where table_name='${tableName}') a
+                    left join
+                (select * from user_col_comments where table_name='${tableName}') b
+                on a.column_name=b.column_name
+                    left join
+            (
+                select table_name,column_name,position from user_cons_columns
+                where
                     table_name='${tableName}'
-                        and constraint_name=(select constraint_name
-                            from
-                            user_constraints where table_name='${tableName}' and constraint_type='P')
-              and owner='${dbName}'
-             ) c
-        on a.column_name=c.column_name
+                  and constraint_name=(select constraint_name
+                                       from
+                                           user_constraints where table_name='${tableName}' and constraint_type='P')
+                  and owner='${dbName}'
+            ) c
+            on a.column_name=c.column_name
         order by a.column_id
-	</select>
+    </select>
+
+    <!-- 查询指定表中所有字段 mssql -->
+    <select id="selectInformationColumns" parameterType="String" resultMap="inforMationColumnsResult" databaseId = "mssql">
+        SELECT
+            C.name AS column_name,
+            T.name AS data_type,
+            isnull( ETP.value, '' ) AS column_comment,
+            CASE
+
+                WHEN EXISTS (
+                        SELECT
+                            1
+                        FROM
+                            sysobjects
+                        WHERE
+                            xtype = 'PK'
+                          AND parent_obj = c.id
+                          AND name IN ( SELECT name FROM sysindexes WHERE indid IN ( SELECT indid FROM sysindexkeys WHERE id = c.id AND colid = c.colid ) )
+                    ) THEN
+                    'PRI' ELSE ''
+                END AS column_key
+        FROM
+            syscolumns C
+                INNER JOIN systypes T ON C.xusertype = T.xusertype
+                LEFT JOIN sys.extended_properties ETP ON ETP.major_id = c.id
+                AND ETP.minor_id = C.colid
+                AND ETP.name = 'MS_Description'
+                LEFT JOIN syscomments CM ON C.cdefault= CM.id
+        WHERE
+            C.id = object_id( '${tableName}' )
+    </select>
+
+    <!-- 查询指定表中所有字段 pgsql -->
+    <select id="selectInformationColumns" parameterType="String" resultMap="inforMationColumnsResult" databaseId = "pgsql">
+        SELECT
+            t1.*,
+            COALESCE(t2.pk_name, '') AS column_key
+        FROM
+            (
+                SELECT A
+                           .attname AS COLUMN_NAME,
+                       pg_type.typname AS data_type,
+                       col_description ( A.attrelid, A.attnum ) AS column_comment
+                FROM
+                    pg_class AS C,
+                    pg_attribute
+                        AS A INNER JOIN pg_type ON pg_type.oid = A.atttypid
+                WHERE
+                    C.relname = '${tableName}'
+                  AND A.attrelid = C.oid
+                  AND A.attnum > 0
+            ) t1
+                LEFT JOIN (
+                SELECT
+                    pg_attribute.attname AS COLUMN_NAME,
+                    CASE WHEN pg_constraint.conname ISNULL THEN '' ELSE 'PRI' END AS pk_name
+                FROM
+                    pg_constraint
+                        INNER JOIN pg_class ON pg_constraint.conrelid = pg_class.oid
+                        INNER JOIN pg_attribute ON pg_attribute.attrelid = pg_class.oid
+                        AND pg_attribute.attnum = pg_constraint.conkey [ 1 ]
+                        INNER JOIN pg_type ON pg_type.oid = pg_attribute.atttypid
+                WHERE
+                    pg_class.relname = '${tableName}'
+                  AND pg_constraint.contype = 'p'
+            ) t2 ON t1.COLUMN_NAME = t2.COLUMN_NAME
+    </select>
+
+    <!-- 查询指定表中所有字段 达梦数据库 -->
+    <select id="selectInformationColumns" parameterType="String" resultMap="inforMationColumnsResult" databaseId = "dm">
+        select
+            a.column_name as column_name,
+            a.data_type as data_type,
+            b.comments as column_comment,
+            case
+                when c.position>0 then 'PRI'
+                else ''
+                end column_key
+        from
+                (select * from user_tab_columns where table_name='${tableName}') a
+                    left join
+                (select * from user_col_comments where table_name='${tableName}') b
+                on a.column_name=b.column_name
+                    left join
+            (
+                select table_name,column_name,position from user_cons_columns
+                where
+                    table_name='${tableName}'
+                  and constraint_name=(select constraint_name
+                                       from
+                                           user_constraints where table_name='${tableName}' and constraint_type='P')
+                  and owner='${dbName}'
+            ) c
+            on a.column_name=c.column_name
+        order by a.column_id
+    </select>
+
+    <!-- 查询指定表中所有字段 人大金仓数据库 -->
+    <select id="selectInformationColumns" parameterType="String" resultMap="inforMationColumnsResult" databaseId = "dm">
+        select
+            a.column_name as column_name,
+            a.data_type as data_type,
+            b.comments as column_comment,
+            case
+                when c.position>0 then 'PRI'
+                else ''
+                end column_key
+        from
+                (select * from user_tab_columns where table_name='${tableName}') a
+                    left join
+                (select * from user_col_comments where table_name='${tableName}') b
+                on a.column_name=b.column_name
+                    left join
+            (
+                select table_name,column_name,position from user_cons_columns
+                where
+                    table_name='${tableName}'
+                  and constraint_name=(select constraint_name
+                                       from
+                                           user_constraints where table_name='${tableName}' and constraint_type='P')
+                  and owner='${dbName}'
+            ) c
+            on a.column_name=c.column_name
+        order by a.column_id
+    </select>
 
 </mapper>