#!/bin/bash

dbname=postgres
secs=10
extra_cols_start=0
extra_cols_end=40
extra_cols_increment=10
psql -c "alter system set max_parallel_workers_per_gather = 0;" $dbname > /dev/null
psql -c "alter system set jit = 0;" $dbname > /dev/null
psql -c "select pg_reload_conf();" $dbname > /dev/null
psql -c "create extension if not exists pg_prewarm;" $dbname > /dev/null

for extracol in ", b int not null default 0" ", b int default null"
do
	for firstcol in "c int not null default 0" "c text not null default '0'" "c int null" "c text null"
	do
		for c in $(seq $extra_cols_start $extra_cols_increment $extra_cols_end)
		do
			psql -c "drop table if exists t1;" $dbname > /dev/null
			sql="create table t1 ($firstcol"
			for i in $(seq 0 $c)
			do
				sql="$sql,c$i int not null default 0"
			done
			sql="$sql,a int not null$extracol);"
			psql -c "$sql" $dbname > /dev/null

			psql -c "insert into t1 (a) select a from generate_series(1,1000000) a;" $dbname > /dev/null
			psql -c "vacuum freeze analyze t1;" $dbname > /dev/null
			psql -c "select pg_prewarm('t1');" $dbname > /dev/null

			echo "select sum(a) from t1;" > bench.sql
			for i in {1..3}
			do
				echo -n "extra_cols $c run $i "
				pgbench -n -f bench.sql -M prepared -T $secs $dbname | grep latency
			done
		done
	done
done