Add iteration option to thread test program.

bmomjian · bmomjian · commit 697f9f09ed55 · 2003-08-07T16:14:03.000Z
diff --git a/src/interfaces/ecpg/test/test_thread.pgc b/src/interfaces/ecpg/test/test_thread.pgc
@@ -5,12 +5,14 @@
 
 
 #include <pthread.h>
+#include <stdlib.h>
 
 void		ins1(void);
 void		ins2(void);
 
 EXEC SQL BEGIN DECLARE SECTION;
 char	   *dbname;
+int	    iterations = 10;
 EXEC SQL END DECLARE SECTION;
 
 int
@@ -22,13 +24,21 @@ EXEC SQL BEGIN DECLARE SECTION;
 	int		rows;
 EXEC SQL END DECLARE SECTION;
 
-	if (argc != 2)
+	if (argc < 2 || argc > 3)
 	{
-		fprintf(stderr, "Usage: %s dbname\n", argv[0]);
+		fprintf(stderr, "Usage: %s dbname [iterations]\n", argv[0]);
 		return 1;
 	}
 	dbname = argv[1];
 
+	if (argc == 3)
+		iterations = atoi(argv[2]);
+	if (iterations % 2 != 0)
+	{
+		fprintf(stderr, "iterations must be an even number\n");
+		return 1;
+	}
+
 	EXEC SQL CONNECT TO:dbname AS test0;
 
 	/* DROP might fail */
@@ -47,7 +57,7 @@ EXEC SQL END DECLARE SECTION;
 	EXEC SQL AT test3 COMMIT WORK;
 	EXEC SQL DISCONNECT test3;
 
-	if (rows == 10)
+	if (rows == iterations)
 		printf("Success.\n");
 	else
 		printf("Failure.\n");
@@ -61,7 +71,7 @@ ins1(void)
 	EXEC SQL WHENEVER sqlerror sqlprint;
 	EXEC SQL CONNECT TO:dbname AS test1;
 
-	for (i = 0; i < 5; i++)
+	for (i = 0; i < iterations / 2; i++)
 	{
 		printf("thread 1 : inserting\n");
 		EXEC SQL AT test1 INSERT INTO test_thread VALUES('thread1');
@@ -82,7 +92,7 @@ ins2(void)
 	EXEC SQL WHENEVER sqlerror sqlprint;
 	EXEC SQL CONNECT TO:dbname AS test2;
 
-	for (i = 0; i < 5; i++)
+	for (i = 0; i < iterations / 2; i++)
 	{
 		printf("thread  2: inserting\n");
 		EXEC SQL AT test2 INSERT INTO test_thread VALUES('thread2');
