From 9afae0df9765958d57b5f37d5121e4e582a99641 Mon Sep 17 00:00:00 2001
From: Lukas Eder <lukas.eder@gmail.com>
Date: Wed, 11 Nov 2020 13:25:49 +0100
Subject: [PATCH] [jOOQ/jOOQ#10913] Add Database.getPrimaryKeys() and overloads
 to jOOQ-meta

---
 .../java/org/jooq/meta/AbstractDatabase.java  | 29 ++++++++++++++++++-
 .../src/main/java/org/jooq/meta/Database.java | 10 +++++++
 2 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/jOOQ-meta/src/main/java/org/jooq/meta/AbstractDatabase.java b/jOOQ-meta/src/main/java/org/jooq/meta/AbstractDatabase.java
index 196b093d30..0abf3a6692 100644
--- a/jOOQ-meta/src/main/java/org/jooq/meta/AbstractDatabase.java
+++ b/jOOQ-meta/src/main/java/org/jooq/meta/AbstractDatabase.java
@@ -208,6 +208,7 @@ public abstract class AbstractDatabase implements Database {
     private List<SequenceDefinition>                                         sequences;
     private List<IdentityDefinition>                                         identities;
     private List<IndexDefinition>                                            indexes;
+    private List<UniqueKeyDefinition>                                        primaryKeys;
     private List<UniqueKeyDefinition>                                        uniqueKeys;
     private List<UniqueKeyDefinition>                                        keys;
     private List<ForeignKeyDefinition>                                       foreignKeys;
@@ -226,6 +227,7 @@ public abstract class AbstractDatabase implements Database {
     private transient Map<SchemaDefinition, List<IdentityDefinition>>        identitiesBySchema;
     private transient Map<SchemaDefinition, List<IndexDefinition>>           indexesBySchema;
     private transient Map<TableDefinition, List<IndexDefinition>>            indexesByTable;
+    private transient Map<SchemaDefinition, List<UniqueKeyDefinition>>       primaryKeysBySchema;
     private transient Map<SchemaDefinition, List<UniqueKeyDefinition>>       uniqueKeysBySchema;
     private transient Map<SchemaDefinition, List<UniqueKeyDefinition>>       keysBySchema;
     private transient Map<SchemaDefinition, List<ForeignKeyDefinition>>      foreignKeysBySchema;
@@ -1609,7 +1611,7 @@ public abstract class AbstractDatabase implements Database {
         if (uniqueKeys == null) {
             uniqueKeys = new ArrayList<>();
 
-            if (getIncludeUniqueKeys() || getIncludePrimaryKeys())
+            if (getIncludeUniqueKeys())
                 for (SchemaDefinition s : getSchemata())
                     for (TableDefinition table : getTables(s))
                         for (UniqueKeyDefinition uniqueKey : table.getUniqueKeys())
@@ -1621,6 +1623,31 @@ public abstract class AbstractDatabase implements Database {
         return uniqueKeys;
     }
 
+    @Override
+    public final List<UniqueKeyDefinition> getPrimaryKeys(SchemaDefinition schema) {
+        if (primaryKeysBySchema == null)
+            primaryKeysBySchema = new LinkedHashMap<>();
+
+        return filterSchema(getPrimaryKeys(), schema, primaryKeysBySchema);
+    }
+
+    @Override
+    public final List<UniqueKeyDefinition> getPrimaryKeys() {
+        if (primaryKeys == null) {
+            primaryKeys = new ArrayList<>();
+
+            if (getIncludePrimaryKeys())
+                for (SchemaDefinition s : getSchemata())
+                    for (TableDefinition table : getTables(s))
+                        if (table.getPrimaryKey() != null)
+                            primaryKeys.add(table.getPrimaryKey());
+
+            sort(primaryKeys);
+        }
+
+        return primaryKeys;
+    }
+
     @Override
     public final List<UniqueKeyDefinition> getUniqueKeys(SchemaDefinition schema) {
         if (uniqueKeysBySchema == null)
diff --git a/jOOQ-meta/src/main/java/org/jooq/meta/Database.java b/jOOQ-meta/src/main/java/org/jooq/meta/Database.java
index e02686af3e..79ca417e41 100644
--- a/jOOQ-meta/src/main/java/org/jooq/meta/Database.java
+++ b/jOOQ-meta/src/main/java/org/jooq/meta/Database.java
@@ -134,6 +134,16 @@ public interface Database extends AutoCloseable {
      */
     List<UniqueKeyDefinition> getUniqueKeys(SchemaDefinition schema);
 
+    /**
+     * The primary keys contained in this database.
+     */
+    List<UniqueKeyDefinition> getPrimaryKeys();
+
+    /**
+     * The primary keys contained in this database.
+     */
+    List<UniqueKeyDefinition> getPrimaryKeys(SchemaDefinition schema);
+
     /**
      * The unique and primary keys contained in this database.
      */