Update dtmbench to support pathman

knizhnik · knizhnik · commit 160b6061efad · 2017-04-19T16:51:32.000+03:00
diff --git a/contrib/postgres_fdw/tests/dtmbench.cpp b/contrib/postgres_fdw/tests/dtmbench.cpp
@@ -156,13 +156,19 @@ void* writer(void* arg)
 void initializeDatabase()
 {
     connection conn(cfg.connection);
-	int accountsPerShard = (cfg.nAccounts + cfg.nShards - 1)/cfg.nShards;
-	for (int i = 0; i < cfg.nShards; i++)
-	{
+	if (cfg.nShards == 0) { 
 		work txn(conn);
-		exec(txn, "alter table t_fdw%i add check (u between %d and %d)", i+1, accountsPerShard*i, accountsPerShard*(i+1)-1);
-		exec(txn, "insert into t_fdw%i (select generate_series(%d,%d), %d)", i+1, accountsPerShard*i, accountsPerShard*(i+1)-1, 0);
+		exec(txn, "insert into t (select generate_series(1,%d), 0)", cfg.nAccounts);
 		txn.commit();
+	} else { 
+		int accountsPerShard = (cfg.nAccounts + cfg.nShards - 1)/cfg.nShards;
+		for (int i = 0; i < cfg.nShards; i++)
+		{
+			work txn(conn);
+			exec(txn, "alter table t_fdw%i add check (u between %d and %d)", i+1, accountsPerShard*i, accountsPerShard*(i+1)-1);
+			exec(txn, "insert into t_fdw%i (select generate_series(%d,%d), %d)", i+1, accountsPerShard*i, accountsPerShard*(i+1)-1, 0);
+			txn.commit();
+		}
 	}
 }
 
