Add pgbench option to add foreign key constraints to the standard scenario.

The option --foreign-keys, used at initialization time, will create foreign
key constraints for the columns that represent references to other tables'
primary keys.  This can help in benchmarking FK performance.

Jeff Janes

diff --git a/contrib/pgbench/pgbench.c b/contrib/pgbench/pgbench.c
index f2fdc6c56f67acb9f56a72eb23d08349fd301cd3..f6cd4aad159a9add1a758ddf15b3cb5bd1d8e722 100644 (file)
--- a/contrib/pgbench/pgbench.c
+++ b/contrib/pgbench/pgbench.c
@@ -119,6 +119,11 @@ int            scale = 1;
  */
 int            fillfactor = 100;
 
+/*
+ * create foreign key constraints on the tables?
+ */
+int            foreign_keys = 0;
+
 /*
  * use unlogged tables?
  */
@@ -343,6 +348,8 @@ usage(const char *progname)
           "  -i           invokes initialization mode\n"
           "  -F NUM       fill factor\n"
           "  -s NUM       scaling factor\n"
+          "  --foreign-keys\n"
+          "               create foreign key constraints between tables\n"
           "  --index-tablespace=TABLESPACE\n"
           "               create indexes in the specified tablespace\n"
           "  --tablespace=TABLESPACE\n"
@@ -1275,9 +1282,9 @@ init(void)
    };
    struct ddlinfo DDLs[] = {
        {
-           "pgbench_branches",
-           "bid int not null,bbalance int,filler char(88)",
-           1
+           "pgbench_history",
+           "tid int,bid int,aid int,delta int,mtime timestamp,filler char(22)",
+           0
        },
        {
            "pgbench_tellers",
@@ -1290,9 +1297,9 @@ init(void)
            1
        },
        {
-           "pgbench_history",
-           "tid int,bid int,aid int,delta int,mtime timestamp,filler char(22)",
-           0
+           "pgbench_branches",
+           "bid int not null,bbalance int,filler char(88)",
+           1
        }
    };
    static char *DDLAFTERs[] = {
@@ -1300,6 +1307,13 @@ init(void)
        "alter table pgbench_tellers add primary key (tid)",
        "alter table pgbench_accounts add primary key (aid)"
    };
+   static char *DDLKEYs[] = {
+       "alter table pgbench_tellers add foreign key (bid) references pgbench_branches",
+       "alter table pgbench_accounts add foreign key (bid) references pgbench_branches",
+       "alter table pgbench_history add foreign key (bid) references pgbench_branches",
+       "alter table pgbench_history add foreign key (tid) references pgbench_tellers",
+       "alter table pgbench_history add foreign key (aid) references pgbench_accounts"
+   };
 
    PGconn     *con;
    PGresult   *res;
@@ -1403,7 +1417,7 @@ init(void)
    /*
     * create indexes
     */
-   fprintf(stderr, "set primary key...\n");
+   fprintf(stderr, "set primary keys...\n");
    for (i = 0; i < lengthof(DDLAFTERs); i++)
    {
        char        buffer[256];
@@ -1424,6 +1438,18 @@ init(void)
        executeStatement(con, buffer);
    }
 
+   /*
+    * create foreign keys
+    */
+   if (foreign_keys)
+   {
+       fprintf(stderr, "set foreign keys...\n");
+       for (i = 0; i < lengthof(DDLKEYs); i++)
+       {
+           executeStatement(con, DDLKEYs[i]);
+       }
+   }
+
    /* vacuum */
    fprintf(stderr, "vacuum...");
    executeStatement(con, "vacuum analyze pgbench_branches");
@@ -1864,6 +1890,7 @@ main(int argc, char **argv)
    int         i;
 
    static struct option long_options[] = {
+       {"foreign-keys", no_argument, &foreign_keys, 1},
        {"index-tablespace", required_argument, NULL, 3},
        {"tablespace", required_argument, NULL, 2},
        {"unlogged-tables", no_argument, &unlogged_tables, 1},

diff --git a/doc/src/sgml/pgbench.sgml b/doc/src/sgml/pgbench.sgml
index 4a80b46416d4e610b6fbc6d174731f99b85653bb..5ce66a6ac04a52b09a6a5fe7d1dde24cbd0d5bfa 100644 (file)
--- a/doc/src/sgml/pgbench.sgml
+++ b/doc/src/sgml/pgbench.sgml
@@ -180,6 +180,15 @@ pgbench  options  dbname
       
      
 
+     
+      --foreign-keys
+      
+       
+        Create foreign key constraints between the standard tables.
+       
+      
+     
+
      
       --index-tablespace=index_tablespace