Function initCreatePKeys

src/bin/pgbench/pgbench.c:4367–4416

* Create primary keys on the standard tables */

 * Create primary keys on the standard tables
 */
static void
initCreatePKeys(PGconn *con)
{
	static const char *const DDLINDEXes[] = {
		"alter table pgbench_branches add primary key (bid)",
		"alter table pgbench_tellers add primary key (tid)",
		"alter table pgbench_accounts add primary key (aid)"
	};
	static const char *const NON_UNIQUE_INDEX_DDLINDEXes[] = {
		"CREATE INDEX branch_idx ON pgbench_branches (bid) ",
		"CREATE INDEX teller_idx ON pgbench_tellers (tid) ",
		"CREATE INDEX account_idx ON pgbench_accounts (aid) "
	};
	StaticAssertStmt(lengthof(DDLINDEXes) == lengthof(NON_UNIQUE_INDEX_DDLINDEXes),
					 "NON_UNIQUE_INDEX_DDLINDEXes must have same size as DDLINDEXes");
	int			i;
	PQExpBufferData query;

	if (use_unique_key)
		fprintf(stderr, "creating primary keys...\n");
	else
		fprintf(stderr, "creating non-unique keys...\n");

	initPQExpBuffer(&query);

	for (i = 0; i < lengthof(DDLINDEXes); i++)
	{

		resetPQExpBuffer(&query);
		if (use_unique_key)
			appendPQExpBufferStr(&query, DDLINDEXes[i]);
		else
			appendPQExpBufferStr(&query, NON_UNIQUE_INDEX_DDLINDEXes[i]);


		if (index_tablespace != NULL)
		{
			char	   *escape_tablespace;

			escape_tablespace = PQescapeIdentifier(con, index_tablespace,
												   strlen(index_tablespace));
			appendPQExpBuffer(&query, " using index tablespace %s", escape_tablespace);
			PQfreemem(escape_tablespace);
		}

		executeStatement(con, query.data);
	}

	termPQExpBuffer(&query);
}

/*
 * Create foreign key constraints between the standard tables