#include <stdio.h>
#include <stdlib.h>
#include <string.h>

static void
pf(const char *str)
{
	printf("%s;\n", str);
	fflush(stdout);
}

int
main(int argc, char **argv)
{
	int scenario = atoi(argv[1]);
	int count = atoi(argv[2]);
	int i;

	switch (scenario)
	{
		case 0:
			pf("set lock_timeout = 1");
			while (count-- > 0)
			{
				pf("begin");
				for (i = 0; i < 10; i++)
				{
					pf("savepoint mysp");
					pf("insert into lock_bug values (1, 42)");
					pf("rollback to savepoint mysp");
				}
				pf("rollback");
			}
			break;
		case 1:
			while (count-- > 0)
			{
				pf("create index lock_bug_idx_f1 on lock_bug(f1)");
				pf("drop index lock_bug_idx_f1");
			}
			break;
	}

	return 0;
}